/*
Copyright (c) 2025 Miranda NG team (https://miranda-ng.org)

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation version 2
of the License.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#include "stdafx.h"

////////////////////////////////////////////////////////////////////////////////////////
// File receiving

static void __cdecl DownloadCallack(size_t iProgress, void *pParam)
{
	auto *ofd = (OFDTHREAD *)pParam;

	DBVARIANT dbv = { DBVT_DWORD };
	dbv.dVal = unsigned(iProgress);
	db_event_setJson(ofd->hDbEvent, "ft", &dbv);
}

void CTeamsProto::ReceiveFileThread(void *param)
{
	auto *ofd = (OFDTHREAD *)param;

	DB::EventInfo dbei(ofd->hDbEvent);
	if (IsOnline() && dbei && !strcmp(dbei.szModule, m_szModuleName) && dbei.eventType == EVENTTYPE_FILE) {
		DB::FILE_BLOB blob(dbei);

		if (ofd->bCopy) {
			ofd->wszPath = Utf2T(blob.getUrl()).get();
			ofd->pCallback->Invoke(*ofd);
		}
		else {
			CMStringA szCookie, szUrl;
			szCookie.AppendFormat("skypetoken_asm=%s", m_szSkypeToken.c_str());

			auto &json = dbei.getJson();
			auto skft = json["skft"].as_string();
			{
				const char *preview;
				if (skft == "Picture.1")
					preview = "imgpsh_mobile_save_anim";
				else if (skft == "Video.1")
					preview = "video";
				else
					preview = "original";

				MHttpRequest nlhr(REQUEST_GET);
				nlhr.flags = NLHRF_HTTP11 | NLHRF_NOUSERAGENT;
				nlhr.m_szUrl = blob.getUrl();
				nlhr.m_szUrl.AppendFormat("/views/%s/status", preview);
				nlhr.AddHeader("Accept", "*/*");
				nlhr.AddHeader("Accept-Encoding", "gzip, deflate");
				nlhr.AddHeader("Cookie", szCookie);
				NLHR_PTR response(Netlib_HttpTransaction(m_hNetlibUser, &nlhr));
				if (response) {
					TeamsReply reply(response);
					if (!reply.error()) {
						auto &root = reply.data();
						if (root["content_state"].as_string() == "ready")
							szUrl = root["view_location"].as_string().c_str();
					}
				}
			}

			if (!szUrl.IsEmpty()) {
				MHttpRequest nlhr(REQUEST_GET);
				nlhr.flags = NLHRF_HTTP11 | NLHRF_NOUSERAGENT;
				nlhr.m_szUrl = blob.getUrl();
				if (skft == "Picture.1")
					nlhr.m_szUrl += "/views/imgpsh_fullsize_anim";
				else if (skft == "Video.1")
					nlhr.m_szUrl += "/views/video";
				else
					nlhr.m_szUrl += "/views/original";

				nlhr.AddHeader("Accept", "*/*");
				nlhr.AddHeader("Accept-Encoding", "gzip, deflate");
				nlhr.AddHeader("Cookie", szCookie);

				NLHR_PTR reply(Netlib_DownloadFile(m_hNetlibUser, &nlhr, ofd->wszPath, DownloadCallack, ofd));
				if (reply && reply->resultCode == 200) {
					struct _stat st;
					_wstat(ofd->wszPath, &st);

					DBVARIANT dbv = { DBVT_DWORD };
					dbv.dVal = st.st_size;
					db_event_setJson(ofd->hDbEvent, "ft", &dbv);

					ofd->Finish();
				}
			}
		}
	}

	delete ofd;
}

INT_PTR CTeamsProto::SvcOfflineFile(WPARAM param, LPARAM)
{
	ForkThread(&CTeamsProto::ReceiveFileThread, (void *)param);
	return 0;
}

////////////////////////////////////////////////////////////////////////////////////////
// File sending

#define FILETRANSFER_FAILED(fup) { ProtoBroadcastAck(fup->hContact, ACKTYPE_FILE, ACKRESULT_FAILED, (HANDLE)fup); delete fup; fup = nullptr;} 

