From c098eb31d870aafd96ef055d725f2e127043f6ab Mon Sep 17 00:00:00 2001 From: "C. McEnroe" Date: Sun, 1 Dec 2019 19:22:38 -0500 Subject: Implement database file search --- litterbox.h | 68 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 68 insertions(+) create mode 100644 litterbox.h (limited to 'litterbox.h') diff --git a/litterbox.h b/litterbox.h new file mode 100644 index 0000000..48d707c --- /dev/null +++ b/litterbox.h @@ -0,0 +1,68 @@ +/* Copyright (C) 2019 C. McEnroe + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#define DATABASE_PATH "litterbox/database.sqlite" + +static inline sqlite3 *dbOpenPath(char *path, int flags) { + char *base = strrchr(path, '/'); + if (flags & SQLITE_OPEN_CREATE && base) { + *base = '\0'; + int error = mkdir(path, 0700); + if (error && errno != EEXIST) err(EX_CANTCREAT, "%s", path); + *base = '/'; + } + sqlite3 *db; + int error = sqlite3_open_v2(path, &db, flags, NULL); + if (!error) return db; + if (error == SQLITE_CANTOPEN) { + sqlite3_close(db); + return NULL; + } + errx(EX_NOINPUT, "%s: %s", path, sqlite3_errmsg(db)); +} + +static inline sqlite3 *dbOpen(int flags) { + const char *home = getenv("HOME"); + const char *dataHome = getenv("XDG_DATA_HOME"); + const char *dataDirs = getenv("XDG_DATA_DIRS"); + char path[PATH_MAX]; + if (dataHome) { + snprintf(path, sizeof(path), "%s/" DATABASE_PATH, dataHome); + } else { + snprintf(path, sizeof(path), "%s/.local/share/" DATABASE_PATH, home); + } + sqlite3 *db = dbOpenPath(path, flags); + if (db) return db; + while (dataDirs && *dataDirs) { + size_t len = strcspn(dataDirs, ":"); + snprintf(path, sizeof(path), "%.*s/" DATABASE_PATH, (int)len, dataDirs); + db = dbOpenPath(path, flags); + if (db) return db; + dataDirs += len; + if (*dataDirs) dataDirs++; + } + return NULL; +} -- cgit 1.4.1