tarballs/nob.c at main · tsoding/tarballs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#define NOB_IMPLEMENTATION
#include "nob.h"

#define BUILD_FOLDER "build/"

bool delete_walk_entry(Walk_Entry entry)
{
    return delete_file(entry.path);
}

// TODO: we eventually should make this part of nob.h
bool delete_directory_recursively(const char *dir_path)
{
    return walk_dir(dir_path, delete_walk_entry, .post_order = true);
}

bool build_exec(Cmd *cmd, Procs *procs, const char *src_path, const char *bin_path)
{
    cmd_append(cmd, "clang");
    cmd_append(cmd, "-Wall");
    cmd_append(cmd, "-Wextra");
    cmd_append(cmd, "-ggdb");
    cmd_append(cmd, "-fsanitize=memory");
    cmd_append(cmd, "-o", bin_path);
    cmd_append(cmd, src_path);
    return cmd_run(cmd, .async = procs);
}

bool test_unpacker(Cmd *cmd)
{
    cmd_append(cmd, BUILD_FOLDER"unpacker");
    cmd_append(cmd, BUILD_FOLDER"foo.tar");
    cmd_append(cmd, BUILD_FOLDER);
    if (!cmd_run(cmd)) return false;

    cmd_append(cmd, "diff");
    cmd_append(cmd, "-u");
    cmd_append(cmd, "-r");
    cmd_append(cmd, "./foo/");
    cmd_append(cmd, BUILD_FOLDER"foo/");
    if (!cmd_run(cmd)) return false;

    return true;
}

bool test_packer(Cmd *cmd)
{
    cmd_append(cmd, BUILD_FOLDER"packer");
    cmd_append(cmd, BUILD_FOLDER"bar.tar");
    if (!cmd_run(cmd)) return false;

    cmd_append(cmd, "tar");
    cmd_append(cmd, "fvx");
    cmd_append(cmd, BUILD_FOLDER"bar.tar");
    cmd_append(cmd, "-C", BUILD_FOLDER);
    if (!cmd_run(cmd)) return false;

    return true;
}

int main(int argc, char **argv)
{
    GO_REBUILD_URSELF(argc, argv);

    Cmd cmd = {0};
    Procs procs = {0};

    if (file_exists(BUILD_FOLDER)) {
        if (!delete_directory_recursively(BUILD_FOLDER)) return false;
    }
    if (!mkdir_if_not_exists(BUILD_FOLDER)) return 1;

    cmd_append(&cmd, "tar");
    cmd_append(&cmd, "--create");
    cmd_append(&cmd, "--file");
    cmd_append(&cmd, BUILD_FOLDER"foo.tar");
    cmd_append(&cmd, "./foo/");
    if (!cmd_run(&cmd, .async = &procs)) return 1;

    if (!build_exec(&cmd, &procs, "unpacker.c", BUILD_FOLDER"unpacker")) return 1;
    if (!build_exec(&cmd, &procs, "packer.c", BUILD_FOLDER"packer")) return 1;

    if (!procs_flush(&procs)) return 1;

    if (!test_unpacker(&cmd)) return 1;
    if (!test_packer(&cmd)) return 1;

    return 0;
}