2 * Copyright 2006 the original author or authors.
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 package org.wamblee.mythtv;
20 import java.io.IOException;
21 import java.text.SimpleDateFormat;
22 import java.util.HashMap;
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;
33 public class LinkStructure implements Listener {
35 private static final Log LOG = LogFactory.getLog(LinkStructure.class);
37 private String _monitorDir;
39 private File _linkDir;
41 private RecordingDatabase _database;
43 private SimpleDateFormat _format;
45 private Map<File,Recording> _recordings;
47 public LinkStructure(String aMonitorDir, File aLinkDir,
48 RecordingDatabase aDatabase) {
49 _monitorDir = aMonitorDir + "/";
52 _database = aDatabase;
53 _format = new SimpleDateFormat("yyyy-MM-dd-HH:mm");
54 _recordings = new HashMap<File,Recording>();
57 private void deleteDir(File aFile) {
58 for (File file: aFile.listFiles()) {
59 if ( file.isDirectory()) {
62 LOG.info("File deleted " + file + ": " + file.delete());
69 * @see org.wamblee.io.DirectoryMonitor.Listener#fileChanged(java.io.File)
71 public void fileChanged(File aFile) {
72 LOG.debug("file changed " + aFile);
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);
81 if (exists(dir + "/" + getFilename(recording, type))) {
85 for (FileType t : FileType.values()) {
86 rmlink(dir + "/" + getFilename(recording, t));
88 createSymLink(_monitorDir + aFile.getName(), dir + "/"
89 + getFilename(recording, type));
96 * @see org.wamblee.io.DirectoryMonitor.Listener#fileCreated(java.io.File)
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 + " "
105 recording.setFilesize(aFile.length());
106 _database.update(recording);
107 String dir = getDirectory(recording);
109 createSymLink(_monitorDir + aFile.getName(), dir + "/"
110 + getFilename(recording, getFileType(aFile)));
116 * @see org.wamblee.io.DirectoryMonitor.Listener#fileDeleted(java.io.File)
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));
130 private String getDirectory(Recording aRecording) {
131 return aRecording.getTitle().replaceAll("/", "-");
134 private FileType getFileType(File aFile) {
135 SimpleProcess process = new SimpleProcess(new File(_monitorDir), new String[] {
136 "file", aFile.getName() });
139 if (process.getStdout().contains("RIFF")) {
144 } catch (IOException e) {
145 LOG.error("Determining filetype for " + aFile + " failed", e);
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("/", "-");
157 private boolean exists(String aPath) {
158 LOG.debug("exists " + aPath);
159 return new File(_linkDir, aPath).exists();
162 private void rmlink(String aPath) {
163 LOG.debug("rmlink " + aPath);
164 File link = new File(_linkDir, aPath);
165 //if ( !link.exists()) {
168 if (!link.delete()) {
169 LOG.warn("Delete failed: " + aPath);
171 LOG.info("Removed link " + link);
175 private void mkdir(String aDir) {
176 LOG.debug("mkdir " + aDir);
177 File dir = new File(_linkDir, aDir);
178 if ( dir.isDirectory()) {
182 LOG.warn("Could not create directory path: " + aDir);
184 LOG.info("Created directory " + dir);
188 private void rmdir(String aDir) {
189 LOG.debug("rmdir " + aDir);
190 File dir = new File(_linkDir, aDir);
192 LOG.warn("Directory not deleted (still recordings left): " + aDir);
194 LOG.info("Directory " + dir + " deleted.");
198 private void createSymLink(String aTarget, String aSource) {
200 SimpleProcess process = new SimpleProcess(_linkDir, new String[] {
201 "ln", "-s", aTarget, aSource });
203 LOG.info("Created symlink " + aSource + " -> " + aTarget);
204 } catch (IOException e) {
206 "Could not create symlink: " + aTarget + " <- " + aSource,