void CTeamsProto::SendFile(CFileUploadParam *fup)
{
	auto *pwszFileName = &fup->arFileName[0];
	if (!IsOnline() || _waccess(pwszFileName, 0)) {
		FILETRANSFER_FAILED(fup);
		return;
	}

	if (auto *pBitmap = FreeImage_LoadU(FreeImage_GetFIFFromFilenameU(pwszFileName), pwszFileName)) {
		fup->isPicture = true;
		fup->width = FreeImage_GetWidth(pBitmap);
		fup->height = FreeImage_GetHeight(pBitmap);
		FreeImage_Unload(pBitmap);
	}
	else fup->isPicture = false;

	ProtoBroadcastAck(fup->hContact, ACKTYPE_FILE, ACKRESULT_CONNECTING, (HANDLE)fup);

	// create upload slot
	auto *pReq = new AsyncHttpRequest(REQUEST_POST, HOST_OTHER, "https://api.asm.skype.com/v1/objects", &CTeamsProto::OnASMObjectCreated);
	pReq->flags &= (~NLHRF_DUMPASTEXT);
	pReq->pUserInfo = fup;

	pReq->AddHeader("Authorization", CMStringA(FORMAT, "skype_token %s", m_szSkypeToken.c_str()));
	pReq->AddHeader("Content-Type", "application/json");
	pReq->AddHeader("X-Client-Version", "0/0.0.0.0");

	CMStringA szContact(getId(fup->hContact));
	T2Utf uszFileName(&fup->arFileName[0]);
	const char *szFileName = strrchr(uszFileName.get() + 1, '\\');

	JSONNode node;
	if (fup->isPicture)
		node << CHAR_PARAM("type", "pish/image");
	else
		node << CHAR_PARAM("type", "sharing/file");

	JSONNode jPermission(JSON_ARRAY); jPermission.set_name(szContact.c_str()); jPermission << CHAR_PARAM("", "read");
	JSONNode jPermissions; jPermissions.set_name("permissions"); jPermissions << jPermission;
	node << CHAR_PARAM("filename", szFileName) << jPermissions;
	pReq->m_szParam = node.write().c_str();
	PushRequest(pReq);
}

void CTeamsProto::OnASMObjectCreated(MHttpResponse *response, AsyncHttpRequest *pRequest)
{
	auto *fup = (CFileUploadParam*)pRequest->pUserInfo;
	if (response == nullptr || response->body.IsEmpty()) {
LBL_Error:
		FILETRANSFER_FAILED(fup);
		return;
	}

	if (response->resultCode != 200 && response->resultCode != 201) {
		debugLogA("Object creation failed with error code %d", response->resultCode);
		goto LBL_Error;
	}

	JSONNode node = JSONNode::parse(response->body);
	std::string strObjectId = node["id"].as_string();
	if (strObjectId.empty()) {
		debugLogA("Invalid server response (empty object id)");
		goto LBL_Error;
	}
	
	fup->uid = mir_strdup(strObjectId.c_str());
	FILE *pFile = _wfopen(&fup->arFileName[0], L"rb");
	if (pFile == nullptr)
		goto LBL_Error;

	fseek(pFile, 0, SEEK_END);
	long lFileLen = ftell(pFile);
	if (lFileLen < 1) {
		fclose(pFile);
		goto LBL_Error;
	}

	fseek(pFile, 0, SEEK_SET);

	mir_ptr<uint8_t> pData((uint8_t*)mir_alloc(lFileLen));
	long lBytes = (long)fread(pData, sizeof(uint8_t), lFileLen, pFile);
	fclose(pFile);

	if (lBytes != lFileLen)
		goto LBL_Error;

	fup->size = lBytes;
	ProtoBroadcastAck(fup->hContact, ACKTYPE_FILE, ACKRESULT_INITIALISING, (HANDLE)fup);

	// upload file to the previously created slot
	auto *pReq = new AsyncHttpRequest(REQUEST_PUT, HOST_OTHER, 0, &CTeamsProto::OnASMObjectUploaded);
	pReq->m_szUrl.Format("https://api.asm.skype.com/v1/objects/%s/content/%s",
		strObjectId.c_str(), fup->isPicture ? "imgpsh" : "original");
	pReq->pUserInfo = fup;

	pReq->AddHeader("Authorization", CMStringA(FORMAT, "skype_token %s", m_szSkypeToken.c_str()));
	pReq->AddHeader("Content-Type", fup->isPicture ? "application" : "application/octet-stream");

	pReq->m_szParam.Truncate(lBytes);
	memcpy(pReq->m_szParam.GetBuffer(), pData, lBytes);
	PushRequest(pReq);
}

