use arena everywhere + bunch of cool new stuff

deprecated/dir.c

@@ -0,0 +1,204 @@
+#include "dir.h"
+#include "tracelog.h"
+
+#if COLLA_WIN
+#include "win32_slim.h"
+#include <stdlib.h>
+#include <assert.h>
+
+#include "strstream.h"
+
+typedef struct {
+    dir_entry_t cur;
+    dir_entry_t next;
+    HANDLE handle;
+} _dir_internal_t;
+
+static dir_entry_t _fillDirEntry(WIN32_FIND_DATAW *data) {
+    return (dir_entry_t) {
+        .type = 
+            data->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ? 
+            FS_TYPE_DIR : FS_TYPE_FILE,
+        .name = strFromWChar(data->cFileName, 0)
+    };
+}
+
+static _dir_internal_t _getFirst(const wchar_t *path) {
+    _dir_internal_t res = {0};
+    WIN32_FIND_DATAW data = {0};
+
+    res.handle = FindFirstFileW(path, &data);
+
+    if(res.handle != INVALID_HANDLE_VALUE) {
+        res.next = _fillDirEntry(&data);
+    }
+
+    return res;
+}
+
+static void _getNext(_dir_internal_t *ctx) {
+    WIN32_FIND_DATAW data = {0};
+
+    BOOL result = FindNextFileW(ctx->handle, &data);
+    if(!result) {
+        if(GetLastError() == ERROR_NO_MORE_FILES) {
+            FindClose(ctx->handle);
+            ctx->handle = NULL;
+            return;
+        }
+    }
+    ctx->next = _fillDirEntry(&data);
+}
+
+dir_t dirOpen(const char *path) {
+    DWORD n = GetFullPathName(path, 0, NULL, NULL);
+    outstream_t out = ostrInitLen(n + 3);
+    n = GetFullPathName(path, n, out.buf, NULL);
+    assert(n > 0);
+    out.len += n;
+    switch(ostrBack(out)) {
+    case '\\':
+    case '/':
+    case ':':
+        // directory ends in path separator
+        // NOP
+        break;
+    default:
+        ostrPutc(&out, '\\');
+    }
+    ostrPutc(&out, '*');
+
+    _dir_internal_t *dir = malloc(sizeof(_dir_internal_t));
+    if(dir) {
+        wchar_t *wpath = strToWCHAR(ostrAsStr(out));
+        assert(wpath);
+        *dir = _getFirst(wpath);
+        free(wpath);
+    }
+    ostrFree(out);
+
+    return dir;
+}
+
+void dirClose(dir_t ctx) {
+    free(ctx);
+}
+
+bool dirValid(dir_t ctx) {
+    _dir_internal_t *dir = (_dir_internal_t*)ctx;
+    return dir->handle != INVALID_HANDLE_VALUE;
+}
+
+dir_entry_t *dirNext(dir_t ctx) {
+    _dir_internal_t *dir = (_dir_internal_t*)ctx;
+    strFree(dir->cur.name);
+    if(!dir->handle) return NULL;
+    dir->cur = dir->next;
+    _getNext(dir);
+    return &dir->cur;
+}
+
+void dirCreate(const char *path) {
+    CreateDirectoryA(path, NULL);
+}
+
+#else
+
+#include <dirent.h>
+#include <stdlib.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+
+// taken from https://sites.uclouvain.be/SystInfo/usr/include/dirent.h.html
+// hopefully shouldn't be needed
+#ifndef DT_DIR
+    #define DT_DIR 4
+#endif
+#ifndef DT_REG
+    #define DT_REG 8
+#endif
+
+typedef struct {
+    DIR *dir;
+    dir_entry_t next;
+} _dir_internal_t;
+
+dir_t dirOpen(const char *path) {
+    _dir_internal_t *ctx = (_dir_internal_t *)calloc(1, sizeof(_dir_internal_t));
+    if(ctx) ctx->dir = opendir(path);
+    return ctx;
+}
+
+void dirClose(dir_t ctx) {
+    if(ctx) {
+        _dir_internal_t *in = (_dir_internal_t *)ctx;
+        closedir(in->dir);
+        free(in);
+    }
+}
+
+bool dirValid(dir_t ctx) {
+    _dir_internal_t *dir = (_dir_internal_t*)ctx;
+    return dir->dir != NULL;
+}
+
+dir_entry_t *dirNext(dir_t ctx) {
+    if(!ctx) return NULL;
+    _dir_internal_t *in = (_dir_internal_t *)ctx;
+    strFree(in->next.name);
+    struct dirent *dp = readdir(in->dir);
+    if(!dp) return NULL;
+    
+    switch(dp->d_type) {
+    case DT_DIR: in->next.type = FS_TYPE_DIR; break;
+    case DT_REG: in->next.type = FS_TYPE_FILE; break;
+    default: in->next.type = FS_TYPE_UNKNOWN; break;
+    }
+
+    in->next.name = strFromStr(dp->d_name);
+    return &in->next;
+}
+
+void dirCreate(const char *path) {
+    mkdir(path, 0700);
+}
+
+#endif
+
+#include <stdio.h>
+
+bool dirRemove(const char *path) {
+    dir_t dir = dirOpen(path);
+    if (!dirValid(dir)) return false;
+    dir_entry_t *it = NULL;
+    while((it = dirNext(dir))) {
+        if (it->type == FS_TYPE_FILE) {
+            str_t file_path = strFromFmt("%s/%s", path, it->name.buf);
+            if (remove(file_path.buf)) {
+                err("couldn't remove %s > %s", file_path.buf, strerror(errno));
+            }
+            strFree(file_path);
+        }
+        else if (it->type == FS_TYPE_DIR) {
+            if (strcmp(it->name.buf, ".") == 0 || strcmp(it->name.buf, "..") == 0) {
+                continue;
+            }
+            str_t new_path = strFromFmt("%s/%s", path, it->name.buf);
+            info("new path: %s--%s -> %s", path, it->name.buf, new_path.buf);
+            if (!dirRemove(new_path.buf)) {
+                err("couldn't delete folder %s", new_path.buf);
+                break;
+            }
+            strFree(new_path);
+        }
+        else {
+            err("%d -> %s", it->type, it->name.buf);
+        }
+    }
+    dirClose(dir);
+#if COLLA_WIN
+    return RemoveDirectoryA(path);
+#else
+    return rmdir(path) == 0;
+#endif
+}