#include <stdio.h>
#include <sys/stat.h>
#include <string.h>

#include "unicode/locid.h"
#include "utils/Log.h"

#include <vector>
#include <minikin/Hyphenator.h>

using minikin::HyphenationType;
using minikin::Hyphenator;

Hyphenator* loadHybFile(const char* fn, int minPrefix, int minSuffix) {
    struct stat statbuf;
    int status = stat(fn, &statbuf);
    if (status < 0) {
        fprintf(stderr, "error opening %s\n", fn);
        return nullptr;
    }
    size_t size = statbuf.st_size;
    FILE* f = fopen(fn, "rb");
    if (f == NULL) {
        fprintf(stderr, "error opening %s\n", fn);
        return nullptr;
    }
    uint8_t* buf = new uint8_t[size];
    size_t read_size = fread(buf, 1, size, f);
    fclose(f);
    if (read_size < size) {
        fprintf(stderr, "error reading %s\n", fn);
        delete[] buf;
        return nullptr;
    }
    return Hyphenator::loadBinary(buf, minPrefix, minSuffix);
}

int main(int argc, char** argv) {
    Hyphenator* hyph = loadHybFile("/tmp/en.hyb", 2, 3);  // should also be configurable
    std::vector<HyphenationType> result;
    std::vector<uint16_t> word;
    if (argc < 2) {
        fprintf(stderr, "usage: hyphtool word\n");
        return 1;
    }
    char* asciiword = argv[1];
    size_t len = strlen(asciiword);
    for (size_t i = 0; i < len; i++) {
        uint32_t c = asciiword[i];
        if (c == '-') {
            c = 0x00AD;
        }
        // ASCII (or possibly ISO Latin 1), but kinda painful to do utf conversion :(
        word.push_back(c);
    }
    hyph->hyphenate(&result, word.data(), word.size(), icu::Locale::getUS());
    for (size_t i = 0; i < len; i++) {
        if (result[i] != HyphenationType::DONT_BREAK) {
            printf("-");
        }
        printf("%c", word[i]);
    }
    printf("\n");
    return 0;
}