wamblee.org Git - utils/blob - mythtv/monitor/src/main/java/org/wamblee/mythtv/LinkStructure.java

   1 /*
   2  * Copyright 2006 the original author or authors.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.wamblee.mythtv;
  18
  19 import java.io.File;
  20 import java.io.IOException;
  21 import java.text.SimpleDateFormat;
  22 import java.util.HashMap;
  23 import java.util.Map;
  24
  25 import org.apache.commons.logging.Log;
  26 import org.apache.commons.logging.LogFactory;
  27 import org.wamblee.io.SimpleProcess;
  28 import org.wamblee.io.DirectoryMonitor.Listener;
  29
  30 /**
  31  * Link structure.
  32  */
  33 public class LinkStructure implements Listener {
  34
  35     private static final Log LOG = LogFactory.getLog(LinkStructure.class);
  36
  37     private String _monitorDir;
  38
  39     private File _linkDir;
  40
  41     private RecordingDatabase _database;
  42
  43     private SimpleDateFormat _format;
  44
  45     private Map<File,Recording> _recordings;
  46
  47     public LinkStructure(String aMonitorDir, File aLinkDir,
  48             RecordingDatabase aDatabase) {
  49         _monitorDir = aMonitorDir + "/";
  50         deleteDir(aLinkDir);
  51         _linkDir = aLinkDir;
  52         _database = aDatabase;
  53         _format = new SimpleDateFormat("yyyy-MM-dd-HH:mm");
  54         _recordings = new HashMap<File,Recording>();
  55     }
  56
  57     private void deleteDir(File aFile) {
  58         for (File file: aFile.listFiles()) {
  59             if ( file.isDirectory()) {
  60                 deleteDir(file);
  61             }
  62             LOG.info("File deleted " + file + ": " + file.delete());
  63         }
  64     }
  65
  66     /*
  67      * (non-Javadoc)
  68      *
  69      * @see org.wamblee.io.DirectoryMonitor.Listener#fileChanged(java.io.File)
  70      */
  71     public void fileChanged(File aFile) {
  72         LOG.debug("file changed " + aFile);
  73
  74         // Re-assess file type
  75         Recording recording = _recordings.get(aFile);
  76         LOG.debug("Recording changed " + recording);
  77         recording.setFilesize(aFile.length());
  78         _database.update(recording);
  79         String dir = getDirectory(recording);
  80         FileType type = getFileType(aFile);
  81         String path = dir + "/" + getFilename(recording, type);
  82
  83         if (exists(dir + "/" + getFilename(recording, type))) {
  84             // Nothing to do.
  85         } else {
  86             mkdir(dir);
  87             for (FileType t : FileType.values()) {
  88                 rmlink(dir + "/" + getFilename(recording, t));
  89             }
  90             createSymLink(_monitorDir + aFile.getName(), dir + "/"
  91                     + getFilename(recording, type));
  92         }
  93     }
  94
  95     /*
  96      * (non-Javadoc)
  97      *
  98      * @see org.wamblee.io.DirectoryMonitor.Listener#fileCreated(java.io.File)
  99      */
 100     public void fileCreated(File aFile) {
 101         LOG.debug("file created " + aFile);
 102         Recording recording = _database.findRecording(aFile.getName());
 103         _recordings.put(aFile, recording);
 104         LOG.info("New recording detected " + aFile + " "
 105                 + recording);
 106
 107         recording.setFilesize(aFile.length());
 108         _database.update(recording);
 109         String dir = getDirectory(recording);
 110         mkdir(dir);
 111         createSymLink(_monitorDir + aFile.getName(), dir + "/"
 112                 + getFilename(recording, getFileType(aFile)));
 113     }
 114
 115     /*
 116      * (non-Javadoc)
 117      *
 118      * @see org.wamblee.io.DirectoryMonitor.Listener#fileDeleted(java.io.File)
 119      */
 120     public void fileDeleted(File aFile) {
 121         LOG.debug("file deleted " + aFile);
 122         Recording recording = _recordings.get(aFile);
 123         _recordings.remove(recording);
 124         // mythtv will remove the recording from its database itself.
 125         LOG.info("recording deleted " + recording);
 126         String dir = getDirectory(recording);
 127         for (FileType t: FileType.values()) {
 128             rmlink(dir + "/" + getFilename(recording, t));
 129         }
 130         rmdir(dir);
 131     }
 132
 133     private String getDirectory(Recording aRecording) {
 134         return aRecording.getTitle().replaceAll("/", "-");
 135     }
 136
 137     private FileType getFileType(File aFile) {
 138         SimpleProcess process = new SimpleProcess(new File(_monitorDir), new String[] {
 139                 "file", aFile.getName() });
 140         try {
 141             process.run();
 142             if (process.getStdout().contains("RIFF")) {
 143                 return FileType.AVI;
 144             } else {
 145                 return FileType.MPG;
 146             }
 147         } catch (IOException e) {
 148             LOG.error("Determining filetype for " + aFile + " failed", e);
 149             return FileType.MPG;
 150         }
 151     }
 152
 153     private String getFilename(Recording aRecording, FileType aType) {
 154         return (_format.format(aRecording.getProgstart()) + "-"
 155                 + aRecording.getSubtitle() + "-"
 156                 + aRecording.getChannel().getName() + "."
 157                 + aType.toString().toLowerCase()).replaceAll("/", "-");
 158     }
 159
 160     private boolean exists(String aPath) {
 161         LOG.debug("exists " + aPath);
 162         return new File(_linkDir, aPath).exists();
 163     }
 164
 165     private void rmlink(String aPath) {
 166         LOG.debug("rmlink " + aPath);
 167         File link = new File(_linkDir, aPath);
 168         //if ( !link.exists()) {
 169         //    return;
 170        // }
 171         if (!link.delete()) {
 172             LOG.warn("Delete failed: " + aPath);
 173         } else {
 174             LOG.info("Removed link " + link);
 175         }
 176     }
 177
 178     private void mkdir(String aDir) {
 179         LOG.debug("mkdir " + aDir);
 180         File dir = new File(_linkDir, aDir);
 181         if ( dir.isDirectory()) {
 182             return;
 183         }
 184         if (!dir.mkdirs()) {
 185             LOG.warn("Could not create directory path: " + aDir);
 186         } else {
 187             LOG.info("Created directory " + dir);
 188         }
 189     }
 190
 191     private void rmdir(String aDir) {
 192         LOG.debug("rmdir " + aDir);
 193         File dir = new File(_linkDir, aDir);
 194         if (!dir.delete()) {
 195             LOG.warn("Directory not deleted (still recordings left): " + aDir);
 196         } else {
 197             LOG.info("Directory " + dir + " deleted.");
 198         }
 199     }
 200
 201     private void createSymLink(String aTarget, String aSource) {
 202         try {
 203             SimpleProcess process = new SimpleProcess(_linkDir, new String[] {
 204                     "ln", "-s", aTarget, aSource });
 205             process.run();
 206             LOG.info("Created symlink " + aSource + " -> " + aTarget);
 207         } catch (IOException e) {
 208             LOG.error(
 209                     "Could not create symlink: " + aTarget + " <- " + aSource,
 210                     e);
 211         }
 212     }
 213 }