summaryrefslogtreecommitdiff
path: root/src/modules/m_remoteinclude.cpp
blob: 7e6ba8df338c12b2efe30fcc2f80aedbb51c83c5 (plain)
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
103
104
105
106
107
108
/*       +------------------------------------+
 *       | Inspire Internet Relay Chat Daemon |
 *       +------------------------------------+
 *
 *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
 * See: http://www.inspircd.org/wiki/index.php/Credits
 *
 * This program is free but copyrighted software; see
 *            the file COPYING for details.
 *
 * ---------------------------------------------------
 */

#include "inspircd.h"
#include "httpclient.h"

/* $ModDesc: The base module for remote includes */

class ModuleRemoteInclude : public Module
{
	std::map<std::string, std::stringstream*> assoc;

 public:
	ModuleRemoteInclude(InspIRCd* Me)
		: Module(Me)
	{
		ServerInstance->Modules->Attach(I_OnDownloadFile, this);
		ServerInstance->Modules->Attach(I_OnRequest, this);
	}
	
	virtual ~ModuleRemoteInclude()
	{
	}
	
	virtual Version GetVersion()
	{
		// this method instantiates a class of type Version, and returns
		// the modules version information using it.
	
		return Version(1,1,0,1,VF_VENDOR,API_VERSION);
	}

	char* OnRequest(Request* req)
	{
		if (!strcmp(req->GetId(), HTTP_CLIENT_RESPONSE))
		{
			HTTPClientResponse* resp = (HTTPClientResponse*)req;
			ServerInstance->Log(DEBUG, "Got http file for %s", resp->GetURL().c_str());

			std::map<std::string, std::stringstream*>::iterator n = assoc.find(resp->GetURL());

			if (n == assoc.end())
				ServerInstance->Config->Complete(resp->GetURL(), true);
			
			*(n->second) << resp->GetData();

			ServerInstance->Log(DEBUG, "Got data: %s", resp->GetData().c_str());

			ServerInstance->Log(DEBUG, "Flag file complete without error");
			ServerInstance->Config->Complete(resp->GetURL(), false);

			/* Erase from our association map, but dont delete the pointer.
			 * the core will want to access this pointer for the file data.
			 */
			assoc.erase(n);
		}
		else if (!strcmp(req->GetId(), HTTP_CLIENT_ERROR))
		{
			HTTPClientError* resp = (HTTPClientError*)req;

			ServerInstance->Log(DEBUG, "Got http error when accessing %s", resp->GetURL().c_str());
			ServerInstance->Config->Complete(resp->GetURL(), true);

			std::map<std::string, std::stringstream*>::iterator n = assoc.find(resp->GetURL());

			if (n != assoc.end())
				assoc.erase(n);
		}
		return NULL;
	}

	int OnDownloadFile(const std::string &name, std::istream* &filedata)
	{
		if (name.substr(0, 7) == "http://")
		{
			Module* target = ServerInstance->Modules->Find("m_http_client.so");
			if (target)
			{
				ServerInstance->Log(DEBUG,"Claiming schema http://, making fetch request");

				HTTPClientRequest req(ServerInstance, this, target, name);
				req.Send();

				assoc[name] = new std::stringstream();
				delete filedata;
				filedata = assoc[name];

				return true;
			}
		}

		return false;
	}
};


MODULE_INIT(ModuleRemoteInclude)