#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <pthread.h>
#include <dirent.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <time.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netdb.h>
#include <arpa/inet.h>

char *pbin_version_string = "0.0.5";
char *homedir = "/srv/files/tmp";
char *log_filename = "/var/log/pbin.log";

void *DeleteStale(void *argp) {
	DIR *d;
	struct dirent *de;
	struct statx st;
	char *dirname = "/srv/files/tmp";
	char fullname[1024];
	char buffer[1028];
	time_t t0;
	while (1) {
		d = opendir(dirname);
		if (d == NULL) {
			fprintf(stderr, "pbin error: Cannot open %s: %s\n", dirname,
				strerror(errno));
			sleep(60);
			continue;
		}

		while (1) {
			de = readdir(d);
			if (de == NULL)
				break;
			else if (strcmp(de->d_name, ".") == 0 || strcmp(de->d_name, "..") == 0)
				continue;

			sprintf(fullname, "%s/%s", dirname, de->d_name);
			if (statx(0, fullname, 0, STATX_BTIME, &st) > -1) {
				t0 = time(NULL);
				if (st.stx_btime.tv_sec < t0 - 60*60*24) {
					sprintf(buffer, "rm %s", fullname);
					system(buffer);
				}
			}
		}
		
		closedir(d);
		sleep(15);
	}
	return NULL;
}

char *GenUniqueFilename(void) {
	char *name = malloc(6);
	struct timeval tv0;
	struct stat st;
	while (1) {
		gettimeofday(&tv0, NULL);
		srand((unsigned int)tv0.tv_usec);
		sprintf(name, "%04d", rand()%10000);
		if (stat(name, &st) == -1)
			break;
	}

	return name;
}

int main(int argc, char **argv) {
	if (chdir(homedir) < 0) {
		fprintf(stderr, "pbin error: Cannot change directory: %s\n",
			strerror(errno));
		return 1;
	}

	pthread_t thr;
	pthread_attr_t attr;
	pthread_attr_init(&attr);
	pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
	pthread_create(&thr, &attr, DeleteStale, NULL);
	pthread_detach(thr);
	pthread_attr_destroy(&attr);

	int sock = socket(AF_INET, SOCK_STREAM, 0);
	if (sock < 0) {
		fprintf(stderr, "pbin error: Cannot open socket: %s\n",
			strerror(errno));
		return 1;
	}

	struct sockaddr_in addr;
	bzero(&addr, sizeof(addr));
	addr.sin_addr.s_addr = inet_addr("0.0.0.0");
	addr.sin_family = AF_INET;
	addr.sin_port = htons(9999);
	if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
		fprintf(stderr, "pbin error: Cannot bind socket: %s\n",
			strerror(errno));
		close(sock);
		return 1;
	}

while (1) {
	if (listen(sock, 1) < 0) {
		fprintf(stderr, "pbin error: Cannot listen: %s\n",
			strerror(errno));
		close(sock);
		return 1;
	}

	struct sockaddr_in peer_addr;
	bzero(&addr, sizeof(peer_addr));
	socklen_t peer_addr_size = sizeof(peer_addr);
	int peer_sock = accept(sock, (struct sockaddr *)&peer_addr, &peer_addr_size);
	if (peer_sock < 0) {
		fprintf(stderr, "pbin error: Cannot accept connection: %s\n",
			strerror(errno));
		close(sock);
		return 1;
	}

	char *filename = GenUniqueFilename();
	FILE *fw = fopen(filename, "w");
	if (fw == NULL) {
		fprintf(stderr, "pbin error: Cannot open %s: %s\n", filename,
			strerror(errno));
		close(sock);
		return 1;
	}

	ssize_t bytes_read;
	char buffer[4096001];
	while (1) {
		memset(buffer, 0, 4096001);
		errno = 0;
		printf("reading...\n");
		bytes_read = read(peer_sock, buffer, 1096);
		printf("bytes_read: %ld\n", bytes_read);
		if (bytes_read == -1) {
			if (errno)
				fprintf(stderr, "pbin error: Cannot read(): %s\n",
					strerror(errno));
			close(sock);
			break;;
		}
		else if (bytes_read < 1096 && bytes_read > 0) {
			fputs(buffer, fw);
			break;
		}
		else if (bytes_read == 0)
			break;;

		fputs(buffer, fw);
	}
	fclose(fw);
	printf("loop done\n");

	sprintf(buffer, "https://esselfe.ca/tmp/%s\n", filename);
	write(peer_sock, buffer, strlen(buffer));
	close(peer_sock);

	fw = fopen(log_filename, "a+");
	if (fw == NULL) {
		fprintf(stderr, "pbin error: Cannot open %s: %s\n", log_filename,
			strerror(errno));
	}
	else {
		struct stat st;
		stat(filename, &st);
		time_t t0 = time(NULL);
		struct tm *tm0 = localtime(&t0);
		char buffer[1024];
		memset(buffer, 0, 1024);
		sprintf(buffer, "%02d%02d%02d-%02d%02d%02d %s %ld %s\n",
			tm0->tm_year+1900-2000, tm0->tm_mon+1, tm0->tm_mday, tm0->tm_hour,
			tm0->tm_min, tm0->tm_sec, filename, st.st_size, inet_ntoa(peer_addr.sin_addr));
		fputs(buffer, fw);
		fclose(fw);
	}

	free(filename);
}
	close(sock);
}