mirror of
https://github.com/fastfetch-cli/fastfetch.git
synced 2026-10-31 01:56:28 +01:00
李通洲
115 lines
3.4 KiB
C
115 lines
3.4 KiB
C
#include "packages.h"
|
|
#include "common/processing.h"
|
|
#include "util/stringUtils.h"
|
|
|
|
#include <handleapi.h>
|
|
#include <fileapi.h>
|
|
|
|
static uint32_t getNumElements(const char* searchPath /* including `\*` suffix */, DWORD type, const char* ignore)
|
|
{
|
|
uint32_t counter = 0;
|
|
bool flag = ignore == NULL;
|
|
WIN32_FIND_DATAA wfd;
|
|
HANDLE hFind = FindFirstFileA(searchPath, &wfd);
|
|
|
|
if (hFind != INVALID_HANDLE_VALUE)
|
|
{
|
|
do // Managed to locate and create an handle to that folder.
|
|
{
|
|
if(!(wfd.dwFileAttributes & type)) continue;
|
|
if(!flag && ffStrEqualsIgnCase(ignore, wfd.cFileName))
|
|
{
|
|
flag = true;
|
|
continue;
|
|
}
|
|
counter++;
|
|
} while (FindNextFileA(hFind, &wfd));
|
|
FindClose(hFind);
|
|
|
|
if(type == FILE_ATTRIBUTE_DIRECTORY && counter >= 2)
|
|
counter -= 2; // accounting for . and ..
|
|
}
|
|
|
|
return counter;
|
|
}
|
|
|
|
static void detectScoop(FFPackagesResult* result)
|
|
{
|
|
FF_STRBUF_AUTO_DESTROY scoopPath = ffStrbufCreateA(MAX_PATH + 3);
|
|
|
|
const char* scoopEnv = getenv("SCOOP");
|
|
if(ffStrSet(scoopEnv))
|
|
{
|
|
ffStrbufAppendS(&scoopPath, scoopEnv);
|
|
ffStrbufAppendS(&scoopPath, "/apps/*");
|
|
}
|
|
else
|
|
{
|
|
ffStrbufAppendS(&scoopPath, instance.state.platform.homeDir.chars);
|
|
ffStrbufAppendS(&scoopPath, "/scoop/apps/*");
|
|
}
|
|
result->scoop = getNumElements(scoopPath.chars, FILE_ATTRIBUTE_DIRECTORY, "scoop");
|
|
}
|
|
|
|
static void detectChoco(FF_MAYBE_UNUSED FFPackagesResult* result)
|
|
{
|
|
const char* chocoInstall = getenv("ChocolateyInstall");
|
|
if(!chocoInstall || chocoInstall[0] == '\0')
|
|
return;
|
|
|
|
char chocoPath[MAX_PATH + 3];
|
|
strcpy(chocoPath, chocoInstall);
|
|
strncat(chocoPath, "/lib/*", sizeof(chocoPath) - 1 - strlen(chocoPath));
|
|
result->choco = getNumElements(chocoPath, FILE_ATTRIBUTE_DIRECTORY, "choco");
|
|
}
|
|
|
|
static void detectPacman(FFPackagesResult* result)
|
|
{
|
|
const char* msystemPrefix = getenv("MSYSTEM_PREFIX");
|
|
if(!msystemPrefix)
|
|
return;
|
|
|
|
// MSYS2
|
|
char pacmanPath[MAX_PATH + 3];
|
|
strcpy(pacmanPath, msystemPrefix);
|
|
strncat(pacmanPath, "/../var/lib/pacman/local/*", sizeof(pacmanPath) - 1 - strlen(pacmanPath));
|
|
result->pacman = getNumElements(pacmanPath, FILE_ATTRIBUTE_DIRECTORY, NULL);
|
|
}
|
|
|
|
static void detectWinget(FFPackagesResult* result)
|
|
{
|
|
FF_STRBUF_AUTO_DESTROY buffer = ffStrbufCreate();
|
|
if (ffProcessAppendStdOut(&buffer, (char* []) {
|
|
"winget.exe",
|
|
"list",
|
|
"--disable-interactivity",
|
|
NULL,
|
|
}))
|
|
return;
|
|
|
|
uint32_t index = ffStrbufFirstIndexS(&buffer, "--\r\n"); // Ignore garbage and table headers
|
|
if (index == buffer.length)
|
|
return;
|
|
|
|
uint32_t count = 0;
|
|
for (
|
|
index += strlen("--\r\n");
|
|
(index = ffStrbufNextIndexC(&buffer, index, '\n')) < buffer.length;
|
|
++index
|
|
)
|
|
++count;
|
|
|
|
if (buffer.chars[buffer.length - 1] != '\n') // count last line
|
|
++count;
|
|
|
|
result->winget = count;
|
|
}
|
|
|
|
void ffDetectPackagesImpl(FFPackagesResult* result, FFPackagesOptions* options)
|
|
{
|
|
if (!(options->disabled & FF_PACKAGES_FLAG_SCOOP_BIT)) detectScoop(result);
|
|
if (!(options->disabled & FF_PACKAGES_FLAG_CHOCO_BIT)) detectChoco(result);
|
|
if (!(options->disabled & FF_PACKAGES_FLAG_PACMAN_BIT)) detectPacman(result);
|
|
if (!(options->disabled & FF_PACKAGES_FLAG_WINGET_BIT)) detectWinget(result);
|
|
}
|