CaptureStream/downloadthread-hls.h at master · CSReviser/CaptureStream · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
	Copyright (C) 2009-2013 jakago

	This file is part of CaptureStream, the flv downloader for NHK radio
	language courses.

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 2 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef DOWNLOADTHREAD_H
#define DOWNLOADTHREAD_H

#define USE_FFMPEG_HLS 1

#include <QThread>
#include <QStringList>
#include <QHash>
#include <QProcess>

#include "mainwindow.h"

class DownloadThread : public QThread {
	Q_OBJECT

public:
	DownloadThread( Ui::MainWindowClass* ui );
	~DownloadThread() {}
	void cancel() { isCanceled = true; }

protected:
	void run();

signals:
	void critical( QString message );
	void information( QString message );
	void current( QString );
	void messageWithoutBreak( QString );

private:
	QStringList getAttribute( QString url, QString attribute );
	bool checkExecutable( QString path );
	bool isFfmpegAvailable( QString& path );
#if !USE_FFMPEG_HLS
	bool isOpensslAvailable( QString& path );
#endif
	bool checkOutputDir( QString dirPath );
	void downloadENews( bool re_read );

#if !USE_FFMPEG_HLS
	QString getMasterM3u8( QString file );
	QString getIndexM3u8( QString masterM3u8 );
	QByteArray getCryptKey( QString indexM3u8 );
	QStringList getSegmentUrlList( QString indexM3u8 );
	QString downloadSegment( QString outputDir, QString segmentUrl );
	bool decryptSegment( QString outputDir, QString segmentName, int index, QString cryptKey );
	bool mergeSegments( QString outputDir, QStringList segmentNames, QString mp4Name );
	bool convertFormat( QString outputDir, QString mp4Name, QString outBasename, QString extension, QString id3tagTitle, QString kouza, QString hdate, QString file );
#endif
	bool captureStream( QString kouza, QString hdate, QString file );

	QString formatName( QString format, QString kouza, QString hdate, QString file, bool checkIllegal );
	QStringList getElements( QString url, QString path );
	void downloadShower();

	Ui::MainWindowClass* ui;
	bool isCanceled;
    bool failed1935;

	static QString paths[];
	static QString prefix;
	static QString suffix;

	static QString flv_host;
	static QString flv_app;
	static QString flv_service_prefix;

	static QString flvstreamer;
	static QString ffmpeg;
#if !USE_FFMPEG_HLS
	static QString openssl;
#endif
	static QString scramble;
	static QStringList malformed;
#if USE_FFMPEG_HLS
	static QHash<QString, QString> ffmpegHash;
	static QHash<QProcess::ProcessError, QString>processError;
#endif

};

#endif // DOWNLOADTHREAD_H