#include "wm.h"

#include "common/sysctl.h"
#include "common/mallocHelper.h"
#include "common/strutil.h"
#include "common/apple/version.h"

#include <ctype.h>
#include <libproc.h>
#import <Foundation/Foundation.h>

const char* ffDetectWMPlugin(FFstrbuf* pluginName) {
    int request[] = { CTL_KERN, KERN_PROC, KERN_PROC_ALL };
    u_int requestLength = ARRAY_SIZE(request);

    size_t length = 0;
    FF_AUTO_FREE struct kinfo_proc* processes = ffSysctlGetData(request, requestLength, &length);
    if (processes == NULL) {
        return "sysctl(CTL_KERN, KERN_PROC, KERN_PROC_ALL) failed";
    }
    assert(length % sizeof(struct kinfo_proc) == 0);

    for (size_t i = 0; i < length / sizeof(struct kinfo_proc); i++) {
        const struct kinfo_proc* proc = &processes[i];
        if (proc->kp_eproc.e_ppid != 1) {
            continue;
        }

        const char* comm = proc->kp_proc.p_comm;

        if (
            !ffStrEqualsIgnCase(comm, "rectangle") && // 28.6k
            !ffStrEqualsIgnCase(comm, "yabai") &&     // 28.4k
            !ffStrEqualsIgnCase(comm, "aerospace") && // 19.6k
            !ffStrEqualsIgnCase(comm, "amethyst") &&  // 16k
            !ffStrEqualsIgnCase(comm, "glazewm") &&   // 11.6k

#if 0
            // Unmaintained
            !ffStrEqualsIgnCase(comm, "spectacle") && // 13.6k
            !ffStrEqualsIgnCase(comm, "chunkwm") && // repo deleted; was https://github.com/koekeishiya/chunkwm
            !ffStrEqualsIgnCase(comm, "kwm") && // repo deleted; was https://github.com/koekeishiya/kwm
#endif
            true)
            continue;

        if (instance.config.general.detectVersion) {
            char buf[PROC_PIDPATHINFO_MAXSIZE];
            int length = proc_pidpath(proc->kp_proc.p_pid, buf, ARRAY_SIZE(buf) - strlen("Info.plist"));
            if (length > 0) {
                buf[length] = '\0';
                FF_STRBUF_AUTO_DESTROY pluginVersion = ffStrbufCreate();
                if (ffGetAppNameAndVersion(buf, pluginName, &pluginVersion)) {
                    if (pluginName->length == 0) {
                        ffStrbufSetS(pluginName, comm);
                    }
                    if (pluginVersion.length > 0) {
                        ffStrbufAppendC(pluginName, ' ');
                        ffStrbufAppend(pluginName, &pluginVersion);
                    }
                    break;
                }
            }
        }

        ffStrbufAppendS(pluginName, comm);
        pluginName->chars[0] = (char) toupper(pluginName->chars[0]);
        break;
    }

    return NULL;
}

const char* ffDetectWMVersion(const FFstrbuf* wmName, FFstrbuf* result, FF_A_UNUSED FFWMOptions* options) {
    if (!wmName) {
        return "No WM detected";
    }

    if (ffStrbufEqualS(wmName, "WindowServer")) {
        NSError* error;
        NSDictionary* dict = [NSDictionary dictionaryWithContentsOfURL:[NSURL fileURLWithPath:@"/System/Library/PrivateFrameworks/SkyLight.framework/Resources/version.plist" isDirectory:NO]
                                                                 error:&error];
        if (!dict) {
            dict = [NSDictionary dictionaryWithContentsOfURL:[NSURL fileURLWithPath:@"/System/Library/Frameworks/ApplicationServices.framework/Frameworks/CoreGraphics.framework/Resources/version.plist" isDirectory:NO]
                                                       error:&error];
        }

        if (dict) {
            ffStrbufSetS(result, ((NSString*) dict[@"CFBundleShortVersionString"]).UTF8String);
        }
    }

    return NULL;
}
