proxy2ch/BBS2chProxyHttpHeaders.cpp

#include <algorithm>
#include "BBS2chProxyHttpHeaders.h"

std::string& BBS2chProxyHttpHeaderEntry::getName(void)
{
	return _name;
}

std::string BBS2chProxyHttpHeaderEntry::getValue(void)
{
	std::string ret;
	for (std::vector<std::string>::iterator it = _values.begin(); it != _values.end(); it++) {
		if (!ret.empty()) ret += ", ";
		ret += *it;
 	}
 	return ret;
}

std::string BBS2chProxyHttpHeaderEntry::getFull(void)
{
	std::string ret = _name;
	ret += ": ";
	ret += getValue();
 	return ret;
}

std::vector<std::string>& BBS2chProxyHttpHeaderEntry::getValueList(void)
{
	return _values;
}

void BBS2chProxyHttpHeaderEntry::add(const std::string &value)
{
	_values.push_back(value);
}

void BBS2chProxyHttpHeaderEntry::set(const std::string &value)
{
	_values.clear();
	_values.push_back(value);
}

std::string BBS2chProxyHttpHeaders::get(const std::string &name)
{
	if (!has(name)) return "";
	std::string lowerName = name;
	std::transform(name.begin(), name.end(), lowerName.begin(), tolower);
	return _headers[lowerName]->getValue();
}

bool BBS2chProxyHttpHeaders::has(const std::string &name)
{
	if (_headers.empty()) return false;
	std::string lowerName = name;
	std::transform(name.begin(), name.end(), lowerName.begin(), tolower);
	return _headers.count(lowerName) != 0;
}

void BBS2chProxyHttpHeaders::add(const std::string &name, const std::string &value)
{
	if (name.empty()) return;
	if (!has(name)) return set(name, value);
	std::string lowerName = name;
	std::transform(name.begin(), name.end(), lowerName.begin(), tolower);
	PBBS2chProxyHttpHeaderEntry entry = _headers[lowerName];
	entry->add(value);
}

void BBS2chProxyHttpHeaders::add(const char *field)
{
	const char *ptr = field;
	while (*ptr == ' ' || *ptr == '\t') ptr++;
	const char *start = ptr;
	while (*ptr != ':' && *ptr != 0) ptr++;
	if (*ptr != ':' || ptr == start) return;
	const char *next = ptr + 1;
	ptr--;
	while (*ptr == ' ' && ptr > start) ptr--;
	std::string name(start, ptr-start+1);
	ptr = next;
	while (*ptr == ' ' || *ptr == '\t') ptr++;
	start = ptr;
	while (*ptr != '\r' && *ptr != '\n' && *ptr != 0) ptr++;
	std::string value(start, ptr-start);
	std::string lowerName = name;
	add(name, value);
}

void BBS2chProxyHttpHeaders::set(const std::string &name, const std::string &value)
{
	if (name.empty()) return;
	std::string lowerName = name;
	std::transform(name.begin(), name.end(), lowerName.begin(), tolower);
	_headers.erase(lowerName);
	PBBS2chProxyHttpHeaderEntry entry(new BBS2chProxyHttpHeaderEntry(name, value));
	_headers.insert(std::make_pair(lowerName, entry));
}

void BBS2chProxyHttpHeaders::remove(const std::string &name)
{
	if (name.empty()) return;
	std::string lowerName = name;
	std::transform(name.begin(), name.end(), lowerName.begin(), tolower);
	_headers.erase(lowerName);
}

void BBS2chProxyHttpHeaders::clear(void)
{
	_headers.clear();
}

curl_slist* BBS2chProxyHttpHeaders::appendToCurlSlist(curl_slist *list)
{
	for (std::map<std::string, PBBS2chProxyHttpHeaderEntry>::iterator it = _headers.begin(); it != _headers.end(); it++) {
		std::vector<std::string>& values = it->second->getValueList();
		for (std::vector<std::string>::iterator it2 = values.begin(); it2 != values.end(); it2++) {
			std::string header = it->second->getName();
			header += ": ";
			header += *it2;
			list = curl_slist_append(list, header.c_str());
		}
	}
	return list;
}

