aboutsummaryrefslogtreecommitdiff
path: root/main.c
blob: 1024d17905da7a67c8245356bd3b6a5cbe0ad7d7 (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
109
110
111
112
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <time.h>
#include <unistd.h>
#include <pthread.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/types.h>

typedef struct f_comp_data {
	char *f1;
	char *f2;
} f_comp_data;

void *f1_newer(void *arg) {
	f_comp_data *files = (f_comp_data *) arg;
	char *f1 = files->f1;
	char *f2 = files->f2;
	struct stat f1_attr;
	struct stat f2_attr;
	stat(f1, &f1_attr);
	stat(f2, &f2_attr);
	int diff = difftime(f1_attr.st_mtime, f2_attr.st_mtime);
	int ret = 0;
	if (diff <= 0) {
		pthread_exit(&ret);
	}
	ret = 1;
	pthread_exit(&ret);
}

void *create_make() {
	/* Create simple makefile */
	int fd = creat("makefile", 00644);
	FILE *file = fdopen(fd, "w");
	fprintf(file, "debug: main.c\n\tgcc -Wall main.c -o ./bin/debug\nrelease: main.c\n\tgcc -O2 -march=native main.c -o ./bin/release");
	fclose(file);

	return NULL;
}


int main(int argc, char *argv[]) {
	/* Struct to hold files */
	f_comp_data files;
	files.f1 = "main.c";
	files.f2 = "./bin/debug";

	/* Parse args to determine which mode to run in
	 * Not very robust, needs improvement
	 */
	enum { RUN, BUILD, BUILD_RELEASE } mode = RUN;
	if (argc > 1) {
		switch(argv[1][0]) {
			case 'r': break;
			case 'b': 
				  if (argc > 2) { 
					  mode = BUILD_RELEASE; 
					  files.f2 = "./bin/release";
					  break; 
				  }
				  mode = BUILD; break;
			default: printf("Unknown argument: '%s'\nDefaulting to 'run'\n", argv[1]);
		}
	}

	/* Threads to:
	 * create makefile
	 * check age of source vs. bin
	 */
	pthread_t f_thread, check_thread;
	pthread_create(&f_thread, NULL, create_make, NULL);
	pthread_create(&check_thread, NULL, f1_newer, (void *)&files);

	mkdir("./bin", 0777);


	pthread_join(f_thread, NULL);

	/* Get value from thread which checks if the bin
	 * is older than the source code
	 */
	void *newer;
	pthread_join(check_thread, &newer);
	int recomp = *(int *) newer;

	if (recomp) {
		pid_t make_pid;
		if ((make_pid = fork()) == 0) {
			printf("Recompiling...\n");
			switch (mode) {
				case BUILD_RELEASE:
					execl("/usr/bin/make", "make", "release", NULL); 
					break;
				default: 
					execl("/usr/bin/make", "make", "debug", NULL); 
			}
		}
		waitpid(make_pid, NULL, 0);
	}

	if (mode == RUN) {
		pid_t x_pid;
		if ((x_pid = fork()) == 0) {
			execl(files.f2, "tmp", NULL);
		}
		waitpid(x_pid, NULL, 0);
	}

	return 0;
}