Refactored main.c into many files

1 files changed, 306 insertions, 0 deletions

diff --git a/src/data.c b/src/data.c
new file mode 100644
index 0000000..33cc6eb
--- /dev/null
+++ b/src/data.c
@@ -0,0 +1,306 @@
+#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;
+}
+
+