curl_slist* BBS2chProxyHttpHeaders::appendToCurlSlist(curl_slist *list, const std::string &name)
{
	if (name.empty()) return list;
	std::string lowerName = name;
	std::transform(name.begin(), name.end(), lowerName.begin(), tolower);
	if (_headers.count(lowerName)) {
		PBBS2chProxyHttpHeaderEntry& entry = _headers[lowerName];
		std::vector<std::string>& values = entry->getValueList();
		for (std::vector<std::string>::iterator it = values.begin(); it != values.end(); it++) {
			std::string header = entry->getName();
			header += ": ";
			header += *it;
			list = curl_slist_append(list, header.c_str());
		}
	}
	return list;
}

std::map<std::string, PBBS2chProxyHttpHeaderEntry>& BBS2chProxyHttpHeaders::getMap(void)
{
	return _headers;
}

#ifdef USE_LUA
static bool lua_isValidUserdata(lua_State *l, int idx)
{
	if (!lua_getmetatable(l, idx)) {
		return false;
	}
	if (lua_getfield(l, -1, "_type") != LUA_TSTRING) {
		lua_pop(l, 2);
		return false;
	}
	bool isValid = !strcmp(lua_tostring(l, -1), "HttpHeaders");
	lua_pop(l, 2);
	return isValid;
}

static int lua_httpHeadersGet(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	const char *name = luaL_checkstring(l, 2);
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		lua_pushnil(l);
		return 1;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers && headers->has(name)) {
		lua_pushstring(l, headers->get(name).c_str());
	}
	else {
		lua_pushnil(l);
	}
	return 1;
}

static int lua_httpHeadersSet(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	const char *name = luaL_checkstring(l, 2);
	const char *value = NULL;
	if (!lua_isnoneornil(l, 3)) {
		value = luaL_checkstring(l, 3);
	}
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		return 0;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers) {
		if (value) headers->set(name, value);
		else headers->remove(name);
	}
	return 0;
}

static int lua_httpHeadersAdd(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	const char *name = luaL_checkstring(l, 2);
	const char *value = NULL;
	if (!lua_isnoneornil(l, 3)) {
		value = luaL_checkstring(l, 3);
	}
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		return 0;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers && value) {
		headers->add(name, value);
	}
	return 0;
}

static int lua_httpHeadersHas(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	const char *name = luaL_checkstring(l, 2);
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		lua_pushboolean(l, 0);
		return 1;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers) {
		lua_pushboolean(l, headers->has(name));
	}
	else {
		lua_pushboolean(l, 0);
	}
	return 1;
}

static int lua_httpHeadersRemove(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	const char *name = luaL_checkstring(l, 2);
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		return 0;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers) {
		headers->remove(name);
	}
	return 0;
}

static int lua_httpHeadersClear(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		return 0;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers) {
		headers->clear();
	}
	return 0;
}

static int lua_httpHeadersCount(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		lua_pushinteger(l, 0);
		return 1;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers) {
		lua_pushinteger(l, headers->getMap().size());
	}
	else {
		lua_pushinteger(l, 0);
	}
	return 1;
}

static int lua_httpHeadersNext(lua_State *l)
{
	BBS2chProxyHttpHeaders *headers = NULL;
	const char *index = NULL;
	if (!lua_isnoneornil(l, 2)) {
		index = luaL_checkstring(l, 2);
	}
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		goto fail;
	}
	if (lua_isValidUserdata(l, 1)) {
		headers = *((BBS2chProxyHttpHeaders **)obj);
	}
	if (headers) {
		if (headers->getMap().empty()) {
			goto fail;
		}
		std::map<std::string, PBBS2chProxyHttpHeaderEntry>::iterator it = headers->getMap().begin();
		if (!index) {
			lua_pushstring(l, it->second->getName().c_str());
			lua_pushstring(l, it->second->getValue().c_str());
			return 2;
		}
		for (; it != headers->getMap().end(); it++) {
			if (it->second->getName() == index) {
				break;
			}
		}
		if (++it != headers->getMap().end()) {
			lua_pushstring(l, it->second->getName().c_str());
			lua_pushstring(l, it->second->getValue().c_str());
			return 2;
		}
	}
fail:
	lua_pushnil(l);
	return 1;
}

