skratch/skratch.c

#include <dirent.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <limits.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>
struct Flags
{
	int use_v;
	int use_s;
	char *suffix;            // file suffix
	char skratch_path[4096]; // ~/.skratch
	char file_path[4096];    // ~/.skratch/<filename><suffix>
	char filename[255];      // skratch-XXXXXX
	char editor[255];        // editor to open file with
};

// sets f.editor using env vars
void set_editor(struct Flags *f)
{
	char *ed;
	if (f->use_v)
	        ed = getenv("VISUAL");
	else 
		ed = getenv("EDITOR");
	if (ed)
		strcpy(f->editor, ed);
	else //no env vars set
		strcpy(f->editor, "vi");
}

// sets f.file_path and makes a file in ~/.skratch if needed
int make_skratch_file(struct Flags *f)
{
	char fp[4096];
	int fd = 0;
	if (!f->use_s) f->suffix = "";

	snprintf(fp, sizeof(fp), "%s/%s%s", f->skratch_path, f->filename, f->suffix);
	if (!strcmp(f->filename, "skratch-XXXXXX"))
	{
		int s_len = strlen(f->suffix);
		fd = mkstemps(fp, s_len);
		close(fd);
	}
	if (fd == -1)
	{
		printf("ERROR: Couldn't create skratch file!");
		return 3;
	}
	strcpy(f->file_path, fp);
	return 0;
}

// creates ~/.skratch if needed and sets f.skratch_path
int make_skratch_path(struct Flags *f)
{
	char sp[4096];
	char *home_path = getenv("HOME");
	if (home_path == NULL)
	{
		printf("ERROR: No HOME variable set!");
		return 1;
	}
	snprintf(sp, sizeof(sp), "%s/.skratch", home_path);
	int result = mkdir(sp, 0755);
	if (result < 0 && errno != EEXIST)
	{
		printf("ERROR: Couldn't create: %s\n", sp);
		return 2;
	}
	strcpy(f->skratch_path, sp);
	return 0;
}

// lists files in f.skratch_path, with option to delete
// this should likely have all delete logic inside delete_files()
int list_files(struct Flags *f, int delete)
{
	DIR *d = opendir(f->skratch_path);
	if (d == NULL) return 5;
	struct dirent *dir;
	if (d)
	{
		while ((dir = readdir(d)) != NULL)
		{	
			char fn[4096];
			strcpy(fn,dir->d_name);
			if (!strcmp(".",fn) || !strcmp("..",fn))
			{
				continue;
			}
			if (delete)
			{
				char fp[4096];
				snprintf(fp, sizeof(fp), "%s/%s", f->skratch_path, dir->d_name);
				remove(fp);
			}
			else printf("%s\n", dir->d_name);

		}
		closedir(d);
	}
	return 0;
}

// menu confirmation for deleting files in f.skratch_path
// there is a bug if the user has no input
int delete_files(struct Flags *f)
{
	list_files(f, 0);
	printf("Delete the listed files? [y/N] ");
	char choice = 'N';
	scanf(" %c", &choice);
	if (choice == 'y' || choice == 'Y')
	{
		list_files(f, 1);
	}
	return 0;
}

// handles arguments and runs the editor
int main(int argc, char *argv[])
{
	int opt;
	struct Flags f = {.use_v = 0, .filename="skratch-XXXXXX"};
	make_skratch_path(&f);
	//handle args
	while ((opt = getopt(argc, argv, "cls:v")) != -1)
	{
		switch (opt)
		{
			case 'c': // delete ~/.skratch files
				delete_files(&f);
				return 0;
			case 'l': // list ~/.skratch files
				list_files(&f, 0);
				return 0;
			case 's': // use file suffix, like .txt
				f.use_s = 1;
				f.suffix = optarg;
				break;
			case 'v': // use visual editor
				f.use_v = 1;
				break;
			default:
				printf("error print usage");
		}
	}
	// find optional filename if exists
	for (int i = optind; i < argc; i++)
	{
		if (!strcmp(f.filename,"skratch-XXXXXX"))
		{
			strcpy(f.filename, argv[i]);
		}
		else
		{
			printf("ERROR: Too many file args");
			return 1;
		}
	}
	set_editor(&f);
	make_skratch_file(&f);
	char *args[3];
	args[0] = f.editor;
	args[1] = f.file_path;
	args[2] = NULL;
	execvp(f.editor, args);
	return 0;
}