#include "data.h"

#include "list.h"
#include "log.h"
#include "ref.h"
#include "track.h"
#include "tag.h"

#include <dirent.h>
#include <sys/stat.h>
#include <unistd.h>

#include <stdbool.h>
#include <string.h>

const char *datadir;

struct list tracks;
struct list tags;
struct list tags_sel;

void
data_load(void)
{
	struct dirent *ent;
	struct tag *tag;
	struct stat st;
	char *path;
	DIR *dir;

	list_init(&tracks);
	list_init(&tags);
	list_init(&tags_sel);

	datadir = getenv("TMUS_DATA");
	ASSERT(datadir != NULL);

	dir = opendir(datadir);
	ASSERT(dir != NULL);
	while ((ent = readdir(dir))) {
		if (!strcmp(ent->d_name, "."))
			continue;
		if (!strcmp(ent->d_name, ".."))
			continue;

		path = aprintf("%s/%s", datadir, ent->d_name);
		ASSERT(path != NULL);

		if (!stat(path, &st) && S_ISDIR(st.st_mode)) {
			tag = tag_init(datadir, ent->d_name);
			tracks_load(tag);
			list_push_back(&tags, LINK(tag));
		}

		free(path);
	}
	closedir(dir);

	ASSERT(!list_empty(&tags));
}

void
data_save(void)
{
	struct link *iter;
	struct tag *tag;

	for (LIST_ITER(&tags, iter)) {
		tag = UPCAST(iter, struct tag);
		tracks_save(tag);
	}
}

void
data_free(void)
{
	struct link *track_link;
	struct link *tag_link;
	struct tag *tag;

	log_info("MAIN: data_free()\n");

	refs_free(&tracks);
	refs_free(&tags_sel);

	while (!list_empty(&tags)) {
		tag_link = list_pop_front(&tags);

		tag = UPCAST(tag_link, struct tag);
		while (!list_empty(&tag->tracks)) {
			track_link = list_pop_front(&tag->tracks);
			track_free(UPCAST(track_link, struct ref)->data);
			ref_free(UPCAST(track_link, struct ref));
		}
		tag_free(tag);
	}
}

int
get_fid(const char *path)
{
	struct stat st;
	return stat(path, &st) ? -1 : st.st_ino;
}

void
index_update(struct tag *tag)
{
	struct track *track, *track_iter;
	struct dirent *ent;
	struct link *iter;
	struct ref *ref;
	struct stat st;
	char *path;
	FILE *file;
	DIR *dir;
	int fid;

	path = aprintf("%s/index", tag->fpath);
	ASSERT(path != NULL);

	file = fopen(path, "w+");
	ASSERT(file != NULL);
	free(path);

	dir = opendir(tag->fpath);
	ASSERT(dir != NULL);

	while ((ent = readdir(dir))) {
		if (!strcmp(ent->d_name, "."))
			continue;
		if (!strcmp(ent->d_name, ".."))
			continue;
		if (!strcmp(ent->d_name, "index"))
			continue;

		/* skip files without extension */
		if (!strchr(ent->d_name + 1, '.'))
			continue;

		path = aprintf("%s/%s", tag->fpath, ent->d_name);
		ASSERT(path != NULL);
		fid = get_fid(path);
		free(path);

		fprintf(file, "%i:%s\n", fid, ent->d_name);
	}

	closedir(dir);
	fclose(file);
}

void
tracks_load(struct tag *tag)
{
	char linebuf[1024];
	struct link *link;
	struct track *track;
	struct track *track2;
	struct ref *ref;
	char *index_path;
	char *track_name, *sep;
	int track_fid;
	bool new_track;
	FILE *file;

	printf("Loading files from %s", tag->fpath);
	fflush(stdout);

	index_path = aprintf("%s/index", tag->fpath);
	ASSERT(index_path != NULL);

	file = fopen(index_path, "r");
	if (file == NULL) {
		index_update(tag);
		file = fopen(index_path, "r");
		ASSERT(file != NULL);
	}

	while (fgets(linebuf, sizeof(linebuf), file)) {
		sep = strchr(linebuf, '\n');
		if (sep) *sep = '\0';

		sep = strchr(linebuf, ':');
		ASSERT(sep != NULL);
		*sep = '\0';

		track_fid = atoi(linebuf);
		track_name = sep + 1;
		track = track_alloc(tag->fpath, track_name, track_fid);

		ref = ref_init(tag);
		ASSERT(ref != NULL);
		list_push_back(&track->tags, LINK(ref));

		ref = ref_init(track);
		ASSERT(ref != NULL);
		list_push_back(&tag->tracks, LINK(ref));

		new_track = true;
		for (LIST_ITER(&tracks, link)) {
			track2 = UPCAST(link, struct ref)->data;
			if (track->fid > 0 && track->fid == track2->fid)
				new_track = false;
		}

		if (new_track) {
			ref = ref_init(track);
			ASSERT(ref != NULL);
			list_push_back(&tracks, LINK(ref));
		}
	}

	/* clear line and reset cursor */
	printf("\x1b[0K\r");

	fclose(file);
	free(index_path);
}

void
tracks_save(struct tag *tag)
{
	struct track *track;
	struct link *link;
	char *index_path;
	FILE *file;

	/* write playlist back to index file */

	printf("Saving tracks to %s", tag->fpath);

	index_path = aprintf("%s/index", tag->fpath);
	ASSERT(index != NULL);

	file = fopen(index_path, "w+");
	ASSERT(file != NULL);

	for (LIST_ITER(&tag->tracks, link)) {
		track = UPCAST(link, struct ref)->data;
		fprintf(file, "%i:%s\n", track->fid, track->fname);
	}

	/* clear line and reset cursor */
	printf("\x1b[0K\r");

	fclose(file);
	free(index_path);
}

void
rm_file(const char *path)
{
	ASSERT(unlink(path) == 0);
}

void
copy_file(const char *src, const char *dst)
{
	FILE *in, *out;
	char buf[4096];
	int len, nread;

	in = fopen(src, "r");
	if (in == NULL)
		PANIC("Failed to open file %s", src);

	out = fopen(dst, "w+");
	if (out == NULL)
		PANIC("Failed to open file %s", dst);

	while ((nread = fread(buf, 1, sizeof(buf), in)) > 0) {
		fwrite(buf, 1, nread, out);
	}

	if (nread < 0)
		PANIC("Copy failed!", src, dst);

	fclose(in);
	fclose(out);
}

void
move_file(const char *src, const char *dst)
{
	copy_file(src, dst);
	rm_file(src);
}

struct tag *
tag_find(const wchar_t *query)
{
	struct link *iter;
	struct tag *tag;

	for (LIST_ITER(&tags, iter)) {
		tag = UPCAST(iter, struct tag);
		if (!wcscmp(tag->name, query)) {
			return tag;
		}
	}

	return NULL;
}