now updating file size in the database when the file has changed.
[utils] / 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         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 }