tabby/tabby-electron/src/sftpContextMenu.ts

import * as tmp from 'tmp-promise'
import * as path from 'path'
import * as fs from 'fs'
import { Subject, debounceTime, debounce } from 'rxjs'
import { Injectable } from '@angular/core'
import { MenuItemOptions, TranslateService } from 'tabby-core'
import { SFTPFile, SFTPPanelComponent, SFTPContextMenuItemProvider, SFTPSession } from 'tabby-ssh'
import { ElectronPlatformService } from './services/platform.service'


/** @hidden */
@Injectable()
export class EditSFTPContextMenu extends SFTPContextMenuItemProvider {
    weight = 0

    constructor (
        private translate: TranslateService,
        private platform: ElectronPlatformService,
    ) {
        super()
    }

    async getItems (item: SFTPFile, panel: SFTPPanelComponent): Promise<MenuItemOptions[]> {
        const items: MenuItemOptions[] = [
            {
                click: () => this.platform.setClipboard({
                    text: item.fullPath,
                }),
                label: this.translate.instant('Copy full path'),
            },
        ]
        if (!item.isDirectory) {
            items.push({
                click: () => this.edit(item, panel.sftp),
                label: this.translate.instant('Edit locally'),
            })
        }
        return items
    }

    private async edit (item: SFTPFile, sftp: SFTPSession) {
        const tempDir = (await tmp.dir({ unsafeCleanup: true })).path
        const tempPath = path.join(tempDir, item.name)
        const transfer = await this.platform.startDownload(item.name, item.mode, item.size, tempPath)
        if (!transfer) {
            return
        }
        await sftp.download(item.fullPath, transfer)
        this.platform.openPath(tempPath)

        const events = new Subject<string>()
        fs.chmodSync(tempPath, 0o700)

        // skip the first burst of events
        setTimeout(() => {
            const watcher = fs.watch(tempPath, event => events.next(event))
            events.pipe(debounceTime(1000), debounce(async event => {
                if (event === 'rename') {
                    watcher.close()
                }
                const upload = await this.platform.startUpload({ multiple: false }, [tempPath])
                if (!upload.length) {
                    return
                }
                await sftp.upload(item.fullPath, upload[0])
                await sftp.chmod(item.fullPath, item.mode)
            })).subscribe()
            watcher.on('close', () => events.complete())
            sftp.closed$.subscribe(() => watcher.close())
        }, 1000)
    }
}