frontend/src/saccubus/worker/download/Download.java

   1 package saccubus.worker.download;
   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.worker.Worker;
  17 import saccubus.worker.WorkerListener;
  18 import saccubus.worker.profile.CommentProfile;
  19 import saccubus.worker.profile.GeneralProfile;
  20 import saccubus.worker.profile.OutputProfile;
  21 import saccubus.worker.profile.DownloadProfile;
  22 import saccubus.worker.profile.ProxyProfile;
  23
  24 /**
  25  * <p>タイトル: さきゅばす</p>
  26  *
  27  * <p>説明: ニコニコ動画の動画をコメントつきで保存</p>
  28  *
  29  * <p>著作権: Copyright (c) 2007 PSI</p>
  30  *
  31  * <p>会社名: </p>
  32  *
  33  * @author 未入力
  34  * @version 1.0
  35  */
  36 public class Download extends Worker<DownloadResult, DownloadProgress> {
  37
  38     private static final Logger logger = Logger.getLogger(Download.class.getName());
  39     private final DownloadProfile profile;
  40     private final String videoId;
  41
  42     public Download(DownloadProfile profile, String videoId) {
  43         this(profile, videoId, null);
  44     }
  45
  46     /**
  47      * コンバータを構築します.
  48      * @param videoId 対象となる動画のID.
  49      * @param time
  50      * @param profile
  51      * @param listener
  52      * @param flag
  53      */
  54     public Download(DownloadProfile profile, String videoId, WorkerListener<DownloadProgress> listener) {
  55         // TODO listener登録
  56         super(listener);
  57         this.videoId = videoId;
  58         this.profile = profile;
  59     }
  60
  61 //    @Override
  62 //    public Boolean call() throws Exception {
  63 //        try {
  64 //            final DownloadResult result = doInBackground();
  65 //            return Boolean.valueOf(result.getResultValue());
  66 //        } finally {
  67 //            // TODO 何か処理が必要？
  68 ////            getStopFlag().finished();
  69 //        }
  70 //    }
  71 //    // TODO Runnableを実装しなくなったので削除する
  72 //    public void run() {
  73 //        try {
  74 //            call();
  75 //        } catch (Exception ex) {
  76 //            String text = (ex.getMessage() != null) ? ex.getMessage() : "予期しないエラー発生のため中断しました。";
  77 //            sendText(text);
  78 //            logger.log(Level.SEVERE, null, ex);
  79 //        }
  80 //    }
  81     @Override
  82     public DownloadResult call() throws Exception {
  83
  84         publish(new DownloadProgress("ログイン中"));
  85
  86         NicoHttpClient client = null;
  87         nicobrowser.VideoInfo vi = null;
  88         NamePattern videoNamePattern = null;
  89         WayBackInfo wbi = null;
  90         if (needsLogin()) {
  91             client = createClientAndLogin();
  92             vi = client.getVideoInfo(videoId);
  93
  94             final String name = profile.getVideoProfile().getFileName();
  95             final String replaceFrom = profile.getGeneralProfile().getReplaceFrom();
  96             final String replaceTo = profile.getGeneralProfile().getReplaceTo();
  97             videoNamePattern = new NamePattern(name, replaceFrom, replaceTo, vi.getTitleInWatchPage());
  98
  99             if (needsBackLog()) {
 100                 final String key = client.getWayBackKey(vi);
 101                 wbi = new WayBackInfo(key, profile.getCommentProfile().getBackLogPoint());
 102             }
 103         }
 104
 105         checkStop();
 106
 107         File commentFile;
 108         if (profile.getCommentProfile().isDownload()) {
 109             final CommentProfile prof = profile.getCommentProfile();
 110             final GeneralProfile gene = profile.getGeneralProfile();
 111
 112             final NamePattern pattern = new NamePattern(prof.getFileName(), gene.getReplaceFrom(), gene.getReplaceTo(),
 113                     vi.getTitleInWatchPage());
 114             // TODO コメントファイルに{low}は使えないことをどこかに書くべきか
 115             final String name = pattern.createFileName(videoId, true);
 116             final File file = new File(profile.getCommentProfile().getDir(), name);
 117
 118             commentFile = client.getCommentFile(vi, file.getPath(), wbi, profile.getCommentProfile().
 119                     getLengthRelatedCommentSize(),
 120                     profile.getCommentProfile().isDisablePerMinComment());
 121         } else {
 122             commentFile = profile.getCommentProfile().getLocalFile();
 123         }
 124
 125         checkStop();
 126
 127         File videoFile;
 128         GetFlvResult vf;
 129         if (profile.getVideoProfile().isDownload()) {
 130             vf = client.getFlvFile(vi, profile.getVideoProfile().getDir(), videoNamePattern,
 131                     Status.GET_INFO, true, new ProgressListener() {
 132
 133                 @Override
 134                 public void progress(long fileSize, long downloadSize) {
 135                     final double vol = (double) downloadSize / (double) fileSize * 100.0;
 136                     publish(new DownloadProgress(String.format("ダウンロード%.2f%%", vol)));
 137                 }
 138             });
 139
 140             videoFile = vf.getFile();
 141         } else {
 142             videoFile = profile.getVideoProfile().getLocalFile();
 143         }
 144         return new DownloadResult(true, videoFile, commentFile);
 145
 146
 147         // TODO FFMPEG 実行開始は別タスクとして実装する.
 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
 170 //        boolean res = new FfmpegCommand(getListener(), getStopFlag(), commentFile, videoFile,
 171 //                convertedVideoFile, profile.getFfmpeg(), profile.getGeneralSetting()).execute();
 172 //        return res;
 173     }
 174
 175     /** @return 何かダウンロードするものがあればtrue. */
 176     private static boolean needsDownload(DownloadProfile profile) {
 177         return (profile.getVideoProfile().isDownload() || profile.getCommentProfile().isDownload());
 178     }
 179
 180     // TODO どこかに処理を移す必要がある.
 181     /**
 182      * (ネットワーク設定以外の)設定を検証する.
 183      * @throws IllegalArgumentException 設定に不備がある場合.
 184      */
 185 //    private void validSetting() {
 186 //        if (profile.getOutputFileSetting().isConvert()) {
 187 //            File a = profile.getFfmpeg().getFfmpeg();
 188 //            if (!a.canRead()) {
 189 //                throw new IllegalArgumentException("FFmpegが見つかりません。");
 190 //            }
 191 //            if (profile.getFfmpeg().getVhook().getPath().indexOf(' ') >= 0) {
 192 //                throw new IllegalArgumentException("すいません。現在vhookライブラリには半角空白は使えません。");
 193 //            }
 194 //            a = profile.getFfmpeg().getVhook();
 195 //            if (!a.canRead()) {
 196 //                throw new IllegalArgumentException("Vhookライブラリが見つかりません。");
 197 //            }
 198 //            a = profile.getFfmpeg().getFont();
 199 //            if (!a.canRead()) {
 200 //                throw new IllegalArgumentException("フォントが見つかりません。");
 201 //            }
 202 //        }
 203 //    }
 204     /**
 205      * HttpClientを生成し, ニコニコ動画サーバへログインします.
 206      * @return 生成したHttpClientインスタンス.
 207      * @throws IOException ログイン失敗.
 208      * @throws InterruptedException ログイン失敗.
 209      */
 210     // TODO HttpException を投げるのをやめたい. コンパイル時にHttpComponentが必要になるので.
 211     private NicoHttpClient createClientAndLogin() throws IOException, InterruptedException, HttpException {
 212         final NicoHttpClient client = createClient(profile.getProxyProfile());
 213         final boolean hasLogin;
 214         try {
 215             hasLogin = client.login(profile.getLoginInfo().getMail(), profile.getLoginInfo().getPassword());
 216             if (!hasLogin) {
 217                 throw new IOException("login fail");
 218             }
 219         } catch (URISyntaxException ex) {
 220             throw new IOException("login fail", ex);
 221         }
 222         return client;
 223     }
 224
 225     private NicoHttpClient createClient(ProxyProfile proxy) {
 226         if (proxy.use()) {
 227             return new NicoHttpClient(proxy.getHost(), proxy.getPort());
 228         } else {
 229             return new NicoHttpClient();
 230         }
 231     }
 232
 233     /** @return ログインする必要があればtrue. */
 234     private boolean needsLogin() {
 235         return profile.getVideoProfile().isDownload() || profile.getCommentProfile().isDownload();
 236     }
 237
 238     /** @return 過去ログ取得の必要があればtrue. */
 239     private boolean needsBackLog() {
 240         return profile.getCommentProfile().getBackLogPoint() >= 0L;
 241     }
 242
 243     private void checkStop() throws InterruptedException {
 244         if (Thread.interrupted()) {
 245             throw new InterruptedException("中止要求を受け付けました");
 246         }
 247     }
 248 }