Files
Doppelspaltapp-C-Raylib/nob.h
T
2024-06-11 08:31:59 +02:00

1175 lines
37 KiB
C

// This is a complete backward incompatible rewrite of https://github.com/tsoding/nobuild
// because I'm really unhappy with the direction it is going. It's gonna sit in this repo
// until it's matured enough and then I'll probably extract it to its own repo.
// Copyright 2023 Alexey Kutepov <reximkut@gmail.com>
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#ifndef NOB_H_
#define NOB_H_
#define NOB_ASSERT assert
#define NOB_REALLOC realloc
#define NOB_FREE free
#include <assert.h>
#include <stdbool.h>
#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#ifdef _WIN32
# define WIN32_LEAN_AND_MEAN
# include <windows.h>
# include <direct.h>
# include <shellapi.h>
#else
# include <sys/types.h>
# include <sys/wait.h>
# include <sys/stat.h>
# include <unistd.h>
# include <fcntl.h>
#endif
#ifdef _WIN32
# define NOB_LINE_END "\r\n"
#else
# define NOB_LINE_END "\n"
#endif
#define NOB_ARRAY_LEN(array) (sizeof(array)/sizeof(array[0]))
#define NOB_ARRAY_GET(array, index) \
(NOB_ASSERT(index >= 0), NOB_ASSERT(index < NOB_ARRAY_LEN(array)), array[index])
typedef enum {
NOB_INFO,
NOB_WARNING,
NOB_ERROR,
} Nob_Log_Level;
void nob_log(Nob_Log_Level level, const char *fmt, ...);
// It is an equivalent of shift command from bash. It basically pops a command line
// argument from the beginning.
char *nob_shift_args(int *argc, char ***argv);
typedef struct {
const char **items;
size_t count;
size_t capacity;
} Nob_File_Paths;
typedef enum {
NOB_FILE_REGULAR = 0,
NOB_FILE_DIRECTORY,
NOB_FILE_SYMLINK,
NOB_FILE_OTHER,
} Nob_File_Type;
bool nob_mkdir_if_not_exists(const char *path);
bool nob_copy_file(const char *src_path, const char *dst_path);
bool nob_copy_directory_recursively(const char *src_path, const char *dst_path);
bool nob_read_entire_dir(const char *parent, Nob_File_Paths *children);
bool nob_write_entire_file(const char *path, void *data, size_t size);
Nob_File_Type nob_get_file_type(const char *path);
#define nob_return_defer(value) do { result = (value); goto defer; } while(0)
// Initial capacity of a dynamic array
#define NOB_DA_INIT_CAP 256
// Append an item to a dynamic array
#define nob_da_append(da, item) \
do { \
if ((da)->count >= (da)->capacity) { \
(da)->capacity = (da)->capacity == 0 ? NOB_DA_INIT_CAP : (da)->capacity*2; \
(da)->items = NOB_REALLOC((da)->items, (da)->capacity*sizeof(*(da)->items)); \
NOB_ASSERT((da)->items != NULL && "Buy more RAM lol"); \
} \
\
(da)->items[(da)->count++] = (item); \
} while (0)
#define nob_da_free(da) NOB_FREE((da).items)
// Append several items to a dynamic array
#define nob_da_append_many(da, new_items, new_items_count) \
do { \
if ((da)->count + new_items_count > (da)->capacity) { \
if ((da)->capacity == 0) { \
(da)->capacity = NOB_DA_INIT_CAP; \
} \
while ((da)->count + new_items_count > (da)->capacity) { \
(da)->capacity *= 2; \
} \
(da)->items = NOB_REALLOC((da)->items, (da)->capacity*sizeof(*(da)->items)); \
NOB_ASSERT((da)->items != NULL && "Buy more RAM lol"); \
} \
memcpy((da)->items + (da)->count, new_items, new_items_count*sizeof(*(da)->items)); \
(da)->count += new_items_count; \
} while (0)
typedef struct {
char *items;
size_t count;
size_t capacity;
} Nob_String_Builder;
bool nob_read_entire_file(const char *path, Nob_String_Builder *sb);
// Append a sized buffer to a string builder
#define nob_sb_append_buf(sb, buf, size) nob_da_append_many(sb, buf, size)
// Append a NULL-terminated string to a string builder
#define nob_sb_append_cstr(sb, cstr) \
do { \
const char *s = (cstr); \
size_t n = strlen(s); \
nob_da_append_many(sb, s, n); \
} while (0)
// Append a single NULL character at the end of a string builder. So then you can
// use it a NULL-terminated C string
#define nob_sb_append_null(sb) nob_da_append_many(sb, "", 1)
// Free the memory allocated by a string builder
#define nob_sb_free(sb) NOB_FREE((sb).items)
// Process handle
#ifdef _WIN32
typedef HANDLE Nob_Proc;
#define NOB_INVALID_PROC INVALID_HANDLE_VALUE
#else
typedef int Nob_Proc;
#define NOB_INVALID_PROC (-1)
#endif // _WIN32
typedef struct {
Nob_Proc *items;
size_t count;
size_t capacity;
} Nob_Procs;
bool nob_procs_wait(Nob_Procs procs);
// Wait until the process has finished
bool nob_proc_wait(Nob_Proc proc);
// A command - the main workhorse of Nob. Nob is all about building commands an running them
typedef struct {
const char **items;
size_t count;
size_t capacity;
} Nob_Cmd;
// Render a string representation of a command into a string builder. Keep in mind the the
// string builder is not NULL-terminated by default. Use nob_sb_append_null if you plan to
// use it as a C string.
void nob_cmd_render(Nob_Cmd cmd, Nob_String_Builder *render);
#define nob_cmd_append(cmd, ...) \
nob_da_append_many(cmd, ((const char*[]){__VA_ARGS__}), (sizeof((const char*[]){__VA_ARGS__})/sizeof(const char*)))
// Free all the memory allocated by command arguments
#define nob_cmd_free(cmd) NOB_FREE(cmd.items)
// Run command asynchronously
Nob_Proc nob_cmd_run_async(Nob_Cmd cmd);
// Run command synchronously
bool nob_cmd_run_sync(Nob_Cmd cmd);
#ifndef NOB_TEMP_CAPACITY
#define NOB_TEMP_CAPACITY (8*1024*1024)
#endif // NOB_TEMP_CAPACITY
char *nob_temp_strdup(const char *cstr);
void *nob_temp_alloc(size_t size);
char *nob_temp_sprintf(const char *format, ...);
void nob_temp_reset(void);
size_t nob_temp_save(void);
void nob_temp_rewind(size_t checkpoint);
int is_path1_modified_after_path2(const char *path1, const char *path2);
bool nob_rename(const char *old_path, const char *new_path);
int nob_needs_rebuild(const char *output_path, const char **input_paths, size_t input_paths_count);
int nob_needs_rebuild1(const char *output_path, const char *input_path);
int nob_file_exists(const char *file_path);
// TODO: add MinGW support for Go Rebuild Urself™ Technology
#ifndef NOB_REBUILD_URSELF
# if _WIN32
# if defined(__GNUC__)
# define NOB_REBUILD_URSELF(binary_path, source_path) "gcc", "-o", binary_path, source_path
# elif defined(__clang__)
# define NOB_REBUILD_URSELF(binary_path, source_path) "clang", "-o", binary_path, source_path
# elif defined(_MSC_VER)
# define NOB_REBUILD_URSELF(binary_path, source_path) "cl.exe", source_path
# endif
# else
# define NOB_REBUILD_URSELF(binary_path, source_path) "cc", "-o", binary_path, source_path
# endif
#endif
// Go Rebuild Urself™ Technology
//
// How to use it:
// int main(int argc, char** argv) {
// GO_REBUILD_URSELF(argc, argv);
// // actual work
// return 0;
// }
//
// After your added this macro every time you run ./nobuild it will detect
// that you modified its original source code and will try to rebuild itself
// before doing any actual work. So you only need to bootstrap your build system
// once.
//
// The modification is detected by comparing the last modified times of the executable
// and its source code. The same way the make utility usually does it.
//
// The rebuilding is done by using the REBUILD_URSELF macro which you can redefine
// if you need a special way of bootstraping your build system. (which I personally
// do not recommend since the whole idea of nobuild is to keep the process of bootstrapping
// as simple as possible and doing all of the actual work inside of the nobuild)
//
#define NOB_GO_REBUILD_URSELF(argc, argv) \
do { \
const char *source_path = __FILE__; \
assert(argc >= 1); \
const char *binary_path = argv[0]; \
\
int rebuild_is_needed = nob_needs_rebuild(binary_path, &source_path, 1); \
if (rebuild_is_needed < 0) exit(1); \
if (rebuild_is_needed) { \
Nob_String_Builder sb = {0}; \
nob_sb_append_cstr(&sb, binary_path); \
nob_sb_append_cstr(&sb, ".old"); \
nob_sb_append_null(&sb); \
\
if (!nob_rename(binary_path, sb.items)) exit(1); \
Nob_Cmd rebuild = {0}; \
nob_cmd_append(&rebuild, NOB_REBUILD_URSELF(binary_path, source_path)); \
bool rebuild_succeeded = nob_cmd_run_sync(rebuild); \
nob_cmd_free(rebuild); \
if (!rebuild_succeeded) { \
nob_rename(sb.items, binary_path); \
exit(1); \
} \
\
Nob_Cmd cmd = {0}; \
nob_da_append_many(&cmd, argv, argc); \
if (!nob_cmd_run_sync(cmd)) exit(1); \
exit(0); \
} \
} while(0)
// The implementation idea is stolen from https://github.com/zhiayang/nabs
typedef struct {
size_t count;
const char *data;
} Nob_String_View;
const char *nob_temp_sv_to_cstr(Nob_String_View sv);
Nob_String_View nob_sv_chop_by_delim(Nob_String_View *sv, char delim);
Nob_String_View nob_sv_chop_by_space(Nob_String_View *sv);
Nob_String_View nob_sv_trim(Nob_String_View sv);
bool nob_sv_eq(Nob_String_View a, Nob_String_View b);
Nob_String_View nob_sv_from_cstr(const char *cstr);
Nob_String_View nob_sv_from_parts(const char *data, size_t count);
// printf macros for String_View
#ifndef SV_Fmt
#define SV_Fmt "%.*s"
#endif // SV_Fmt
#ifndef SV_Arg
#define SV_Arg(sv) (int) (sv).count, (sv).data
#endif // SV_Arg
// USAGE:
// String_View name = ...;
// printf("Name: "SV_Fmt"\n", SV_Arg(name));
// minirent.h HEADER BEGIN ////////////////////////////////////////
// Copyright 2021 Alexey Kutepov <reximkut@gmail.com>
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
// ============================================================
//
// minirent — 0.0.1 — A subset of dirent interface for Windows.
//
// https://github.com/tsoding/minirent
//
// ============================================================
//
// ChangeLog (https://semver.org/ is implied)
//
// 0.0.2 Automatically include dirent.h on non-Windows
// platforms
// 0.0.1 First Official Release
#ifndef _WIN32
#include <dirent.h>
#else // _WIN32
#define WIN32_LEAN_AND_MEAN
#include "windows.h"
struct dirent
{
char d_name[MAX_PATH+1];
};
typedef struct DIR DIR;
DIR *opendir(const char *dirpath);
struct dirent *readdir(DIR *dirp);
int closedir(DIR *dirp);
#endif // _WIN32
// minirent.h HEADER END ////////////////////////////////////////
#endif // NOB_H_
#ifdef NOB_IMPLEMENTATION
static size_t nob_temp_size = 0;
static char nob_temp[NOB_TEMP_CAPACITY] = {0};
bool nob_mkdir_if_not_exists(const char *path)
{
#ifdef _WIN32
int result = mkdir(path);
#else
int result = mkdir(path, 0755);
#endif
if (result < 0) {
if (errno == EEXIST) {
nob_log(NOB_INFO, "directory `%s` already exists", path);
return true;
}
nob_log(NOB_ERROR, "could not create directory `%s`: %s", path, strerror(errno));
return false;
}
nob_log(NOB_INFO, "created directory `%s`", path);
return true;
}
bool nob_copy_file(const char *src_path, const char *dst_path)
{
nob_log(NOB_INFO, "copying %s -> %s", src_path, dst_path);
#ifdef _WIN32
if (!CopyFile(src_path, dst_path, FALSE)) {
nob_log(NOB_ERROR, "Could not copy file: %lu", GetLastError());
return false;
}
return true;
#else
int src_fd = -1;
int dst_fd = -1;
size_t buf_size = 32*1024;
char *buf = NOB_REALLOC(NULL, buf_size);
NOB_ASSERT(buf != NULL && "Buy more RAM lol!!");
bool result = true;
src_fd = open(src_path, O_RDONLY);
if (src_fd < 0) {
nob_log(NOB_ERROR, "Could not open file %s: %s", src_path, strerror(errno));
nob_return_defer(false);
}
struct stat src_stat;
if (fstat(src_fd, &src_stat) < 0) {
nob_log(NOB_ERROR, "Could not get mode of file %s: %s", src_path, strerror(errno));
nob_return_defer(false);
}
dst_fd = open(dst_path, O_CREAT | O_TRUNC | O_WRONLY, src_stat.st_mode);
if (dst_fd < 0) {
nob_log(NOB_ERROR, "Could not create file %s: %s", dst_path, strerror(errno));
nob_return_defer(false);
}
for (;;) {
ssize_t n = read(src_fd, buf, buf_size);
if (n == 0) break;
if (n < 0) {
nob_log(NOB_ERROR, "Could not read from file %s: %s", src_path, strerror(errno));
nob_return_defer(false);
}
char *buf2 = buf;
while (n > 0) {
ssize_t m = write(dst_fd, buf2, n);
if (m < 0) {
nob_log(NOB_ERROR, "Could not write to file %s: %s", dst_path, strerror(errno));
nob_return_defer(false);
}
n -= m;
buf2 += m;
}
}
defer:
free(buf);
close(src_fd);
close(dst_fd);
return result;
#endif
}
void nob_cmd_render(Nob_Cmd cmd, Nob_String_Builder *render)
{
for (size_t i = 0; i < cmd.count; ++i) {
const char *arg = cmd.items[i];
if (arg == NULL) break;
if (i > 0) nob_sb_append_cstr(render, " ");
if (!strchr(arg, ' ')) {
nob_sb_append_cstr(render, arg);
} else {
nob_da_append(render, '\'');
nob_sb_append_cstr(render, arg);
nob_da_append(render, '\'');
}
}
}
Nob_Proc nob_cmd_run_async(Nob_Cmd cmd)
{
if (cmd.count < 1) {
nob_log(NOB_ERROR, "Could not run empty command");
return NOB_INVALID_PROC;
}
Nob_String_Builder sb = {0};
nob_cmd_render(cmd, &sb);
nob_sb_append_null(&sb);
nob_log(NOB_INFO, "CMD: %s", sb.items);
nob_sb_free(sb);
memset(&sb, 0, sizeof(sb));
#ifdef _WIN32
// https://docs.microsoft.com/en-us/windows/win32/procthread/creating-a-child-process-with-redirected-input-and-output
STARTUPINFO siStartInfo;
ZeroMemory(&siStartInfo, sizeof(siStartInfo));
siStartInfo.cb = sizeof(STARTUPINFO);
// NOTE: theoretically setting NULL to std handles should not be a problem
// https://docs.microsoft.com/en-us/windows/console/getstdhandle?redirectedfrom=MSDN#attachdetach-behavior
// TODO: check for errors in GetStdHandle
siStartInfo.hStdError = GetStdHandle(STD_ERROR_HANDLE);
siStartInfo.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
siStartInfo.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
siStartInfo.dwFlags |= STARTF_USESTDHANDLES;
PROCESS_INFORMATION piProcInfo;
ZeroMemory(&piProcInfo, sizeof(PROCESS_INFORMATION));
// TODO: use a more reliable rendering of the command instead of cmd_render
// cmd_render is for logging primarily
nob_cmd_render(cmd, &sb);
nob_sb_append_null(&sb);
BOOL bSuccess = CreateProcessA(NULL, sb.items, NULL, NULL, TRUE, 0, NULL, NULL, &siStartInfo, &piProcInfo);
nob_sb_free(sb);
if (!bSuccess) {
nob_log(NOB_ERROR, "Could not create child process: %lu", GetLastError());
return NOB_INVALID_PROC;
}
CloseHandle(piProcInfo.hThread);
return piProcInfo.hProcess;
#else
pid_t cpid = fork();
if (cpid < 0) {
nob_log(NOB_ERROR, "Could not fork child process: %s", strerror(errno));
return NOB_INVALID_PROC;
}
if (cpid == 0) {
// NOTE: This leaks a bit of memory in the child process.
// But do we actually care? It's a one off leak anyway...
Nob_Cmd cmd_null = {0};
nob_da_append_many(&cmd_null, cmd.items, cmd.count);
nob_cmd_append(&cmd_null, NULL);
if (execvp(cmd.items[0], (char * const*) cmd_null.items) < 0) {
nob_log(NOB_ERROR, "Could not exec child process: %s", strerror(errno));
exit(1);
}
NOB_ASSERT(0 && "unreachable");
}
return cpid;
#endif
}
bool nob_procs_wait(Nob_Procs procs)
{
bool success = true;
for (size_t i = 0; i < procs.count; ++i) {
success = nob_proc_wait(procs.items[i]) && success;
}
return success;
}
bool nob_proc_wait(Nob_Proc proc)
{
if (proc == NOB_INVALID_PROC) return false;
#ifdef _WIN32
DWORD result = WaitForSingleObject(
proc, // HANDLE hHandle,
INFINITE // DWORD dwMilliseconds
);
if (result == WAIT_FAILED) {
nob_log(NOB_ERROR, "could not wait on child process: %lu", GetLastError());
return false;
}
DWORD exit_status;
if (!GetExitCodeProcess(proc, &exit_status)) {
nob_log(NOB_ERROR, "could not get process exit code: %lu", GetLastError());
return false;
}
if (exit_status != 0) {
nob_log(NOB_ERROR, "command exited with exit code %lu", exit_status);
return false;
}
CloseHandle(proc);
return true;
#else
for (;;) {
int wstatus = 0;
if (waitpid(proc, &wstatus, 0) < 0) {
nob_log(NOB_ERROR, "could not wait on command (pid %d): %s", proc, strerror(errno));
return false;
}
if (WIFEXITED(wstatus)) {
int exit_status = WEXITSTATUS(wstatus);
if (exit_status != 0) {
nob_log(NOB_ERROR, "command exited with exit code %d", exit_status);
return false;
}
break;
}
if (WIFSIGNALED(wstatus)) {
nob_log(NOB_ERROR, "command process was terminated by %s", strsignal(WTERMSIG(wstatus)));
return false;
}
}
return true;
#endif
}
bool nob_cmd_run_sync(Nob_Cmd cmd)
{
Nob_Proc p = nob_cmd_run_async(cmd);
if (p == NOB_INVALID_PROC) return false;
return nob_proc_wait(p);
}
char *nob_shift_args(int *argc, char ***argv)
{
NOB_ASSERT(*argc > 0);
char *result = **argv;
(*argv) += 1;
(*argc) -= 1;
return result;
}
void nob_log(Nob_Log_Level level, const char *fmt, ...)
{
switch (level) {
case NOB_INFO:
fprintf(stderr, "[INFO] ");
break;
case NOB_WARNING:
fprintf(stderr, "[WARNING] ");
break;
case NOB_ERROR:
fprintf(stderr, "[ERROR] ");
break;
default:
NOB_ASSERT(0 && "unreachable");
}
va_list args;
va_start(args, fmt);
vfprintf(stderr, fmt, args);
va_end(args);
fprintf(stderr, "\n");
}
bool nob_read_entire_dir(const char *parent, Nob_File_Paths *children)
{
bool result = true;
DIR *dir = NULL;
dir = opendir(parent);
if (dir == NULL) {
nob_log(NOB_ERROR, "Could not open directory %s: %s", parent, strerror(errno));
nob_return_defer(false);
}
errno = 0;
struct dirent *ent = readdir(dir);
while (ent != NULL) {
nob_da_append(children, nob_temp_strdup(ent->d_name));
ent = readdir(dir);
}
if (errno != 0) {
nob_log(NOB_ERROR, "Could not read directory %s: %s", parent, strerror(errno));
nob_return_defer(false);
}
defer:
if (dir) closedir(dir);
return result;
}
bool nob_write_entire_file(const char *path, void *data, size_t size)
{
bool result = true;
FILE *f = fopen(path, "wb");
if (f == NULL) {
nob_log(NOB_ERROR, "Could not open file %s for writing: %s\n", path, strerror(errno));
nob_return_defer(false);
}
// len
// v
// aaaaaaaaaa
// ^
// data
char *buf = data;
while (size > 0) {
size_t n = fwrite(buf, 1, size, f);
if (ferror(f)) {
nob_log(NOB_ERROR, "Could not write into file %s: %s\n", path, strerror(errno));
nob_return_defer(false);
}
size -= n;
buf += n;
}
defer:
if (f) fclose(f);
return result;
}
Nob_File_Type nob_get_file_type(const char *path)
{
#ifdef _WIN32
DWORD attr = GetFileAttributesA(path);
if (attr == INVALID_FILE_ATTRIBUTES) {
nob_log(NOB_ERROR, "Could not get file attributes of %s: %lu", path, GetLastError());
return -1;
}
if (attr & FILE_ATTRIBUTE_DIRECTORY) return NOB_FILE_DIRECTORY;
// TODO: detect symlinks on Windows (whatever that means on Windows anyway)
return NOB_FILE_REGULAR;
#else // _WIN32
struct stat statbuf;
if (stat(path, &statbuf) < 0) {
nob_log(NOB_ERROR, "Could not get stat of %s: %s", path, strerror(errno));
return -1;
}
switch (statbuf.st_mode & S_IFMT) {
case S_IFDIR: return NOB_FILE_DIRECTORY;
case S_IFREG: return NOB_FILE_REGULAR;
case S_IFLNK: return NOB_FILE_SYMLINK;
default: return NOB_FILE_OTHER;
}
#endif // _WIN32
}
bool nob_copy_directory_recursively(const char *src_path, const char *dst_path)
{
bool result = true;
Nob_File_Paths children = {0};
Nob_String_Builder src_sb = {0};
Nob_String_Builder dst_sb = {0};
size_t temp_checkpoint = nob_temp_save();
Nob_File_Type type = nob_get_file_type(src_path);
if (type < 0) return false;
switch (type) {
case NOB_FILE_DIRECTORY: {
if (!nob_mkdir_if_not_exists(dst_path)) nob_return_defer(false);
if (!nob_read_entire_dir(src_path, &children)) nob_return_defer(false);
for (size_t i = 0; i < children.count; ++i) {
if (strcmp(children.items[i], ".") == 0) continue;
if (strcmp(children.items[i], "..") == 0) continue;
src_sb.count = 0;
nob_sb_append_cstr(&src_sb, src_path);
nob_sb_append_cstr(&src_sb, "/");
nob_sb_append_cstr(&src_sb, children.items[i]);
nob_sb_append_null(&src_sb);
dst_sb.count = 0;
nob_sb_append_cstr(&dst_sb, dst_path);
nob_sb_append_cstr(&dst_sb, "/");
nob_sb_append_cstr(&dst_sb, children.items[i]);
nob_sb_append_null(&dst_sb);
if (!nob_copy_directory_recursively(src_sb.items, dst_sb.items)) {
nob_return_defer(false);
}
}
} break;
case NOB_FILE_REGULAR: {
if (!nob_copy_file(src_path, dst_path)) {
nob_return_defer(false);
}
} break;
case NOB_FILE_SYMLINK: {
nob_log(NOB_WARNING, "TODO: Copying symlinks is not supported yet");
} break;
case NOB_FILE_OTHER: {
nob_log(NOB_ERROR, "Unsupported type of file %s", src_path);
nob_return_defer(false);
} break;
default: NOB_ASSERT(0 && "unreachable");
}
defer:
nob_temp_rewind(temp_checkpoint);
nob_da_free(src_sb);
nob_da_free(dst_sb);
nob_da_free(children);
return result;
}
char *nob_temp_strdup(const char *cstr)
{
size_t n = strlen(cstr);
char *result = nob_temp_alloc(n + 1);
NOB_ASSERT(result != NULL && "Increase NOB_TEMP_CAPACITY");
memcpy(result, cstr, n);
result[n] = '\0';
return result;
}
void *nob_temp_alloc(size_t size)
{
if (nob_temp_size + size > NOB_TEMP_CAPACITY) return NULL;
void *result = &nob_temp[nob_temp_size];
nob_temp_size += size;
return result;
}
char *nob_temp_sprintf(const char *format, ...)
{
va_list args;
va_start(args, format);
int n = vsnprintf(NULL, 0, format, args);
va_end(args);
NOB_ASSERT(n >= 0);
char *result = nob_temp_alloc(n + 1);
NOB_ASSERT(result != NULL && "Extend the size of the temporary allocator");
// TODO: use proper arenas for the temporary allocator;
va_start(args, format);
vsnprintf(result, n + 1, format, args);
va_end(args);
return result;
}
void nob_temp_reset(void)
{
nob_temp_size = 0;
}
size_t nob_temp_save(void)
{
return nob_temp_size;
}
void nob_temp_rewind(size_t checkpoint)
{
nob_temp_size = checkpoint;
}
const char *nob_temp_sv_to_cstr(Nob_String_View sv)
{
char *result = nob_temp_alloc(sv.count + 1);
NOB_ASSERT(result != NULL && "Extend the size of the temporary allocator");
memcpy(result, sv.data, sv.count);
result[sv.count] = '\0';
return result;
}
int nob_needs_rebuild(const char *output_path, const char **input_paths, size_t input_paths_count)
{
#ifdef _WIN32
BOOL bSuccess;
HANDLE output_path_fd = CreateFile(output_path, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
if (output_path_fd == INVALID_HANDLE_VALUE) {
// NOTE: if output does not exist it 100% must be rebuilt
if (GetLastError() == ERROR_FILE_NOT_FOUND) return 1;
nob_log(NOB_ERROR, "Could not open file %s: %lu", output_path, GetLastError());
return -1;
}
FILETIME output_path_time;
bSuccess = GetFileTime(output_path_fd, NULL, NULL, &output_path_time);
CloseHandle(output_path_fd);
if (!bSuccess) {
nob_log(NOB_ERROR, "Could not get time of %s: %lu", output_path, GetLastError());
return -1;
}
for (size_t i = 0; i < input_paths_count; ++i) {
const char *input_path = input_paths[i];
HANDLE input_path_fd = CreateFile(input_path, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
if (input_path_fd == INVALID_HANDLE_VALUE) {
// NOTE: non-existing input is an error cause it is needed for building in the first place
nob_log(NOB_ERROR, "Could not open file %s: %lu", input_path, GetLastError());
return -1;
}
FILETIME input_path_time;
bSuccess = GetFileTime(input_path_fd, NULL, NULL, &input_path_time);
CloseHandle(input_path_fd);
if (!bSuccess) {
nob_log(NOB_ERROR, "Could not get time of %s: %lu", input_path, GetLastError());
return -1;
}
// NOTE: if even a single input_path is fresher than output_path that's 100% rebuild
if (CompareFileTime(&input_path_time, &output_path_time) == 1) return 1;
}
return 0;
#else
struct stat statbuf = {0};
if (stat(output_path, &statbuf) < 0) {
// NOTE: if output does not exist it 100% must be rebuilt
if (errno == ENOENT) return 1;
nob_log(NOB_ERROR, "could not stat %s: %s", output_path, strerror(errno));
return -1;
}
int output_path_time = statbuf.st_mtime;
for (size_t i = 0; i < input_paths_count; ++i) {
const char *input_path = input_paths[i];
if (stat(input_path, &statbuf) < 0) {
// NOTE: non-existing input is an error cause it is needed for building in the first place
nob_log(NOB_ERROR, "could not stat %s: %s", input_path, strerror(errno));
return -1;
}
int input_path_time = statbuf.st_mtime;
// NOTE: if even a single input_path is fresher than output_path that's 100% rebuild
if (input_path_time > output_path_time) return 1;
}
return 0;
#endif
}
int nob_needs_rebuild1(const char *output_path, const char *input_path)
{
return nob_needs_rebuild(output_path, &input_path, 1);
}
bool nob_rename(const char *old_path, const char *new_path)
{
nob_log(NOB_INFO, "renaming %s -> %s", old_path, new_path);
#ifdef _WIN32
if (!MoveFileEx(old_path, new_path, MOVEFILE_REPLACE_EXISTING)) {
nob_log(NOB_ERROR, "could not rename %s to %s: %lu", old_path, new_path, GetLastError());
return false;
}
#else
if (rename(old_path, new_path) < 0) {
nob_log(NOB_ERROR, "could not rename %s to %s: %s", old_path, new_path, strerror(errno));
return false;
}
#endif // _WIN32
return true;
}
bool nob_read_entire_file(const char *path, Nob_String_Builder *sb)
{
bool result = true;
size_t buf_size = 32*1024;
char *buf = NOB_REALLOC(NULL, buf_size);
NOB_ASSERT(buf != NULL && "Buy more RAM lool!!");
FILE *f = fopen(path, "rb");
if (f == NULL) {
nob_log(NOB_ERROR, "Could not open %s for reading: %s", path, strerror(errno));
nob_return_defer(false);
}
size_t n = fread(buf, 1, buf_size, f);
while (n > 0) {
nob_sb_append_buf(sb, buf, n);
n = fread(buf, 1, buf_size, f);
}
if (ferror(f)) {
nob_log(NOB_ERROR, "Could not read %s: %s\n", path, strerror(errno));
nob_return_defer(false);
}
defer:
NOB_FREE(buf);
if (f) fclose(f);
return result;
}
Nob_String_View nob_sv_chop_by_space(Nob_String_View *sv)
{
size_t i = 0;
while (i < sv->count && !isspace(sv->data[i])) {
i += 1;
}
Nob_String_View result = nob_sv_from_parts(sv->data, i);
if (i < sv->count) {
sv->count -= i + 1;
sv->data += i + 1;
} else {
sv->count -= i;
sv->data += i;
}
return result;
}
Nob_String_View nob_sv_chop_by_delim(Nob_String_View *sv, char delim)
{
size_t i = 0;
while (i < sv->count && sv->data[i] != delim) {
i += 1;
}
Nob_String_View result = nob_sv_from_parts(sv->data, i);
if (i < sv->count) {
sv->count -= i + 1;
sv->data += i + 1;
} else {
sv->count -= i;
sv->data += i;
}
return result;
}
Nob_String_View nob_sv_from_parts(const char *data, size_t count)
{
Nob_String_View sv;
sv.count = count;
sv.data = data;
return sv;
}
Nob_String_View nob_sv_trim_left(Nob_String_View sv)
{
size_t i = 0;
while (i < sv.count && isspace(sv.data[i])) {
i += 1;
}
return nob_sv_from_parts(sv.data + i, sv.count - i);
}
Nob_String_View nob_sv_trim_right(Nob_String_View sv)
{
size_t i = 0;
while (i < sv.count && isspace(sv.data[sv.count - 1 - i])) {
i += 1;
}
return nob_sv_from_parts(sv.data, sv.count - i);
}
Nob_String_View nob_sv_trim(Nob_String_View sv)
{
return nob_sv_trim_right(nob_sv_trim_left(sv));
}
Nob_String_View nob_sv_from_cstr(const char *cstr)
{
return nob_sv_from_parts(cstr, strlen(cstr));
}
bool nob_sv_eq(Nob_String_View a, Nob_String_View b)
{
if (a.count != b.count) {
return false;
} else {
return memcmp(a.data, b.data, a.count) == 0;
}
}
// RETURNS:
// 0 - file does not exists
// 1 - file exists
// -1 - error while checking if file exists. The error is logged
int nob_file_exists(const char *file_path)
{
#if _WIN32
// TODO: distinguish between "does not exists" and other errors
DWORD dwAttrib = GetFileAttributesA(file_path);
return dwAttrib != INVALID_FILE_ATTRIBUTES;
#else
struct stat statbuf;
if (stat(file_path, &statbuf) < 0) {
if (errno == ENOENT) return 0;
nob_log(NOB_ERROR, "Could not check if file %s exists: %s", file_path, strerror(errno));
return -1;
}
return 1;
#endif
}
// minirent.h SOURCE BEGIN ////////////////////////////////////////
#ifdef _WIN32
struct DIR
{
HANDLE hFind;
WIN32_FIND_DATA data;
struct dirent *dirent;
};
DIR *opendir(const char *dirpath)
{
assert(dirpath);
char buffer[MAX_PATH];
snprintf(buffer, MAX_PATH, "%s\\*", dirpath);
DIR *dir = (DIR*)calloc(1, sizeof(DIR));
dir->hFind = FindFirstFile(buffer, &dir->data);
if (dir->hFind == INVALID_HANDLE_VALUE) {
// TODO: opendir should set errno accordingly on FindFirstFile fail
// https://docs.microsoft.com/en-us/windows/win32/api/errhandlingapi/nf-errhandlingapi-getlasterror
errno = ENOSYS;
goto fail;
}
return dir;
fail:
if (dir) {
free(dir);
}
return NULL;
}
struct dirent *readdir(DIR *dirp)
{
assert(dirp);
if (dirp->dirent == NULL) {
dirp->dirent = (struct dirent*)calloc(1, sizeof(struct dirent));
} else {
if(!FindNextFile(dirp->hFind, &dirp->data)) {
if (GetLastError() != ERROR_NO_MORE_FILES) {
// TODO: readdir should set errno accordingly on FindNextFile fail
// https://docs.microsoft.com/en-us/windows/win32/api/errhandlingapi/nf-errhandlingapi-getlasterror
errno = ENOSYS;
}
return NULL;
}
}
memset(dirp->dirent->d_name, 0, sizeof(dirp->dirent->d_name));
strncpy(
dirp->dirent->d_name,
dirp->data.cFileName,
sizeof(dirp->dirent->d_name) - 1);
return dirp->dirent;
}
int closedir(DIR *dirp)
{
assert(dirp);
if(!FindClose(dirp->hFind)) {
// TODO: closedir should set errno accordingly on FindClose fail
// https://docs.microsoft.com/en-us/windows/win32/api/errhandlingapi/nf-errhandlingapi-getlasterror
errno = ENOSYS;
return -1;
}
if (dirp->dirent) {
free(dirp->dirent);
}
free(dirp);
return 0;
}
#endif // _WIN32
// minirent.h SOURCE END ////////////////////////////////////////
#endif