67 lines
2.0 KiB
TypeScript
Vendored
67 lines
2.0 KiB
TypeScript
Vendored
import path from 'path';
|
|
import axios from 'axios';
|
|
import fsExtra from 'fs-extra';
|
|
import chalk from 'chalk';
|
|
import { dir } from 'tmp-promise';
|
|
|
|
import logger from './logger';
|
|
import { npmDirectory } from '@/utils/dir';
|
|
import { IS_LINUX, IS_WIN } from '@/utils/platform';
|
|
import { getSpinner } from '@/utils/info';
|
|
import { fileTypeFromBuffer } from 'file-type';
|
|
import { PakeAppOptions } from '@/types';
|
|
|
|
export async function handleIcon(options: PakeAppOptions) {
|
|
if (options.icon) {
|
|
if (options.icon.startsWith('http')) {
|
|
return downloadIcon(options.icon);
|
|
} else {
|
|
return path.resolve(options.icon);
|
|
}
|
|
} else {
|
|
logger.warn('✼ No icon given, default in use. For a custom icon, use --icon option.');
|
|
const iconPath = IS_WIN
|
|
? 'src-tauri/png/icon_256.ico'
|
|
: IS_LINUX
|
|
? 'src-tauri/png/icon_512.png'
|
|
: 'src-tauri/icons/icon.icns';
|
|
return path.join(npmDirectory, iconPath);
|
|
}
|
|
}
|
|
|
|
export async function downloadIcon(iconUrl: string) {
|
|
const spinner = getSpinner('Downloading icon...');
|
|
try {
|
|
const iconResponse = await axios.get(iconUrl, { responseType: 'arraybuffer' });
|
|
const iconData = await iconResponse.data;
|
|
|
|
if (!iconData) {
|
|
return null;
|
|
}
|
|
|
|
const fileDetails = await fileTypeFromBuffer(iconData);
|
|
if (!fileDetails) {
|
|
return null;
|
|
}
|
|
|
|
const { path: tempPath } = await dir();
|
|
let iconPath = `${tempPath}/icon.${fileDetails.ext}`;
|
|
// Fix this for linux
|
|
if (IS_LINUX) {
|
|
iconPath = 'png/linux_temp.png';
|
|
await fsExtra.outputFile(`${npmDirectory}/src-tauri/${iconPath}`, iconData);
|
|
} else {
|
|
await fsExtra.outputFile(iconPath, iconData);
|
|
}
|
|
await fsExtra.outputFile(iconPath, iconData);
|
|
spinner.succeed(chalk.green('Icon downloaded successfully!'));
|
|
return iconPath;
|
|
} catch (error) {
|
|
spinner.fail(chalk.red('Icon download failed!'));
|
|
if (error.response && error.response.status === 404) {
|
|
return null;
|
|
}
|
|
throw error;
|
|
}
|
|
}
|