static int lua_httpHeadersPairs(lua_State *l)
{
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		return 0;
	}
	if (lua_isValidUserdata(l, 1)) {
		lua_pushcfunction(l, lua_httpHeadersNext);
		lua_pushvalue(l, 1);
		lua_pushnil(l);
		return 3;
	}
	return 0;
}

static int lua_httpHeadersNew(lua_State *l)
{
	void **obj = (void **)lua_newuserdata(l, sizeof(void *));
	*obj = new BBS2chProxyHttpHeaders();
	lua_pushvalue(l, lua_upvalueindex(1));
	lua_setmetatable(l, -2);
	return 1;
}

static int lua_httpHeadersDelete(lua_State *l)
{
	void *obj = lua_touserdata(l, 1);
	if (!obj) {
		return 0;
	}
	if (lua_isValidUserdata(l, 1)) {
		delete *((BBS2chProxyHttpHeaders **)obj);
	}
	return 0;
}

static int lua_functionOrGetter(lua_State *l)
{
	if (!lua_getmetatable(l, 1)) {
		lua_pushnil(l);
		return 1;
	}
	lua_pushvalue(l, 2);
	if (lua_rawget(l, -2) == LUA_TFUNCTION) {
		return 1;
	}
	lua_pop(l, 2);
	return lua_httpHeadersGet(l);
}

static int lua_enableLowLevelFuncs(lua_State *l)
{
	if (!lua_isuserdata(l, 1) || !lua_isValidUserdata(l, 1)) {
		return 0;
	}
	if (!lua_getmetatable(l, 1)) {
		return 0;
	}
	lua_newtable(l);
	lua_pushnil(l);
	while (lua_next(l, -3)) {
		lua_pushvalue(l, -2);
		lua_insert(l, -2);
		lua_settable(l, -4);
	}
	lua_pushcfunction(l, lua_functionOrGetter);
	lua_setfield(l, -2, "__index");
	lua_setmetatable(l, 1);
	lua_pop(l, 1);
	return 0;
}

void BBS2chProxyHttpHeaders::getObjectMetatableForLua(lua_State *l)
{
	lua_newtable(l);
	lua_pushcfunction(l, lua_httpHeadersGet);
	lua_setfield(l, -2, "__index");
	lua_pushcfunction(l, lua_httpHeadersSet);
	lua_setfield(l, -2, "__newindex");
	lua_pushcfunction(l, lua_httpHeadersPairs);
	lua_setfield(l, -2, "__pairs");
	lua_pushcfunction(l, lua_httpHeadersCount);
	lua_setfield(l, -2, "__len");
	lua_pushcfunction(l, lua_httpHeadersGet);
	lua_setfield(l, -2, "get");
	lua_pushcfunction(l, lua_httpHeadersSet);
	lua_setfield(l, -2, "set");
	lua_pushcfunction(l, lua_httpHeadersAdd);
	lua_setfield(l, -2, "add");
	lua_pushcfunction(l, lua_httpHeadersHas);
	lua_setfield(l, -2, "has");
	lua_pushcfunction(l, lua_httpHeadersRemove);
	lua_setfield(l, -2, "remove");
	lua_pushcfunction(l, lua_httpHeadersClear);
	lua_setfield(l, -2, "clear");
	lua_pushcfunction(l, lua_httpHeadersCount);
	lua_setfield(l, -2, "count");
	lua_pushliteral(l, "HttpHeaders");
	lua_setfield(l, -2, "_type");
}

void BBS2chProxyHttpHeaders::getClassDefinitionForLua(lua_State *l)
{
	lua_newtable(l);
	getObjectMetatableForLua(l);
	lua_pushcfunction(l, lua_httpHeadersDelete);
	lua_setfield(l, -2, "__gc");
	lua_pushcclosure(l, lua_httpHeadersNew, 1);
	lua_setfield(l, -2, "new");
	lua_pushcfunction(l, lua_enableLowLevelFuncs);
	lua_setfield(l, -2, "enableLowLevelAccess");
}

void BBS2chProxyHttpHeaders::getUserdataForLua(lua_State *l)
{
	void **obj = (void **)lua_newuserdata(l, sizeof(this));
	*obj = this;
	getObjectMetatableForLua(l);
	lua_setmetatable(l, -2);
}
#endif