rclone-ui/lib/rclone/common.ts

import { appLocalDataDir, sep } from '@tauri-apps/api/path'
import { exists, mkdir, writeTextFile } from '@tauri-apps/plugin-fs'
import { fetch as tauriFetch } from '@tauri-apps/plugin-http'
import { Command } from '@tauri-apps/plugin-shell'
import createRCDClient from 'rclone-sdk'
import { useHostStore } from '../../store/host'
import type { FlagValue } from '../../types/rclone'
import { getConfigParentFolder } from '../format'
import rclone from './client'
import { DOUBLE_BACKSLASH_REGEX } from './constants'

export async function getDefaultPaths() {
    console.log('[getDefaultPaths]')

    const defaultPaths = await rclone('/config/paths')

    console.log('[getDefaultPaths] json', JSON.stringify(defaultPaths, null, 2))

    return {
        cache: defaultPaths.cache ? defaultPaths.cache.replace(DOUBLE_BACKSLASH_REGEX, '\\') : '',
        config: defaultPaths.config
            ? defaultPaths.config.replace(DOUBLE_BACKSLASH_REGEX, '\\')
            : '',
        temp: defaultPaths.temp ? defaultPaths.temp.replace(DOUBLE_BACKSLASH_REGEX, '\\') : '',
    }
}

export async function getSystemConfigPath() {
    console.log('[getSystemConfigPath] running system rclone')

    const instance = Command.create('rclone-system', [
        'rcd',
        '--rc-no-auth',
        '--rc-serve',
        // '-rc-addr',
        // ':5572',
    ])

    if (!instance) {
        console.error('[getSystemConfigPath] failed to create rclone instance')
        throw new Error('Failed to create rclone instance, please try again later.')
    }

    const output = await instance.spawn()

    console.log('[getSystemConfigPath] spawned rclone')

    await new Promise((resolve) => setTimeout(resolve, 200))

    try {
        // no host store at this point
        const client = createRCDClient({
            baseUrl: 'http://localhost:5572',
            fetch: (request: Request) => tauriFetch(request),
        })

        const defaultPaths = await client.POST('/config/paths', {})

        const configPath = defaultPaths.data?.config
        if (!configPath) {
            throw new Error('Failed to fetch config path')
        }

        return configPath.replace(DOUBLE_BACKSLASH_REGEX, '\\')
    } catch (error) {
        console.error('[getSystemConfigPath] error', error)
        if (error instanceof Error) {
            throw error
        }
        throw new Error('Failed to get default path, please try again later.')
    } finally {
        await output.kill()
    }
}

export async function getConfigPath({ id, validate = true }: { id: string; validate?: boolean }) {
    console.log('[getConfigPath]', id, validate)

    const appLocalDataDirPath = await appLocalDataDir()
    console.log('[getConfigPath] appLocalDataDirPath', appLocalDataDirPath)

    let configPath = appLocalDataDirPath + sep() + 'configs' + sep() + id + sep() + 'rclone.conf'

    console.log('[getConfigPath] configPath', configPath)

    if (id == 'default' && (await isSystemRcloneInstalled())) {
        const defaultPath = await getSystemConfigPath()

        configPath = defaultPath
        console.log('[getConfigPath] configPath', configPath)
    }

    if (validate) {
        const configExists = await exists(configPath)
        if (!configExists) {
            console.error('[getConfigPath] config file does not exist')
            throw new Error('Config file does not exist')
        }
    }

    return configPath
}

export async function createConfigFile(path: string) {
    console.log('[createConfigFile] path', path)

    const hasConfig = await exists(path).catch(() => false)
    console.log('[createConfigFile] hasConfig', hasConfig)
    if (!hasConfig) {
        console.log('[createConfigFile] writing space character to default path (1)', path)
        try {
            await writeTextFile(path, '# Empty config file\n')
        } catch (error) {
            console.error('[createConfigFile] error', error)
        }

        if (!(await exists(path).catch(() => false))) {
            console.log(
                '[createConfigFile] failed to write space character to default path (1)',
                path
            )
            const folderPath = getConfigParentFolder(path)
            console.log('[createConfigFile] creating folder', folderPath)
            await mkdir(folderPath, { recursive: true })
            console.log('[createConfigFile] created folder', folderPath)
            console.log('[createConfigFile] writing space character to default path (2)', path)
            await writeTextFile(path, '# Empty config file\n')
            const existsFinally = await exists(path).catch(() => false)
            console.log('[createConfigFile] existsFinally', existsFinally)
        }
    }
}

