regexghost-dotfiles

directory_bookmarks.c (9633B)

---

 #include <stdio.h>
 #include <stdlib.h>
 #include <stdbool.h>
 #include <string.h>
 #include <unistd.h>
 #include <linux/limits.h>
 
 #define BOOKMARK_FILE_LOCATION "/.local/share/regexghost/script_data/directory_bookmarks.txt"
 #define FILE_BOOKMARK_FILE_LOCATION "/.local/share/regexghost/script_data/file_bookmarks.txt"
 #define RANGER_BOOKMARK_FILE_LOCATION "/.local/share/regexghost/script_data/ranger_directory_bookmarks.conf"
 
 // This script is based on apparix: https://github.com/micans/apparix
 
 static char file_loc[256];
 static char file_bookmarks_file_loc[256];
 static char ranger_file_loc[256];
 
 typedef struct {
 	char bookmark_name[11];
 	char directory_name[251];
 } Bookmark;
 
 typedef struct {
 	char bookmark_name[11];
 	char file_path[251];
 } FileBookmark;
 
 void print_help() {
 	printf("Usage:\n");
 	printf("  Add New Bookmark: bookmarks add *name*\n");
 	printf("  Remove Bookmark: bookmarks remove *name*\n");
 	printf("  Get Bookmark: bookmarks get *name*\n");
 	printf("  List Bookmarks: bookmarks list\n");
 	printf("  Get Bookmark Name of Current Directory: bookmarks current\n");
 	printf("Usage (File Bookmarks):\n");
 	printf("  Add New File Bookmark: bookmarks file add *name* *filename*\n");
 	printf("  Remove Bookmark: bookmarks file remove *name*\n");
 	printf("  Get Bookmark: bookmarks file get *name*\n");
 	printf("  List Bookmarks: bookmarks file list\n");
 }
 
 int get_bookmarks(Bookmark* bookmarks) {
 	FILE* f;
 	f = fopen(file_loc, "r");
 	if (f == NULL) {
 		return 0;
 	}
 
 	int i = 0;
 	char buffer[400];
 	
 	while (fgets(buffer, sizeof(buffer), f) != NULL) {
 		if (i > 99) {
 			printf("Error, bookmarks file too big\n");
 			break;
 		}
 
 		char bookmark_name[11] = {0};
 		char directory_name[251] = {0};
 		sscanf(buffer, "%s %[^\n]", bookmark_name, directory_name);
 		strcpy(bookmarks[i].bookmark_name, bookmark_name);
 		strcpy(bookmarks[i].directory_name, directory_name);
 
 		memset(buffer, 0, sizeof(buffer));
 
 		i++;
 	}
 	return i;
 }
 
 int get_file_bookmarks(FileBookmark* bookmarks) {
 	FILE* f;
 	f = fopen(file_bookmarks_file_loc, "r");
 	if (f == NULL) {
 		return 0;
 	}
 
 	int i = 0;
 	char buffer[400];
 	
 	while (fgets(buffer, sizeof(buffer), f) != NULL) {
 		if (i > 99) {
 			printf("Error, file bookmarks file too big\n");
 			break;
 		}
 
 		char bookmark_name[11] = {0};
 		char file_path[251] = {0};
 		sscanf(buffer, "%s %[^\n]", bookmark_name, file_path);
 		strcpy(bookmarks[i].bookmark_name, bookmark_name);
 		strcpy(bookmarks[i].file_path, file_path);
 
 		memset(buffer, 0, sizeof(buffer));
 
 		i++;
 	}
 	return i;
 }
 
 void write_bookmarks(Bookmark* bookmarks, int num_bookmarks) {
 	FILE* f;
 	f = fopen(file_loc, "w");
 	for (int i=0; i<num_bookmarks; i++) {
 		fprintf(f, "%s %s\n", bookmarks[i].bookmark_name, bookmarks[i].directory_name);
 	}
 	fclose(f);
 
 	FILE* g;
 	g = fopen(ranger_file_loc, "w");
 	for (int i=0; i<num_bookmarks; i++) {
 		fprintf(g, "map go%s cd %s\n", bookmarks[i].bookmark_name, bookmarks[i].directory_name);
 	}
 	fclose(g);
 }
 
 void write_file_bookmarks(FileBookmark* bookmarks, int num_bookmarks) {
 	FILE* f;
 	f = fopen(file_bookmarks_file_loc, "w");
 	for (int i=0; i<num_bookmarks; i++) {
 		fprintf(f, "%s %s\n", bookmarks[i].bookmark_name, bookmarks[i].file_path);
 	}
 	fclose(f);
 }
 
 void add_file_bookmark(char* bookmark_name, char* filename) {
 	FileBookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_file_bookmarks(bookmarks);
 
 	if (num_bookmarks > 99) {
 		printf("Error, file bookmarks full\n");
 	}
 
 	char cwd[PATH_MAX] = {0};
 	if (getcwd(cwd, sizeof(cwd)) == NULL) {
 		printf("Error, unable to get current directory\n");
 		return;
 	}
 
 	strcpy(bookmarks[num_bookmarks].bookmark_name, bookmark_name);
 	strcpy(bookmarks[num_bookmarks].file_path, cwd);
 	strcpy(bookmarks[num_bookmarks].file_path, filename);
 	strcat(strcat(strcpy(bookmarks[num_bookmarks].file_path, cwd), "/"), filename);
 
 
 	num_bookmarks++;
 	write_file_bookmarks(bookmarks, num_bookmarks);
 }
 
 void add_bookmark(char* to_add) {
 	Bookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_bookmarks(bookmarks);
 
 	if (num_bookmarks > 99) {
 		printf("Error, bookmarks full\n");
 	}
 
 	char cwd[PATH_MAX] = {0};
 	if (getcwd(cwd, sizeof(cwd)) == NULL) {
 		printf("Error, unable to get current directory\n");
 		return;
 	}
 
 	strcpy(bookmarks[num_bookmarks].bookmark_name, to_add);
 	strcpy(bookmarks[num_bookmarks].directory_name, cwd);
 	
 	num_bookmarks++;
 	write_bookmarks(bookmarks, num_bookmarks);
 }
 
 void resolve_bookmark(char* to_resolve) {
 	Bookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_bookmarks(bookmarks);
 
 	for (int i=0; i<num_bookmarks; i++) {
 		if (!strcmp(bookmarks[i].bookmark_name, to_resolve)) {
 			printf("%s\n", bookmarks[i].directory_name);
 			return;
 		}
 	}
 }
 
 void resolve_file_bookmark(char* to_resolve) {
 	FileBookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_file_bookmarks(bookmarks);
 
 	for (int i=0; i<num_bookmarks; i++) {
 		if (!strcmp(bookmarks[i].bookmark_name, to_resolve)) {
 			printf("%s\n", bookmarks[i].file_path);
 			return;
 		}
 	}
 }
 
 void bookmark_for_current_dir() {
 	Bookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_bookmarks(bookmarks);
 
 	char cwd[PATH_MAX] = {0};
 	if (getcwd(cwd, sizeof(cwd)) == NULL) {
 		printf("Error, unable to get current directory\n");
 		return;
 	}
 
 	for (int i=0; i<num_bookmarks; i++) {
 		if (!strcmp(bookmarks[i].directory_name, cwd)) {
 			printf("-(%s)", bookmarks[i].bookmark_name);
 			return;
 		}
 	}
 }
 
 void list_bookmarks() {
 	Bookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_bookmarks(bookmarks);
 
 	for (int i=0; i<num_bookmarks; i++) {
 		printf("%s: %s\n", bookmarks[i].bookmark_name, bookmarks[i].directory_name);
 	}
 }
 
 void list_file_bookmarks() {
 	FileBookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_file_bookmarks(bookmarks);
 
 	for (int i=0; i<num_bookmarks; i++) {
 		printf("%s: %s\n", bookmarks[i].bookmark_name, bookmarks[i].file_path);
 	}
 }
 
 void remove_bookmark(char* to_remove) {
 	Bookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_bookmarks(bookmarks);
 
 	bool move_backwards = false;
 
 	for (int i=0; i<num_bookmarks; i++) {
 		if (move_backwards) {
 			memset(bookmarks[i-1].bookmark_name, 0, sizeof(bookmarks[i-1].bookmark_name));
 			memset(bookmarks[i-1].directory_name, 0, sizeof(bookmarks[i-1].directory_name));
 			strcpy(bookmarks[i-1].bookmark_name, bookmarks[i].bookmark_name);
 			strcpy(bookmarks[i-1].directory_name, bookmarks[i].directory_name);
 		}
 		else {
 			if (!strcmp(bookmarks[i].bookmark_name, to_remove)) {
 				move_backwards = true;
 			}
 		}
 	}
 	if (move_backwards) {
 		num_bookmarks--;
 	}
 	write_bookmarks(bookmarks, num_bookmarks);
 }
 
 void remove_file_bookmark(char* to_remove) {
 	FileBookmark bookmarks[100];
 	memset(bookmarks, 0, sizeof(bookmarks));
 	int num_bookmarks = get_file_bookmarks(bookmarks);
 
 	bool move_backwards = false;
 
 	for (int i=0; i<num_bookmarks; i++) {
 		if (move_backwards) {
 			memset(bookmarks[i-1].bookmark_name, 0, sizeof(bookmarks[i-1].bookmark_name));
 			memset(bookmarks[i-1].file_path, 0, sizeof(bookmarks[i-1].file_path));
 			strcpy(bookmarks[i-1].bookmark_name, bookmarks[i].bookmark_name);
 			strcpy(bookmarks[i-1].file_path, bookmarks[i].file_path);
 		}
 		else {
 			if (!strcmp(bookmarks[i].bookmark_name, to_remove)) {
 				move_backwards = true;
 			}
 		}
 	}
 	if (move_backwards) {
 		num_bookmarks--;
 	}
 	write_file_bookmarks(bookmarks, num_bookmarks);
 }
 
 int main(int argc, char* argv[]) {
 	if (argc == 1) {
 		print_help();
 		return 1;
 	}
 
 	memset(file_loc, 0, 256);
 	memset(ranger_file_loc, 0, 256);
 	strcat(strcpy(file_loc, getenv("HOME")), BOOKMARK_FILE_LOCATION);
 	strcat(strcpy(file_bookmarks_file_loc, getenv("HOME")), FILE_BOOKMARK_FILE_LOCATION);
 	strcat(strcpy(ranger_file_loc, getenv("HOME")), RANGER_BOOKMARK_FILE_LOCATION);
 
 	// Add a new bookmark for current directory
 	if (!strcmp(argv[1], "file")) {
 		if (!strcmp(argv[2], "add")) {
 				if (argc != 5) {
 					print_help();
 					return 1;
 				}
 				char* to_add = argv[3];
 				char* file_name = argv[4];
 				add_file_bookmark(to_add, file_name);
 			}
 		// Get a bookmark
 			else if (!strcmp(argv[2], "get")) {
 				if (argc != 4) {
 					print_help();
 					return 1;
 				}
 				char* to_resolve = argv[3];
 				resolve_file_bookmark(to_resolve);
 			}
 		// List all file bookmarks
 			else if (!strcmp(argv[2], "list")) {
 				list_file_bookmarks();
 			}
 		// Remove a bookmark by name
 			else if (!strcmp(argv[2], "remove")) {
 				if (argc != 4) {
 					print_help();
 					return 1;
 				}
 				char* to_remove = argv[3];
 				remove_file_bookmark(to_remove);
 			}
 			else {
 				print_help();
 				return 1;
 			}
 	} else {
 		if (!strcmp(argv[1], "add")) {
 			if (argc != 3) {
 				print_help();
 				return 1;
 			}
 			char* to_add = argv[2];
 			add_bookmark(to_add);
 		}
 		// Get a bookmark
 		else if (!strcmp(argv[1], "get")) {
 			if (argc != 3) {
 				print_help();
 				return 1;
 			}
 			char* to_resolve = argv[2];
 			resolve_bookmark(to_resolve);
 		}
 		// Print the current directory's bookmark, if it exists
 		else if (!strcmp(argv[1], "current")) {
 			bookmark_for_current_dir();
 		}
 		// List all bookmarks
 		else if (!strcmp(argv[1], "list")) {
 			list_bookmarks();
 		}
 		// Remove a bookmark by name
 		else if (!strcmp(argv[1], "remove")) {
 			if (argc != 3) {
 				print_help();
 				return 1;
 			}
 			char* to_remove = argv[2];
 			remove_bookmark(to_remove);
 		}
 		else {
 			print_help();
 			return 1;
 		}
 	}
 }