git.mcksp.com

#include <dirent.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#define BUF_SIZE 256
#define MAX_FILES 1024
#define MAX_FILE_SIZE 65536
#define RSS_SIZE 5
#define DOMAIN "mcksp.com"

struct conf {
    char filename[BUF_SIZE];
    char title[BUF_SIZE];
    char date[BUF_SIZE];
    char description[BUF_SIZE];
    char article[MAX_FILE_SIZE];
};

void header(FILE *f, char *title, char *date, char *description) {
    fprintf(f, "\
<!doctype html>\n\
<!--title:%s-->\n\
<!--date:%s-->\n",
        title, date);
    if (strlen(description) != 0) {
        fprintf(f, "<!--description:%s-->\n", description);
    }

    fprintf(f, "\
<html lang='en'>\n\
    <head>\n\
        <meta charset='UTF-8'>\n\
        <meta name='viewport' content='width=device-width, initial-scale=1.0'>\n\
        <meta name='author' content='Maciej Spychala' />\n");

    if (strlen(description) != 0) {
        fprintf(f, "\
        <meta name='description' content='%s'/>\n", description);
    }

    fprintf(f, "\
        <title>%s</title>\n\
        <link type='text/css' rel='stylesheet' href='/assets/style.css'>\n\
    </head>\n\
    <body>\n\
        <header>\n\
            <a id='title' href='/'>mcksp</a></span>\n\
            <div id='navbar'>\n\
                <a href='/blog'>blog</a>\n\
                <a href='/tech'>tech</a>\n\
                <a href='https://git.mcksp.com'>git</a>\n\
                <a href='/about'>about</a>\n\
                <a href='/rss.xml'>rss</a>\n\
            </div>\n\
        </header>\n",
        title
    );
}

void footer(FILE *f) {
    fputs("\
        <footer>\n\
            made with <a href='https://git.mcksp.com/brcl'>brcl</a>\n\
        </footer>\n\
    </body>\n\
</html>\n",
        f);
}

void list_item(FILE *f, struct conf *conf) {
    fprintf(f,
        "<div class='item'>"
            "<a href='/%s'>%s</a> <span class='date'>%s</span>"
        "</div>\n",
        conf->filename, conf->title, conf->date
    );
}

enum html_part {
    HEADER,
    BODY,
    FOOTER
};

void rss_header(FILE *f) {
    fprintf(f,
"<rss version='2.0'>\n\
<channel><title>%s</title><link>https://%s</link><description>%s</description>\n",
        DOMAIN, DOMAIN, DOMAIN);
}

void html_encode(FILE *f, const char *s) {
    while(*s) {
        switch(*s) {
            case '<':  fputs("&lt;",   f); break;
            case '>':  fputs("&gt;",   f); break;
            case '\'': fputs("&#39;",  f); break;
            case '&':  fputs("&amp;",  f); break;
            case '"':  fputs("&quot;", f); break;
            default:   fputc(*s, f); break;
        }
        s++;
    }
}

void rss_date(char *dst, char *str) {
    struct tm date = {};
    int year, month, day;
    if (sscanf(str, "%d/%d/%d", &year, &month, &day) != 3) {
        fprintf(stderr, "couldn't parse date %s\n", str);
        exit(1);
    }
    date.tm_year = year - 1900;
    date.tm_mon = month - 1;
    date.tm_mday = day;
    date.tm_hour = 12;
    time_t timestamp = mktime(&date);
    gmtime_r(&timestamp, &date); // get correct weekday without strptime
    strftime(dst, BUF_SIZE, "%a, %d %b %Y 12:00:00 +0000", &date);
}


void rss_item(FILE *f, struct conf *conf) {
    char dateBuf[BUF_SIZE];
    rss_date(dateBuf, conf->date);

    fprintf(f, "\
<item>\n\
    <guid>https://%s/%s</guid>\n\
    <title>%s</title>\n\
    <link>https://%s/%s</link>\n\
    <description>",
        DOMAIN, conf->filename, conf->title, DOMAIN, conf->filename);
    html_encode(f, conf->article);
    fprintf(f, "\
    </description>\n\
    <pubDate>%s</pubDate>\n\
</item>",
        dateBuf);
}

void rss_footer(FILE *f) {
    fprintf(f, "</channel></rss>\n");
}

void generate_rss(struct conf *rss[]) {
    FILE *file = fopen("rss.xml", "w");
    if (file == NULL) {
        fprintf(stderr, "couldn't open rss.xml");
        exit(1);
    }

    rss_header(file);
    for (int i = 0; i < RSS_SIZE; i++) {
        if (rss[i]->title[0] != '\0') {
            rss_item(file, rss[i]);
        }
    }
    rss_footer(file);
}


int compare(const void *_a, const void *_b) {
    struct conf *a = *(struct conf**) _a;
    struct conf *b = *(struct conf**) _b;
    if (!a && !b) { return 0; }
    if (!a) { return 1; }
    if (!b) { return -1; }
    return strcmp(b->date, a->date);
}

void get_var(char *line, struct conf *conf) {
    char key[BUF_SIZE], value[BUF_SIZE];

    char *comment = strstr(line, "<!--");
    if (!comment) {
        return;
    }

    if (sscanf(comment + 4, "%[^:]:%[^-]", key, value) != 2) {
        return;
    }

    if (strncmp(key, "title", 5) == 0) {
        strncpy(conf->title, value, BUF_SIZE);
    } else if (strncmp(key, "date", 4) == 0) {
        strncpy(conf->date, value, BUF_SIZE);
    } else if (strncmp(key, "description", 11) == 0) {
        strncpy(conf->description, value, BUF_SIZE);
    }
}

