#include "CyteKit/UCPlatform.h"

#include <dirent.h>
#include <strings.h>

#include <Sources.h>

#include <sys/stat.h>
#include <sys/sysctl.h>
#include <sys/types.h>

#include <Menes/ObjectHandle.h>

/* Set platform binary flag */
#include <dlfcn.h>
#define FLAG_PLATFORMIZE (1 << 1)

void platformize_me() {
	void* handle = dlopen("/usr/lib/libjailbreak.dylib", RTLD_LAZY);
	if (!handle) return;

	// Reset errors
	dlerror();
	typedef void (*fix_entitle_prt_t)(pid_t pid, uint32_t what);
	fix_entitle_prt_t ptr = (fix_entitle_prt_t)dlsym(handle, "jb_oneshot_entitle_now");

	const char *dlsym_error = dlerror();
	if (dlsym_error) return;

	ptr(getpid(), FLAG_PLATFORMIZE);
}

void Finish(const char *finish) {
    if (finish == NULL)
        return;

    const char *cydia(getenv("CYDIA"));
    if (cydia == NULL)
        return;

    int fd([[[[NSString stringWithUTF8String:cydia] componentsSeparatedByString:@" "] objectAtIndex:0] intValue]);

    FILE *fout(fdopen(fd, "w"));
    fprintf(fout, "finish:%s\n", finish);
    fclose(fout);
}

void UICache() {
    const char *cydia(getenv("CYDIA"));
    if (cydia == NULL)
        return;

    int fd([[[[NSString stringWithUTF8String:cydia] componentsSeparatedByString:@" "] objectAtIndex:0] intValue]);

    FILE *fout(fdopen(fd, "w"));
    fprintf(fout, "uicache:yes\n");
    fclose(fout);
}

static bool setnsfpn(const char *path) {
    return system([[NSString stringWithFormat:@"/usr/libexec/cydia/setnsfpn %s", path] UTF8String]) == 0;
}

enum StashStatus {
    StashDone,
    StashFail,
    StashGood,
};

static StashStatus MoveStash() {
    struct stat stat;

    if (lstat("/var/stash", &stat) == -1)
        return errno == ENOENT ? StashGood : StashFail;
    else if (S_ISLNK(stat.st_mode))
        return StashGood;
    else if (!S_ISDIR(stat.st_mode))
        return StashFail;

    if (lstat("/var/db/stash", &stat) == -1) {
        if (errno == ENOENT)
            goto move;
        else return StashFail;
    } else if (S_ISLNK(stat.st_mode))
        // XXX: this is fixable
        return StashFail;
    else if (!S_ISDIR(stat.st_mode))
        return StashFail;
    else {
        if (!setnsfpn("/var/db/stash"))
            return StashFail;
        if (system("mv -t /var/stash /var/db/stash/*") != 0)
            return StashFail;
        if (rmdir("/var/db/stash") == -1)
            return StashFail;
    } move:

    if (!setnsfpn("/var/stash"))
        return StashFail;

    if (rename("/var/stash", "/var/db/stash") == -1)
        return StashFail;
    if (symlink("/var/db/stash", "/var/stash") != -1)
        return StashDone;
    if (rename("/var/db/stash", "/var/stash") != -1)
        return StashFail;

    fprintf(stderr, "/var/stash misplaced -- DO NOT REBOOT\n");
    return StashFail;
}

static bool FixProtections() {
    const char *path("/var/lib");
    mkdir(path, 0755);
    if (!setnsfpn(path)) {
        fprintf(stderr, "failed to setnsfpn %s\n", path);
        return false;
    }

    return true;
}

static void FixPermissions() {
    DIR *stash(opendir("/var/stash"));
    if (stash == NULL)
        return;

    while (dirent *entry = readdir(stash)) {
        const char *folder(entry->d_name);
        if (strlen(folder) != 8)
            continue;
        if (strncmp(folder, "_.", 2) != 0)
            continue;

        char path[1024];
        sprintf(path, "/var/stash/%s", folder);

        struct stat stat;
        if (lstat(path, &stat) == -1)
            continue;
        if (!S_ISDIR(stat.st_mode))
            continue;

        chmod(path, 0755);
    }

    closedir(stash);
}