/**
 * Checks if rclone is installed and accessible from the system PATH
 * @returns {Promise<boolean>} True if rclone is installed and working
 */
export async function isSystemRcloneInstalled() {
    console.log('[isSystemRcloneInstalled]')

    try {
        const output = await Command.create('rclone-system').execute()
        return (
            output.stdout.includes('Available commands') ||
            output.stderr.includes('Available commands')
        )
    } catch {
        return false
    }
}

/**
 * Checks if rclone is downloaded by the application in the app's local data directory
 * @returns {Promise<boolean>} True if downloaded rclone is present and working
 */
export async function isInternalRcloneInstalled() {
    console.log('[isInternalRcloneInstalled]')

    try {
        const output = await Command.create('rclone-internal').execute()
        // console.log('[isInternalRcloneInstalled] output', output)
        return (
            output.stdout.includes('Available commands') ||
            output.stderr.includes('Available commands')
        )
    } catch {
        return false
    }
}

export function parseRcloneOptions(options: Record<string, FlagValue>) {
    console.log('[parseRcloneOptions]', options)

    return options
}

export function compareVersions(version1: string, version2: string): number {
    const parseVersion = (version: string) => {
        const parts = version.split('.').map((num) => Number.parseInt(num, 10))
        return {
            major: parts[0] || 0,
            minor: parts[1] || 0,
            patch: parts[2] || 0,
        }
    }

    const v1 = parseVersion(version1)
    const v2 = parseVersion(version2)

    if (v1.major !== v2.major) {
        return v1.major > v2.major ? 1 : -1
    }
    if (v1.minor !== v2.minor) {
        return v1.minor > v2.minor ? 1 : -1
    }
    if (v1.patch !== v2.patch) {
        return v1.patch > v2.patch ? 1 : -1
    }
    return 0
}

const YOURS_VERSION_REGEX = /yours:\s+([^\s]+)/
const LATEST_VERSION_REGEX = /latest:\s+([^\s]+)/

export async function getRcloneVersion(type?: 'system' | 'internal') {
    let instanceType = type
    if (!instanceType) {
        instanceType = (await isSystemRcloneInstalled()) ? 'system' : 'internal'
    }

    const result = await Command.create(
        instanceType === 'system' ? 'rclone-system' : 'rclone-internal',
        ['selfupdate', '--check']
    ).execute()
    const output = result.stdout.trim()
    return parseRcloneVersion(output)
}

export function parseRcloneVersion(output: string) {
    const yoursMatch = output.match(YOURS_VERSION_REGEX)
    const latestMatch = output.match(LATEST_VERSION_REGEX)

    if (!yoursMatch || !latestMatch) {
        return null
    }

    return {
        yours: yoursMatch[1],
        latest: latestMatch[1],
    }
}

export function shouldUpdateRclone(versionData: { yours: string; latest: string } | null) {
    if (!versionData?.yours) return false

    if (!versionData) {
        console.warn('[shouldUpdateRclone] received no version data:', versionData)
        return false
    }

    const currentVersion = versionData.yours
    const latestVersion = versionData.latest

    if (!currentVersion || !latestVersion) {
        console.warn('[shouldUpdateRclone] could not parse version output:', versionData)
        return false
    }

    console.log('[shouldUpdateRclone] current version:', currentVersion)
    console.log('[shouldUpdateRclone] latest version:', latestVersion)

    if (useHostStore.getState().lastSkippedVersion === currentVersion) {
        console.log('[shouldUpdateRclone] current version is in the lastSkippedVersion')
        return false
    }

    // Compare versions using the existing compareVersions function
    const versionComparison = compareVersions(currentVersion, latestVersion)
    if (versionComparison < 0) {
        console.log('[shouldUpdateRclone] internal rclone needs update')
        return true
    }

    console.log('[shouldUpdateRclone] internal rclone is up to date')
    return false
}