khamba/dist/functions/commands.js

A cli tool for sharing files through local network.

// import {Command} from 'commander';
import fs from 'fs';
import path from 'path';
import yargs from 'yargs';
import { hideBin } from 'yargs/helpers';
import { useEffect } from 'react';
import { $action, $baseInfo, $errorMsg, $isDev, $sendingFiles, } from '../stores/baseStore.js';
import { cleanFileName, getFileSize, getFileType, getFolderSize, getMissingFiles, } from '../functions/helper.js';
import { v4 as uuidv4 } from 'uuid';
import { logError } from '../functions/log.js';
import { SEND_PATH } from '../functions/variables.js';
import { hashFile, hashFolder } from '../functions/useHashCheck.js';
export const useCommands = () => {
    useEffect(() => {
        yargs(hideBin(process.argv))
            .command('chat', 'start chat', yargs => {
            return yargs.positional('name', {
                describe: 'a future chat feature',
                type: 'string',
            });
        }, argv => {
            console.info('Chat command:', argv);
        })
            .command('* [files...]', 'Handles file operations', (yargs) => {
            return yargs.positional('files', {
                describe: 'Optional list of files',
                type: 'string',
            });
        }, async (argv) => {
            if (argv.dev_mode) {
                $isDev.set(true);
            }
            if (argv.name && argv.name != '') {
                $baseInfo.setKey('MY_NAME', argv.name);
            }
            if (argv.files && argv.files?.length > 0) {
                try {
                    // NOTE: // ! [-KEEP COMMENT-]
                    // ! Get all files from the command line
                    // const files: any = argv.files?.flatMap(item => {
                    // 	if (isDirectory(item)) {
                    // 		return getAllFiles(`${SEND_PATH}/${item}`);
                    // 	} else {
                    // 		return [item];
                    // 	}
                    // });
                    const files = argv.files;
                    // ! Check if file exists
                    const missingFiles = await getMissingFiles(files);
                    if (missingFiles.length) {
                        const errorMessage = missingFiles.length === 1
                            ? `The file "${missingFiles[0]}" is missing.`
                            : `The following files are missing: ${missingFiles.join(', ')}.`;
                        $errorMsg.set(errorMessage);
                        return;
                    }
                    // ! Create peer transfer info
                    const peerTransferInfo = await Promise.all(files?.map(async (uncleanFileName) => {
                        const fileName = cleanFileName(uncleanFileName);
                        const filePath = path.join(SEND_PATH, fileName);
                        const fileType = getFileType(filePath);
                        const stats = fs.statSync(filePath);
                        const fileHash = (stats.isDirectory()
                            ? await hashFolder(filePath)
                            : await hashFile(filePath));
                        const fileSize = fileType === 'folder'
                            ? getFolderSize(filePath)
                            : getFileSize(filePath);
                        return {
                            id: uuidv4(),
                            data: { fileName, fileHash, fileSize, fileType },
                        };
                    })).then(results => results.reduce((acc, { id, data }) => {
                        acc[id] = data;
                        return acc;
                    }, {}));
                    $sendingFiles.set(peerTransferInfo);
                    $action.set('SEND');
                }
                catch (error) {
                    logError(error);
                }
            }
            else {
                $action.set('RECEIVE');
            }
        })
            .option('debug', {
            alias: '-d',
            type: 'boolean',
            description: 'Start with debug mode.',
        })
            .option('name', {
            alias: 'n',
            type: 'string',
            description: 'Enter name for this instance.',
        })
            .parse();
        return () => { };
    }, []);
    return {};
};