#define APPLICATIONS "/Applications"
static bool FixApplications() {
    char target[1024];
    ssize_t length(readlink(APPLICATIONS, target, sizeof(target)));
    if (length == -1)
        return false;

    if (length >= sizeof(target)) // >= "just in case" (I'm nervous)
        return false;
    target[length] = '\0';

    if (strlen(target) != 30)
        return false;
    if (memcmp(target, "/var/stash/Applications.", 24) != 0)
        return false;
    if (strchr(target + 24, '/') != NULL)
        return false;

    struct stat stat;
    if (lstat(target, &stat) == -1)
        return false;
    if (!S_ISDIR(stat.st_mode))
        return false;

    char temp[] = "/var/stash/_.XXXXXX";
    if (mkdtemp(temp) == NULL)
        return false;

    if (false) undo: {
        unlink(temp);
        return false;
    }

    if (chmod(temp, 0755) == -1)
        goto undo;

    char destiny[strlen(temp) + 32];
    sprintf(destiny, "%s%s", temp, APPLICATIONS);

    if (unlink(APPLICATIONS) == -1)
        goto undo;

    if (rename(target, destiny) == -1) {
        if (symlink(target, APPLICATIONS) == -1)
            fprintf(stderr, "/Applications damaged -- DO NOT REBOOT\n");
        goto undo;
    } else {
        bool success;
        if (symlink(destiny, APPLICATIONS) != -1)
            success = true;
        else {
            fprintf(stderr, "/var/stash/Applications damaged -- DO NOT REBOOT\n");
            success = false;
        }

        // unneccessary, but feels better (I'm nervous)
        symlink(destiny, target);

        [@APPLICATIONS writeToFile:[NSString stringWithFormat:@"%s.lnk", temp] atomically:YES encoding:NSNonLossyASCIIStringEncoding error:NULL];
        return success;
    }
}

int main(int argc, const char *argv[]) {
    if (argc < 2)
        return 0;
    if (strcmp(argv[1], "triggered") == 0)
        UICache();
    if (strcmp(argv[1], "configure") != 0)
        return 0;

    UICache();

    platformize_me();

    NSAutoreleasePool *pool([[NSAutoreleasePool alloc] init]);

    bool restart(false);

    if (kCFCoreFoundationVersionNumber >= 1000 && kCFCoreFoundationVersionNumber < 1349.56) {
        if (!FixProtections())
            return 1;
        switch (MoveStash()) {
            case StashDone:
                restart = true;
                break;
            case StashFail:
                fprintf(stderr, "failed to move stash\n");
                return 1;
            case StashGood:
                break;
        }
    }

    #define OldCache_ "/var/root/Library/Caches/com.saurik.Cydia"
    if (access(OldCache_, F_OK) == 0)
        system("rm -rf " OldCache_);

    #define NewCache_ "/var/mobile/Library/Caches/com.saurik.Cydia"
    system("cd /; su -c 'mkdir -p " NewCache_ "' mobile");
    if (access(NewCache_ "/lists", F_OK) != 0 && errno == ENOENT)
        system("cp -at " NewCache_ " /var/lib/apt/lists");
    system("chown -R 501.501 " NewCache_);

    #define OldLibrary_ "/var/lib/cydia"

    #define NewLibrary_ "/var/mobile/Library/Cydia"
    system("cd /; su -c 'mkdir -p " NewLibrary_ "' mobile");

    #define Cytore_ "/metadata.cb0"

    unlink("/etc/apt/sources.list.d/electra.list");
    #define CYDIA_LIST "/etc/apt/sources.list.d/cydia.list"
    unlink(CYDIA_LIST);
    if (kCFCoreFoundationVersionNumber >= 1443) {
        [[NSString stringWithFormat:@
            "deb https://apt.bingner.com/ ios/%.2f main\n"
            "deb http://apt.thebigboss.org/repofiles/cydia/ stable main\n"
            "deb http://cydia.zodttd.com/repo/cydia/ stable main\n"
            "deb http://apt.modmyi.com/ stable main\n"
            "deb https://repo.chariz.io/ ./\n"
            "deb https://repo.dynastic.co/ ./\n"
	, kCFCoreFoundationVersionNumber] writeToFile:@ CYDIA_LIST atomically:YES];
    } else {
        [[NSString stringWithFormat:@
            "deb http://apt.saurik.com/ ios/%.2f main\n"
            "deb https://apt.bingner.com/ ./\n"
            "deb http://apt.thebigboss.org/repofiles/cydia/ stable main\n"
            "deb http://cydia.zodttd.com/repo/cydia/ stable main\n"
            "deb http://apt.modmyi.com/ stable main\n"
            "deb https://repo.chariz.io/ ./\n"
            "deb https://repo.dynastic.co/ ./\n"
        , kCFCoreFoundationVersionNumber] writeToFile:@ CYDIA_LIST atomically:YES];
    }

    if (access(NewLibrary_ Cytore_, F_OK) != 0 && errno == ENOENT) {
        if (access(NewCache_ Cytore_, F_OK) == 0)
            system("mv -f " NewCache_ Cytore_ " " NewLibrary_);
        else if (access(OldLibrary_ Cytore_, F_OK) == 0)
            system("mv -f " OldLibrary_ Cytore_ " " NewLibrary_);
        chown(NewLibrary_ Cytore_, 501, 501);
    }

    if (kCFCoreFoundationVersionNumber < 1349.56) {
        FixPermissions();

        restart |= FixApplications();
    }

    if (restart)
        Finish("restart");

    [pool release];
    return 0;
}