✨ feat：初始化 QuickJump 项目，实现完整的目录跳转和配置管理功能
This commit is contained in:
395
app/Commands.hs
Normal file
395
app/Commands.hs
Normal file
@@ -0,0 +1,395 @@
|
||||
{-# LANGUAGE OverloadedStrings #-}
|
||||
|
||||
module Commands
|
||||
( runCommand
|
||||
, runJump
|
||||
, runQuick
|
||||
, runConfigCmd
|
||||
, printShellIntegration
|
||||
) where
|
||||
|
||||
import Control.Monad (forM_, unless, when)
|
||||
import Data.List (intercalate)
|
||||
import Data.Map (Map)
|
||||
import qualified Data.Map as M
|
||||
import Data.Maybe (fromMaybe, isJust)
|
||||
import Data.Text (Text)
|
||||
import qualified Data.Text as T
|
||||
import System.Directory (doesDirectoryExist, doesFileExist)
|
||||
import System.Environment (lookupEnv)
|
||||
import System.Exit (exitFailure, exitSuccess)
|
||||
import System.FilePath ((</>))
|
||||
import System.Info (os)
|
||||
import System.IO (hFlush, stdout)
|
||||
import System.Process (callCommand, spawnCommand, waitForProcess)
|
||||
|
||||
import Config
|
||||
import Types
|
||||
import Utils
|
||||
|
||||
-- | 运行主命令
|
||||
runCommand :: Command -> IO ()
|
||||
runCommand cmd = case cmd of
|
||||
Jump name quiet -> runJump name quiet
|
||||
JumpInteractive quiet -> runJumpInteractive quiet
|
||||
Quick action quiet -> runQuick action quiet
|
||||
ConfigCmd action quiet -> runConfigCmd action quiet
|
||||
ShellIntegration -> printShellIntegration
|
||||
Version -> putStrLn "quickjump version 0.1.0.0"
|
||||
|
||||
-- | 跳转到指定条目
|
||||
runJump :: Text -> Bool -> IO ()
|
||||
runJump name quiet = do
|
||||
cfg <- ensureConfigExists
|
||||
case findEntry name cfg of
|
||||
Nothing -> do
|
||||
unless quiet $ do
|
||||
putStrLn $ "Unknown jump target: " ++ T.unpack name
|
||||
putStrLn "Use 'quickjump config list' to see available targets"
|
||||
exitFailure
|
||||
Just entry -> do
|
||||
expanded <- expandPath (path entry)
|
||||
exists <- doesDirectoryExist expanded
|
||||
if exists
|
||||
then do
|
||||
-- 输出 cd 命令供 shell 执行
|
||||
-- 使用引号包裹路径以处理包含空格的路径
|
||||
let cdCmd = case os of
|
||||
"mingw32" -> "cd \"" ++ expanded ++ "\""
|
||||
"mingw64" -> "cd \"" ++ expanded ++ "\""
|
||||
_ -> "cd " ++ show expanded
|
||||
putStrLn cdCmd
|
||||
else do
|
||||
unless quiet $ putStrLn $ "Directory does not exist: " ++ expanded
|
||||
exitFailure
|
||||
|
||||
-- | 交互式选择跳转
|
||||
runJumpInteractive :: Bool -> IO ()
|
||||
runJumpInteractive quiet = do
|
||||
cfg <- ensureConfigExists
|
||||
let sorted = getSortedEntries cfg
|
||||
if null sorted
|
||||
then unless quiet $ do
|
||||
putStrLn "No jump targets configured"
|
||||
putStrLn "Use 'quickjump config add <name> <path>' to add one"
|
||||
else do
|
||||
unless quiet $ do
|
||||
putStrLn "Available jump targets:"
|
||||
putStrLn ""
|
||||
forM_ (zip [1..] sorted) $ \(i, (name, entry)) -> do
|
||||
let desc = fromMaybe "" (description entry)
|
||||
putStrLn $ " " ++ show (i :: Int) ++ ". " ++ T.unpack name
|
||||
++ " -> " ++ path entry
|
||||
++ if T.null desc then "" else " (" ++ T.unpack desc ++ ")"
|
||||
putStr "\nSelect target (number or name): "
|
||||
hFlush stdout
|
||||
selection <- getLine
|
||||
case reads selection of
|
||||
[(n, "")] | n > 0 && n <= length sorted ->
|
||||
runJump (fst $ sorted !! (n - 1)) quiet
|
||||
_ -> runJump (T.pack selection) quiet
|
||||
|
||||
-- | 运行快速操作
|
||||
runQuick :: QuickAction -> Bool -> IO ()
|
||||
runQuick action quiet = do
|
||||
cfg <- ensureConfigExists
|
||||
case action of
|
||||
QuickOpen name -> do
|
||||
case findEntry name cfg of
|
||||
Nothing -> do
|
||||
unless quiet $ putStrLn $ "Unknown quick target: " ++ T.unpack name
|
||||
exitFailure
|
||||
Just entry -> openPath (path entry) cfg quiet
|
||||
|
||||
QuickOpenPath p -> openPath p cfg quiet
|
||||
|
||||
QuickList -> do
|
||||
let sorted = getSortedEntries cfg
|
||||
unless quiet $ do
|
||||
putStrLn "Quick access targets:"
|
||||
putStrLn ""
|
||||
forM_ sorted $ \(name, entry) -> do
|
||||
let desc = fromMaybe "" (description entry)
|
||||
putStrLn $ " " ++ padRight 15 (T.unpack name)
|
||||
++ " -> " ++ padRight 30 (path entry)
|
||||
++ if T.null desc then "" else " # " ++ T.unpack desc
|
||||
|
||||
QuickDefault ->
|
||||
case defaultPath cfg of
|
||||
Nothing -> do
|
||||
unless quiet $ do
|
||||
putStrLn "No default path configured"
|
||||
putStrLn "Use 'quickjump config set-default <path>' to set one"
|
||||
exitFailure
|
||||
Just p -> openPath p cfg quiet
|
||||
|
||||
-- | 打开路径（使用文件管理器或 cd）
|
||||
openPath :: FilePath -> Config -> Bool -> IO ()
|
||||
openPath p cfg quiet = do
|
||||
expanded <- expandPath p
|
||||
exists <- doesDirectoryExist expanded
|
||||
unless exists $ do
|
||||
unless quiet $ putStrLn $ "Directory does not exist: " ++ expanded
|
||||
exitFailure
|
||||
|
||||
-- 尝试使用配置的文件管理器，或者自动检测
|
||||
let fm = fileManager cfg
|
||||
case fm of
|
||||
Just cmd -> runFileManager cmd expanded quiet
|
||||
Nothing -> autoDetectAndOpen expanded quiet
|
||||
|
||||
-- | 自动检测并打开文件管理器
|
||||
autoDetectAndOpen :: FilePath -> Bool -> IO ()
|
||||
autoDetectAndOpen path quiet = do
|
||||
let (cmd, args) = case os of
|
||||
"darwin" -> ("open", [path])
|
||||
"mingw32" -> ("explorer", [path])
|
||||
"mingw64" -> ("explorer", [path])
|
||||
"cygwin" -> ("cygstart", [path])
|
||||
_ -> ("xdg-open", [path]) -- Linux and others
|
||||
|
||||
-- 检查命令是否存在
|
||||
exists <- commandExists cmd
|
||||
if exists
|
||||
then do
|
||||
_ <- spawnCommand (unwords (cmd : map show args)) >>= waitForProcess
|
||||
return ()
|
||||
else do
|
||||
unless quiet $ do
|
||||
putStrLn $ "Cannot open file manager. Please configure one:"
|
||||
putStrLn $ " quickjump config set-file-manager <command>"
|
||||
-- 输出 cd 命令作为备选
|
||||
putStrLn $ "cd " ++ show path
|
||||
|
||||
-- | 运行文件管理器
|
||||
runFileManager :: FilePath -> FilePath -> Bool -> IO ()
|
||||
runFileManager cmd path quiet = do
|
||||
expanded <- expandPath path
|
||||
let fullCmd = cmd ++ " " ++ show expanded
|
||||
_ <- spawnCommand fullCmd >>= waitForProcess
|
||||
return ()
|
||||
|
||||
-- | 运行配置命令
|
||||
runConfigCmd :: ConfigAction -> Bool -> IO ()
|
||||
runConfigCmd action quiet = do
|
||||
cfg <- ensureConfigExists
|
||||
case action of
|
||||
ConfigAdd name path mDesc -> do
|
||||
expanded <- expandPath path
|
||||
exists <- doesDirectoryExist expanded
|
||||
unless exists $ do
|
||||
unless quiet $ putStrLn $ "Warning: Directory does not exist: " ++ expanded
|
||||
let entry = JumpEntry
|
||||
{ path = path
|
||||
, description = mDesc
|
||||
, priority = 100
|
||||
}
|
||||
newCfg = cfg { entries = M.insert name entry (entries cfg) }
|
||||
saveConfig newCfg
|
||||
unless quiet $ putStrLn $ "Added '" ++ T.unpack name ++ "' -> " ++ path
|
||||
|
||||
ConfigRemove name -> do
|
||||
if M.member name (entries cfg)
|
||||
then do
|
||||
let newCfg = cfg { entries = M.delete name (entries cfg) }
|
||||
saveConfig newCfg
|
||||
unless quiet $ putStrLn $ "Removed '" ++ T.unpack name ++ "'"
|
||||
else do
|
||||
unless quiet $ putStrLn $ "No such entry: '" ++ T.unpack name ++ "'"
|
||||
exitFailure
|
||||
|
||||
ConfigList -> do
|
||||
let sorted = getSortedEntries cfg
|
||||
unless quiet $ do
|
||||
if null sorted
|
||||
then putStrLn "No entries configured"
|
||||
else do
|
||||
putStrLn "Configured jump entries:"
|
||||
putStrLn ""
|
||||
forM_ sorted $ \(name, entry) -> do
|
||||
let desc = fromMaybe "" (description entry)
|
||||
putStrLn $ " " ++ padRight 15 (T.unpack name)
|
||||
++ " -> " ++ padRight 30 (path entry)
|
||||
++ if T.null desc then "" else " # " ++ T.unpack desc
|
||||
|
||||
ConfigSetDefault path -> do
|
||||
expanded <- expandPath path
|
||||
exists <- doesDirectoryExist expanded
|
||||
unless exists $ do
|
||||
unless quiet $ putStrLn $ "Warning: Directory does not exist: " ++ expanded
|
||||
let newCfg = cfg { defaultPath = Just path }
|
||||
saveConfig newCfg
|
||||
unless quiet $ putStrLn $ "Set default path to: " ++ path
|
||||
|
||||
ConfigSetEditor cmd -> do
|
||||
let newCfg = cfg { editor = Just cmd }
|
||||
saveConfig newCfg
|
||||
unless quiet $ putStrLn $ "Set editor to: " ++ cmd
|
||||
|
||||
ConfigSetFileManager cmd -> do
|
||||
let newCfg = cfg { fileManager = Just cmd }
|
||||
saveConfig newCfg
|
||||
unless quiet $ putStrLn $ "Set file manager to: " ++ cmd
|
||||
|
||||
ConfigExport path -> do
|
||||
saveConfigTo path cfg
|
||||
unless quiet $ putStrLn $ "Config exported to: " ++ path
|
||||
|
||||
ConfigImport path merge -> do
|
||||
imported <- loadConfigFrom path
|
||||
let merged = mergeConfigs cfg imported merge
|
||||
saveConfig merged
|
||||
unless quiet $
|
||||
if merge
|
||||
then putStrLn "Config imported and merged successfully"
|
||||
else putStrLn "Config imported (replaced existing)"
|
||||
|
||||
ConfigEdit -> do
|
||||
let ed = fromMaybe (defaultEditor os) (editor cfg)
|
||||
configPath <- getConfigPath
|
||||
_ <- spawnCommand (ed ++ " " ++ show configPath) >>= waitForProcess
|
||||
return ()
|
||||
|
||||
ConfigShow -> do
|
||||
configPath <- getConfigPath
|
||||
unless quiet $ do
|
||||
putStrLn $ "Config location: " ++ configPath
|
||||
putStrLn $ "Version: " ++ T.unpack (version cfg)
|
||||
putStrLn $ "Entries: " ++ show (M.size $ entries cfg)
|
||||
putStrLn $ "Default path: " ++ fromMaybe "(not set)" (defaultPath cfg)
|
||||
putStrLn $ "Editor: " ++ fromMaybe "(not set)" (editor cfg)
|
||||
putStrLn $ "File manager: " ++ fromMaybe "(auto-detect)" (fileManager cfg)
|
||||
|
||||
-- | 获取默认编辑器
|
||||
defaultEditor :: String -> String
|
||||
defaultEditor platform = case platform of
|
||||
"darwin" -> "open -t"
|
||||
"mingw32" -> "notepad"
|
||||
"mingw64" -> "notepad"
|
||||
_ -> "vi"
|
||||
|
||||
-- | 打印 shell 集成脚本
|
||||
printShellIntegration :: IO ()
|
||||
printShellIntegration = do
|
||||
putStrLn $ shellScript os
|
||||
|
||||
-- | 获取对应 shell 的集成脚本
|
||||
shellScript :: String -> String
|
||||
shellScript platform = case platform of
|
||||
"mingw32" -> windowsPowerShellScript
|
||||
"mingw64" -> windowsPowerShellScript
|
||||
"cygwin" -> bashScript
|
||||
"darwin" -> bashScript
|
||||
_ -> bashScript
|
||||
|
||||
-- | Bash/Zsh 集成脚本
|
||||
bashScript :: String
|
||||
bashScript = intercalate "\n"
|
||||
[ "# QuickJump Shell Integration for Bash/Zsh"
|
||||
, "# Add this to your shell profile (.bashrc, .zshrc, etc.)"
|
||||
, "#"
|
||||
, "# eval \"$(quickjump shell-integration)\""
|
||||
, ""
|
||||
, "# Bash/Zsh function for directory jumping"
|
||||
, "qj() {"
|
||||
, " local output=$(quickjump jump \"$1\")"
|
||||
, " if [[ $output == cd* ]]; then"
|
||||
, " eval \"$output\""
|
||||
, " else"
|
||||
, " echo \"$output\""
|
||||
, " fi"
|
||||
, "}"
|
||||
, ""
|
||||
, "# Quick open function"
|
||||
, "qo() {"
|
||||
, " quickjump quick \"$1\""
|
||||
, "}"
|
||||
, ""
|
||||
, "# Quiet mode function"
|
||||
, "qjq() {"
|
||||
, " quickjump --quiet jump \"$1\""
|
||||
, "}"
|
||||
, ""
|
||||
, "# Tab completion for bash"
|
||||
, "if [ -n \"$BASH_VERSION\" ]; then"
|
||||
, " _qj_complete() {"
|
||||
, " local cur=\"${COMP_WORDS[COMP_CWORD]}\""
|
||||
, " local entries=$(quickjump config list 2>/dev/null | grep '^ ' | awk '{print $1}')"
|
||||
, " COMPREPLY=($(compgen -W \"$entries\" -- \"$cur\"))"
|
||||
, " }"
|
||||
, " complete -F _qj_complete qj"
|
||||
, " complete -F _qj_complete qjq"
|
||||
, "fi"
|
||||
, ""
|
||||
, "# Tab completion for zsh"
|
||||
, "if [ -n \"$ZSH_VERSION\" ]; then"
|
||||
, " _qj_complete() {"
|
||||
, " local -a entries"
|
||||
, " entries=(${(f)\"$(quickjump config list 2>/dev/null | grep '^ ' | awk '{print $1}')\"})"
|
||||
, " _describe 'jump targets' entries"
|
||||
, " }"
|
||||
, " compdef _qj_complete qj"
|
||||
, " compdef _qj_complete qjq"
|
||||
, "fi"
|
||||
]
|
||||
|
||||
-- | Windows PowerShell 集成脚本
|
||||
windowsPowerShellScript :: String
|
||||
windowsPowerShellScript = intercalate "\n"
|
||||
[ "# QuickJump Shell Integration for PowerShell"
|
||||
, "# Add this to your PowerShell profile ($PROFILE)"
|
||||
, "#"
|
||||
, "# To edit your profile: notepad $PROFILE"
|
||||
, "#"
|
||||
, "# . (quickjump shell-integration | Out-String)"
|
||||
, ""
|
||||
, "# Function for directory jumping"
|
||||
, "function qj {"
|
||||
, " param([string]$name)"
|
||||
, " $output = quickjump jump $name"
|
||||
, " if ($output -like 'cd *') {"
|
||||
, " # Remove 'cd ' prefix and execute"
|
||||
, " $path = $output -replace '^cd \"?([^\"\"]*)\"?$', '$1'"
|
||||
, " Set-Location $path"
|
||||
, " } else {"
|
||||
, " Write-Output $output"
|
||||
, " }"
|
||||
, "}"
|
||||
, ""
|
||||
, "# Quiet mode function"
|
||||
, "function qjq {"
|
||||
, " param([string]$name)"
|
||||
, " $output = quickjump --quiet jump $name"
|
||||
, " if ($output -like 'cd *') {"
|
||||
, " $path = $output -replace '^cd \"?([^\"\"]*)\"?$', '$1'"
|
||||
, " Set-Location $path"
|
||||
, " }"
|
||||
, "}"
|
||||
, ""
|
||||
, "# Quick open function"
|
||||
, "function qo {"
|
||||
, " param([string]$name)"
|
||||
, " quickjump quick $name"
|
||||
, "}"
|
||||
, ""
|
||||
, "# Tab completion"
|
||||
, "Register-ArgumentCompleter -CommandName qj -ScriptBlock {"
|
||||
, " param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)"
|
||||
, " $entries = quickjump config list 2>$null | Select-String '^ ' | ForEach-Object {"
|
||||
, " $_.ToString().Trim().Split()[0]"
|
||||
, " }"
|
||||
, " $entries | Where-Object { $_ -like \"$wordToComplete*\" } | ForEach-Object {"
|
||||
, " [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)"
|
||||
, " }"
|
||||
, "}"
|
||||
, ""
|
||||
, "Register-ArgumentCompleter -CommandName qjq -ScriptBlock {"
|
||||
, " param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter)"
|
||||
, " $entries = quickjump config list 2>$null | Select-String '^ ' | ForEach-Object {"
|
||||
, " $_.ToString().Trim().Split()[0]"
|
||||
, " }"
|
||||
, " $entries | Where-Object { $_ -like \"$wordToComplete*\" } | ForEach-Object {"
|
||||
, " [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)"
|
||||
, " }"
|
||||
, "}"
|
||||
]
|
||||
Reference in New Issue
Block a user