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 = "/mnt/vcr";
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 + "/";
51 _database = aDatabase;
52 _format = new SimpleDateFormat("yyyy-MM-dd-HH:mm");
53 _recordings = new HashMap<File,Recording>();
59 * @see org.wamblee.io.DirectoryMonitor.Listener#fileChanged(java.io.File)
61 public void fileChanged(File aFile) {
62 LOG.debug("file changed " + aFile);
64 // Re-assess file type
65 Recording recording = _recordings.get(aFile);
66 LOG.debug("Recording changed " + recording);
67 String dir = getDirectory(recording);
68 FileType type = getFileType(aFile);
69 String path = dir + "/" + getFilename(recording, type);
71 if (exists(dir + "/" + getFilename(recording, type))) {
75 for (FileType t : FileType.values()) {
76 rmlink(dir + "/" + getFilename(recording, t));
78 createSymLink(_monitorDir + aFile.getName(), dir + "/"
79 + getFilename(recording, type));
86 * @see org.wamblee.io.DirectoryMonitor.Listener#fileCreated(java.io.File)
88 public void fileCreated(File aFile) {
89 LOG.debug("file created " + aFile);
90 Recording recording = _database.findRecording(aFile.getName());
91 _recordings.put(aFile, recording);
92 LOG.info("New recording detected " + aFile + " "
95 recording.setFilesize(aFile.length());
96 _database.update(recording);
97 String dir = getDirectory(recording);
99 createSymLink(_monitorDir + aFile.getName(), dir + "/"
100 + getFilename(recording, getFileType(aFile)));
106 * @see org.wamblee.io.DirectoryMonitor.Listener#fileDeleted(java.io.File)
108 public void fileDeleted(File aFile) {
109 LOG.debug("file deleted " + aFile);
110 Recording recording = _recordings.get(aFile);
111 _recordings.remove(recording);
112 LOG.info("recording deleted " + recording);
113 String dir = getDirectory(recording);
114 for (FileType t: FileType.values()) {
115 rmlink(dir + "/" + getFilename(recording, t));
120 private String getDirectory(Recording aRecording) {
121 return aRecording.getTitle();
124 private FileType getFileType(File aFile) {
125 SimpleProcess process = new SimpleProcess(new File(_monitorDir), new String[] {
126 "file", aFile.getName() });
129 if (process.getStdout().contains("RIFF")) {
134 } catch (IOException e) {
135 LOG.error("Determining filetype for " + aFile + " failed", e);
140 private String getFilename(Recording aRecording, FileType aType) {
141 return (_format.format(aRecording.getProgstart()) + "-"
142 + aRecording.getSubtitle() + "-"
143 + aRecording.getChannel().getName() + "."
144 + aType.toString().toLowerCase()).replaceAll("/", "-");
147 private boolean exists(String aPath) {
148 LOG.debug("exists " + aPath);
149 return new File(_linkDir, aPath).exists();
152 private void rmlink(String aPath) {
153 LOG.debug("rmlink " + aPath);
154 File link = new File(_linkDir, aPath);
155 //if ( !link.exists()) {
158 if (!link.delete()) {
159 LOG.warn("Delete failed: " + aPath);
161 LOG.info("Removed link " + link);
165 private void mkdir(String aDir) {
166 LOG.debug("mkdir " + aDir);
167 File dir = new File(_linkDir, aDir);
168 if ( dir.isDirectory()) {
172 LOG.warn("Could not create directory path: " + aDir);
174 LOG.info("Created directory " + dir);
178 private void rmdir(String aDir) {
179 LOG.debug("rmdir " + aDir);
180 File dir = new File(_linkDir, aDir);
182 LOG.warn("Directory not deleted (still recordings left): " + aDir);
184 LOG.info("Directory " + dir + " deleted.");
188 private void createSymLink(String aTarget, String aSource) {
190 SimpleProcess process = new SimpleProcess(_linkDir, new String[] {
191 "ln", "-s", aTarget, aSource });
193 LOG.info("Created symlink " + aSource + " -> " + aTarget);
194 } catch (IOException e) {
196 "Could not create symlink: " + aTarget + " <- " + aSource,