secretive/Sources/Secretive/Controllers/ShellConfigurationController.swift

import Foundation
import Cocoa
import SecretKit

struct ShellConfigurationController {
    
    let socketPath = (NSHomeDirectory().replacingOccurrences(of: Bundle.main.hostBundleID, with: Bundle.main.agentBundleID) as NSString).appendingPathComponent("socket.ssh") as String
    
    var shellInstructions: [ShellConfigInstruction] {
        [
            ShellConfigInstruction(shell: "global",
                                   shellConfigDirectory: "~/.ssh/",
                                   shellConfigFilename: "config",
                                   text: "Host *\n\tIdentityAgent \(socketPath)"),
            ShellConfigInstruction(shell: "zsh",
                                   shellConfigDirectory: "~/",
                                   shellConfigFilename: ".zshrc",
                                   text: "export SSH_AUTH_SOCK=\(socketPath)"),
            ShellConfigInstruction(shell: "bash",
                                   shellConfigDirectory: "~/",
                                   shellConfigFilename: ".bashrc",
                                   text: "export SSH_AUTH_SOCK=\(socketPath)"),
            ShellConfigInstruction(shell: "fish",
                                   shellConfigDirectory: "~/.config/fish",
                                   shellConfigFilename: "config.fish",
                                   text: "set -x SSH_AUTH_SOCK \(socketPath)"),
        ]
        
    }
    
    
    @MainActor func addToShell(shellInstructions: ShellConfigInstruction) -> Bool {
        let openPanel = NSOpenPanel()
        // This is sync, so no need to strongly retain
        let delegate = Delegate(name: shellInstructions.shellConfigFilename)
        openPanel.delegate = delegate
        openPanel.message = "Select \(shellInstructions.shellConfigFilename) to let Secretive configure your shell automatically."
        openPanel.prompt = "Add to \(shellInstructions.shellConfigFilename)"
        openPanel.canChooseFiles = true
        openPanel.canChooseDirectories = false
        openPanel.showsHiddenFiles = true
        openPanel.directoryURL = URL(fileURLWithPath: shellInstructions.shellConfigDirectory)
        openPanel.nameFieldStringValue = shellInstructions.shellConfigFilename
        openPanel.allowedContentTypes = [.symbolicLink, .data, .plainText]
        openPanel.runModal()
        guard let fileURL = openPanel.urls.first else { return false }
        let handle: FileHandle
        do {
            handle = try FileHandle(forUpdating: fileURL)
            guard let existing = try handle.readToEnd(),
                  let existingString = String(data: existing, encoding: .utf8) else { return false }
            guard !existingString.contains(shellInstructions.text) else {
                return true
            }
            try handle.seekToEnd()
        } catch {
            return false
        }
        handle.write("\n# Secretive Config\n\(shellInstructions.text)\n".data(using: .utf8)!)
        return true
    }
    
}