wamblee.org Git - utils/blob - mythtv/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         String dir = getDirectory(recording);
  78         FileType type = getFileType(aFile);
  79         String path = dir + "/" + getFilename(recording, type);
  80
  81         if (exists(dir + "/" + getFilename(recording, type))) {
  82             // Nothing to do.
  83         } else {
  84             mkdir(dir);
  85             for (FileType t : FileType.values()) {
  86                 rmlink(dir + "/" + getFilename(recording, t));
  87             }
  88             createSymLink(_monitorDir + aFile.getName(), dir + "/"
  89                     + getFilename(recording, type));
  90         }
  91     }
  92
  93     /*
  94      * (non-Javadoc)
  95      *
  96      * @see org.wamblee.io.DirectoryMonitor.Listener#fileCreated(java.io.File)
  97      */
  98     public void fileCreated(File aFile) {
  99         LOG.debug("file created " + aFile);
 100         Recording recording = _database.findRecording(aFile.getName());
 101         _recordings.put(aFile, recording);
 102         LOG.info("New recording detected " + aFile + " "
 103                 + recording);
 104
 105         recording.setFilesize(aFile.length());
 106         _database.update(recording);
 107         String dir = getDirectory(recording);
 108         mkdir(dir);
 109         createSymLink(_monitorDir + aFile.getName(), dir + "/"
 110                 + getFilename(recording, getFileType(aFile)));
 111     }
 112
 113     /*
 114      * (non-Javadoc)
 115      *
 116      * @see org.wamblee.io.DirectoryMonitor.Listener#fileDeleted(java.io.File)
 117      */
 118     public void fileDeleted(File aFile) {
 119         LOG.debug("file deleted " + aFile);
 120         Recording recording = _recordings.get(aFile);
 121         _recordings.remove(recording);
 122         LOG.info("recording deleted " + recording);
 123         String dir = getDirectory(recording);
 124         for (FileType t: FileType.values()) {
 125             rmlink(dir + "/" + getFilename(recording, t));
 126         }
 127         rmdir(dir);
 128     }
 129
 130     private String getDirectory(Recording aRecording) {
 131         return aRecording.getTitle().replaceAll("/", "-");
 132     }
 133
 134     private FileType getFileType(File aFile) {
 135         SimpleProcess process = new SimpleProcess(new File(_monitorDir), new String[] {
 136                 "file", aFile.getName() });
 137         try {
 138             process.run();
 139             if (process.getStdout().contains("RIFF")) {
 140                 return FileType.AVI;
 141             } else {
 142                 return FileType.MPG;
 143             }
 144         } catch (IOException e) {
 145             LOG.error("Determining filetype for " + aFile + " failed", e);
 146             return FileType.MPG;
 147         }
 148     }
 149
 150     private String getFilename(Recording aRecording, FileType aType) {
 151         return (_format.format(aRecording.getProgstart()) + "-"
 152                 + aRecording.getSubtitle() + "-"
 153                 + aRecording.getChannel().getName() + "."
 154                 + aType.toString().toLowerCase()).replaceAll("/", "-");
 155     }
 156
 157     private boolean exists(String aPath) {
 158         LOG.debug("exists " + aPath);
 159         return new File(_linkDir, aPath).exists();
 160     }
 161
 162     private void rmlink(String aPath) {
 163         LOG.debug("rmlink " + aPath);
 164         File link = new File(_linkDir, aPath);
 165         //if ( !link.exists()) {
 166         //    return;
 167        // }
 168         if (!link.delete()) {
 169             LOG.warn("Delete failed: " + aPath);
 170         } else {
 171             LOG.info("Removed link " + link);
 172         }
 173     }
 174
 175     private void mkdir(String aDir) {
 176         LOG.debug("mkdir " + aDir);
 177         File dir = new File(_linkDir, aDir);
 178         if ( dir.isDirectory()) {
 179             return;
 180         }
 181         if (!dir.mkdirs()) {
 182             LOG.warn("Could not create directory path: " + aDir);
 183         } else {
 184             LOG.info("Created directory " + dir);
 185         }
 186     }
 187
 188     private void rmdir(String aDir) {
 189         LOG.debug("rmdir " + aDir);
 190         File dir = new File(_linkDir, aDir);
 191         if (!dir.delete()) {
 192             LOG.warn("Directory not deleted (still recordings left): " + aDir);
 193         } else {
 194             LOG.info("Directory " + dir + " deleted.");
 195         }
 196     }
 197
 198     private void createSymLink(String aTarget, String aSource) {
 199         try {
 200             SimpleProcess process = new SimpleProcess(_linkDir, new String[] {
 201                     "ln", "-s", aTarget, aSource });
 202             process.run();
 203             LOG.info("Created symlink " + aSource + " -> " + aTarget);
 204         } catch (IOException e) {
 205             LOG.error(
 206                     "Could not create symlink: " + aTarget + " <- " + aSource,
 207                     e);
 208         }
 209     }
 210 }