bs-manager/src/main/services/folder-linker.service.ts

import { InstallationLocationService } from "./installation-location.service";
import log from "electron-log";
import { deleteFolder, ensureFolderExist, moveFolderContent, pathExist, unlinkPath } from "../helpers/fs.helpers";
import { lstat, symlink } from "fs/promises";
import path from "path";
import { copy, readlink } from "fs-extra";

export class FolderLinkerService {

    private static instance: FolderLinkerService;

    public static getInstance(): FolderLinkerService {
        if (!FolderLinkerService.instance) {
            FolderLinkerService.instance = new FolderLinkerService();
        }
        return FolderLinkerService.instance;
    }

    private readonly installLocationService = InstallationLocationService.getInstance();

    private constructor(){
        this.installLocationService = InstallationLocationService.getInstance();
    }

    private get sharedFolder(): string{
        return this.installLocationService.sharedContentPath;
    }

    private getSharedFolder(folderPath: string, intermediateFolder?: string): string {
        return path.join(this.sharedFolder, intermediateFolder ?? "", path.basename(folderPath));
    }

    private getBackupFolder(folderPath: string): string {
        return folderPath + "_backup";
    }

    private async backupFolder(folderPath: string): Promise<void> {
        if(!await pathExist(folderPath)){ return; }
        return copy(folderPath, this.getBackupFolder(folderPath), { overwrite: true, errorOnExist: false });
    }

    private async restoreFolder(folderPath: string): Promise<void> {
        if(!await pathExist( this.getBackupFolder(folderPath) )){ return; }
        return copy(this.getBackupFolder(folderPath), folderPath, { overwrite: true, errorOnExist: false }).then(() => {
            return deleteFolder(this.getBackupFolder(folderPath));
        });
    }

    public async linkFolder(folderPath: string, options?: LinkOptions): Promise<void> {

        const sharedPath = this.getSharedFolder(folderPath, options?.intermediateFolder);

        if(await this.isFolderSymlink(folderPath)){
            const isTargetedToSharedPath = await readlink(folderPath).then(target => target === sharedPath).catch(() => false);
            if(isTargetedToSharedPath){ return; }
            await unlinkPath(folderPath);
            return symlink(sharedPath, folderPath, "junction");
        }

        await ensureFolderExist(sharedPath);

        if(options?.backup === true){
            await this.backupFolder(folderPath);
        }

        await ensureFolderExist(folderPath);

        if(options?.keepContents !== false){
            await moveFolderContent(folderPath, sharedPath).toPromise();
        }

        await deleteFolder(folderPath);

        return symlink(sharedPath, folderPath, "junction");
    }

    public async unlinkFolder(folderPath: string, options?: UnlinkOptions): Promise<void> {

        if(!(await this.isFolderSymlink(folderPath))){ return; }
        await unlinkPath(folderPath);

        const sharedPath = this.getSharedFolder(folderPath, options?.intermediateFolder);

        await ensureFolderExist(folderPath);

        if(options?.backup === true){
            return this.restoreFolder(folderPath);
        }

        if(options.moveContents === true){
            return moveFolderContent(sharedPath, folderPath).toPromise().then(() => {});
        }

        if(options?.keepContents === false){ return; }

        await ensureFolderExist(sharedPath);
        
        return copy(sharedPath, folderPath, { errorOnExist: false, recursive: true });
    }

    public async isFolderSymlink(folder: string): Promise<boolean> {
        try{
            if(!(await pathExist(folder))){ return false; }
            return lstat(folder).then(stat => stat.isSymbolicLink());
        }
        catch(e){
            log.error(e);
        }
        return false;
    }

}

export interface LinkOptions {
    keepContents?: boolean,
    intermediateFolder?: string,
    backup?: boolean,
}

export interface UnlinkOptions extends LinkOptions {
    moveContents?: boolean,
};