#include "packages.h"
#include "common/io.h"
#include "common/parsing.h"
#include "common/processing.h"
#include "common/strutil.h"

static void countBrewPackages(FFstrbuf* baseDir, FFPackagesResult* result) {
    uint32_t baseDirLength = baseDir->length;

    ffStrbufAppendS(baseDir, "/Caskroom");
    result->brewCask += ffPackagesGetNumElements(baseDir->chars, true);
    ffStrbufSubstrBefore(baseDir, baseDirLength);

    ffStrbufAppendS(baseDir, "/Cellar");
    result->brew += ffPackagesGetNumElements(baseDir->chars, true);
    ffStrbufSubstrBefore(baseDir, baseDirLength);
}

static uint32_t getMacPortsPackages(FFstrbuf* baseDir) {
    ffStrbufAppendS(baseDir, "/var/macports/software");
    return ffPackagesGetNumElements(baseDir->chars, true);
}

void ffDetectPackagesImpl(FFPackagesResult* result, FFPackagesOptions* options) {
    FF_STRBUF_AUTO_DESTROY baseDir = ffStrbufCreate();
    if (FF_PACKAGES_IS_ENABLED(options, BREW)) {
        const char* prefix = getenv("HOMEBREW_PREFIX");
        if (ffStrSet(prefix)) {
            ffStrbufSetS(&baseDir, prefix);
        } else {
#ifdef __aarch64__
            ffStrbufSetS(&baseDir, FASTFETCH_TARGET_DIR_ROOT "/opt/homebrew");
#else
            ffStrbufSetS(&baseDir, FASTFETCH_TARGET_DIR_USR "/local");
#endif
        }
        countBrewPackages(&baseDir, result);
    }
    if (FF_PACKAGES_IS_ENABLED(options, MACPORTS)) {
        const char* prefix = getenv("MACPORTS_PREFIX");
        if (ffStrSet(prefix)) {
            ffStrbufSetS(&baseDir, prefix);
        } else {
            ffStrbufSetS(&baseDir, FASTFETCH_TARGET_DIR_ROOT "/opt/local");
        }

        result->macports = getMacPortsPackages(&baseDir);
    }
    if (FF_PACKAGES_IS_ENABLED(options, NIX)) {
        ffStrbufSetS(&baseDir, FASTFETCH_TARGET_DIR_ROOT);
        result->nixDefault += ffPackagesGetNix(&baseDir, "/nix/var/nix/profiles/default");
        result->nixSystem += ffPackagesGetNix(&baseDir, "/run/current-system");
        ffStrbufSet(&baseDir, &instance.state.platform.homeDir);
        result->nixUser = ffPackagesGetNix(&baseDir, "/.nix-profile");
    }
}