#include <tools/ccanlint/ccanlint.h>
#include <tools/tools.h>
#include <ccan/talloc/talloc.h>
#include <ccan/str/str.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <limits.h>
#include <errno.h>
#include <stdlib.h>
#include <stdio.h>
#include <err.h>
#include <string.h>
#include <ctype.h>

/* Note: we already test safe_mode in run_tests.c */
static const char *can_run_coverage(struct manifest *m)
{
	unsigned int timeleft = default_timeout_ms;
	char *output = run_command(m, &timeleft, "gcov -h");

	if (output)
		return talloc_asprintf(m, "No gcov support: %s", output);
	return NULL;
}

static bool build_module_objs_with_coverage(struct manifest *m, bool keep,
					    struct score *score,
					    char **modobjs)
{
	struct ccan_file *i;

	*modobjs = talloc_strdup(m, "");
	list_for_each(&m->c_files, i, list) {
		char *err;
		char *fullfile = talloc_asprintf(m, "%s/%s", m->dir, i->name);

		i->cov_compiled = maybe_temp_file(m, "", keep, fullfile);
		err = compile_object(m, fullfile, ccan_dir, "",
				     i->cov_compiled);
		if (err) {
			score_file_error(score, i, 0, err);
			talloc_free(i->cov_compiled);
			i->cov_compiled = NULL;
			return false;
		}
		*modobjs = talloc_asprintf_append(*modobjs,
						  " %s", i->cov_compiled);
	}
	return true;
}

static char *obj_list(const struct manifest *m, const char *modobjs)
{
	char *list;
	struct ccan_file *i;

	/* We expect to be linked with tap, unless that's us. */
	if (!streq(m->basename, "tap"))
		list = talloc_asprintf(m, "%s/ccan/tap.o", ccan_dir);
	else
		list = talloc_strdup(m, "");

	/* Objects from any other C files. */
	list_for_each(&m->other_test_c_files, i, list)
		list = talloc_asprintf_append(list, " %s", i->compiled);

	if (modobjs)
		list = talloc_append_string(list, modobjs);

	/* Other ccan modules (don't need coverage versions of those). */
	list_for_each(&m->dep_dirs, i, list) {
		if (i->compiled)
			list = talloc_asprintf_append(list, " %s", i->compiled);
	}

	return list;
}

static char *lib_list(const struct manifest *m)
{
	unsigned int i, num;
	char **libs = get_libs(m, m->dir, &num, &m->info_file->compiled);
	char *ret = talloc_strdup(m, "");

	for (i = 0; i < num; i++)
		ret = talloc_asprintf_append(ret, "-l%s ", libs[i]);
	return ret;
}

static char *cov_compile(const void *ctx,
			 struct manifest *m,
			 struct ccan_file *file,
			 const char *modobjs,
			 bool keep)
{
	char *errmsg;

	file->cov_compiled = maybe_temp_file(ctx, "", keep, file->fullname);
	errmsg = compile_and_link(ctx, file->fullname, ccan_dir,
				  obj_list(m, modobjs),
				  COVERAGE_CFLAGS,
				  lib_list(m), file->cov_compiled);
	if (errmsg) {
		talloc_free(file->cov_compiled);
		file->cov_compiled = NULL;
		return errmsg;
	}

	return NULL;
}

/* FIXME: Coverage from testable examples as well. */
static void do_compile_coverage_tests(struct manifest *m,
				      bool keep,
				      unsigned int *timeleft,
				      struct score *score)
{
	char *cmdout, *modobjs = NULL;
	struct ccan_file *i;

	if (!list_empty(&m->api_tests)
	    && !build_module_objs_with_coverage(m, keep, score, &modobjs)) {
		score->error = "Failed to compile module objects with coverage";
		return;
	}

	list_for_each(&m->run_tests, i, list) {
		cmdout = cov_compile(m, m, i, NULL, keep);
		if (cmdout) {
			score->error = "Failed to compile test with coverage";
			score_file_error(score, i, 0, cmdout);
		}
	}

	list_for_each(&m->api_tests, i, list) {
		cmdout = cov_compile(m, m, i, modobjs, keep);
		if (cmdout) {
			score->error = "Failed to compile test with coverage";
			score_file_error(score, i, 0, cmdout);
		}
	}
	if (!score->error) {
		score->pass = true;
		score->score = score->total;
	}
}

struct ccanlint compile_coverage_tests = {
	.key = "compile-coverage-tests",
	.name = "Module tests compile with profiling",
	.check = do_compile_coverage_tests,
	.can_run = can_run_coverage,
};

REGISTER_TEST(compile_coverage_tests, &compile_tests, NULL);