makefile.c

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <unistd.h>
#include <libgen.h>
#include <sys/wait.h>
#include "parser/colors.h"


#include "deps/hash/hash.h"

#include "package/package.h"
#include "package/export.h"
#include "package/import.h"
#include "package/atomic-stream.h"
#include "utils/utils.h"
#include "deps/stream/stream.h"

static const char * ops[] = {
	":=",
	"?=",
	"+=",
};

typedef struct {
	char * makefile;
	char * makefile_dir;
	char * makefile_base;
	char * target;
	char * cwd;
} makevars;

makevars get_makevars(package_t * pkg, char * makefile) {
	makevars v = {
		.makefile      = makefile,
		.makefile_base = strdup(basename(makefile)),
		.makefile_dir  = strdup(dirname(makefile)),
		.target        = strdup(basename(pkg->generated)),
		.cwd           = getcwd(NULL, 0),
	};

	char * ext = strrchr(v.target, '.');
	if (strcmp(pkg->name, "main") == 0){
		*ext = 0;
	} else {
		ext[1] = 'a';
	}

	chdir(v.makefile_dir);
	return v;
}

int clear_makevars(makevars v, int result, char * cmd) {
	chdir(v.cwd);

	free(v.cwd);
	free(v.target);
	free(v.makefile_dir);
	free(v.makefile_base);
	free(v.makefile);
	free(cmd);

	if (result == 0 || result == -1) return result;
	if (result == 127) return -1;
	return WEXITSTATUS(result);
}

int makefile_make(package_t * pkg, char * makefile) {
	if (pkg == NULL) return -1;
	makevars v = get_makevars(pkg, makefile);

	char * cmd;
	asprintf(&cmd, "make -f %s %s", v.makefile_base,  v.target);

	return clear_makevars(v, system(cmd), cmd);
}

int makefile_clean(package_t * pkg, char * makefile) {
	if (pkg == NULL) return -1;
	makevars v = get_makevars(pkg, makefile);

	char * cmd;
	asprintf(&cmd, "make -f %s CLEAN_%s", v.makefile_base,  v.target);

	return clear_makevars(v, system(cmd), cmd);
}

char * write_deps(package_t * pkg, package_t * root, stream_t * out, char * deps) {
	if (pkg == NULL || pkg->exported) return deps;
	pkg->exported = true;

	char * source = utils_relative(root->generated, pkg->generated);
	char * object = strdup(source);
	object[strlen(source) - 1] = 'o';

	size_t len = deps == NULL ? 0 : strlen(deps);
	deps = realloc(deps, len + strlen(object) + 2);
	sprintf(deps + len, " %s", object);

	char * package_rel = utils_relative(root->source_abs, pkg->generated);
	stream_printf(out, "#dependencies for package '%s'\n", package_rel);
	free(package_rel);

	int i;
	for (i = 0; i < pkg->n_variables; i++) {
		package_var_t v = pkg->variables[i];
		stream_printf(out, "%s %s %s\n", v.name, ops[v.operation], v.value);
	}

	stream_printf(out, "%s: %s", object, source);
	free(source);
	free(object);

	if (pkg->deps == NULL) {
	stream_printf(out,"\n\n");
		return deps;
	}

	hash_each_val(pkg->deps, {
			package_import_t * dep = (package_import_t *) val;
			if (dep && dep->pkg && dep->pkg->header) {
				char * path = utils_relative(root->source_abs, dep->pkg->header);
				stream_printf(out, " %s", path);
				free(path);
			}
	});
	stream_printf(out,"\n\n");

	hash_each_val(pkg->deps, {
			package_import_t * dep = (package_import_t *)val;
			deps = write_deps(dep->pkg, root, out, deps);
	});

	return deps;
}

char * get_makefile_name(const char * path) {
	char * name = strdup(path);
	name[strlen(name) - strlen("module.c")] = 0;
	strcat(name, "mk");
	return name;
}

char * makefile_write(package_t * pkg, const char * name) {
	char * target = NULL;
	char * mkfile_name = get_makefile_name(name);
	stream_t * mkfile = atomic_stream_open(mkfile_name);
	char * deps = write_deps(pkg, pkg, mkfile, NULL);

	if (strcmp(pkg->name, "main") == 0) {
		char * buf = strdup(pkg->generated);
		char * base = basename(buf);
		asprintf(&target, "%.*s", (int)strlen(base) - 2, base);
		free(buf);

		stream_printf(mkfile, "%s:%s\n",target, deps);
		stream_printf(mkfile, "\t$(CC) $(CFLAGS) $(LDFLAGS) %s -o %s $(LDLIBS)\n\n", deps, target);
	} else {
		target = malloc(strlen(pkg->name) + 3);
		strcpy(target, pkg->name);
		strcat(target, ".a");

		stream_printf(mkfile, "%s:%s\n",target, deps);
		stream_printf(mkfile, "\tar rcs $@ $^\n\n", deps, target);
	}

	stream_printf(mkfile, "CLEAN_%s:\n", target);
	stream_printf(mkfile, "\trm -rf %s%s\n", target, deps);
	free(target);

	stream_close(mkfile);
	free(deps);

	return mkfile_name;
}