|
0
|
1 // Filename : main.cpp
|
|
13
|
2 // Last Change: 2018-11-14 水 12:54:16.
|
|
0
|
3 //
|
|
9
|
4
|
|
10
|
5 #include <wx/socket.h>
|
|
11
|
6 #include <wx/sstream.h>
|
|
|
7 #include <wx/protocol/http.h>
|
|
4
|
8 #include "id.h"
|
|
0
|
9 #include "main.h"
|
|
|
10 #include "auth.h"
|
|
11
|
11 #include "net.h"
|
|
0
|
12 #include "rsearcher.h"
|
|
|
13
|
|
|
14 IMPLEMENT_APP( MyApp )
|
|
|
15
|
|
|
16 IMPLEMENT_CLASS( MyApp, wxApp )
|
|
|
17
|
|
|
18 MyApp::MyApp()
|
|
|
19 {
|
|
1
|
20 develop = true;
|
|
0
|
21 }
|
|
|
22 MyApp::~MyApp()
|
|
|
23 {
|
|
|
24 }
|
|
|
25
|
|
|
26 bool MyApp::OnInit()
|
|
|
27 {
|
|
|
28 if ( !wxApp::OnInit() ) return false;
|
|
|
29
|
|
11
|
30 InitSetting();
|
|
|
31
|
|
|
32 // Check New-Version
|
|
|
33 wxString newver = GetVersion();
|
|
|
34 if ( newver.Cmp( RSVER ) == 1 ) { // 1.9 -> 1.91 -> 1.92 ...
|
|
|
35 wxString sv = wxString::Format( wxT( "%s:%d" ), m_serveraddr, m_serverport );
|
|
|
36
|
|
|
37 wxArrayString args;
|
|
|
38 args.Add( wxT( "upgrade.bat" ) );
|
|
|
39 args.Add( sv );
|
|
|
40 args.Add( RSVER );
|
|
|
41 args.Add( newver );
|
|
|
42
|
|
|
43 wxExecute( wxJoin( args, ' ', '\\' ) );
|
|
|
44 return false;
|
|
|
45 }
|
|
|
46
|
|
0
|
47 wxImage::AddHandler( new wxJPEGHandler );
|
|
|
48 wxImage::AddHandler( new wxPNGHandler );
|
|
9
|
49 wxFileSystem::AddHandler( new wxZipFSHandler );
|
|
0
|
50
|
|
7
|
51 // Main Window
|
|
|
52 MainFrame *mainframe = new MainFrame( NULL, ID_MAIN, wxEmptyString, wxPoint( rect.x, rect.y ), rect.GetSize(), wxDEFAULT_FRAME_STYLE );
|
|
|
53 mainframe->SetServer( m_serveraddr, m_serverport );
|
|
13
|
54 mainframe->SetDBdir( m_dbdir );
|
|
10
|
55 mainframe->InDevelop( true );
|
|
13
|
56 if ( !mainframe->GetDB( 1, 1, 1 ) ) {
|
|
9
|
57 mainframe->Destroy();
|
|
|
58 return true;
|
|
|
59 }
|
|
6
|
60
|
|
5
|
61 // Splash Screen
|
|
9
|
62 SplashScreen( splash );
|
|
3
|
63
|
|
10
|
64 // Mode
|
|
|
65 if ( !unlock_key.IsEmpty() ) {
|
|
|
66 wxIPV4address addr;
|
|
|
67 addr.Hostname( wxGetFullHostName() );
|
|
|
68
|
|
|
69 wxString key = wxT( "re:searcher" ) + addr.IPAddress();
|
|
|
70 wxArrayString args;
|
|
|
71 args.Add( wxT( "crypto.exe" ) );
|
|
|
72 args.Add( wxT( "-a" ) );
|
|
|
73 args.Add( key );
|
|
|
74
|
|
|
75 wxArrayString output, errors;
|
|
|
76 wxExecute( wxJoin( args, ' ', '\\' ), output, errors );
|
|
|
77
|
|
|
78 if ( unlock_key.IsSameAs( output[0] ) ) {
|
|
|
79 mainframe->SetUser( wxT( "root" ) );
|
|
|
80 mainframe->SetTitle( wxT( "Re:Searcher - root" ) );
|
|
|
81 mainframe->LoadDB();
|
|
|
82 mainframe->Show( true );
|
|
|
83 return true;
|
|
|
84 }
|
|
|
85 }
|
|
|
86
|
|
|
87 // User Dialog
|
|
7
|
88 AuthDialog *authdlg = new AuthDialog( NULL, wxID_ANY, wxT( "Who are you ?" ), wxDefaultPosition, wxDefaultSize, wxCAPTION );
|
|
5
|
89 if ( !authdlg->LoadDB() ) {
|
|
|
90 authdlg->Destroy();
|
|
|
91 return false;
|
|
|
92 }
|
|
1
|
93
|
|
3
|
94 if ( authdlg->ShowModal() == wxID_OK ) {
|
|
5
|
95 wxString ui = wxString::Format( wxT( "./image/" ) + authdlg->GetUser() + wxT( ".jpg" ) );
|
|
6
|
96 if ( wxFileExists( ui ) ) {
|
|
|
97 wxCopyFile( ui, wxT( "./image/hello.jpg" ), true );
|
|
|
98 } else {
|
|
|
99 unsigned int seed = (unsigned int)time( 0 );
|
|
|
100 srand( seed );
|
|
|
101 int n = rand() % 13;
|
|
|
102 wxFileSystem* fs = new wxFileSystem();
|
|
9
|
103 wxString archive = wxT( "file:///./myapp.bin" );
|
|
6
|
104 wxFSFile* file = fs->OpenFile( archive + wxString::Format( wxT( "#zip:%02d.jpg" ), n ) );
|
|
|
105 if ( file ) {
|
|
|
106 wxInputStream* s = file->GetStream();
|
|
|
107 wxFileOutputStream of( wxT( "./image/hello.jpg" ) );
|
|
|
108 s->Read( of );
|
|
|
109 delete file;
|
|
|
110 }
|
|
|
111 delete fs;
|
|
|
112 }
|
|
5
|
113
|
|
7
|
114 mainframe->SetUser( authdlg->GetUser() );
|
|
|
115 mainframe->SetTitle( wxT( "Re:Searcher - " ) + authdlg->GetUser() );
|
|
|
116 mainframe->LoadDB();
|
|
3
|
117 mainframe->Show( true );
|
|
7
|
118 } else {
|
|
|
119 mainframe->Destroy();
|
|
3
|
120 }
|
|
|
121 authdlg->Destroy();
|
|
0
|
122
|
|
|
123 return true;
|
|
|
124 }
|
|
|
125
|
|
|
126 int MyApp::OnExit()
|
|
|
127 {
|
|
13
|
128 RemoveFile( wxT( "hhs.csv.tmp" ) );
|
|
|
129 RemoveFile( wxT( "hhs.csv" ) );
|
|
|
130 RemoveFile( wxT( "index.db.tmp" ) );
|
|
6
|
131 RemoveFile( wxT( "auth.db" ) );
|
|
|
132 RemoveFile( wxT( "hhs.db" ) );
|
|
|
133 RemoveFile( wxT( ".cache/*" ) );
|
|
0
|
134 SaveSetting();
|
|
|
135 return 0;
|
|
|
136 }
|
|
|
137
|
|
|
138 void MyApp::InitSetting()
|
|
|
139 {
|
|
2
|
140 conf_file = wxGetCwd() + wxFILE_SEP_PATH + wxT( "app.conf" );
|
|
|
141 config = new wxFileConfig( wxT( "MyApp" ), wxT( "T.Mutoh" ), conf_file, wxEmptyString, wxCONFIG_USE_LOCAL_FILE );
|
|
|
142
|
|
|
143 config->SetPath( wxT( "/Geometry" ) );
|
|
|
144 config->Read( wxT( "x" ), &rect.x );
|
|
|
145 config->Read( wxT( "y" ), &rect.y );
|
|
|
146 config->Read( wxT( "w" ), &rect.width );
|
|
|
147 config->Read( wxT( "h" ), &rect.height );
|
|
0
|
148
|
|
13
|
149 config->SetPath( wxT( "/DBManage" ) );
|
|
|
150 config->Read( wxT( "dbdir" ), &m_dbdir );
|
|
|
151
|
|
2
|
152 config->SetPath( wxT( "/Server" ) );
|
|
7
|
153 config->Read( wxT( "proxy_address" ), &m_serveraddr );
|
|
|
154 config->Read( wxT( "proxy_port" ), &m_serverport );
|
|
|
155 if ( m_serveraddr.IsSameAs( wxEmptyString, false ) ) {
|
|
|
156 config->Read( wxT( "address" ), &m_serveraddr );
|
|
|
157 config->Read( wxT( "port" ), &m_serverport );
|
|
|
158 }
|
|
2
|
159
|
|
7
|
160 if ( m_serveraddr.IsSameAs( wxEmptyString, false ) ) {
|
|
|
161 m_serveraddr = wxT( "192.168.21.151" ); // nginx
|
|
|
162 m_serverport = 80;
|
|
|
163 }
|
|
9
|
164
|
|
|
165 config->SetPath( wxT( "/Misc" ) );
|
|
|
166 config->Read( wxT( "splash" ), &splash );
|
|
10
|
167 config->Read( wxT( "unlock_key" ), &unlock_key );
|
|
9
|
168
|
|
|
169 delete config;
|
|
0
|
170 }
|
|
|
171
|
|
|
172 void MyApp::SaveSetting()
|
|
|
173 {
|
|
3
|
174 config = new wxFileConfig( wxT( "MyApp" ), wxT( "T.Mutoh" ), conf_file, wxEmptyString, wxCONFIG_USE_LOCAL_FILE );
|
|
0
|
175
|
|
3
|
176 config->SetPath( wxT( "/Geometry" ) );
|
|
|
177 config->Write( wxT( "x" ), rect.x );
|
|
|
178 config->Write( wxT( "y" ), rect.y );
|
|
|
179 config->Write( wxT( "w" ), rect.width );
|
|
|
180 config->Write( wxT( "h" ), rect.height );
|
|
0
|
181 delete config;
|
|
|
182 }
|
|
|
183
|
|
9
|
184 void MyApp::SplashScreen( int ms )
|
|
|
185 {
|
|
|
186 if ( ms < 0 ) return;
|
|
|
187
|
|
|
188 wxFileSystem* fs = new wxFileSystem();
|
|
|
189 wxString archive = wxT( "file:///./myapp.bin" );
|
|
|
190 for ( int i = 0; i < 11; i++ ) {
|
|
|
191 wxFSFile* file = fs->OpenFile( archive + wxString::Format( wxT( "#zip:startup%02d.png" ), i ) );
|
|
|
192 if ( file ) {
|
|
|
193 wxInputStream* s = file->GetStream();
|
|
|
194 wxImage image( *s, wxBITMAP_TYPE_PNG );
|
|
|
195 wxBitmap bmp = wxBitmap( image );
|
|
|
196 wxSplashScreen* splash = new wxSplashScreen( bmp, wxSPLASH_CENTRE_ON_SCREEN|wxSPLASH_TIMEOUT, 2000, NULL, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxBORDER_NONE|wxSTAY_ON_TOP );
|
|
|
197 delete file;
|
|
|
198 }
|
|
|
199 wxMilliSleep( ms );
|
|
|
200 }
|
|
|
201 delete fs;
|
|
|
202 }
|
|
|
203
|
|
11
|
204 wxString MyApp::GetVersion( void )
|
|
|
205 {
|
|
|
206 wxHTTP get;
|
|
|
207 get.SetTimeout( 30 );
|
|
|
208 get.SetFlags( wxSOCKET_WAITALL|wxSOCKET_BLOCK );
|
|
|
209 while ( !get.Connect( m_serveraddr, m_serverport ) )
|
|
|
210 wxSleep( 1 );
|
|
|
211
|
|
|
212 wxString version;
|
|
|
213 wxString url = wxT( "/release/version" );
|
|
|
214 wxInputStream *http_istream = get.GetInputStream( url );
|
|
|
215 if ( get.GetError() == wxPROTO_NOERR ) {
|
|
|
216 wxStringOutputStream out_stream( &version );
|
|
|
217 http_istream->Read( out_stream );
|
|
|
218 }
|
|
|
219
|
|
|
220 wxDELETE( http_istream );
|
|
|
221 get.Close();
|
|
|
222
|
|
|
223 return version;
|
|
|
224 }
|
|
|
225
|
|
6
|
226 void MyApp::RemoveFile( wxString pattern )
|
|
|
227 {
|
|
|
228 wxString file = wxFindFirstFile( pattern );
|
|
|
229 while ( !file.empty() ) {
|
|
|
230 wxRemoveFile( file );
|
|
|
231 file = wxFindNextFile();
|
|
|
232 }
|
|
|
233 }
|
|
|
234
|
