OSDN Git Service

Downloadテスト実行時の問題修正
[coroid/inqubus.git] / frontend / src / saccubus / converter / Download.java
1 package saccubus.converter;
2
3 import java.io.File;
4 import java.io.IOException;
5 import java.net.URISyntaxException;
6 import java.util.concurrent.Callable;
7 import java.util.logging.Logger;
8 import javax.swing.SwingWorker;
9 import nicobrowser.GetFlvResult;
10 import nicobrowser.NamePattern;
11 import nicobrowser.NicoHttpClient;
12 import nicobrowser.ProgressListener;
13 import nicobrowser.WayBackInfo;
14 import nicobrowser.entity.NicoContent.Status;
15 import org.apache.http.HttpException;
16 import saccubus.converter.profile.CommentProfile;
17 import saccubus.converter.profile.FfmpegProfile;
18 import saccubus.converter.profile.GeneralProfile;
19 import saccubus.converter.profile.OutputProfile;
20 import saccubus.converter.profile.Profile;
21 import saccubus.converter.profile.ProxyProfile;
22
23 /**
24  * <p>タイトル: さきゅばす</p>
25  *
26  * <p>説明: ニコニコ動画の動画をコメントつきで保存</p>
27  *
28  * <p>著作権: Copyright (c) 2007 PSI</p>
29  *
30  * <p>会社名: </p>
31  *
32  * @author 未入力
33  * @version 1.0
34  */
35 public class Download extends SwingWorker<DownloadResult, DownloadProgress> implements Callable<Boolean> {
36
37     private static final Logger logger = Logger.getLogger(Download.class.getName());
38     private final Profile profile;
39     private final String movieId;
40
41     /**
42      * コンバータを構築します.
43      * @param id 対象となる動画のID.
44      * @param time
45      * @param profile
46      * @param listener
47      * @param flag
48      */
49     public Download(String id, Profile profile) {
50         this.movieId = id;
51         this.profile = profile;
52     }
53
54     @Override
55     public Boolean call() throws Exception {
56         try {
57             final DownloadResult result = doInBackground();
58             return Boolean.valueOf(result.getResultValue());
59         } finally {
60             // TODO 何か処理が必要?
61 //            getStopFlag().finished();
62         }
63     }
64
65 //    // TODO Runnableを実装しなくなったので削除する
66 //    public void run() {
67 //        try {
68 //            call();
69 //        } catch (Exception ex) {
70 //            String text = (ex.getMessage() != null) ? ex.getMessage() : "予期しないエラー発生のため中断しました。";
71 //            sendText(text);
72 //            logger.log(Level.SEVERE, null, ex);
73 //        }
74 //    }
75     @Override
76     protected DownloadResult doInBackground() throws Exception {
77         if (!shouldRun(profile)) {
78             publish(new DownloadProgress("何もすることがありません"));
79             return new DownloadResult(true);
80         }
81
82         validSetting();
83
84         // TODO ログインしないで良い場合もある.
85         publish(new DownloadProgress("ログイン中"));
86
87
88
89         NicoHttpClient client = null;
90         nicobrowser.VideoInfo vi = null;
91         NamePattern videoNamePattern = null;
92         WayBackInfo wbi = null;
93         if (needsLogin()) {
94             client = createClientAndLogin();
95             vi = client.getVideoInfo(movieId);
96
97             final String name = profile.getVideoSetting().getFileName();
98             final String replaceFrom = profile.getGeneralSetting().getReplaceFrom();
99             final String replaceTo = profile.getGeneralSetting().getReplaceTo();
100             videoNamePattern = new NamePattern(name, replaceFrom, replaceTo, vi.getTitleInWatchPage());
101
102             if (needsBackLog()) {
103                 final String key = client.getWayBackKey(vi);
104                 wbi = new WayBackInfo(key, profile.getCommentSetting().getBackLogPoint());
105             }
106         }
107
108         checkStop();
109
110         File commentFile;
111         if (profile.getCommentSetting().isDownload()) {
112             final CommentProfile prof = profile.getCommentSetting();
113             final GeneralProfile gene = profile.getGeneralSetting();
114
115             final NamePattern pattern = new NamePattern(prof.getFileName(), gene.getReplaceFrom(), gene.getReplaceTo(),
116                     vi.getTitleInWatchPage());
117             // TODO コメントファイルに{low}は使えないことをどこかに書くべきか
118             final String name = pattern.createFileName(movieId, true);
119             final File file = new File(profile.getCommentSetting().getDir(), name);
120
121             commentFile = client.getCommentFile(vi, file.getPath(), wbi, profile.getCommentSetting().
122                     getLengthRelatedCommentSize(),
123                     profile.getCommentSetting().isDisablePerMinComment());
124         } else {
125             commentFile = profile.getCommentSetting().getLocalFile();
126         }
127
128         checkStop();
129
130         File videoFile;
131         GetFlvResult vf = null;
132         if (profile.getVideoSetting().isDownload()) {
133             vf = client.getFlvFile(vi, profile.getVideoSetting().getDir(), videoNamePattern,
134                     Status.GET_INFO, true, new ProgressListener() {
135
136                 @Override
137                 public void progress(long fileSize, long downloadSize) {
138                     final double vol = (double) downloadSize / (double) fileSize * 100.0;
139                     publish(new DownloadProgress(String.format("ダウンロード%.2f%%", vol)));
140                 }
141             });
142
143             videoFile = vf.getFile();
144         } else {
145             videoFile = profile.getVideoSetting().getLocalFile();
146         }
147
148         if (!profile.getOutputFileSetting().isConvert()) {
149             publish(new DownloadProgress("動画・コメントを保存し、変換は行いませんでした。"));
150             return new DownloadResult(true);
151         }
152
153         if (!videoFile.isFile()) {
154             throw new IOException("入力動画ファイルが存在しません:" + videoFile.getPath());
155         }
156
157         if (profile.getOutputFileSetting().isAddComment()) {
158             if (!commentFile.isFile()) {
159                 throw new IOException("入力コメントファイルが存在しません:" + commentFile.getPath());
160             }
161         } else {
162             commentFile = null;
163         }
164
165         /*ビデオ名の確定*/
166         final boolean isNotLow = (vf == null) ? true : (vf.getStatus() != Status.GET_LOW);
167         File convertedVideoFile = getOutputFileName(vi.getTitleInWatchPage(), isNotLow);
168
169         return new DownloadResult(true);
170
171         // TODO FFMPEG 実行開始は別タスクとして実装する.
172 //        boolean res = new FfmpegCommand(getListener(), getStopFlag(), commentFile, videoFile,
173 //                convertedVideoFile, profile.getFfmpeg(), profile.getGeneralSetting()).execute();
174 //        return res;
175     }
176
177     /** @return 何か実行すべき処理があればtrue. */
178     private static boolean shouldRun(Profile profile) {
179         return profile.getOutputFileSetting().isConvert() || needsDownload(profile);
180     }
181
182     /** @return 何かダウンロードするものがあればtrue. */
183     private static boolean needsDownload(Profile profile) {
184         return (profile.getVideoSetting().isDownload() || profile.getCommentSetting().isDownload());
185     }
186
187     /**
188      * (ネットワーク設定以外の)設定を検証する.
189      * @throws IllegalArgumentException 設定に不備がある場合.
190      */
191     private void validSetting() {
192         if (profile.getOutputFileSetting().isConvert()) {
193             File a = profile.getFfmpeg().getFfmpeg();
194             if (!a.canRead()) {
195                 throw new IllegalArgumentException("FFmpegが見つかりません。");
196             }
197             if (profile.getFfmpeg().getVhook().getPath().indexOf(' ') >= 0) {
198                 throw new IllegalArgumentException("すいません。現在vhookライブラリには半角空白は使えません。");
199             }
200             a = profile.getFfmpeg().getVhook();
201             if (!a.canRead()) {
202                 throw new IllegalArgumentException("Vhookライブラリが見つかりません。");
203             }
204             a = profile.getFfmpeg().getFont();
205             if (!a.canRead()) {
206                 throw new IllegalArgumentException("フォントが見つかりません。");
207             }
208         }
209     }
210
211     /**
212      * HttpClientを生成し, ニコニコ動画サーバへログインします.
213      * @return 生成したHttpClientインスタンス.
214      * @throws IOException ログイン失敗.
215      * @throws InterruptedException ログイン失敗.
216      */
217     // TODO HttpException を投げるのをやめたい. コンパイル時にHttpComponentが必要になるので.
218     private NicoHttpClient createClientAndLogin() throws IOException, InterruptedException, HttpException {
219         final NicoHttpClient client = createClient(profile.getProxySetting());
220         final boolean hasLogin;
221         try {
222             hasLogin = client.login(profile.getLoginInfo().getMail(), profile.getLoginInfo().getPassword());
223             if (!hasLogin) {
224                 throw new IOException("login fail");
225             }
226         } catch (URISyntaxException ex) {
227             throw new IOException("login fail", ex);
228         }
229         return client;
230     }
231
232     private NicoHttpClient createClient(ProxyProfile proxy) {
233         if (proxy.use()) {
234             return new NicoHttpClient(proxy.getHost(), proxy.getPort());
235         } else {
236             return new NicoHttpClient();
237         }
238     }
239
240     /** @return ログインする必要があればtrue. */
241     private boolean needsLogin() {
242         return profile.getVideoSetting().isDownload() || profile.getCommentSetting().isDownload();
243     }
244
245     /** @return 過去ログ取得の必要があればtrue. */
246     private boolean needsBackLog() {
247         return profile.getCommentSetting().getBackLogPoint() >= 0L;
248     }
249
250     /**
251      * 出力ファイルの保存先を確定します.
252      * @param title 動画タイトル.
253      * @param isNotLow 入力動画ファイルがlowでないかどうか.
254      * @return 出力ファイルの保存先.
255      */
256     private File getOutputFileName(String title, boolean isNotLow) {
257         final OutputProfile prof = profile.getOutputFileSetting();
258         final GeneralProfile gene = profile.getGeneralSetting();
259         final File dir = prof.getDir();
260         final String replaceFrom = gene.getReplaceFrom();
261         final String replaceTo = gene.getReplaceTo();
262         final NamePattern pattern = new NamePattern(prof.getFileName(), replaceFrom, replaceTo, title);
263         final String name = pattern.createFileName(movieId, isNotLow);
264         return new File(dir, name);
265     }
266
267     private void checkStop() throws InterruptedException {
268         if (Thread.interrupted()) {
269             throw new InterruptedException("中止要求を受け付けました");
270         }
271     }
272 }