void CTeamsProto::OnASMObjectUploaded(MHttpResponse *response, AsyncHttpRequest *pRequest)
{
	auto *fup = (CFileUploadParam *)pRequest->pUserInfo;
	if (response == nullptr) {
		FILETRANSFER_FAILED(fup);
		return;
	}

	wchar_t *tszFile = wcsrchr(&fup->arFileName[0], L'\\') + 1;

	TiXmlDocument doc;
	auto *pRoot = doc.NewElement("URIObject");
	doc.InsertEndChild(pRoot);

	pRoot->SetAttribute("doc_id", fup->uid.get());
	pRoot->SetAttribute("uri", CMStringA(FORMAT, "https://api.asm.skype.com/v1/objects/%s", fup->uid.get()));

	// is that a picture?
	CMStringA href;
	if (fup->isPicture) {
		pRoot->SetAttribute("type", "Picture.1");
		pRoot->SetAttribute("url_thumbnail", CMStringA(FORMAT, "https://api.asm.skype.com/v1/objects/%s/views/imgt1_anim", fup->uid.get()));
		pRoot->SetAttribute("width", fup->width);
		pRoot->SetAttribute("height", fup->height);
		pRoot->SetText("To view this shared photo, go to:");

		href.Format("https://login.skype.com/login/sso?go=xmmfallback?pic=%s", fup->uid.get());
	}
	else {
		pRoot->SetAttribute("type", "File.1");
		pRoot->SetAttribute("url_thumbnail", CMStringA(FORMAT, "https://api.asm.skype.com/v1/objects/%s/views/original", fup->uid.get()));
		pRoot->SetText("To view this file, go to:");

		href.Format("https://login.skype.com/login/sso?go=webclient.xmm&docid=%s", fup->uid.get());
	}

	auto *xmlA = doc.NewElement("a"); xmlA->SetText(href);
	xmlA->SetAttribute("href", href);
	pRoot->InsertEndChild(xmlA);

	auto *xmlOrigName = doc.NewElement("OriginalName"); xmlOrigName->SetAttribute("v", tszFile); pRoot->InsertEndChild(xmlOrigName);
	auto *xmlSize = doc.NewElement("FileSize"); xmlSize->SetAttribute("v", (int)fup->size); pRoot->InsertEndChild(xmlSize);

	if (fup->isPicture) {
		auto xmlMeta = doc.NewElement("meta");
		xmlMeta->SetAttribute("type", "photo"); xmlMeta->SetAttribute("originalName", tszFile);
		pRoot->InsertEndChild(xmlMeta);
	}

	tinyxml2::XMLPrinter printer(0, true);
	doc.Print(&printer);

	// create a new file transfer event using previously filled slot
	auto *pReq = new AsyncHttpRequest(REQUEST_POST, HOST_CHATS);
	pReq->m_szUrl.AppendFormat("/users/ME/conversations/%s/messages", mir_urlEncode(getId(fup->hContact)).c_str());
	pReq->hContact = fup->hContact;

	JSONNode ref(JSON_ARRAY); ref.set_name("amsreferences"); ref << CHAR_PARAM("", fup->uid);

	JSONNode node;
	if (fup->isPicture)
		node << CHAR_PARAM("messagetype", "RichText/UriObject");
	else
		node << CHAR_PARAM("messagetype", "RichText/Media_GenericFile");

	node << INT64_PARAM("clientmessageid", getRandomId()) << CHAR_PARAM("contenttype", "text") << CHAR_PARAM("content", printer.CStr()) << ref;
	pReq->m_szParam = node.write().c_str();

	PushRequest(pReq);

	// if that's last file in the queue, finish file transfer, or proceed with the next file
	if (fup->arFileName.getCount() == 1) {
		ProtoBroadcastAck(fup->hContact, ACKTYPE_FILE, ACKRESULT_SUCCESS, (HANDLE)fup);
		delete fup;
	}
	else {
		fup->arFileName.remove(int(0));
		ProtoBroadcastAck(fup->hContact, ACKTYPE_FILE, ACKRESULT_NEXTFILE, (HANDLE)fup);
		SendFile(fup);
	}
}

HANDLE CTeamsProto::SendFile(MCONTACT hContact, const wchar_t *szDescription, wchar_t **ppszFiles)
{
	if (!IsOnline())
		return INVALID_HANDLE_VALUE;

	CFileUploadParam *fup = new CFileUploadParam(hContact, ppszFiles, szDescription);
	SendFile(fup);
	return fup;
}