polychora/ScreenSaverModeHandler.cpp at master · shorstok/polychora · 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
#include "ScreenSaverModeHandler.h"

#include <iostream>
#include <string>
#include <sstream>
#include <algorithm>
#include <iterator>

#include <vector>


CScreenSaverModeHandler::CScreenSaverModeHandler(void) : isValid(false),parentWindowHandle(nullptr)
{
}

CScreenSaverModeHandler::CScreenSaverModeHandler( Mode _mode, std::wstring token ) : mode(_mode), isValid(true), parentWindowHandle(nullptr)
{
	auto idx = token.find(L":");

	if(token.length()<1)
		parentWindowHandle = nullptr;
	else if (std::wstring::npos == idx)
		parentWindowHandle = GetForegroundWindow();
	else
	{
		auto handleString = token.substr(idx+1,std::wstring::npos);
		parentWindowHandle = (HWND)_wtoi(handleString.c_str());
	}
}


CScreenSaverModeHandler::~CScreenSaverModeHandler(void)
{
}

CScreenSaverModeHandler * CScreenSaverModeHandler::FromCommandLine( LPWSTR cmdLine )
{
	std::vector<std::wstring> elems;

	std::wstringstream ss(cmdLine);
	std::wstring item;
	while (std::getline(ss, item, L' '))
		elems.push_back(item);

	if(elems.empty() || elems[0].length() < 2)
		return InvalidMode();

	CScreenSaverModeHandler * ret = new CScreenSaverModeHandler();

	switch(elems[0][1])
	{
		case L'p':	case L'P':
		case L'l':	case L'L':
			return new CScreenSaverModeHandler(Preview,elems.size()>1?elems[0]+L":"+elems[1]:elems[0]);

		case L's':	case L'S':
			return new CScreenSaverModeHandler(Screensaver,L"");

		case L'c':	case L'C':
			return new CScreenSaverModeHandler(Configure,elems[0]);

		case L'a':	case L'A':
			return new CScreenSaverModeHandler(Password,elems[0]);
	}

	return InvalidMode();
}

bool CScreenSaverModeHandler::IsValid()
{
	return isValid;
}

CScreenSaverModeHandler * CScreenSaverModeHandler::InvalidMode()
{
	return new CScreenSaverModeHandler();
}

CScreenSaverModeHandler::Mode CScreenSaverModeHandler::CurrentMode()
{
	return mode;
}