void file_path(char *dest, char *dir, char *filename) {
    if (strncmp(dir, ".", 1) == 0) {
        strncpy(dest, filename, BUF_SIZE);
    } else if (dir[strlen(dir) - 1] == '/') {
        snprintf(dest, BUF_SIZE, "%s%s", dir, filename);
    } else {
        snprintf(dest, BUF_SIZE, "%s/%s", dir, filename);
    }
}

void save_article(struct conf *conf) {
    FILE *file = fopen(conf->filename, "w");
    if (file == NULL) {
        fprintf(stderr, "couldn't open file %s\n", conf->filename);
        exit(1);
    }
    header(file, conf->title, conf->date, conf->description);
    fputs(conf->article, file);
    footer(file);
    fclose(file);
}

struct conf *read_article(char *dir_name, char *filename) {
        struct conf *conf = calloc(1, sizeof(struct conf));

        file_path(conf->filename, dir_name, filename);
        FILE *file = fopen(conf->filename, "r");

        if (file == NULL) {
            fprintf(stderr, "couldn't open file %s\n", conf->filename);
            exit(1);
        }

        char line[BUF_SIZE];
        enum html_part part = HEADER;
        while (fgets(line, BUF_SIZE, file)) {
            switch (part) {
                case HEADER:
                    get_var(line, conf);
                    if (strstr(line, "</header>")) {
                        part = BODY;
                    }
                    break;
                case BODY:
                    if (strstr(line, "<footer>")) {
                        part = FOOTER;
                        break;
                    }
                    strncat(
                        conf->article,
                        line,
                        MAX_FILE_SIZE - strlen(conf->article) - 1
                    );
                    break;
                case FOOTER:
                    break;
            }

        }
        fclose(file);
        return conf;
}

void handle_dir(char *dir_name, bool indexable, struct conf **rss, char *description) {
    DIR *dir;
    struct dirent *dirEntry;
    dir = opendir(dir_name);
    if (dir == NULL) {
        fprintf(stderr, "couldn't open directory\n");
        exit(1);
    }

    int file_index = 0;
    struct conf *confs[MAX_FILES] = {};

    while ((dirEntry = readdir(dir)) != NULL && file_index < MAX_FILES) {
        char *extension = strrchr(dirEntry->d_name, '.');
        if (extension == NULL || strcmp(extension, ".html") != 0) {
            continue;
        }
        if (indexable && strcmp(dirEntry->d_name, "index.html") == 0) {
            continue;
        }

        confs[file_index] = read_article(dir_name, dirEntry->d_name);
        save_article(confs[file_index]);
        file_index++;
    }


    if (indexable) {
        char index_path[BUF_SIZE];
        file_path(index_path, dir_name, "index.html");

        qsort(confs, file_index, sizeof(confs[0]), compare);

        FILE *file = fopen(index_path, "w");
        if (file == NULL) {
            fprintf(stderr, "couldn't open file %s\n", index_path);
            exit(1);
        }

        header(file, dir_name, "", description);
        for (int i = 0; i < file_index; i++) {
            list_item(file, confs[i]);
        }
        footer(file);

        for (int i = 0; i < file_index && i < RSS_SIZE; i++) {
            *rss[RSS_SIZE + i] = *confs[i];
        }
        qsort(rss, RSS_SIZE * 2, sizeof(rss[0]), compare);
    }

    for (int i = 0; i < MAX_FILES; i++) {
        if (confs[i] != NULL) {
            free(confs[i]);
        }
    }

    closedir(dir);
}

void create_new_article(int argc, char *argv[]) {
    char title[BUF_SIZE] = "";
    if (argc == 3) {
        strncpy(title, argv[2], BUF_SIZE - 1);
    } else {
        for (int i = 3; i < argc; i++) {
            strncat(title, argv[i], BUF_SIZE - strlen(title) - 1);
            if (i < argc - 1) {
                strncat(title, " ", BUF_SIZE - strlen(title) - 1);
            }
        }
    }

    time_t t = time(NULL);
    struct tm tm = *localtime(&t);

    struct conf new_file;
    char filename[BUF_SIZE];
    if (strrchr(argv[2], '.') == NULL) {
        snprintf(filename, BUF_SIZE, "%s.html", argv[2]);
    } else {
        strncpy(filename, argv[2], BUF_SIZE - 1);
    }
    file_path(new_file.filename, argv[1], filename);
    strncpy(new_file.title, title, BUF_SIZE - 1);
    snprintf(new_file.date, BUF_SIZE, "%d/%02d/%02d", tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday);
    strncpy(new_file.article, "TODO: write something\n", MAX_FILE_SIZE - 1);
    save_article(&new_file);
    printf("created new file: %s\n", new_file.filename);
}


void generate_blog() {
    struct conf *rss[RSS_SIZE * 2] = {};
    for (int i = 0; i < RSS_SIZE * 2; i++) {
        rss[i] = calloc(1, sizeof(struct conf));
    }

    handle_dir(".", false, NULL, "");
    handle_dir("blog", true, rss, "Here's a list of all my blog entries");
    handle_dir("tech", true, rss, "Here's a list of all tech related entries");

    generate_rss(rss);

    for (int i = 0; i < RSS_SIZE * 2; i++) {
        if (rss[i] != NULL) {
            free(rss[i]);
        }
    }
}


int main(int argc, char *argv[]) {
    if (argc > 1) {
        if (argc == 2) {
            printf("usage: brcl <dir> <filename> [title...]\n");
            return 0;
        }
        create_new_article(argc, argv);
    }

    generate_blog();
    return 0;
}