Subversion Repositories Kolibri OS

#define MEMSIZE 1024 * 40

//Copyright 2020 - 2021 by Leency

#include "../lib/gui.h"

#include "../lib/random.h"

#include "../lib/obj/box_lib.h"

#include "../lib/obj/http.h"

#include "const.h"

bool exit_param = false;

bool open_file = false;

dword speed;

_http http;

checkbox autoclose = { T_AUTOCLOSE, false };

char uEdit[URL_SIZE];

char filepath[URL_SIZE+96];

progress_bar pb = {0, GAPX, 58, 380, 17, 0, NULL, NULL, 0xFFFfff, 0x74DA00, NULL};

edit_box ed = {WIN_W-GAPX-GAPX,GAPX,20,0xffffff,0x94AECE,0xffffff,0xffffff,

	0x10000000, sizeof(uEdit)-2,#uEdit,0,ed_focus,19,19};

void main()

{

	dword shared_url;

	load_dll(boxlib,  #box_lib_init,0);

	load_dll(libHTTP, #http_lib_init,1);

	if (!dir_exists(#save_dir)) CreateDir(#save_dir);

	SetCurDir(#save_dir);

	if (param) {

		if (streqrp(#param, "-e ")) {

			exit_param = true;

			param += 3;

		}

		if (streqrp(#param, "-eo ")) {

			exit_param = true;

			open_file = true;

			param += 4;

		}

		if (!strncmp(#param, "-mem", 5)) {

			//shared_url = memopen(#dl_shared, URL_SIZE+1, SHM_OPEN + SHM_WRITE);

			strcpy(#uEdit, shared_url);

		} else {

			strcpy(#uEdit, #param);

		}

		if (streq(#param, "-test")) {

			strcpy(#uEdit, URL_SPEED_TEST);

		}

	}

	if (uEdit[0]) StartDownloading(); else {

		edit_box_set_text stdcall (#ed, "http://");

	}

	@SetEventMask(EVM_REDRAW + EVM_KEY + EVM_BUTTON + EVM_MOUSE + EVM_MOUSE_FILTER + EVM_STACK);

	loop() switch(@WaitEvent())

	{

		case evMouse:  edit_box_mouse stdcall (#ed); break;

		case evButton: ProcessButtonClick(@GetButtonID()); break;

		case evKey:    ProcessKeyPress(); break;

		case evReDraw: DrawWindow(); break;

		default:       MonitorProgress();

	}

}

void ProcessButtonClick(int id)

{

	autoclose.click(id);

	if (id==BTN_EXIT) { StopDownloading(); ExitProcess(); }

	if (id==BTN_START) StartDownloading();

	if (id==BTN_STOP) StopDownloading();

	if (id==BTN_DIR) { RunProgram("/sys/File managers/Eolite", #filepath); ExitProcess(); }

	if (id==BTN_RUN) { RunProgram("/sys/@open", #filepath); ExitProcess(); }

}

void ProcessKeyPress()

{

	@GetKey();

	edit_box_key stdcall(#ed);

	EAX >>= 16;

	if (AL == SCAN_CODE_ENTER) StartDownloading();

	if (AL == SCAN_CODE_ESC) StopDownloading();

}

void DrawWindow()

{

	sc.get();

	pb.frame_color = sc.work_dark;

	DefineAndDrawWindow(110 + random(300), 100 + random(300), WIN_W+9,

		WIN_H + 5 + skin_h, 0x34, sc.work, DL_WINDOW_HEADER, 0);

	#define BUT_Y 58;

	//autoclose.draw(WIN_W-135, BUT_Y+6);

	if (!http.transfer)

	{

		DrawStandartCaptButton(GAPX, BUT_Y, BTN_START, T_DOWNLOAD);

		if (filepath)

		{

			DrawStandartCaptButton(GAPX+102, BUT_Y, BTN_DIR, T_OPEN_DIR);

			DrawStandartCaptButton(GAPX+276, BUT_Y, BTN_RUN, T_RUN);

		}

	} else {

		DrawStandartCaptButton(WIN_W - 120, BUT_Y, BTN_STOP, T_CANCEL);

		DrawDownloadingProgress();

	}

	//ed.offset=0; //DEL?

	DrawEditBox(#ed);

}

void StartDownloading()

{

	char get_url[URL_SIZE+33];

	if (http.transfer > 0) return;

	ResetDownloadSpeed();

	filepath = '\0';

	if (!strncmp(#uEdit,"https:",6)) {

		miniprintf(#get_url, "http://gate.aspero.pro/?site=%s", #uEdit);

		//notify("'HTTPS for download is not supported, trying to download the file via HTTP' -W");

		//miniprintf(#http_url, "http://%s", #uEdit+8);

		//if (!downloader.Start(#http_url)) {

		//	notify("'Download failed.' -E");

		//	StopDownloading();

		//}

	} else {

		strcpy(#get_url, #uEdit);

	}

	if (http.get(#get_url)) {

		ed.blur_border_color = 0xCACACA;

		EditBox_UpdateText(#ed, ed_disabled);

		pb.value = 0;

		DrawWindow();

	} else {

		notify(T_ERROR_STARTING_DOWNLOAD);

		StopDownloading();

		if (exit_param) ExitProcess();

	}

}

void DrawDownloadingProgress()

{

	char bytes_received[70];

	dword gotkb = http.content_received/1024;

	EDI = http.content_received / 100;

	if (pb.value == EDI) return;

	pb.value = EDI;

	pb.max = http.content_length / 100;

	progressbar_draw stdcall(#pb);

	CalculateDownloadSpeed();

	sprintf(#bytes_received, KB_RECEIVED, gotkb/1024, gotkb%1024/103, speed);

	WriteTextWithBg(GAPX, pb.top + 22, 0xD0, sc.work_text, #bytes_received, sc.work);

}

void StopDownloading()

{

	http.stop();

	if (http.content_pointer) http.content_pointer = free(http.content_pointer);

	http.content_received = http.content_length = 0;

	ed.blur_border_color = 0xFFFfff;

	EditBox_UpdateText(#ed, ed_focus);

	DrawWindow();

}

void MonitorProgress()

{

	char redirect_url[URL_SIZE];

	if (http.transfer <= 0) return;

	http.receive();

	if (!http.content_length) http.content_length = http.content_received * 20; //MOVE?

	if (http.receive_result) {

		DrawDownloadingProgress();

	} else {

		if (http.status_code >= 300) && (http.status_code < 400) {

			http.header_field("location", #redirect_url, URL_SIZE);

			get_absolute_url(#redirect_url, #uEdit);

			edit_box_set_text stdcall (#ed, #redirect_url);

			StopDownloading();

			StartDownloading();

			return;

		}

		SaveFile(0);

		if (exit_param) ExitProcess();

		StopDownloading();

		DrawWindow();

	}

}

void SaveFile(int attempt)

{

	int i, fi=0;

	char notify_message[4296];

	char file_name[URL_SIZE+96];

	strcpy(#filepath, #save_dir);

	chrcat(#filepath, '/');

	if (attempt > 9) {

		notify("'Too many saving attempts' -E");

		return;

	}

	if (attempt > 0) {

		chrcat(#filepath, attempt+'0');

		chrcat(#filepath, '_');

	}

	//Content-Disposition: attachment; filename="RealFootball_2018_Nokia_5800_EN_IGP_EU_TS_101.zip"

	if (http.header_field("content-disposition", #file_name, sizeof(file_name))) {

		if (EDX = strstr(#file_name,"filename=\"")) {

			strcat(#filepath, EDX+10);

			ESBYTE[strchr(#filepath,'\"')] = '\0';

		}

	} else {

		// Clean all slashes at the end

		strcpy(#file_name, #uEdit);

		while (file_name[strlen(#file_name)-1] == '/') {

			file_name[strlen(#file_name)-1] = 0;

		}

		strcat(#filepath, #file_name+strrchr(#file_name, '/'));

	}

	for (i=0; i

	while (file_exists(#filepath)) {

		SaveFile(attempt+1);

		return;

	}

	if (CreateFile(http.content_received, http.content_pointer, #filepath)==0) {

		miniprintf(#notify_message, FILE_SAVED_AS, #filepath);

	} else {

		miniprintf(#notify_message, FILE_NOT_SAVED, #filepath);

	}

	if (!exit_param) notify(#notify_message);

	if (!strcmpi(#filepath+strlen(#filepath)-4, ".zip"))

	|| (!strcmpi(#filepath+strlen(#filepath)-3, ".7z")) {

		Unarchive(#filepath);

	}

	if (open_file) ProcessButtonClick(BTN_RUN);

	if (autoclose.checked) ExitProcess();

}

void Unarchive(dword _arc)

{

	char folder_name[4096];

	strcpy(#folder_name, "/tmp0/1/Downloads/");

	strcpy(#folder_name, #filepath+strrchr(#filepath, '/'));

	folder_name[strlen(#folder_name)-4] = '\0';

	CreateDir(#folder_name);

	strcpy(#param, "-o \"");

	strcat(#param, #folder_name);

	strcat(#param, "\" -h \"");

	strcat(#param, #filepath);

	chrcat(#param, '\"');

	RunProgram("/sys/unz", #param);

}

struct TIME

{

	dword old;

	dword cur;

	dword gone;

} time = {0,0,0};

dword netdata_received;

void ResetDownloadSpeed()

{

	time.old = 0;

	netdata_received = 0;

}

void CalculateDownloadSpeed()

{

	time.cur = GetStartTime();

	if (time.old) {

		time.gone = time.cur - time.old;

		if (time.gone >= 200) {

			speed = http.content_received - netdata_received / time.gone / 10;

			time.old = time.cur;

			netdata_received = http.content_received;

		}

	} else {

		time.old = time.cur;

	}

}