# ff - fastfetch with logo flags
# Usage: ff [-logo]
# Examples: ff, ff -apple, ff -arch, ff -linux

function ff
    if test (count $argv) -eq 0
        fastfetch
    else
        # Get the flag without the dash and convert to lowercase
        set -l raw $argv[1]
        set -l flag (string lower -- (string replace -r -- '^-' '' $raw))

        switch $flag
            case apple
                fastfetch --logo Apple
            case arch
                fastfetch --logo Arch
            case linux
                fastfetch --logo Linux
            case debian
                fastfetch --logo Debian
            case ubuntu
                fastfetch --logo Ubuntu
            case fedora
                fastfetch --logo Fedora
            case gentoo
                fastfetch --logo Gentoo
            case nixos
                fastfetch --logo NixOS
            case void
                fastfetch --logo Void
            case manjaro
                fastfetch --logo Manjaro
            case mint
                fastfetch --logo Mint
            case pop
                fastfetch --logo Pop
            case kali
                fastfetch --logo Kali
            case parrot
                fastfetch --logo Parrot
            case '*'
                # Try to use the argument as a logo name directly (capitalize first letter)
                set -l logo (string sub -l 1 -- $flag | string upper)(string sub -s 2 -- $flag)
                fastfetch --logo $logo
        end
    end
end