source: trunk/src/Halite.cpp @ 384

Revision 384, 5.1 KB checked in by Eoin, 12 years ago (diff)

Added 'Open Download Folder'.

Line 
1
2//         Copyright Eóin O'Callaghan 2006 - 2008.
3// Distributed under the Boost Software License, Version 1.0.
4//    (See accompanying file LICENSE_1_0.txt or copy at
5//          http://www.boost.org/LICENSE_1_0.txt)
6
7#define HALITE_GUID L"HALITE-{E3A8BF7D-962F-476E-886B-FECEDD2F0FC7}"
8#define WMU_ARE_YOU_ME_STRING  L"WMU_ARE_YOU_ME_HALITE-{E3A8BF7D-962F-476E-886B-FECEDD2F0FC7}"
9#pragma comment(linker, "\"/manifestdependency:type='Win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='*' publicKeyToken='6595b64144ccf1df' language='*'\"")
10
11#include "stdAfx.hpp"
12
13#include "WinAPIMutex.hpp"
14
15#include "global/ini.hpp"
16#include "global/logger.hpp"
17#include "halConfig.hpp"
18
19#include "HaliteWindow.hpp"
20#include "SplashDialog.hpp"
21
22#include "Halite.hpp"
23
24Halite& halite()
25{
26        static Halite h;
27        return h;
28}
29
30namespace fs = boost::filesystem;
31using fs::ifstream;
32using fs::ofstream;
33       
34static class halite_log_file : public boost::signals::trackable
35{
36public: 
37        void operator()(shared_ptr<hal::EventDetail> event)
38        {
39                if (halite().logToFile())
40                {
41                        if (!wofs.is_open()) wofs.open(hal::app().working_directory()/L"HaliteLog.txt");
42                       
43                        wofs << (wformat(L"%1% %2%, %3%\r\n") 
44                                % event->timeStamp() % hal::Event::eventLevelToStr(event->level()) 
45                                % event->msg()).str();
46                       
47                        wofs.flush();
48                }
49        }
50       
51        void connect() 
52        { 
53                conn_ = hal::event().attach(bind(&halite_log_file::operator(), this, _1)); 
54                assert(conn_.connected());
55        }
56       
57        void disconnect() { conn_.disconnect(); }
58       
59private:
60        fs::wofstream wofs;
61        boost::signals::connection conn_;
62       
63} halite_log_file_;
64
65static const unsigned WMU_ARE_YOU_ME = ::RegisterWindowMessage(WMU_ARE_YOU_ME_STRING);
66
67static BOOL CALLBACK hwndSearcher(HWND hWnd, LPARAM lParam)
68{
69        ULONG_PTR result;
70        LRESULT ok = ::SendMessageTimeout(hWnd, WMU_ARE_YOU_ME,
71                0, 0, SMTO_BLOCK | SMTO_ABORTIFHUNG, 200, &result);
72       
73        if (ok == 0)
74                return false;
75       
76        if (result == WMU_ARE_YOU_ME)
77        {
78                HWND* target = (HWND*)lParam;
79                *target = hWnd;
80                return false;
81        }
82       
83        return true;
84}
85
86int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
87{
88
89//      HAL_DEV_MSG(L"Hello");
90//      hal::event().post(shared_ptr<hal::EventDetail>(new hal::EventDebug(hal::Event::info, L"Hello")));
91
92        ::SetProcessAffinityMask(::GetCurrentProcess(), (DWORD_PTR)0x1);
93       
94        try 
95        {
96       
97        boost::filesystem::path::default_name_check(boost::filesystem::native);
98
99        AtlInitCommonControls(ICC_COOL_CLASSES | ICC_BAR_CLASSES);     
100        HINSTANCE hInstRich = ::LoadLibrary(CRichEditCtrl::GetLibraryName());
101        ATLASSERT(hInstRich != NULL);
102   
103        int nRet = 0;   
104        HRESULT hRes = _Module.Init(NULL, hInstance);
105        assert (SUCCEEDED(hRes));       
106       
107        hal::ini().load_data();
108        halite_log_file_.connect();
109       
110        { WinAPIMutex oneInstance(HALITE_GUID);
111       
112        if (!oneInstance.owner() && halite().oneInst)
113        {
114                WinAPIMutexLock lock(oneInstance, 5000L);
115               
116                HWND hOther = NULL;
117                ::EnumWindows(static_cast<WNDENUMPROC>(hwndSearcher), (LPARAM)&hOther);
118               
119                if (hOther != NULL)
120                {
121                        ::SetForegroundWindow(hOther);
122                       
123                        if (::IsIconic(hOther))
124                                ::ShowWindow(hOther, SW_RESTORE);
125                       
126                        if (!hal::app().command_args().empty())
127                        {
128                                COPYDATASTRUCT cmdLine; 
129                                cmdLine.dwData = HALITE_SENDING_CMD; 
130                                cmdLine.cbData = 
131                                        hal::app().command_args().front().length()*sizeof(wchar_t); 
132                                cmdLine.lpData = const_cast<wchar_t*>(
133                                        hal::app().command_args().front().c_str());
134                               
135                                ::SendMessage(hOther, WM_COPYDATA, 0, (LPARAM)(LPVOID)&cmdLine);
136                        }
137                }                               
138        }
139        else
140        {
141//              INI().LoadData();       
142                hal::event().post(shared_ptr<hal::EventDetail>(
143                        new hal::EventMsg(wformat(L"Exe Path: %1%.") % hal::app().exe_path())));               
144               
145                hal::event().post(shared_ptr<hal::EventDetail>(
146                        new hal::EventMsg(wformat(L"Initial Path: %1%.") % hal::app().initial_path())));               
147               
148                hal::event().post(shared_ptr<hal::EventDetail>(
149                        new hal::EventMsg((wformat(L"Working Directory: %1%.") % hal::app().working_directory()), hal::Event::info)));         
150               
151                CMessageLoop theLoop;
152                _Module.AddMessageLoop(&theLoop);
153                       
154                        if (halite().dll() != L"") hal::app().res_set_dll(halite().dll());
155                       
156                        HaliteWindow wndMain(WMU_ARE_YOU_ME);
157                        if (wndMain.CreateEx() == NULL)
158                                return 1;
159                       
160                        oneInstance.release();
161                       
162                        if (!hal::app().command_args().empty())
163                                wndMain.ProcessFile(hal::app().command_args().front().c_str());
164                       
165                        wndMain.SetIcon(LoadIcon(hInstance, MAKEINTRESOURCE(IDR_APP_ICON)), false);     
166                        wndMain.ShowWindow(nCmdShow);
167                       
168                        nRet = theLoop.Run();                           
169               
170                _Module.RemoveMessageLoop();
171               
172                hal::bittorrent().stopEventReceiver();
173               
174                if (halite().showMessage)
175                {
176                        SplashDialog splDlg;
177                        splDlg.DoModal();
178                }
179                else
180                {
181                        hal::bittorrent().closeAll();
182                        hal::bittorrent().shutDownSession();           
183                }
184               
185//              INI().SaveData();
186               
187                halite().save();
188                hal::ini().save_data();
189        }
190        }
191
192        ::FreeLibrary(hInstRich);       
193        _Module.Term();
194       
195        return nRet;
196       
197        }
198        catch (const std::exception& e)
199        {
200        MessageBoxA(0, e.what(), "Exception Thrown!", 0);
201       
202        return -1;
203        }
204       
205}
Note: See TracBrowser for help on using the repository browser.