Compare commits
16 Commits
master
...
26-integra
| Author | SHA1 | Date | |
|---|---|---|---|
| b381e6fa7f | |||
| d87cd468cc | |||
| b6a6e63358 | |||
| 1230a3ea07 | |||
| a324935005 | |||
| d963d4ff88 | |||
| 89cd5232cf | |||
| 7c64383d72 | |||
| 85a9cd9ae5 | |||
| ca7be76937 | |||
| 6b0b648d67 | |||
| 4095ee32f6 | |||
| 8315549838 | |||
| 6c7e718748 | |||
| 0632d7b0a9 | |||
| 85b0b3dbbf |
498
Cargo.lock
generated
498
Cargo.lock
generated
@ -1,498 +0,0 @@
|
||||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 4
|
||||
|
||||
[[package]]
|
||||
name = "aho-corasick"
|
||||
version = "1.1.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8e60d3430d3a69478ad0993f19238d2df97c507009a52b3c10addcd7f6bcb916"
|
||||
dependencies = [
|
||||
"memchr",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anstream"
|
||||
version = "0.6.21"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "43d5b281e737544384e969a5ccad3f1cdd24b48086a0fc1b2a5262a26b8f4f4a"
|
||||
dependencies = [
|
||||
"anstyle",
|
||||
"anstyle-parse",
|
||||
"anstyle-query",
|
||||
"anstyle-wincon",
|
||||
"colorchoice",
|
||||
"is_terminal_polyfill",
|
||||
"utf8parse",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anstyle"
|
||||
version = "1.0.13"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5192cca8006f1fd4f7237516f40fa183bb07f8fbdfedaa0036de5ea9b0b45e78"
|
||||
|
||||
[[package]]
|
||||
name = "anstyle-parse"
|
||||
version = "0.2.7"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4e7644824f0aa2c7b9384579234ef10eb7efb6a0deb83f9630a49594dd9c15c2"
|
||||
dependencies = [
|
||||
"utf8parse",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anstyle-query"
|
||||
version = "1.1.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9e231f6134f61b71076a3eab506c379d4f36122f2af15a9ff04415ea4c3339e2"
|
||||
dependencies = [
|
||||
"windows-sys",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anstyle-wincon"
|
||||
version = "3.0.10"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3e0633414522a32ffaac8ac6cc8f748e090c5717661fddeea04219e2344f5f2a"
|
||||
dependencies = [
|
||||
"anstyle",
|
||||
"once_cell_polyfill",
|
||||
"windows-sys",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "bitflags"
|
||||
version = "2.9.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2261d10cca569e4643e526d8dc2e62e433cc8aba21ab764233731f8d369bf394"
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
version = "1.0.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2fd1289c04a9ea8cb22300a459a72a385d7c73d3259e2ed7dcb2af674838cfa9"
|
||||
|
||||
[[package]]
|
||||
name = "cfg_aliases"
|
||||
version = "0.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "613afe47fcd5fac7ccf1db93babcb082c5994d996f20b8b159f2ad1658eb5724"
|
||||
|
||||
[[package]]
|
||||
name = "clap"
|
||||
version = "4.5.48"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e2134bb3ea021b78629caa971416385309e0131b351b25e01dc16fb54e1b5fae"
|
||||
dependencies = [
|
||||
"clap_builder",
|
||||
"clap_derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "clap_builder"
|
||||
version = "4.5.48"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c2ba64afa3c0a6df7fa517765e31314e983f51dda798ffba27b988194fb65dc9"
|
||||
dependencies = [
|
||||
"anstream",
|
||||
"anstyle",
|
||||
"clap_lex",
|
||||
"strsim",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "clap_derive"
|
||||
version = "4.5.47"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "bbfd7eae0b0f1a6e63d4b13c9c478de77c2eb546fba158ad50b4203dc24b9f9c"
|
||||
dependencies = [
|
||||
"heck",
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "clap_lex"
|
||||
version = "0.7.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b94f61472cee1439c0b966b47e3aca9ae07e45d070759512cd390ea2bebc6675"
|
||||
|
||||
[[package]]
|
||||
name = "colorchoice"
|
||||
version = "1.0.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b05b61dc5112cbb17e4b6cd61790d9845d13888356391624cbe7e41efeac1e75"
|
||||
|
||||
[[package]]
|
||||
name = "heck"
|
||||
version = "0.5.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
|
||||
|
||||
[[package]]
|
||||
name = "homedir"
|
||||
version = "0.3.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "68df315d2857b2d8d2898be54a85e1d001bbbe0dbb5f8ef847b48dd3a23c4527"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"nix",
|
||||
"widestring",
|
||||
"windows",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "is_terminal_polyfill"
|
||||
version = "1.70.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf"
|
||||
|
||||
[[package]]
|
||||
name = "lazy-regex"
|
||||
version = "3.4.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "60c7310b93682b36b98fa7ea4de998d3463ccbebd94d935d6b48ba5b6ffa7126"
|
||||
dependencies = [
|
||||
"lazy-regex-proc_macros",
|
||||
"once_cell",
|
||||
"regex",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "lazy-regex-proc_macros"
|
||||
version = "3.4.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4ba01db5ef81e17eb10a5e0f2109d1b3a3e29bac3070fdbd7d156bf7dbd206a1"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"regex",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "libc"
|
||||
version = "0.2.176"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "58f929b4d672ea937a23a1ab494143d968337a5f47e56d0815df1e0890ddf174"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.7.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f52b00d39961fc5b2736ea853c9cc86238e165017a493d1d5c8eac6bdc4cc273"
|
||||
|
||||
[[package]]
|
||||
name = "multi-ssh"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"clap",
|
||||
"homedir",
|
||||
"lazy-regex",
|
||||
"shell-words",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "nix"
|
||||
version = "0.30.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "74523f3a35e05aba87a1d978330aef40f67b0304ac79c1c00b294c9830543db6"
|
||||
dependencies = [
|
||||
"bitflags",
|
||||
"cfg-if",
|
||||
"cfg_aliases",
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "once_cell"
|
||||
version = "1.21.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
|
||||
|
||||
[[package]]
|
||||
name = "once_cell_polyfill"
|
||||
version = "1.70.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a4895175b425cb1f87721b59f0f286c2092bd4af812243672510e1ac53e2e0ad"
|
||||
|
||||
[[package]]
|
||||
name = "proc-macro2"
|
||||
version = "1.0.101"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "89ae43fd86e4158d6db51ad8e2b80f313af9cc74f5c0e03ccb87de09998732de"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "quote"
|
||||
version = "1.0.41"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ce25767e7b499d1b604768e7cde645d14cc8584231ea6b295e9c9eb22c02e1d1"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "regex"
|
||||
version = "1.11.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8b5288124840bee7b386bc413c487869b360b2b4ec421ea56425128692f2a82c"
|
||||
dependencies = [
|
||||
"aho-corasick",
|
||||
"memchr",
|
||||
"regex-automata",
|
||||
"regex-syntax",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "regex-automata"
|
||||
version = "0.4.11"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "833eb9ce86d40ef33cb1306d8accf7bc8ec2bfea4355cbdebb3df68b40925cad"
|
||||
dependencies = [
|
||||
"aho-corasick",
|
||||
"memchr",
|
||||
"regex-syntax",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "regex-syntax"
|
||||
version = "0.8.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "caf4aa5b0f434c91fe5c7f1ecb6a5ece2130b02ad2a590589dda5146df959001"
|
||||
|
||||
[[package]]
|
||||
name = "shell-words"
|
||||
version = "1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "24188a676b6ae68c3b2cb3a01be17fbf7240ce009799bb56d5b1409051e78fde"
|
||||
|
||||
[[package]]
|
||||
name = "strsim"
|
||||
version = "0.11.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f"
|
||||
|
||||
[[package]]
|
||||
name = "syn"
|
||||
version = "2.0.106"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ede7c438028d4436d71104916910f5bb611972c5cfd7f89b8300a8186e6fada6"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "unicode-ident"
|
||||
version = "1.0.19"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f63a545481291138910575129486daeaf8ac54aee4387fe7906919f7830c7d9d"
|
||||
|
||||
[[package]]
|
||||
name = "utf8parse"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821"
|
||||
|
||||
[[package]]
|
||||
name = "widestring"
|
||||
version = "1.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "dd7cf3379ca1aac9eea11fba24fd7e315d621f8dfe35c8d7d2be8b793726e07d"
|
||||
|
||||
[[package]]
|
||||
name = "windows"
|
||||
version = "0.61.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9babd3a767a4c1aef6900409f85f5d53ce2544ccdfaa86dad48c91782c6d6893"
|
||||
dependencies = [
|
||||
"windows-collections",
|
||||
"windows-core",
|
||||
"windows-future",
|
||||
"windows-link 0.1.3",
|
||||
"windows-numerics",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-collections"
|
||||
version = "0.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3beeceb5e5cfd9eb1d76b381630e82c4241ccd0d27f1a39ed41b2760b255c5e8"
|
||||
dependencies = [
|
||||
"windows-core",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-core"
|
||||
version = "0.61.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c0fdd3ddb90610c7638aa2b3a3ab2904fb9e5cdbecc643ddb3647212781c4ae3"
|
||||
dependencies = [
|
||||
"windows-implement",
|
||||
"windows-interface",
|
||||
"windows-link 0.1.3",
|
||||
"windows-result",
|
||||
"windows-strings",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-future"
|
||||
version = "0.2.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fc6a41e98427b19fe4b73c550f060b59fa592d7d686537eebf9385621bfbad8e"
|
||||
dependencies = [
|
||||
"windows-core",
|
||||
"windows-link 0.1.3",
|
||||
"windows-threading",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-implement"
|
||||
version = "0.60.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "edb307e42a74fb6de9bf3a02d9712678b22399c87e6fa869d6dfcd8c1b7754e0"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-interface"
|
||||
version = "0.59.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c0abd1ddbc6964ac14db11c7213d6532ef34bd9aa042c2e5935f59d7908b46a5"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-link"
|
||||
version = "0.1.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5e6ad25900d524eaabdbbb96d20b4311e1e7ae1699af4fb28c17ae66c80d798a"
|
||||
|
||||
[[package]]
|
||||
name = "windows-link"
|
||||
version = "0.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "45e46c0661abb7180e7b9c281db115305d49ca1709ab8242adf09666d2173c65"
|
||||
|
||||
[[package]]
|
||||
name = "windows-numerics"
|
||||
version = "0.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9150af68066c4c5c07ddc0ce30421554771e528bde427614c61038bc2c92c2b1"
|
||||
dependencies = [
|
||||
"windows-core",
|
||||
"windows-link 0.1.3",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-result"
|
||||
version = "0.3.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "56f42bd332cc6c8eac5af113fc0c1fd6a8fd2aa08a0119358686e5160d0586c6"
|
||||
dependencies = [
|
||||
"windows-link 0.1.3",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-strings"
|
||||
version = "0.4.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "56e6c93f3a0c3b36176cb1327a4958a0353d5d166c2a35cb268ace15e91d3b57"
|
||||
dependencies = [
|
||||
"windows-link 0.1.3",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-sys"
|
||||
version = "0.60.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f2f500e4d28234f72040990ec9d39e3a6b950f9f22d3dba18416c35882612bcb"
|
||||
dependencies = [
|
||||
"windows-targets",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-targets"
|
||||
version = "0.53.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2d42b7b7f66d2a06854650af09cfdf8713e427a439c97ad65a6375318033ac4b"
|
||||
dependencies = [
|
||||
"windows-link 0.2.0",
|
||||
"windows_aarch64_gnullvm",
|
||||
"windows_aarch64_msvc",
|
||||
"windows_i686_gnu",
|
||||
"windows_i686_gnullvm",
|
||||
"windows_i686_msvc",
|
||||
"windows_x86_64_gnu",
|
||||
"windows_x86_64_gnullvm",
|
||||
"windows_x86_64_msvc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows-threading"
|
||||
version = "0.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b66463ad2e0ea3bbf808b7f1d371311c80e115c0b71d60efc142cafbcfb057a6"
|
||||
dependencies = [
|
||||
"windows-link 0.1.3",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "windows_aarch64_gnullvm"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "86b8d5f90ddd19cb4a147a5fa63ca848db3df085e25fee3cc10b39b6eebae764"
|
||||
|
||||
[[package]]
|
||||
name = "windows_aarch64_msvc"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c7651a1f62a11b8cbd5e0d42526e55f2c99886c77e007179efff86c2b137e66c"
|
||||
|
||||
[[package]]
|
||||
name = "windows_i686_gnu"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c1dc67659d35f387f5f6c479dc4e28f1d4bb90ddd1a5d3da2e5d97b42d6272c3"
|
||||
|
||||
[[package]]
|
||||
name = "windows_i686_gnullvm"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9ce6ccbdedbf6d6354471319e781c0dfef054c81fbc7cf83f338a4296c0cae11"
|
||||
|
||||
[[package]]
|
||||
name = "windows_i686_msvc"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "581fee95406bb13382d2f65cd4a908ca7b1e4c2f1917f143ba16efe98a589b5d"
|
||||
|
||||
[[package]]
|
||||
name = "windows_x86_64_gnu"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "2e55b5ac9ea33f2fc1716d1742db15574fd6fc8dadc51caab1c16a3d3b4190ba"
|
||||
|
||||
[[package]]
|
||||
name = "windows_x86_64_gnullvm"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "0a6e035dd0599267ce1ee132e51c27dd29437f63325753051e71dd9e42406c57"
|
||||
|
||||
[[package]]
|
||||
name = "windows_x86_64_msvc"
|
||||
version = "0.53.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "271414315aff87387382ec3d271b52d7ae78726f5d44ac98b4f4030c91880486"
|
||||
@ -84,7 +84,3 @@ Once you have that installed, just run
|
||||
cargo build --release
|
||||
```
|
||||
and you will find an executable in `target/release`.
|
||||
|
||||
### Unit tests
|
||||
|
||||
In order for the unit tests to pass, you will need `python3`.
|
||||
|
||||
62
flake.lock
62
flake.lock
@ -1,62 +0,0 @@
|
||||
{
|
||||
"nodes": {
|
||||
"nixpkgs": {
|
||||
"locked": {
|
||||
"lastModified": 1759580034,
|
||||
"narHash": "sha256-YWo57PL7mGZU7D4WeKFMiW4ex/O6ZolUS6UNBHTZfkI=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "3bcc93c5f7a4b30335d31f21e2f1281cba68c318",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "NixOS",
|
||||
"ref": "nixos-25.05",
|
||||
"repo": "nixpkgs",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"nixpkgs_2": {
|
||||
"locked": {
|
||||
"lastModified": 1744536153,
|
||||
"narHash": "sha256-awS2zRgF4uTwrOKwwiJcByDzDOdo3Q1rPZbiHQg/N38=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "18dd725c29603f582cf1900e0d25f9f1063dbf11",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "NixOS",
|
||||
"ref": "nixpkgs-unstable",
|
||||
"repo": "nixpkgs",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"root": {
|
||||
"inputs": {
|
||||
"nixpkgs": "nixpkgs",
|
||||
"rust-overlay": "rust-overlay"
|
||||
}
|
||||
},
|
||||
"rust-overlay": {
|
||||
"inputs": {
|
||||
"nixpkgs": "nixpkgs_2"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1759631821,
|
||||
"narHash": "sha256-V8A1L0FaU/aSXZ1QNJScxC12uP4hANeRBgI4YdhHeRM=",
|
||||
"owner": "oxalica",
|
||||
"repo": "rust-overlay",
|
||||
"rev": "1d7cbdaad90f8a5255a89a6eddd8af24dc89cafe",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "oxalica",
|
||||
"repo": "rust-overlay",
|
||||
"type": "github"
|
||||
}
|
||||
}
|
||||
},
|
||||
"root": "root",
|
||||
"version": 7
|
||||
}
|
||||
116
flake.nix
116
flake.nix
@ -1,116 +0,0 @@
|
||||
{
|
||||
description = "Flake of https://dev.stupstech.de/Mr_Steppy/multi-ssh";
|
||||
|
||||
inputs = {
|
||||
nixpkgs.url = "github:NixOS/nixpkgs/nixos-25.05";
|
||||
rust-overlay.url = "github:oxalica/rust-overlay";
|
||||
};
|
||||
|
||||
outputs = { self, nixpkgs, rust-overlay, ... }:
|
||||
let
|
||||
systems = nixpkgs.lib.systems.flakeExposed;
|
||||
forAllSystems = nixpkgs.lib.genAttrs systems;
|
||||
in
|
||||
{
|
||||
packages = forAllSystems (system:
|
||||
let
|
||||
overlays = [ rust-overlay.overlays.default ];
|
||||
pkgs = import nixpkgs { inherit system; overlays = overlays; };
|
||||
rustToolchain = pkgs.rust-bin.stable."1.90.0".minimal;
|
||||
rustPlatform = pkgs.makeRustPlatform { cargo = rustToolchain; rustc = rustToolchain; };
|
||||
in {
|
||||
multiSsh = rustPlatform.buildRustPackage {
|
||||
pname = "multi-ssh";
|
||||
version = (builtins.fromTOML (builtins.readFile ./Cargo.toml)).package.version;
|
||||
src = ./.;
|
||||
cargoLock = {
|
||||
lockFile = ./Cargo.lock;
|
||||
};
|
||||
nativeCheckInputs = [ pkgs.python3 ];
|
||||
release = true;
|
||||
};
|
||||
default = self.packages.${system}.multiSsh;
|
||||
}
|
||||
);
|
||||
|
||||
apps = forAllSystems (system:
|
||||
let
|
||||
pkg = self.packages.${system}.multiSsh;
|
||||
in {
|
||||
default = {
|
||||
type = "app";
|
||||
program = "${pkg}/bin/multi-ssh";
|
||||
};
|
||||
}
|
||||
);
|
||||
|
||||
nixosModules.multiSsh = { lib, pkgs, config, ... }: {
|
||||
options = {
|
||||
programs.multiSsh = {
|
||||
enable = lib.mkEnableOption "multi-ssh";
|
||||
editor = lib.mkOption {
|
||||
type = lib.types.nullOr lib.types.str;
|
||||
description = "Editor to forward to multi-ssh (MSSH_EDITOR). If null, the variable is not set.";
|
||||
default = null;
|
||||
};
|
||||
package = lib.mkOption {
|
||||
type = lib.types.package;
|
||||
description = "Derivation to install for multi-ssh";
|
||||
default = self.packages.${pkgs.system}.multiSsh;
|
||||
};
|
||||
servers = {
|
||||
local = {
|
||||
dir = lib.mkOption {
|
||||
type = lib.types.nullOr lib.types.str;
|
||||
default = null;
|
||||
description = ''Local server directory, relative to the user's HOME. If set, a local entry ".:<dir>" is added to MSSH_SERVERS.'';
|
||||
};
|
||||
};
|
||||
server = lib.mkOption {
|
||||
type = lib.types.attrsOf (lib.types.submodule ({ name, ... }: {
|
||||
options = {
|
||||
dir = lib.mkOption {
|
||||
type = lib.types.str;
|
||||
description = "Server directory for ${name} (remote ssh host), as a string.";
|
||||
};
|
||||
};
|
||||
}));
|
||||
default = {};
|
||||
description = "Attribute set of servers where each attribute name is the ssh host and its value provides a 'dir' string.";
|
||||
};
|
||||
};
|
||||
msshAlias = lib.mkOption {
|
||||
type = lib.types.bool;
|
||||
default = true;
|
||||
description = "If true, install an 'mssh' wrapper (alias) that calls 'multi-ssh' and sets env vars.";
|
||||
};
|
||||
};
|
||||
};
|
||||
|
||||
config = lib.mkIf config.programs.multiSsh.enable (let
|
||||
# Build MSSH_SERVERS from options
|
||||
serverEntries = lib.mapAttrsToList (name: value: "${name}:${value.dir}") config.programs.multiSsh.servers.server;
|
||||
localEntry = lib.optional (config.programs.multiSsh.servers.local.dir != null) ".:${config.programs.multiSsh.servers.local.dir}";
|
||||
msshServers = lib.concatStringsSep "," (localEntry ++ serverEntries);
|
||||
|
||||
# wrapper that sets env and execs the real binary (installed as 'multi-ssh')
|
||||
multiSshWrapper = pkgs.writeShellScriptBin "multi-ssh" ''
|
||||
#!${pkgs.runtimeShell}
|
||||
${lib.optionalString (config.programs.multiSsh.editor != null) ''export MSSH_EDITOR="${config.programs.multiSsh.editor}"''}
|
||||
${lib.optionalString (msshServers != "") ''export MSSH_SERVERS="${msshServers}"''}
|
||||
exec ${config.programs.multiSsh.package}/bin/multi-ssh "$@"
|
||||
'';
|
||||
|
||||
# optional 'mssh' alias that delegates to the wrapped 'multi-ssh'
|
||||
msshWrapper = pkgs.writeShellScriptBin "mssh" ''
|
||||
#!${pkgs.runtimeShell}
|
||||
exec ${multiSshWrapper}/bin/multi-ssh "$@"
|
||||
'';
|
||||
in {
|
||||
environment.systemPackages = [
|
||||
multiSshWrapper
|
||||
] ++ lib.optional config.programs.multiSsh.msshAlias msshWrapper;
|
||||
});
|
||||
};
|
||||
};
|
||||
}
|
||||
200
src/command.rs
200
src/command.rs
@ -1,200 +0,0 @@
|
||||
use crate::log;
|
||||
use crate::logger::{LogLevel, Logger};
|
||||
use std::error::Error;
|
||||
use std::fmt::{Debug, Display, Formatter};
|
||||
use std::io;
|
||||
use std::iter::once;
|
||||
use std::process::{Command, ExitStatus, Output};
|
||||
|
||||
pub trait LogRunnable {
|
||||
fn run(&mut self, logger: &Logger) -> Result<(), CommandSpecificError<ExecutionError>>;
|
||||
fn collect_output(&mut self) -> Result<Output, CommandSpecificError<ExecutionError>>;
|
||||
fn collect_full_output(&mut self) -> Result<Output, CommandSpecificError<StartError>>;
|
||||
}
|
||||
|
||||
impl LogRunnable for Command {
|
||||
fn run(&mut self, logger: &Logger) -> Result<(), CommandSpecificError<ExecutionError>> {
|
||||
run(self, logger).map_err(|error| CommandSpecificError {
|
||||
command: self,
|
||||
error,
|
||||
})
|
||||
}
|
||||
|
||||
fn collect_output(&mut self) -> Result<Output, CommandSpecificError<ExecutionError>> {
|
||||
collect_output(self, None).map_err(|error| CommandSpecificError {
|
||||
command: self,
|
||||
error,
|
||||
})
|
||||
}
|
||||
|
||||
fn collect_full_output(&mut self) -> Result<Output, CommandSpecificError<StartError>> {
|
||||
collect_full_output(self).map_err(|error| CommandSpecificError {
|
||||
command: self,
|
||||
error,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
fn run(command: &mut Command, logger: &Logger) -> Result<(), ExecutionError> {
|
||||
match logger.level {
|
||||
LogLevel::Debug | LogLevel::Info => {
|
||||
let status = command.status()?;
|
||||
if !status.success() {
|
||||
Err(status)?;
|
||||
}
|
||||
}
|
||||
LogLevel::Error => {
|
||||
collect_output(command, Some(logger))?;
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn collect_output(
|
||||
command: &mut Command,
|
||||
logger: Option<&Logger>,
|
||||
) -> Result<Output, ExecutionError> {
|
||||
let output = collect_full_output(command)?;
|
||||
if !output.status.success() {
|
||||
if let Some(logger) = logger {
|
||||
log!(logger, error, "{}", String::from_utf8_lossy(&output.stdout));
|
||||
log!(logger, error, "{}", String::from_utf8_lossy(&output.stderr));
|
||||
}
|
||||
Err(output.status)?;
|
||||
}
|
||||
Ok(output)
|
||||
}
|
||||
|
||||
fn collect_full_output(command: &mut Command) -> Result<Output, StartError> {
|
||||
Ok(command.output()?)
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct CommandSpecificError<'a, E> {
|
||||
pub command: &'a Command,
|
||||
pub error: E,
|
||||
}
|
||||
|
||||
impl<E> Display for CommandSpecificError<'_, E>
|
||||
where
|
||||
E: Display,
|
||||
{
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
write!(
|
||||
f,
|
||||
"Failed to execute command '{}': {}",
|
||||
command_to_string(self.command),
|
||||
self.error
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
fn command_to_string(command: &Command) -> String {
|
||||
once(command.get_program().to_string_lossy().to_string())
|
||||
.chain(command.get_args().map(|arg| {
|
||||
let arg_str = arg.to_string_lossy();
|
||||
if arg_str.contains(' ') {
|
||||
format!("\"{arg_str}\"")
|
||||
} else {
|
||||
arg_str.to_string()
|
||||
}
|
||||
}))
|
||||
.collect::<Vec<_>>()
|
||||
.join(" ")
|
||||
}
|
||||
|
||||
impl<E> Error for CommandSpecificError<'_, E> where E: Debug + Display {}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct StartError(io::Error);
|
||||
|
||||
impl From<io::Error> for StartError {
|
||||
fn from(value: io::Error) -> Self {
|
||||
StartError(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for StartError {
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
write!(f, "Failed to start command: {}", self.0)
|
||||
}
|
||||
}
|
||||
|
||||
impl Error for StartError {}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum ExecutionError {
|
||||
StartError(StartError),
|
||||
BadExitStatus(ExitStatus),
|
||||
}
|
||||
|
||||
impl From<io::Error> for ExecutionError {
|
||||
fn from(value: io::Error) -> Self {
|
||||
Self::StartError(StartError(value))
|
||||
}
|
||||
}
|
||||
|
||||
impl From<StartError> for ExecutionError {
|
||||
fn from(value: StartError) -> Self {
|
||||
Self::StartError(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl From<ExitStatus> for ExecutionError {
|
||||
fn from(value: ExitStatus) -> Self {
|
||||
Self::BadExitStatus(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for ExecutionError {
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
ExecutionError::StartError(e) => Display::fmt(e, f),
|
||||
ExecutionError::BadExitStatus(status) => write!(f, "Command failed with {}", status),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Error for ExecutionError {}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
use crate::command::{CommandSpecificError, ExecutionError, LogRunnable};
|
||||
use crate::logger::Logger;
|
||||
use std::path::PathBuf;
|
||||
use std::process::Command;
|
||||
|
||||
#[test]
|
||||
fn test_unknown_command() {
|
||||
let mut command = Command::new("python7");
|
||||
let Err(
|
||||
e @ CommandSpecificError {
|
||||
error: ExecutionError::StartError(_),
|
||||
..
|
||||
},
|
||||
) = command
|
||||
.args([PathBuf::from("test-ressources/python/exit_1.py")])
|
||||
.run(&Logger::default())
|
||||
else {
|
||||
panic!("command shouldn't exist");
|
||||
};
|
||||
assert_eq!(e.to_string(), "Failed to execute command 'python7 test-ressources/python/exit_1.py': Failed to start command: No such file or directory (os error 2)");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_error() {
|
||||
let mut command = Command::new("python3");
|
||||
let Err(
|
||||
e @ CommandSpecificError {
|
||||
error: ExecutionError::BadExitStatus(_),
|
||||
..
|
||||
},
|
||||
) = command
|
||||
.arg("test-ressources/python/exit_1.py")
|
||||
.run(&Logger::default())
|
||||
else {
|
||||
panic!("command should return exit-code 1")
|
||||
};
|
||||
assert_eq!(e.to_string(), "Failed to execute command 'python3 test-ressources/python/exit_1.py': Command failed with exit status: 1");
|
||||
}
|
||||
}
|
||||
96
src/environment.rs
Normal file
96
src/environment.rs
Normal file
@ -0,0 +1,96 @@
|
||||
use crate::shell_interface::{
|
||||
build_command_from_shell_command, CommandOutput, CommandResult, ExitStatus, ShellCommand,
|
||||
ShellInterface, StartError,
|
||||
};
|
||||
use std::env::VarError;
|
||||
use std::ffi::{OsStr, OsString};
|
||||
use std::path::PathBuf;
|
||||
use std::{env, io};
|
||||
|
||||
pub trait Environment {
|
||||
fn args_os(&self) -> Vec<OsString>;
|
||||
fn var_os<K>(&self, key: K) -> Option<OsString>
|
||||
where
|
||||
K: AsRef<OsStr>;
|
||||
fn var<K>(&self, key: K) -> Result<String, VarError>
|
||||
where
|
||||
K: AsRef<OsStr>,
|
||||
{
|
||||
self
|
||||
.var_os(key)
|
||||
.ok_or(VarError::NotPresent)
|
||||
.and_then(|s| s.into_string().map_err(VarError::NotUnicode))
|
||||
}
|
||||
fn set_var<K, V>(&mut self, key: K, value: V)
|
||||
where
|
||||
K: AsRef<OsStr>,
|
||||
V: AsRef<OsStr>;
|
||||
fn get_home_directory(&self) -> Option<PathBuf>;
|
||||
fn read_line(&mut self) -> Result<String, io::Error>;
|
||||
}
|
||||
|
||||
#[derive(Debug, Default)]
|
||||
pub struct Prod;
|
||||
|
||||
impl Environment for Prod {
|
||||
fn args_os(&self) -> Vec<OsString> {
|
||||
env::args_os().collect()
|
||||
}
|
||||
|
||||
fn var_os<K>(&self, key: K) -> Option<OsString>
|
||||
where
|
||||
K: AsRef<OsStr>
|
||||
{
|
||||
env::var_os(key)
|
||||
}
|
||||
|
||||
fn var<K>(&self, key: K) -> Result<String, VarError>
|
||||
where
|
||||
K: AsRef<OsStr>,
|
||||
{
|
||||
env::var(key)
|
||||
}
|
||||
|
||||
fn set_var<K, V>(&mut self, key: K, value: V)
|
||||
where
|
||||
K: AsRef<OsStr>,
|
||||
V: AsRef<OsStr>,
|
||||
{
|
||||
env::set_var(key, value);
|
||||
}
|
||||
|
||||
fn get_home_directory(&self) -> Option<PathBuf> {
|
||||
homedir::my_home().ok().flatten()
|
||||
}
|
||||
|
||||
fn read_line(&mut self) -> Result<String, io::Error> {
|
||||
let mut buffer = String::new();
|
||||
io::stdin().read_line(&mut buffer)?;
|
||||
Ok(buffer.trim().to_string())
|
||||
}
|
||||
}
|
||||
|
||||
impl ShellInterface for Prod {
|
||||
fn run_command(&mut self, command: ShellCommand) -> CommandResult<ExitStatus, StartError> {
|
||||
CommandResult {
|
||||
result: build_command_from_shell_command(&command)
|
||||
.status()
|
||||
.map(ExitStatus::from)
|
||||
.map_err(StartError::from),
|
||||
command,
|
||||
}
|
||||
}
|
||||
|
||||
fn collect_command_output(
|
||||
&mut self,
|
||||
command: ShellCommand,
|
||||
) -> CommandResult<CommandOutput, StartError> {
|
||||
CommandResult {
|
||||
result: build_command_from_shell_command(&command)
|
||||
.output()
|
||||
.map(CommandOutput::from)
|
||||
.map_err(StartError::from),
|
||||
command,
|
||||
}
|
||||
}
|
||||
}
|
||||
90
src/integration_test.rs
Normal file
90
src/integration_test.rs
Normal file
@ -0,0 +1,90 @@
|
||||
use crate::environment::Environment;
|
||||
use crate::shell_interface::{
|
||||
CommandOutput, CommandResult, ExitStatus, ShellCommand, ShellInterface, StartError,
|
||||
};
|
||||
use std::collections::{HashMap, VecDeque};
|
||||
use std::ffi::{OsStr, OsString};
|
||||
use std::io::Error;
|
||||
use std::path::PathBuf;
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct TestEnvironment {
|
||||
/// passed command line arguments
|
||||
args_os: Vec<OsString>,
|
||||
/// set environment variables - we assume a pure environment by default
|
||||
env_vars: HashMap<OsString, OsString>,
|
||||
/// home directory, relative to the target/test folder
|
||||
home_dir: PathBuf,
|
||||
/// pending lines of std input
|
||||
stdin: VecDeque<String>,
|
||||
/// whether an ssh agent has been started successfully
|
||||
ssh_agent_started: bool,
|
||||
// TODO ssh servers and local server
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct SshServer {
|
||||
pub name: String,
|
||||
pub home_dir: FsEntry,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct FsEntry {
|
||||
pub name: OsString,
|
||||
pub kind: FsEntryKind,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum FsEntryKind {
|
||||
Directory(Dir),
|
||||
File {
|
||||
contents: String,
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Dir {
|
||||
pub contents: Vec<FsEntry>,
|
||||
}
|
||||
|
||||
impl Environment for TestEnvironment {
|
||||
fn args_os(&self) -> Vec<OsString> {
|
||||
self.args_os.clone()
|
||||
}
|
||||
|
||||
fn var_os<K>(&self, key: K) -> Option<OsString>
|
||||
where
|
||||
K: AsRef<OsStr>
|
||||
{
|
||||
self.env_vars.get(key.as_ref()).map(|s| s.into())
|
||||
}
|
||||
|
||||
fn set_var<K, V>(&mut self, key: K, value: V)
|
||||
where
|
||||
K: AsRef<OsStr>,
|
||||
V: AsRef<OsStr>,
|
||||
{
|
||||
self.env_vars.insert(key.as_ref().into(), value.as_ref().into());
|
||||
}
|
||||
|
||||
fn get_home_directory(&self) -> Option<PathBuf> {
|
||||
PathBuf::from("target/integration_test").join(&self.home_dir).into()
|
||||
}
|
||||
|
||||
fn read_line(&mut self) -> Result<String, Error> {
|
||||
self.stdin.pop_front().ok_or_else(|| Error::other("Unexpected call to read_line: No input prepared"))
|
||||
}
|
||||
}
|
||||
|
||||
impl ShellInterface for TestEnvironment {
|
||||
fn run_command(&mut self, command: ShellCommand) -> CommandResult<ExitStatus, StartError> {
|
||||
todo!()
|
||||
}
|
||||
|
||||
fn collect_command_output(
|
||||
&mut self,
|
||||
command: ShellCommand,
|
||||
) -> CommandResult<CommandOutput, StartError> {
|
||||
todo!()
|
||||
}
|
||||
}
|
||||
507
src/main.rs
507
src/main.rs
@ -1,18 +1,22 @@
|
||||
mod action;
|
||||
mod command;
|
||||
mod environment;
|
||||
mod file;
|
||||
#[cfg(test)]
|
||||
mod integration_test;
|
||||
mod logger;
|
||||
mod os_string_builder;
|
||||
mod os_str_extension;
|
||||
mod os_string_builder;
|
||||
mod server;
|
||||
mod shell_interface;
|
||||
|
||||
use crate::action::{Action, FileAction, ServerActions};
|
||||
use crate::command::{CommandSpecificError, ExecutionError, LogRunnable};
|
||||
use crate::environment::{Environment, Prod};
|
||||
use crate::file::{FileMatcher, FileNameInfo};
|
||||
use crate::logger::{LogLevel, Logger};
|
||||
use crate::os_str_extension::OsStrExtension;
|
||||
use crate::os_string_builder::ReplaceWithOsStr;
|
||||
use crate::server::{RelativeLocalPathAnker, ServerAddress};
|
||||
use crate::shell_interface::{ScpParam, ServerCommand, ShellCommand, ShellInterface};
|
||||
use clap::{Parser, Subcommand, ValueEnum};
|
||||
use lazy_regex::{lazy_regex, Lazy, Regex};
|
||||
use server::{Server, ServerReference};
|
||||
@ -22,15 +26,12 @@ use std::hash::Hash;
|
||||
use std::io::Write;
|
||||
use std::iter::once;
|
||||
use std::path::{Path, PathBuf};
|
||||
use std::str::FromStr;
|
||||
use std::{env, fs, io};
|
||||
|
||||
const SERVERS_ENV_VAR: &str = "MSSH_SERVERS";
|
||||
const EDITOR_ENV_VAR: &str = "MSSH_EDITOR";
|
||||
const FILE_PLACEHOLDER: &str = "<file>";
|
||||
|
||||
type ShellCmd = std::process::Command;
|
||||
|
||||
/// Uploads a file or executes a command on multiple configured servers
|
||||
///
|
||||
/// Servers must either be configured via environment variable or denote their server directory with
|
||||
@ -41,13 +42,13 @@ type ShellCmd = std::process::Command;
|
||||
/// Use `MSSH_SERVERS="crea:home/crea,sky:sky,lobby:,city:city2"` to configure servers.
|
||||
#[derive(Parser, Debug)]
|
||||
#[command(version, about, long_about)]
|
||||
struct Args {
|
||||
pub struct Args {
|
||||
/// The action to perform
|
||||
#[command(subcommand)]
|
||||
command: Command,
|
||||
/// The ssh names and optionally home directories of the servers to perform the action on
|
||||
#[arg(num_args = 0.., value_parser = ServerReference::from_str)]
|
||||
servers: Vec<ServerReference>,
|
||||
#[arg(num_args = 0..)]
|
||||
servers: Vec<String>,
|
||||
/// How verbose logging output should be
|
||||
#[arg(long, default_value = "info", conflicts_with_all = ["quiet", "info"])]
|
||||
log_level: LogLevel,
|
||||
@ -71,7 +72,7 @@ enum Command {
|
||||
/// When this option is set, the file path must be absolute, or relative to the server directory.
|
||||
/// The upload-directory has no influence on where the file will be taken from.
|
||||
#[arg(short = 'S', long)]
|
||||
file_server: Option<ServerReference>,
|
||||
file_server: Option<String>,
|
||||
/// How to handle older versions of the file
|
||||
#[arg(short = 'a', long, default_value = "delete", default_missing_value = "archive", num_args = 0..=1)]
|
||||
old_version_policy: OldVersionPolicy,
|
||||
@ -137,24 +138,26 @@ enum OldVersionPolicy {
|
||||
Delete,
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! input {
|
||||
($prompt: tt) => {{
|
||||
print!($prompt);
|
||||
io::stdout().flush().expect("failed to flush stdout");
|
||||
let mut buf = String::new();
|
||||
io::stdin()
|
||||
.read_line(&mut buf)
|
||||
.expect("failed to read stdin");
|
||||
buf.trim().to_string()
|
||||
}};
|
||||
() => {
|
||||
input!()
|
||||
};
|
||||
#[derive(Debug, Default)]
|
||||
pub struct Application<E> {
|
||||
pub environment: E,
|
||||
}
|
||||
|
||||
fn main() -> Result<(), String> {
|
||||
let args = Args::parse();
|
||||
impl<E> Application<E>
|
||||
where
|
||||
E: Environment + ShellInterface,
|
||||
{
|
||||
pub fn run(&mut self) -> Result<(), String> {
|
||||
let args = Args::try_parse_from(self.environment.args_os()).map_err(|e| e.to_string())?;
|
||||
self.run_with_args(args)
|
||||
}
|
||||
|
||||
pub fn run_with_args(&mut self, args: Args) -> Result<(), String> {
|
||||
macro_rules! env {
|
||||
() => {
|
||||
&mut self.environment
|
||||
};
|
||||
}
|
||||
|
||||
let logger = Logger {
|
||||
//all the below options are conflicting with each other so an if else is fine
|
||||
@ -167,10 +170,16 @@ fn main() -> Result<(), String> {
|
||||
},
|
||||
};
|
||||
|
||||
let mut configured_servers = LazyCell::new(parse_server_configuration_from_env);
|
||||
let mut configured_servers = LazyCell::new(|| self.parse_server_configuration_from_env());
|
||||
let servers = args
|
||||
.servers
|
||||
.iter()
|
||||
.into_iter()
|
||||
.map(|ref_str| {
|
||||
ServerReference::from_str(&ref_str, || self.get_home_directory())
|
||||
.map_err(|e| format!("Invalid server reference '{ref_str}': {e}"))
|
||||
})
|
||||
.collect::<Result<Vec<_>, _>>()?
|
||||
.into_iter()
|
||||
.map(|server_reference| {
|
||||
let server_identifier = server_reference.get_identifier();
|
||||
server_reference
|
||||
@ -196,13 +205,15 @@ fn main() -> Result<(), String> {
|
||||
pure,
|
||||
exclude,
|
||||
} => {
|
||||
require_non_empty_servers(&servers)?;
|
||||
require_non_empty(&files, "files to upload")?;
|
||||
start_ssh_agent(&logger)?;
|
||||
Self::require_non_empty_servers(&servers)?;
|
||||
Self::require_non_empty(&files, "files to upload")?;
|
||||
|
||||
//resolve file server
|
||||
let file_server = match file_server {
|
||||
Some(server_reference) => {
|
||||
Some(ref_str) => {
|
||||
let server_reference =
|
||||
ServerReference::from_str(&ref_str, || self.get_home_directory())
|
||||
.map_err(|e| format!("Invalid file-server reference '{ref_str}': {e}"))?;
|
||||
let file_server_identifier = server_reference.get_identifier().to_string();
|
||||
let server = server_reference.try_resolve_lazy(&mut configured_servers)
|
||||
.map_err(|e| format!("Can't resolve server directory for file-server '{file_server_identifier}': {e}"))?
|
||||
@ -212,6 +223,8 @@ fn main() -> Result<(), String> {
|
||||
None => None,
|
||||
};
|
||||
|
||||
self.start_ssh_agent(&logger)?;
|
||||
|
||||
//make sure files exist
|
||||
match &file_server {
|
||||
Some(file_server) => match &file_server.address {
|
||||
@ -220,23 +233,29 @@ fn main() -> Result<(), String> {
|
||||
files = files
|
||||
.iter()
|
||||
.map(|file| {
|
||||
let output = ShellCmd::new("ssh")
|
||||
.arg(ssh_address)
|
||||
.arg(osf!("realpath -e ") + file_server.server_directory_path.join(file))
|
||||
.collect_full_output()
|
||||
let output = ShellCommand::Ssh {
|
||||
address: ssh_address.to_string(),
|
||||
server_command: ServerCommand::Realpath {
|
||||
path: file_server.server_directory_path.join(file),
|
||||
},
|
||||
}
|
||||
.in_env(env!())
|
||||
.output()
|
||||
.into_result()
|
||||
.map_err(|e| format!("Failed to canonicalize files: {e}"))?;
|
||||
|
||||
if !output.status.success() {
|
||||
if !output.status.success {
|
||||
Err(format!(
|
||||
"Path doesn't match any files on file-server: {}",
|
||||
file.to_string_lossy()
|
||||
))?;
|
||||
}
|
||||
|
||||
let denoted_files = osstring_from_ssh_output(output.stdout)
|
||||
let denoted_files = output
|
||||
.stdout
|
||||
.split(b'\n') //split at line breaks
|
||||
.into_iter()
|
||||
.filter(|bytes| !bytes.is_empty()) //needed since realpath sometimes gives us empty lines
|
||||
.filter(|file_name| !file_name.is_empty()) //needed since realpath sometimes gives us empty lines
|
||||
.map(PathBuf::from)
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
@ -250,9 +269,9 @@ fn main() -> Result<(), String> {
|
||||
ServerAddress::Localhost => files
|
||||
.iter()
|
||||
.map(|file| file_server.server_directory_path.join(file))
|
||||
.try_for_each(check_local_file_exists)?,
|
||||
.try_for_each(Self::check_local_file_exists)?,
|
||||
},
|
||||
None => files.iter().try_for_each(check_local_file_exists)?,
|
||||
None => files.iter().try_for_each(Self::check_local_file_exists)?,
|
||||
}
|
||||
|
||||
let file_details = files
|
||||
@ -294,26 +313,32 @@ fn main() -> Result<(), String> {
|
||||
server,
|
||||
actions: {
|
||||
let present_file_names: Vec<OsString> = match &server.address {
|
||||
ServerAddress::Ssh { ssh_address } => osstring_from_ssh_output(
|
||||
ShellCmd::new("ssh")
|
||||
.arg(ssh_address)
|
||||
.arg(osf!("ls ") + &working_directory)
|
||||
.collect_output()
|
||||
ServerAddress::Ssh { ssh_address } => ShellCommand::Ssh {
|
||||
address: ssh_address.to_string(),
|
||||
server_command: ServerCommand::Ls {
|
||||
dir: working_directory.clone(),
|
||||
},
|
||||
}
|
||||
.in_env(env!())
|
||||
.output()
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| {
|
||||
format!(
|
||||
"Failed to query present files on server {}: {e}",
|
||||
server.get_name()
|
||||
)
|
||||
})?
|
||||
.stdout,
|
||||
)
|
||||
.stdout
|
||||
.split(b'\n')
|
||||
.into_iter()
|
||||
.map(OsString::from)
|
||||
.collect(),
|
||||
ServerAddress::Localhost => fs::read_dir(&working_directory)
|
||||
.map_err(|e| format!("Failed to get files in working directory: {e}"))?
|
||||
.map(|entry| entry.map_err(|e| format!("Failed to access directory entry: {e}")))
|
||||
.map(|entry| {
|
||||
entry.map_err(|e| format!("Failed to access directory entry: {e}"))
|
||||
})
|
||||
.collect::<Result<Vec<_>, _>>()?
|
||||
.into_iter()
|
||||
.filter_map(|entry| {
|
||||
@ -341,7 +366,8 @@ fn main() -> Result<(), String> {
|
||||
|
||||
let file_name = file_name_info.to_full_file_name();
|
||||
|
||||
let add_action = FileAction::new(file, Action::Add).expect("path points to file");
|
||||
let add_action =
|
||||
FileAction::new(file, Action::Add).expect("path points to file");
|
||||
|
||||
if pure && present_file_names.iter().any(|file| *file == file_name) {
|
||||
log!(
|
||||
@ -359,8 +385,7 @@ fn main() -> Result<(), String> {
|
||||
if !present_file_names.iter().any(|file| *file == file_name) {
|
||||
vec![add_action] //file doesn't exist yet
|
||||
} else {
|
||||
vec![FileAction::new(file, Action::Replace)
|
||||
.expect("path points to file")]
|
||||
vec![FileAction::new(file, Action::Replace).expect("path points to file")]
|
||||
}
|
||||
}
|
||||
OldVersionPolicy::Archive => present_file_names
|
||||
@ -392,7 +417,8 @@ fn main() -> Result<(), String> {
|
||||
if *present_file == file_name {
|
||||
FileAction::new(file, Action::Replace).expect("path points to file")
|
||||
} else {
|
||||
FileAction::new(present_file, Action::Delete).expect("path points to file")
|
||||
FileAction::new(present_file, Action::Delete)
|
||||
.expect("path points to file")
|
||||
}
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
@ -427,15 +453,10 @@ fn main() -> Result<(), String> {
|
||||
}
|
||||
}
|
||||
|
||||
if !no_confirm {
|
||||
match input!("Continue? [Y|n] ").to_lowercase().as_str() {
|
||||
"n" | "no" => {
|
||||
if !no_confirm && !self.confirm("Continue?", true) {
|
||||
log!(logger, "Aborting...");
|
||||
return Ok(());
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
|
||||
for server_actions in actions {
|
||||
let server = server_actions.server;
|
||||
@ -443,31 +464,36 @@ fn main() -> Result<(), String> {
|
||||
for file_action in server_actions.actions {
|
||||
match file_action.kind {
|
||||
Action::Add | Action::Replace => {
|
||||
let scp_source = match &file_server {
|
||||
Some(file_server) => {
|
||||
osf!(match &file_server.address {
|
||||
ServerAddress::Ssh { ssh_address } => format!("{ssh_address}:"),
|
||||
ServerAddress::Localhost => "".to_string(),
|
||||
}) + file_server.server_directory_path.join(&file_action.file)
|
||||
}
|
||||
None => osf!(&file_action.file),
|
||||
let source = match &file_server {
|
||||
Some(file_server) => ScpParam::from((
|
||||
file_server,
|
||||
file_server.server_directory_path.join(&file_action.file),
|
||||
)),
|
||||
None => ScpParam::from(file_action.file.as_path()),
|
||||
};
|
||||
let scp_target = osf!(match &server.address {
|
||||
ServerAddress::Ssh { ssh_address } => format!("{ssh_address}:"),
|
||||
ServerAddress::Localhost => "".to_string(),
|
||||
}) + &server_actions.working_directory;
|
||||
ShellCmd::new("scp")
|
||||
.arg(scp_source)
|
||||
.arg(scp_target)
|
||||
.run(&logger)
|
||||
let destination = ScpParam::from((server, &server_actions.working_directory));
|
||||
ShellCommand::Scp {
|
||||
source,
|
||||
destination,
|
||||
}
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| format!("upload failure: {e}"))?;
|
||||
}
|
||||
Action::Delete => match &server.address {
|
||||
ServerAddress::Ssh { ssh_address } => {
|
||||
ShellCmd::new("ssh")
|
||||
.arg(ssh_address)
|
||||
.arg(osf!("rm ") + server_actions.working_directory.join(&file_action.file))
|
||||
.run(&logger)
|
||||
ShellCommand::Ssh {
|
||||
address: ssh_address.to_string(),
|
||||
server_command: ServerCommand::Rm {
|
||||
file: server_actions.working_directory.join(&file_action.file),
|
||||
},
|
||||
}
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| format!("failed to delete old version: {e}"))?;
|
||||
}
|
||||
ServerAddress::Localhost => {
|
||||
@ -477,15 +503,17 @@ fn main() -> Result<(), String> {
|
||||
},
|
||||
Action::Rename { new_name } => match &server.address {
|
||||
ServerAddress::Ssh { ssh_address } => {
|
||||
ShellCmd::new("ssh")
|
||||
.arg(ssh_address)
|
||||
.arg(
|
||||
osf!("mv ")
|
||||
+ server_actions.working_directory.join(&file_action.file)
|
||||
+ " "
|
||||
+ server_actions.working_directory.join(&new_name),
|
||||
)
|
||||
.run(&logger)
|
||||
ShellCommand::Ssh {
|
||||
address: ssh_address.to_string(),
|
||||
server_command: ServerCommand::Mv {
|
||||
source: server_actions.working_directory.join(&file_action.file),
|
||||
destination: server_actions.working_directory.join(&new_name),
|
||||
},
|
||||
}
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| format!("failed to rename: {e}"))?;
|
||||
}
|
||||
ServerAddress::Localhost => {
|
||||
@ -501,26 +529,39 @@ fn main() -> Result<(), String> {
|
||||
log!(logger, "Done!");
|
||||
}
|
||||
Command::Command { command } => {
|
||||
start_ssh_agent(&logger)?;
|
||||
require_non_empty_servers(&servers)?;
|
||||
self.start_ssh_agent(&logger)?;
|
||||
Self::require_non_empty_servers(&servers)?;
|
||||
for server in servers {
|
||||
log!(logger, "Running command on '{}'...", server.get_name());
|
||||
match &server.address {
|
||||
ServerAddress::Ssh { ssh_address } => {
|
||||
ShellCmd::new("ssh")
|
||||
.arg(ssh_address)
|
||||
.arg(osf!("cd ") + server.server_directory_path + "; " + &command)
|
||||
.run(&logger)
|
||||
ShellCommand::Ssh {
|
||||
address: ssh_address.to_string(),
|
||||
server_command: ServerCommand::Execute {
|
||||
working_directory: server.server_directory_path.clone(),
|
||||
command: OsString::from(&command),
|
||||
},
|
||||
}
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| format!("{e}"))?;
|
||||
}
|
||||
ServerAddress::Localhost => {
|
||||
let mut command_args =
|
||||
shell_words::split(&command).map_err(|e| format!("failed to parse command: {e}"))?;
|
||||
ShellCmd::new(command_args.remove(0))
|
||||
.args(&command_args)
|
||||
.current_dir(&server.server_directory_path)
|
||||
.run(&logger)
|
||||
.map_err(|e| format!("{e}"))?;
|
||||
let command = shell_words::split(&command)
|
||||
.map_err(|e| format!("failed to parse command: {e}"))?
|
||||
.into_iter()
|
||||
.map(OsString::from)
|
||||
.collect();
|
||||
ShellCommand::Execute {
|
||||
working_directory: server.server_directory_path.clone(),
|
||||
command,
|
||||
}
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)?;
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -536,8 +577,9 @@ fn main() -> Result<(), String> {
|
||||
let download_directory = match download_directory {
|
||||
Some(download_directory) => download_directory,
|
||||
None => {
|
||||
let home_dir =
|
||||
get_home_directory().map_err(|e| format!("Can't determine download directory: {e}"))?;
|
||||
let home_dir = self
|
||||
.get_home_directory()
|
||||
.map_err(|e| format!("Missing download-directory: {e}"))?;
|
||||
home_dir.join("Downloads")
|
||||
}
|
||||
};
|
||||
@ -567,14 +609,13 @@ fn main() -> Result<(), String> {
|
||||
download_directory.to_string_lossy()
|
||||
);
|
||||
|
||||
if !args.quiet {
|
||||
match input!("{duplication_notification}. Do you want to replace it? [N|y] ")
|
||||
.to_lowercase()
|
||||
.as_str()
|
||||
if !args.quiet
|
||||
&& self.confirm(
|
||||
format!("{duplication_notification}. Do you want to replace it?"),
|
||||
false,
|
||||
)
|
||||
{
|
||||
"y" | "yes" => break 'duplicate_check,
|
||||
_ => {}
|
||||
}
|
||||
break 'duplicate_check;
|
||||
}
|
||||
|
||||
return Err(format!(
|
||||
@ -583,23 +624,24 @@ fn main() -> Result<(), String> {
|
||||
}
|
||||
}
|
||||
|
||||
require_non_empty_servers(&servers)?;
|
||||
start_ssh_agent(&logger)?;
|
||||
Self::require_non_empty_servers(&servers)?;
|
||||
self.start_ssh_agent(&logger)?;
|
||||
|
||||
for server in servers {
|
||||
log!(logger, "Getting file from {}...", server.get_name());
|
||||
let file_source = osf!(match &server.address {
|
||||
ServerAddress::Ssh { ssh_address } => format!("{ssh_address}:"),
|
||||
ServerAddress::Localhost => "".to_string(),
|
||||
}) + server.server_directory_path.join(&file);
|
||||
ShellCmd::new("scp")
|
||||
.arg(&file_source)
|
||||
.arg(&download_directory)
|
||||
.run(&logger)
|
||||
let source = ScpParam::from((&server, server.server_directory_path.join(&file)));
|
||||
ShellCommand::Scp {
|
||||
source: source.clone(),
|
||||
destination: ScpParam::from(download_directory.as_path()),
|
||||
}
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| format!("download failure: {e}"))?;
|
||||
|
||||
//open file in editor
|
||||
let mut editor_command_args = shell_words::split(&editor)
|
||||
let editor_command = shell_words::split(&editor)
|
||||
.map_err(|e| format!("failed to parse editor command: {e}"))?
|
||||
.into_iter()
|
||||
.map(|part| {
|
||||
@ -607,17 +649,22 @@ fn main() -> Result<(), String> {
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
|
||||
let command = editor_command_args.remove(0);
|
||||
ShellCmd::new(command)
|
||||
.args(editor_command_args)
|
||||
.run(&logger)
|
||||
ShellCommand::Editor(editor_command)
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| format!("failed to open file in editor: {e}"))?;
|
||||
|
||||
//upload file again
|
||||
ShellCmd::new("scp")
|
||||
.arg(download_directory.join(file_name))
|
||||
.arg(&file_source)
|
||||
.run(&logger)
|
||||
ShellCommand::Scp {
|
||||
source: ScpParam::from(download_directory.join(file_name).as_path()),
|
||||
destination: source,
|
||||
}
|
||||
.in_env(env!())
|
||||
.run_logged(&logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(&logger)
|
||||
.map_err(|e| format!("failed to re-upload file: {e}"))?;
|
||||
}
|
||||
|
||||
@ -628,20 +675,6 @@ fn main() -> Result<(), String> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn osstring_from_ssh_output(output: Vec<u8>) -> OsString {
|
||||
#[cfg(unix)]
|
||||
{
|
||||
use std::os::unix::ffi::OsStringExt;
|
||||
OsString::from_vec(output)
|
||||
}
|
||||
|
||||
#[cfg(windows)]
|
||||
{
|
||||
use std::os::windows::ffi::OsStringExt;
|
||||
OsString::from_wide(output.iter().map(|&b| b as u16).collect())
|
||||
}
|
||||
}
|
||||
|
||||
fn check_local_file_exists<P>(path: P) -> Result<(), String>
|
||||
where
|
||||
P: AsRef<Path>,
|
||||
@ -657,45 +690,8 @@ where
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
fn check_file_exists_on_server<P, S, D>(
|
||||
path: P,
|
||||
ssh_address: S,
|
||||
server_directory: D,
|
||||
) -> Result<(), String>
|
||||
where
|
||||
P: AsRef<Path>,
|
||||
S: AsRef<str>,
|
||||
D: AsRef<Path>,
|
||||
{
|
||||
let full_path = server_directory.as_ref().join(path);
|
||||
match &ShellCmd::new("ssh")
|
||||
.arg(ssh_address.as_ref())
|
||||
.arg(osf!("test -f ") + &full_path)
|
||||
.collect_output()
|
||||
{
|
||||
Ok(_) => Ok(()), //file exists on file server
|
||||
Err(CommandSpecificError {
|
||||
error: ExecutionError::BadExitStatus(_), //test failed
|
||||
..
|
||||
}) => Err(format!(
|
||||
"File '{}' doesn't exist on file-server",
|
||||
full_path.to_string_lossy()
|
||||
)),
|
||||
Err(e) => Err(format!(
|
||||
"Failed to check whether file exists on file-server: {e}"
|
||||
)),
|
||||
}
|
||||
}
|
||||
|
||||
fn get_home_directory() -> Result<PathBuf, String> {
|
||||
homedir::my_home()
|
||||
.map_err(|e| format!("Failed to determine home directory: {e}"))
|
||||
.and_then(|home_dir| home_dir.ok_or("Failed to find home directory".to_string()))
|
||||
}
|
||||
|
||||
fn require_non_empty_servers<T>(servers: &[T]) -> Result<(), String> {
|
||||
require_non_empty(servers, "servers for this operation")
|
||||
Self::require_non_empty(servers, "servers for this operation")
|
||||
}
|
||||
|
||||
fn require_non_empty<T>(slice: &[T], slice_name: &str) -> Result<(), String> {
|
||||
@ -707,42 +703,97 @@ fn require_non_empty<T>(slice: &[T], slice_name: &str) -> Result<(), String> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn start_ssh_agent(logger: &Logger) -> Result<(), String> {
|
||||
fn start_ssh_agent(&mut self, logger: &Logger) -> Result<(), String> {
|
||||
let env = &mut self.environment;
|
||||
|
||||
//start the ssh agent
|
||||
let agent_output = ShellCmd::new("ssh-agent")
|
||||
.arg("-s")
|
||||
.collect_output()
|
||||
.map_err(|e| format!("failed to start ssh agent: {e}"))?;
|
||||
let agent_stdout = String::from_utf8_lossy(&agent_output.stdout);
|
||||
if !agent_output.status.success() {
|
||||
return Err("failed to start ssh agent; maybe try to run ssh-agent manually?".to_string());
|
||||
}
|
||||
let agent_output = ShellCommand::SshAgent
|
||||
.in_env(env)
|
||||
.output()
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(logger)
|
||||
.map_err(|e| format!("Failed to start ssh agent: {e}"))?;
|
||||
let agent_stdout = &agent_output
|
||||
.stdout
|
||||
.into_string()
|
||||
.map_err(|_| "ssh-agent returned invalid utf-8 - how did this even happen?")?;
|
||||
|
||||
//set the env vars from the agent
|
||||
static ENV_VAR_REGEX: Lazy<Regex> = lazy_regex!("(.+?)=(.+?);");
|
||||
for capture in ENV_VAR_REGEX.captures_iter(&agent_stdout) {
|
||||
for capture in ENV_VAR_REGEX.captures_iter(agent_stdout) {
|
||||
let (_, [env_var, value]) = capture.extract();
|
||||
env::set_var(env_var, value);
|
||||
env.set_var(env_var, value);
|
||||
}
|
||||
|
||||
//add the ssh key
|
||||
ShellCmd::new("ssh-add")
|
||||
.run(logger)
|
||||
ShellCommand::ShhAdd
|
||||
.in_env(env)
|
||||
.run_logged(logger)
|
||||
.and_expect_success()
|
||||
.into_result_with_error_logging(logger)
|
||||
.map_err(|e| format!("failed to add ssh-key: {e}"))?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn parse_server_configuration_from_env() -> Result<Vec<Server>, String> {
|
||||
env::var(SERVERS_ENV_VAR)
|
||||
fn parse_server_configuration_from_env(&self) -> Result<Vec<Server>, String> {
|
||||
self
|
||||
.environment
|
||||
.var(SERVERS_ENV_VAR)
|
||||
.map_err(|_| format!("Missing environment variable {}", SERVERS_ENV_VAR))
|
||||
.and_then(|value| parse_server_configuration(&value))
|
||||
.and_then(|value| parse_server_configuration(&value, || self.get_home_directory()))
|
||||
}
|
||||
|
||||
fn parse_server_configuration(config_str: &str) -> Result<Vec<Server>, String> {
|
||||
fn get_home_directory(&self) -> Result<PathBuf, String> {
|
||||
self
|
||||
.environment
|
||||
.get_home_directory()
|
||||
.ok_or("Failed to find your home directory".to_string())
|
||||
}
|
||||
|
||||
fn confirm<S>(&mut self, prompt: S, default_value: bool) -> bool
|
||||
where
|
||||
S: ToString,
|
||||
{
|
||||
loop {
|
||||
print!(
|
||||
"{}[{}]",
|
||||
prompt.to_string(),
|
||||
if default_value { "Y|n" } else { "y|N" }
|
||||
);
|
||||
io::stdout().flush().expect("failed to flush stdout");
|
||||
let line = self
|
||||
.environment
|
||||
.read_line()
|
||||
.expect("Failed to read console input");
|
||||
match line.to_lowercase().as_str() {
|
||||
"" => return default_value,
|
||||
"y" | "yes" => return true,
|
||||
"n" | "no" => return false,
|
||||
_ => println!("Invalid input, please choose one of the provided options"),
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn main() -> Result<(), String> {
|
||||
Application::<Prod>::default().run()
|
||||
}
|
||||
|
||||
fn parse_server_configuration<F>(
|
||||
config_str: &str,
|
||||
get_home_directory: F,
|
||||
) -> Result<Vec<Server>, String>
|
||||
where
|
||||
F: Fn() -> Result<PathBuf, String>,
|
||||
{
|
||||
config_str
|
||||
.split(',')
|
||||
.map(|server_entry| {
|
||||
Server::from_str(server_entry, RelativeLocalPathAnker::Home)
|
||||
Server::from_str(
|
||||
server_entry,
|
||||
RelativeLocalPathAnker::Home,
|
||||
&get_home_directory,
|
||||
)
|
||||
.map_err(|e| format!("Invalid server entry '{server_entry}': {e}"))
|
||||
})
|
||||
.collect()
|
||||
@ -757,52 +808,15 @@ mod test {
|
||||
|
||||
#[test]
|
||||
fn test_parse_server_configuration() {
|
||||
use std::env;
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
//setup directory structure for test
|
||||
let home_dir = PathBuf::from("target/test");
|
||||
const LOCAL_SERVER_DIR: &str = "fizz/buzz";
|
||||
fs::create_dir_all(home_dir.join(LOCAL_SERVER_DIR)).expect("failed to create server directory");
|
||||
|
||||
// Create an isolated temporary HOME directory to avoid touching the real home.
|
||||
// Ensure the environment is restored even if the test fails by using a guard.
|
||||
struct EnvVarGuard {
|
||||
key: &'static str,
|
||||
prev: Option<std::ffi::OsString>,
|
||||
}
|
||||
impl EnvVarGuard {
|
||||
fn set_path(key: &'static str, value: &std::path::Path) -> Self {
|
||||
let prev = env::var_os(key);
|
||||
env::set_var(key, value);
|
||||
Self { key, prev }
|
||||
}
|
||||
}
|
||||
impl Drop for EnvVarGuard {
|
||||
fn drop(&mut self) {
|
||||
match &self.prev {
|
||||
Some(v) => env::set_var(self.key, v),
|
||||
None => env::remove_var(self.key),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Build a unique temp directory under the OS temp dir
|
||||
let unique = format!(
|
||||
"multi_ssh_test_home_{}_{}",
|
||||
std::process::id(),
|
||||
SystemTime::now().duration_since(UNIX_EPOCH).unwrap().as_nanos()
|
||||
);
|
||||
let temp_home = env::temp_dir().join(unique);
|
||||
fs::create_dir_all(&temp_home).expect("create temp HOME");
|
||||
|
||||
// Override HOME (and USERPROFILE for Windows) for the duration of the test
|
||||
let _home_guard = EnvVarGuard::set_path("HOME", &temp_home);
|
||||
let _userprofile_guard = EnvVarGuard::set_path("USERPROFILE", &temp_home);
|
||||
|
||||
// Prepare the local dir inside the temp HOME
|
||||
let local_dir = temp_home.join("fizz/buzz");
|
||||
fs::create_dir_all(&local_dir).expect("mkdir -p temp local dir");
|
||||
let canonical_local_dir =
|
||||
fs::canonicalize(&local_dir).expect("canonicalize temp local dir");
|
||||
|
||||
let servers =
|
||||
parse_server_configuration("foo:bar,.:fizz/buzz").expect("valid server configuration");
|
||||
let servers = parse_server_configuration(&format!("foo:bar,.:{LOCAL_SERVER_DIR}"), || {
|
||||
Ok(home_dir.clone())
|
||||
})
|
||||
.expect("valid server configuration");
|
||||
assert_eq!(
|
||||
vec![
|
||||
Server {
|
||||
@ -813,7 +827,10 @@ mod test {
|
||||
},
|
||||
Server {
|
||||
address: ServerAddress::Localhost,
|
||||
server_directory_path: canonical_local_dir,
|
||||
server_directory_path: home_dir
|
||||
.join(LOCAL_SERVER_DIR)
|
||||
.canonicalize()
|
||||
.expect("home dir exists"),
|
||||
}
|
||||
],
|
||||
servers
|
||||
|
||||
@ -1,4 +1,3 @@
|
||||
use crate::get_home_directory;
|
||||
use std::cell::LazyCell;
|
||||
use std::error::Error;
|
||||
use std::fmt::{Display, Formatter};
|
||||
@ -6,7 +5,6 @@ use std::fs;
|
||||
use std::hash::{Hash, Hasher};
|
||||
use std::ops::Deref;
|
||||
use std::path::PathBuf;
|
||||
use std::str::FromStr;
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub enum ServerReference {
|
||||
@ -15,6 +13,19 @@ pub enum ServerReference {
|
||||
}
|
||||
|
||||
impl ServerReference {
|
||||
pub fn from_str<F>(s: &str, get_home_directory: F) -> Result<Self, ServerReferenceParseError>
|
||||
where
|
||||
F: FnOnce() -> Result<PathBuf, String>,
|
||||
{
|
||||
Server::from_str(
|
||||
s,
|
||||
RelativeLocalPathAnker::CurrentDirectory,
|
||||
get_home_directory,
|
||||
)
|
||||
.map(Self::Resolved)
|
||||
.or_else(|_| Ok(Self::Identifier(s.to_string())))
|
||||
}
|
||||
|
||||
pub fn get_identifier(&self) -> &str {
|
||||
match self {
|
||||
ServerReference::Resolved(server) => server.address.identifier(),
|
||||
@ -68,16 +79,6 @@ impl ServerReference {
|
||||
}
|
||||
}
|
||||
|
||||
impl FromStr for ServerReference {
|
||||
type Err = ServerReferenceParseError;
|
||||
|
||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||
Server::from_str(s, RelativeLocalPathAnker::CurrentDirectory)
|
||||
.map(Self::Resolved)
|
||||
.or_else(|_| Ok(Self::Identifier(s.to_string())))
|
||||
}
|
||||
}
|
||||
|
||||
impl PartialEq for ServerReference {
|
||||
fn eq(&self, other: &Self) -> bool {
|
||||
self.get_identifier() == other.get_identifier()
|
||||
@ -126,10 +127,14 @@ impl Server {
|
||||
}
|
||||
}
|
||||
|
||||
pub fn from_str(
|
||||
pub fn from_str<F>(
|
||||
s: &str,
|
||||
relative_local_path_anker: RelativeLocalPathAnker,
|
||||
) -> Result<Self, ServerParseError> {
|
||||
get_home_directory: F,
|
||||
) -> Result<Self, ServerParseError>
|
||||
where
|
||||
F: FnOnce() -> Result<PathBuf, String>,
|
||||
{
|
||||
s.split_once(':')
|
||||
.ok_or(ServerParseError::MissingServerDirectory)
|
||||
.and_then(|(identifier, server_directory)| {
|
||||
@ -245,13 +250,12 @@ impl Error for ServerParseError {}
|
||||
mod test_server_reference {
|
||||
use crate::server::{Server, ServerAddress, ServerReference};
|
||||
use std::path::PathBuf;
|
||||
use std::str::FromStr;
|
||||
|
||||
#[test]
|
||||
fn test_from_str() {
|
||||
assert_eq!(
|
||||
ServerReference::Identifier("foo".to_string()),
|
||||
ServerReference::from_str("foo").unwrap()
|
||||
ServerReference::from_str("foo", || panic!("shouldn't be called")).unwrap()
|
||||
);
|
||||
assert_eq!(
|
||||
ServerReference::Resolved(Server {
|
||||
@ -260,7 +264,7 @@ mod test_server_reference {
|
||||
},
|
||||
server_directory_path: PathBuf::from("server/creative2")
|
||||
}),
|
||||
ServerReference::from_str("crea:server/creative2").unwrap()
|
||||
ServerReference::from_str("crea:server/creative2", || panic!("shouldn't be called")).unwrap()
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
562
src/shell_interface.rs
Normal file
562
src/shell_interface.rs
Normal file
@ -0,0 +1,562 @@
|
||||
use crate::logger::{LogLevel, Logger};
|
||||
use crate::server::{Server, ServerAddress};
|
||||
use crate::{log, osf};
|
||||
use std::error::Error;
|
||||
use std::ffi::OsString;
|
||||
use std::fmt::{Debug, Display, Formatter};
|
||||
use std::iter::once;
|
||||
use std::path::{Path, PathBuf};
|
||||
use std::process::{Command, Output};
|
||||
use std::{io, process};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct EnvCommand<'a, E> {
|
||||
command: ShellCommand,
|
||||
environment: &'a mut E,
|
||||
}
|
||||
|
||||
impl<E> EnvCommand<'_, E>
|
||||
where
|
||||
E: ShellInterface,
|
||||
{
|
||||
pub fn run(self) -> CommandResult<ExitStatus, StartError> {
|
||||
self.environment.run_command(self.command)
|
||||
}
|
||||
|
||||
pub fn output(self) -> CommandResult<CommandOutput, StartError> {
|
||||
self.environment.collect_command_output(self.command)
|
||||
}
|
||||
|
||||
pub fn run_logged(self, logger: &Logger) -> CommandResult<LoggedRunOutput, StartError>
|
||||
where
|
||||
Self: Sized,
|
||||
{
|
||||
match logger.level {
|
||||
LogLevel::Debug | LogLevel::Info => {
|
||||
let res = self.run();
|
||||
CommandResult {
|
||||
result: res.result.map(LoggedRunOutput::from),
|
||||
command: res.command,
|
||||
}
|
||||
}
|
||||
LogLevel::Error => {
|
||||
let res = self.output();
|
||||
CommandResult {
|
||||
result: res.result.map(LoggedRunOutput::from),
|
||||
command: res.command,
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub enum ShellCommand {
|
||||
Ssh {
|
||||
address: String,
|
||||
server_command: ServerCommand,
|
||||
},
|
||||
Scp {
|
||||
source: ScpParam,
|
||||
destination: ScpParam,
|
||||
},
|
||||
SshAgent,
|
||||
ShhAdd,
|
||||
Editor(Vec<OsString>),
|
||||
Execute {
|
||||
working_directory: PathBuf,
|
||||
command: Vec<OsString>,
|
||||
},
|
||||
}
|
||||
|
||||
impl ShellCommand {
|
||||
pub fn in_env<E>(self, environment: &mut E) -> EnvCommand<E> {
|
||||
EnvCommand {
|
||||
command: self,
|
||||
environment,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub enum ServerCommand {
|
||||
Realpath {
|
||||
path: PathBuf,
|
||||
},
|
||||
Ls {
|
||||
dir: PathBuf,
|
||||
},
|
||||
Rm {
|
||||
file: PathBuf,
|
||||
},
|
||||
Mv {
|
||||
source: PathBuf,
|
||||
destination: PathBuf,
|
||||
},
|
||||
Execute {
|
||||
working_directory: PathBuf,
|
||||
command: OsString,
|
||||
},
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ScpParam {
|
||||
pub server: Option<String>,
|
||||
pub path: PathBuf,
|
||||
}
|
||||
|
||||
impl<P> From<(&Server, P)> for ScpParam
|
||||
where
|
||||
P: AsRef<Path>,
|
||||
{
|
||||
fn from((server, path): (&Server, P)) -> Self {
|
||||
Self {
|
||||
server: match &server.address {
|
||||
ServerAddress::Ssh { ssh_address } => Some(ssh_address.into()),
|
||||
ServerAddress::Localhost => None,
|
||||
},
|
||||
path: PathBuf::from(path.as_ref()),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<&Path> for ScpParam {
|
||||
fn from(value: &Path) -> Self {
|
||||
Self {
|
||||
server: None,
|
||||
path: PathBuf::from(value),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<&ScpParam> for OsString {
|
||||
fn from(value: &ScpParam) -> Self {
|
||||
let mut builder = osf!();
|
||||
if let Some(server) = &value.server {
|
||||
builder += format!("{server}:");
|
||||
}
|
||||
builder += &value.path;
|
||||
builder.build()
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for ShellCommand {
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
write!(
|
||||
f,
|
||||
"{}",
|
||||
command_to_string(&build_command_from_shell_command(self))
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
fn command_to_string(command: &Command) -> String {
|
||||
once(command.get_program().to_string_lossy().to_string())
|
||||
.chain(command.get_args().map(|arg| {
|
||||
let arg_str = arg.to_string_lossy();
|
||||
if arg_str.contains(' ') {
|
||||
format!("\"{arg_str}\"")
|
||||
} else {
|
||||
arg_str.to_string()
|
||||
}
|
||||
}))
|
||||
.collect::<Vec<_>>()
|
||||
.join(" ")
|
||||
}
|
||||
|
||||
macro_rules! cmd {
|
||||
($programm: expr $(, $arg:expr )*)=> {{
|
||||
#[allow(unused_mut)]
|
||||
let mut cmd = Command::new($programm);
|
||||
$( cmd.arg($arg); )*
|
||||
cmd
|
||||
}};
|
||||
}
|
||||
|
||||
pub fn build_command_from_shell_command(shell_command: &ShellCommand) -> Command {
|
||||
match shell_command {
|
||||
ShellCommand::Ssh {
|
||||
address,
|
||||
server_command,
|
||||
} => cmd!(
|
||||
"ssh",
|
||||
address,
|
||||
match server_command {
|
||||
ServerCommand::Realpath { path } => osf!("realpath -e ") + path,
|
||||
ServerCommand::Ls { dir } => osf!("ls ") + dir,
|
||||
ServerCommand::Rm { file } => osf!("rm ") + file,
|
||||
ServerCommand::Mv {
|
||||
source,
|
||||
destination,
|
||||
} => osf!("mv ") + source + " " + destination,
|
||||
ServerCommand::Execute {
|
||||
working_directory,
|
||||
command,
|
||||
} => osf!("cd ") + working_directory + "; " + command,
|
||||
}
|
||||
),
|
||||
ShellCommand::Scp {
|
||||
source,
|
||||
destination,
|
||||
} => cmd!("scp", OsString::from(source), OsString::from(destination)),
|
||||
ShellCommand::SshAgent => cmd!("ssh-agent", "-s"),
|
||||
ShellCommand::ShhAdd => cmd!("ssh-add"),
|
||||
ShellCommand::Editor(args) => {
|
||||
let mut args = args.clone();
|
||||
let mut cmd = cmd!(args.remove(0));
|
||||
cmd.args(args);
|
||||
cmd
|
||||
}
|
||||
ShellCommand::Execute {
|
||||
working_directory,
|
||||
command,
|
||||
} => {
|
||||
let mut args = command.clone();
|
||||
let mut cmd = cmd!(args.remove(0));
|
||||
cmd.args(args).current_dir(working_directory);
|
||||
cmd
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub trait ShellInterface {
|
||||
fn run_command(&mut self, command: ShellCommand) -> CommandResult<ExitStatus, StartError>;
|
||||
fn collect_command_output(
|
||||
&mut self,
|
||||
command: ShellCommand,
|
||||
) -> CommandResult<CommandOutput, StartError>;
|
||||
}
|
||||
|
||||
pub trait MaybeCast<T> {
|
||||
fn maybe_cast(&self) -> Option<&T>;
|
||||
}
|
||||
|
||||
impl<T> MaybeCast<T> for T {
|
||||
fn maybe_cast(&self) -> Option<&T> {
|
||||
Some(self)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum LoggedRunOutput {
|
||||
ExitStatus(ExitStatus),
|
||||
CommandOutput(CommandOutput),
|
||||
}
|
||||
|
||||
impl From<ExitStatus> for LoggedRunOutput {
|
||||
fn from(value: ExitStatus) -> Self {
|
||||
Self::ExitStatus(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl From<CommandOutput> for LoggedRunOutput {
|
||||
fn from(value: CommandOutput) -> Self {
|
||||
Self::CommandOutput(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<ExitStatus> for LoggedRunOutput {
|
||||
fn as_ref(&self) -> &ExitStatus {
|
||||
match self {
|
||||
LoggedRunOutput::ExitStatus(status) => status,
|
||||
LoggedRunOutput::CommandOutput(output) => output.as_ref(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl MaybeCast<CommandOutput> for LoggedRunOutput {
|
||||
fn maybe_cast(&self) -> Option<&CommandOutput> {
|
||||
match self {
|
||||
LoggedRunOutput::ExitStatus(_) => None,
|
||||
LoggedRunOutput::CommandOutput(output) => Some(output),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct CommandResult<T, E> {
|
||||
pub command: ShellCommand,
|
||||
pub result: Result<T, E>,
|
||||
}
|
||||
|
||||
impl<T, E> CommandResult<T, E> {
|
||||
pub fn into_result(self) -> Result<T, CommandError<E>> {
|
||||
self.result.map_err(|error| CommandError {
|
||||
command: self.command,
|
||||
error,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> CommandResult<T, StartError> {
|
||||
pub fn and_expect_success(self) -> CommandResult<T, ExecutionError<T>>
|
||||
where
|
||||
T: AsRef<ExitStatus>,
|
||||
{
|
||||
CommandResult {
|
||||
result: self.result.map_err(ExecutionError::from).and_then(|t| {
|
||||
if t.as_ref().success {
|
||||
Ok(t)
|
||||
} else {
|
||||
Err(ExecutionError::BadExitStatus(t))
|
||||
}
|
||||
}),
|
||||
command: self.command,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> CommandResult<T, ExecutionError<T>> {
|
||||
pub fn into_result_with_error_logging(
|
||||
self,
|
||||
logger: &Logger,
|
||||
) -> Result<T, CommandError<ExecutionError<T>>>
|
||||
where
|
||||
T: MaybeCast<CommandOutput>,
|
||||
{
|
||||
self.result.map_err(|error| {
|
||||
if let ExecutionError::BadExitStatus(t) = &error {
|
||||
if let Some(output) = t.maybe_cast() {
|
||||
log!(logger, error, "{}", output.stdout.to_string_lossy());
|
||||
log!(logger, error, "{}", output.stderr.to_string_lossy());
|
||||
}
|
||||
}
|
||||
CommandError {
|
||||
command: self.command,
|
||||
error,
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct CommandOutput {
|
||||
pub stdout: OsString,
|
||||
pub stderr: OsString,
|
||||
pub status: ExitStatus,
|
||||
}
|
||||
|
||||
impl From<Output> for CommandOutput {
|
||||
fn from(value: Output) -> Self {
|
||||
Self {
|
||||
stdout: os_string_from_ssh_output(value.stdout),
|
||||
stderr: os_string_from_ssh_output(value.stderr),
|
||||
status: value.status.into(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn os_string_from_ssh_output(output: Vec<u8>) -> OsString {
|
||||
#[cfg(unix)]
|
||||
{
|
||||
use std::os::unix::ffi::OsStringExt;
|
||||
OsString::from_vec(output)
|
||||
}
|
||||
|
||||
#[cfg(windows)]
|
||||
{
|
||||
use std::os::windows::ffi::OsStringExt;
|
||||
OsString::from_wide(output.iter().map(|&b| b as u16).collect())
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<ExitStatus> for CommandOutput {
|
||||
fn as_ref(&self) -> &ExitStatus {
|
||||
&self.status
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ExitStatus {
|
||||
pub success: bool,
|
||||
pub string_form: String,
|
||||
#[allow(dead_code)]
|
||||
pub code: Option<i32>,
|
||||
}
|
||||
|
||||
impl From<process::ExitStatus> for ExitStatus {
|
||||
fn from(value: process::ExitStatus) -> Self {
|
||||
Self {
|
||||
success: value.success(),
|
||||
string_form: value.to_string(),
|
||||
code: value.code(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<ExitStatus> for ExitStatus {
|
||||
fn as_ref(&self) -> &ExitStatus {
|
||||
self
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for ExitStatus {
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
Display::fmt(&self.string_form, f)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct CommandError<E> {
|
||||
pub command: ShellCommand,
|
||||
pub error: E,
|
||||
}
|
||||
|
||||
impl<E> From<CommandError<E>> for String
|
||||
where
|
||||
E: Display,
|
||||
{
|
||||
fn from(value: CommandError<E>) -> Self {
|
||||
value.to_string()
|
||||
}
|
||||
}
|
||||
|
||||
impl<E> Display for CommandError<E>
|
||||
where
|
||||
E: Display,
|
||||
{
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
write!(
|
||||
f,
|
||||
"Error while running command '{}': {}",
|
||||
self.command, self.error
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
impl<E> Error for CommandError<E> where E: Error {}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct StartError(io::Error);
|
||||
|
||||
impl From<io::Error> for StartError {
|
||||
fn from(value: io::Error) -> Self {
|
||||
StartError(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for StartError {
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
write!(f, "Failed to run command: {}", self.0)
|
||||
}
|
||||
}
|
||||
|
||||
impl Error for StartError {}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum ExecutionError<T> {
|
||||
StartError(StartError),
|
||||
BadExitStatus(T),
|
||||
}
|
||||
|
||||
impl<T> From<StartError> for ExecutionError<T> {
|
||||
fn from(value: StartError) -> Self {
|
||||
Self::StartError(value)
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> Display for ExecutionError<T>
|
||||
where
|
||||
T: AsRef<ExitStatus>,
|
||||
{
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
ExecutionError::StartError(e) => Display::fmt(e, f),
|
||||
ExecutionError::BadExitStatus(status) => {
|
||||
write!(f, "execution failed with {}", status.as_ref())
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> Error for ExecutionError<T> where T: AsRef<ExitStatus> + Debug {}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test_commands {
|
||||
use crate::shell_interface::{ScpParam, ServerCommand, ShellCommand};
|
||||
use std::path::PathBuf;
|
||||
|
||||
#[test]
|
||||
fn test_to_string() {
|
||||
assert_eq!(
|
||||
ShellCommand::Ssh {
|
||||
address: "crea".to_string(),
|
||||
server_command: ServerCommand::Realpath {
|
||||
path: PathBuf::from("plugins/*.jar")
|
||||
}
|
||||
}
|
||||
.to_string(),
|
||||
r#"ssh crea "realpath -e plugins/*.jar""#
|
||||
);
|
||||
assert_eq!(
|
||||
ShellCommand::Ssh {
|
||||
address: "crea".to_string(),
|
||||
server_command: ServerCommand::Ls {
|
||||
dir: PathBuf::from("creative/plugins")
|
||||
}
|
||||
}
|
||||
.to_string(),
|
||||
r#"ssh crea "ls creative/plugins""#
|
||||
);
|
||||
assert_eq!(
|
||||
ShellCommand::Ssh {
|
||||
address: "crea".to_string(),
|
||||
server_command: ServerCommand::Rm {
|
||||
file: PathBuf::from("foo.txt")
|
||||
},
|
||||
}
|
||||
.to_string(),
|
||||
r#"ssh crea "rm foo.txt""#
|
||||
);
|
||||
assert_eq!(
|
||||
ShellCommand::Ssh {
|
||||
address: "crea".to_string(),
|
||||
server_command: ServerCommand::Mv {
|
||||
source: PathBuf::from("foo"),
|
||||
destination: PathBuf::from("bar")
|
||||
}
|
||||
}
|
||||
.to_string(),
|
||||
r#"ssh crea "mv foo bar""#
|
||||
);
|
||||
assert_eq!(
|
||||
ShellCommand::Ssh {
|
||||
address: "crea".to_string(),
|
||||
server_command: ServerCommand::Execute {
|
||||
working_directory: PathBuf::from(".."),
|
||||
command: "sudo rm -rf *".into(),
|
||||
}
|
||||
}
|
||||
.to_string(),
|
||||
r#"ssh crea "cd ..; sudo rm -rf *""#
|
||||
);
|
||||
assert_eq!(
|
||||
ShellCommand::Scp {
|
||||
source: ScpParam {
|
||||
server: None,
|
||||
path: PathBuf::from("target/mssh")
|
||||
},
|
||||
destination: ScpParam {
|
||||
server: Some("crea".into()),
|
||||
path: PathBuf::from("/usr/bin")
|
||||
},
|
||||
}
|
||||
.to_string(),
|
||||
r#"scp target/mssh crea:/usr/bin"#
|
||||
);
|
||||
assert_eq!(ShellCommand::SshAgent.to_string(), r#"ssh-agent -s"#);
|
||||
assert_eq!(ShellCommand::ShhAdd.to_string(), r#"ssh-add"#);
|
||||
assert_eq!(
|
||||
ShellCommand::Editor(vec!["kate".into(), "-b".into(), "test.txt".into()]).to_string(),
|
||||
r#"kate -b test.txt"#
|
||||
);
|
||||
assert_eq!(
|
||||
ShellCommand::Execute {
|
||||
working_directory: PathBuf::from("/home/me/server"),
|
||||
command: vec!["java".into(), "-jar".into(), "paper.jar".into()]
|
||||
}
|
||||
.to_string(),
|
||||
r#"java -jar paper.jar"#
|
||||
);
|
||||
}
|
||||
}
|
||||
@ -1 +0,0 @@
|
||||
exit(1)
|
||||
Loading…
Reference in New Issue
Block a user