first commit

src/util.c

@@ -0,0 +1,189 @@
+#include <dirent.h>
+#include <errno.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <time.h>
+#include <unistd.h>
+
+#include "util.h"
+
+bool eq(char *s1, char *s2) {
+  if (strlen(s1) != strlen(s2))
+    return false;
+  return strcmp(s1, s2) == 0;
+}
+
+bool joinhome(char *res, char *path) {
+  char *homedir = getenv("HOME");
+
+  if (NULL == homedir)
+    return false;
+
+  snprintf(res, PATH_MAX, "%s/%s", homedir, path);
+  return true;
+}
+
+bool exists(char *path) {
+  return access(path, F_OK) == 0;
+}
+
+bool endswith(const char *str, const char *suf) {
+  int strl = strlen(str);
+  int sufl = strlen(suf);
+
+  if (sufl > strl)
+    return false;
+
+  return strncmp(str + strl - sufl, suf, sufl) == 0;
+}
+
+bool startswith(char *str, char *sub) {
+  if (strlen(sub) > strlen(str))
+    return false;
+  return strncmp(str, sub, strlen(sub)) == 0;
+}
+
+char *check_path(char *bin) {
+  char *path = getenv("PATH");
+  char *res  = NULL;
+
+  if (NULL == path)
+    return NULL;
+  path = strdup(path);
+
+  char *p = strtok(path, ":");
+  while (NULL != p) {
+    char *fp = malloc(PATH_MAX);
+    snprintf(fp, PATH_MAX, "%s/%s", p, bin);
+    if (exists(fp)) {
+      res = fp;
+      break;
+    }
+    p = strtok(NULL, ":");
+  }
+
+  free(path);
+  return res;
+}
+
+int join(char *res, const char *base, const char *pth) {
+  int blen = strlen(base);
+
+  if ((base[blen - 1] == '/' && pth[0] != '/') || (base[blen - 1] != '/' && pth[0] == '/')) {
+    return sprintf(res, "%s%s", base, pth);
+  } else if (base[blen - 1] != '/' && pth[0] != '/') {
+    return sprintf(res, "%s/%s", base, pth);
+  } else if (base[blen - 1] == '/' && pth[0] == '/') {
+    char *basedup     = strdup(base);
+    basedup[blen - 1] = '\0';
+
+    return sprintf(res, "%s%s", basedup, pth);
+  }
+
+  return -1;
+}
+
+int randint(int min, int max) {
+  srand(time(NULL));
+  return rand() % max + min;
+}
+
+bool is_dir(char *path) {
+  DIR *dir = opendir(path);
+  if (dir) {
+    closedir(dir);
+    return true;
+  }
+
+  return false;
+}
+
+bool mksubdirsp(char *path, int perms) {
+  char cur[PATH_MAX];
+  bzero(cur, PATH_MAX);
+  int indx = 0;
+
+  for (char *c = path; *c != '\0'; c++) {
+    if (*c == '/' && indx != 0) {
+      cur[indx] = '\0';
+      if (!exists(cur) && mkdir(cur, perms) < 0)
+        return false;
+    }
+
+    cur[indx] = *c;
+    indx++;
+  }
+
+  if (!exists(cur) && mkdir(cur, perms) < 0)
+    return false;
+  return true;
+}
+
+bool rmrf(char *path) {
+  if (!exists(path))
+    return true;
+
+  if (!is_dir(path))
+    return unlink(path) == 0;
+
+  DIR           *dir          = opendir(path);
+  struct dirent *ent          = NULL;
+  char           fp[PATH_MAX] = {};
+
+  while ((ent = readdir(dir)) != NULL) {
+    if (eq(ent->d_name, ".") || eq(ent->d_name, ".."))
+      continue;
+
+    join(fp, path, ent->d_name);
+    if (!rmrf(fp))
+      return false;
+  }
+
+  closedir(dir);
+  rmdir(path);
+  return true;
+}
+
+void clist_init(clist_t *l) {
+  l->c = NULL;
+  l->s = 0;
+}
+
+void clist_from_str(clist_t *l, char *str) {
+  char *save = NULL, *el = NULL;
+  char *strdp = strdup(str);
+  el          = strtok_r(strdp, ",", &save);
+
+  while (NULL != el) {
+    clist_add(l, strdup(el));
+    el = strtok_r(NULL, ",", &save);
+  }
+
+  free(strdp);
+}
+
+void clist_free(clist_t *l) {
+  if (NULL == l->c || 0 == l->s)
+    return;
+
+  for (int i = 0; i < l->s; i++)
+    free(l->c[i]);
+  free(l->c);
+  l->s = 0;
+}
+
+void clist_add(clist_t *l, char *en) {
+  if (NULL == l->c || 0 == l->s) {
+    l->c       = malloc(sizeof(char *));
+    l->c[l->s] = en;
+    l->s++;
+    return;
+  }
+
+  l->c       = realloc(l->c, sizeof(char *) * (l->s + 1));
+  l->c[l->s] = en;
+  l->s++;
+}