Compare commits
512 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 0048d71b74 | |||
| c633347ecc | |||
|
|
ecd6a0ec32 | ||
| 30507d9d01 | |||
| 6245a794d5 | |||
| 5c822e4890 | |||
| cab06fe43f | |||
| b4ff5f3351 | |||
| 0834802481 | |||
| 3afd74dcbf | |||
| b1a0bdfbb6 | |||
| 6d38bc5e1d | |||
| 5ba1ba15c9 | |||
| db05d2abfb | |||
| 1840c4e39a | |||
| c5a3f424d6 | |||
| 04aa6b81b5 | |||
| 5ff3b9b996 | |||
| 228e4a61a4 | |||
| df38ea5413 | |||
| 709f6ca6ca | |||
| b012fc29e4 | |||
| bdad723aef | |||
| f97d46cec3 | |||
| 7381eaef57 | |||
| 72c922b311 | |||
|
|
fd14a8152c | ||
| 5cb60c317d | |||
| 847de75713 | |||
| 58723cf3e8 | |||
| c613168bfb | |||
|
|
6f83de77f2 | ||
|
|
3f6ef3beb4 | ||
| 14e50c1465 | |||
| 0aa9fdca14 | |||
|
|
dc50820abc | ||
|
|
81afce78ad | ||
| 8a0b912601 | |||
| 85105a953e | |||
| 40f3452d08 | |||
| a287a5c903 | |||
| f30e5270d8 | |||
| 104bcd7bb2 | |||
| fd6fcfc98f | |||
| f87e02cd7c | |||
| 9b63b10118 | |||
| 111485e7c4 | |||
|
|
0167753cfe | ||
|
|
73131cc518 | ||
| 25576757bb | |||
| 105c8f3a82 | |||
| 5164d81492 | |||
| 319e5f1ac2 | |||
|
|
b24c2fbeb1 | ||
| bc5053c39c | |||
|
|
f06a031c93 | ||
|
|
8d450dea5a | ||
| c4ace8c53f | |||
| 78f104f558 | |||
| e8a6f740b9 | |||
|
|
6f3c6ec840 | ||
|
|
47a3ef1d8b | ||
|
|
367e9bf33b | ||
|
|
f122b02424 | ||
|
|
3a09c17f0a | ||
|
|
6773abb04e | ||
| b757d66d7a | |||
| 81cb7a750c | |||
| 3be59108a9 | |||
| fac9c45aee | |||
| 184bd2b510 | |||
| fda69178b9 | |||
| 652bbcd5d4 | |||
| fd35106df8 | |||
|
|
5ead5bc3d6 | ||
| 3ce0003315 | |||
|
|
ee94059a15 | ||
| 844742053d | |||
|
|
7ed9cfa018 | ||
|
|
71791afca0 | ||
| a52eddfdac | |||
| 0474978ac0 | |||
| 4b94a0ce2a | |||
| 0d1eac7610 | |||
| b129e5d5a4 | |||
| 9f5c22890d | |||
|
|
74f4a19003 | ||
|
|
717d9872dc | ||
| 439270fe2d | |||
|
|
601fd55435 | ||
|
|
341c5254f1 | ||
| cb2701ada4 | |||
| 28fcccce98 | |||
| 2ba3e56772 | |||
|
|
f3fa3401f1 | ||
|
|
820f339982 | ||
| afd9dd34a7 | |||
| 4a4e5d2cf4 | |||
| 029b00532e | |||
|
|
0b052684c2 | ||
|
|
3dda80b50f | ||
| 9dede1e45d | |||
| 6dc64c6edf | |||
| c21748df1e | |||
| 27d024cca2 | |||
| b0dd2575d9 | |||
| 7a004c2cfc | |||
| 53ab164379 | |||
| bf4fb3a55d | |||
| 2235f1d0d9 | |||
| 17dcb2b10b | |||
| 4c7c62eb0b | |||
| 81dcf4b003 | |||
| 3fffdd20a8 | |||
| 078d75473b | |||
| 8b7ff58c7d | |||
| ff67eebcea | |||
|
|
e585eb46ea | ||
| 01507f88da | |||
| 6344b6dba2 | |||
| d3042cf724 | |||
| fd4b9a4f15 | |||
| 2879a2aa67 | |||
| 5cd8cb66f1 | |||
| 47c206ca1d | |||
| 8f22b64a0a | |||
| 7941b3d16c | |||
| e9ded4bde4 | |||
| 7b02472f67 | |||
| a370d67121 | |||
| 4f6e64083a | |||
| 276a672df4 | |||
| e55a157977 | |||
| e1e91d1add | |||
| 8102b5933f | |||
| bc60128679 | |||
| fc3a7ab789 | |||
| 93d3e6cec7 | |||
| 95779f1ac2 | |||
| 7bc57d7696 | |||
| 7e7b75f378 | |||
| a958350b2d | |||
| c502b3d08f | |||
|
|
e602b66188 | ||
| 12fba15bcf | |||
| 7e36ad4e8a | |||
| 33249f509f | |||
| ed645dd0d5 | |||
| b12c635c27 | |||
| c16ecfb188 | |||
| 18a8b81631 | |||
| 1d404d4d2c | |||
| 42479ced21 | |||
| 1193b8c848 | |||
| ec8d748991 | |||
| bafaf7ca7a | |||
| f7315a3bec | |||
| f655ca989d | |||
| fcb87a6779 | |||
| 924f8d5eff | |||
| 64ecc38073 | |||
| 5f94dbcabe | |||
| 2ecc591966 | |||
| 30ba1f3317 | |||
| 4fdf9b3df1 | |||
| 22fe1a8f73 | |||
| 38c0ad29dd | |||
| 89d106c03e | |||
| 3e36bcf307 | |||
| acf983c07c | |||
| fedb78fb88 | |||
| db64a0968b | |||
| aece20af47 | |||
| 6c5a73f78f | |||
| 906e093152 | |||
| 478b4ae3c0 | |||
| 23971cbb76 | |||
| 43410fac60 | |||
| cb8035a2ce | |||
| 8d071c7674 | |||
| 965c488468 | |||
| ede7f64c4b | |||
| ba38dcdc15 | |||
| 92d9222b05 | |||
| 4c396c3442 | |||
| e1d5139e36 | |||
| 1ad35652f8 | |||
| 9a9b13d604 | |||
| 77b8b61079 | |||
| bdf48d1bf4 | |||
| f8792ea012 | |||
| 4afde8b750 | |||
| f5614995c7 | |||
| 9ea6dbec20 | |||
| d73dfb9fc7 | |||
| a7da73300c | |||
| a308b8fe95 | |||
| 1d1e42aeb1 | |||
| 1f81061152 | |||
| 368d5d3db7 | |||
| 0612a02d68 | |||
| a0d470087b | |||
| 3ecaf04eb4 | |||
| df0811985d | |||
| 057ff0fef1 | |||
| 14c46f88ab | |||
| e38e430c77 | |||
|
|
93cd235aef | ||
|
|
df9bba32cb | ||
|
|
28a8f9b2fa | ||
| e49b366d77 | |||
| 9a0963ca2c | |||
| 17737a06a4 | |||
|
|
464779cc17 | ||
|
|
f25c0889a3 | ||
| 90170cb3d5 | |||
| 4dcb141f3a | |||
| 133721917f | |||
| 766e23d265 | |||
| 77d8e84e14 | |||
| 00c1cca412 | |||
| 4968833d05 | |||
| d172fa17f6 | |||
| 3c99b38db7 | |||
|
|
1128937cac | ||
| 6fc1228173 | |||
| b48a2efb7d | |||
| 412cb2408e | |||
| 682bc91855 | |||
| f03120e5a1 | |||
| 8dd63b30e8 | |||
| 54006c378f | |||
| 9269b66aa8 | |||
| cfac433861 | |||
| a28f8c3dd2 | |||
| 4001dee1bd | |||
| d1ffd0d77f | |||
| 91ad50350d | |||
| a88d43807e | |||
| 98619664cf | |||
| 39f8ad2106 | |||
| 82ce38d7b5 | |||
| 12eb453fc7 | |||
| a84324d3bc | |||
| ed2211586e | |||
| c09950d0af | |||
| e9a30382a3 | |||
| 7bf3311102 | |||
| cbad40245f | |||
| 75c4fcbb9e | |||
| f3b7f155b7 | |||
| 6427a80bd1 | |||
| 5b65e87225 | |||
| 1b8b19fde5 | |||
| 03d7aed258 | |||
| 23d149093f | |||
| 27f12716d9 | |||
| 048877bbb6 | |||
| 87a652d911 | |||
| d6863dc1fd | |||
| f1d934b0a6 | |||
| 5850f7a621 | |||
| dd23e84ccf | |||
| b060518778 | |||
| de95f13feb | |||
| 0205f13e53 | |||
| b4de97dfe2 | |||
| 35bc6cf31c | |||
| c58e8b1a00 | |||
| accdf99503 | |||
| 47b609369b | |||
| 23b1ca4371 | |||
| 73d666d1f5 | |||
| b27c13cf74 | |||
| bd1a4f0939 | |||
| 5abed23cf2 | |||
| 9d0948e124 | |||
| 678bc77a23 | |||
| 00cdeee5c6 | |||
| 2d251554ad | |||
| 1b5d70ae2d | |||
| 2d2901f6dc | |||
| a0b27ec105 | |||
| 093ef136e7 | |||
| 8660de530d | |||
| bda6f253e0 | |||
| 4eb974567f | |||
| 4f5bad5874 | |||
| 1c6e798632 | |||
| 3186fb42e7 | |||
| 4b7185fbb0 | |||
| f0d8555a8a | |||
| f338dfcb12 | |||
| 188d781b0d | |||
| adb1f07fd0 | |||
| 82e51be096 | |||
| d7f6d12f59 | |||
| 0db57fbff1 | |||
| b1bdc19afb | |||
| b75a95a708 | |||
| c3fb5dcd5f | |||
| 21911f93d1 | |||
| f7c96d81e9 | |||
| 9b2040059d | |||
| 08f190fc6e | |||
| 4d1b0fe301 | |||
| f139db07d9 | |||
|
|
73a4129000 | ||
| 1ddf797e28 | |||
|
|
18280f0478 | ||
|
|
4348705a0a | ||
| 5a1b92547d | |||
|
|
9e44713985 | ||
|
|
e93837fef7 | ||
| 6006c9d0e8 | |||
|
|
c93543186a | ||
|
|
e1b74d7a36 | ||
| 6375bc3413 | |||
|
|
9c99e0d2ef | ||
|
|
518ccaadc8 | ||
| a766b395c1 | |||
| 1746869b45 | |||
|
|
533366b90b | ||
| 5c68df7246 | |||
| cc26d9a655 | |||
| 02a1303557 | |||
|
|
817ffcf2b5 | ||
| 4a8b7eb837 | |||
| fcf81ad7d9 | |||
| 89a1b0dec7 | |||
| face51ae0c | |||
| 5ca6757fb0 | |||
| 4f4633c6b3 | |||
| 17c70dfef2 | |||
| 92daaebf9d | |||
| 0c70191687 | |||
|
|
e2a5f2f1b5 | ||
| 1893abd773 | |||
| 41bb08418f | |||
| cf1794145c | |||
| 7d9a25e599 | |||
|
|
cd2175b5ad | ||
| 98ff5184e1 | |||
|
|
da785355d6 | ||
| 4cabd5f0d0 | |||
| a41f03c6b2 | |||
| cde86cf9fd | |||
| 8d4981f10f | |||
| 93d023c54a | |||
| bae450f23e | |||
| 42339e65d4 | |||
| 0bb839c0a0 | |||
|
|
6381d7b742 | ||
| 2efbdad4f0 | |||
| 30bf0c22fa | |||
| 169d6c7364 | |||
| e1fb5f570e | |||
| 92362a5d8c | |||
| 7c88901185 | |||
| 5bce9b240e | |||
| e675168798 | |||
| 8aa88d7343 | |||
|
|
9e6879c0f2 | ||
| eb856e28d7 | |||
| 866b0c7537 | |||
| 3ef5c1911d | |||
| 80787d1187 | |||
| ad0b3989ed | |||
| c7a0e33485 | |||
| ee312a21eb | |||
| 3af22cceac | |||
| c29e2ca9ae | |||
| 06c9baf8df | |||
| 97dc5054e9 | |||
| d43862a3a7 | |||
| 1dd4cd74c3 | |||
| 4f86cce497 | |||
| 3968983002 | |||
| 4c7e8f0cf6 | |||
| 1e3141e4ee | |||
| 45542cd3a9 | |||
| da3bb795b7 | |||
| 53a59cdb4c | |||
| 8125bd5ae0 | |||
| 5ba3f2b1ba | |||
| c98828aec7 | |||
| 5ed278ec9c | |||
| c8a2fea9cd | |||
| cac54c5447 | |||
| 374819b4f3 | |||
| 4d92c350de | |||
| 3be9321df6 | |||
| 746064c430 | |||
| ffc00691cb | |||
| 1b5979c36c | |||
| 5ed3372ae2 | |||
| 8002a5aa1e | |||
| 896c50909a | |||
| cea4632a22 | |||
| 7fdec15ba9 | |||
| eb06787bb2 | |||
| c3577a0724 | |||
| 8864e2c867 | |||
| 581975b941 | |||
| b8e4deb80f | |||
| 40bb22ef7c | |||
| 74e9ea17ac | |||
| a11bce603d | |||
| c754275af3 | |||
| 3497a54c39 | |||
| 8807adea83 | |||
| 6896fcc134 | |||
| 68830a8789 | |||
| 2dce587ea8 | |||
| 9403bdcbcb | |||
| aa13735533 | |||
| 33db3efacf | |||
| 8df74585bc | |||
| 16ca8841a1 | |||
| 22fbe025d9 | |||
| c54bd2bab0 | |||
| 539ad75fe6 | |||
| 9476caa392 | |||
| df3cf70682 | |||
| a881d1f33a | |||
| d96316577a | |||
| eefe6392df | |||
| 1cc95e2cd1 | |||
| c5328917de | |||
| 208acafc73 | |||
| 57eced64c0 | |||
| ce701c1ab7 | |||
| b24e3bf9db | |||
| 1227796e78 | |||
| bb1c08277e | |||
| 16538a3158 | |||
| f4c647342b | |||
| 72cb334b6a | |||
| 1a65a7f3e7 | |||
| 6a0049eb8f | |||
| d2e3750de6 | |||
| 4cdad182ef | |||
| 4ed1e99a15 | |||
| 71870d9396 | |||
| fa4ec709c0 | |||
| d7d223400e | |||
| 34157ef32f | |||
| f5631376af | |||
| df1eea22ab | |||
| 86d93377ac | |||
| 5872a6ba72 | |||
| 6da1ae93ef | |||
| b8c60bf59a | |||
| bd2d2875a5 | |||
| 9d782af020 | |||
| a711c3d16c | |||
| 268cc13d27 | |||
| a8328d3636 | |||
| d82a7f7674 | |||
| 5e63c34a9f | |||
| 540db5993b | |||
| 16bf06426f | |||
| cc02832512 | |||
| 2876913f48 | |||
| 6b64b5ecc4 | |||
| 9ceb55a314 | |||
| 7870bb4b5b | |||
| 4fc2d3c94b | |||
| a012945df2 | |||
| f094cf5ad3 | |||
| d8979221c8 | |||
| aaa4e67f43 | |||
|
|
fff38704ab | ||
| d5e6d64d0f | |||
| 003f319385 | |||
| e14b7072c6 | |||
| 1fe95d057b | |||
| 6dffc90e92 | |||
| 295cd56a1f | |||
| 214c89e8b5 | |||
| 29047c3007 | |||
| a8f3bed402 | |||
| 1ca9265a2a | |||
| 60d61b9e31 | |||
| b6f5b9d08c | |||
| 28f7bc6a4c | |||
| 5b42129f55 | |||
| 4f06b2b947 | |||
| 0f98050a12 | |||
| 14839642dc | |||
| eccc1a2df1 | |||
| 9df929a8e3 | |||
| 1e008f9778 | |||
|
|
fa811da5c2 | ||
| 48ad17c6f1 | |||
| 3cd15f34cd | |||
| 53ca14e64d | |||
| 0d8803d35d | |||
| 8c90221a81 | |||
| a708f71d57 | |||
| 2a13f74a2b | |||
| 2a97c49a8e | |||
| 8c155ce656 | |||
|
|
5245ba6d98 | ||
|
|
f9789ecc9b | ||
| 9936ce1ab5 | |||
| 650c9783a6 | |||
| b253a389eb | |||
| 5023fbd3d1 | |||
| fdb08fbd34 | |||
| b125d3341a | |||
|
|
f73e3a4817 |
@@ -4,6 +4,9 @@ set -e
|
|||||||
|
|
||||||
echo "Running pre-push hook:"
|
echo "Running pre-push hook:"
|
||||||
|
|
||||||
|
echo "Executing: cargo fmt"
|
||||||
|
cargo fmt
|
||||||
|
|
||||||
echo "Executing: make lint"
|
echo "Executing: make lint"
|
||||||
make lint
|
make lint
|
||||||
|
|
||||||
|
|||||||
@@ -4,6 +4,9 @@ set -e
|
|||||||
|
|
||||||
echo "Running pre-push hook:"
|
echo "Running pre-push hook:"
|
||||||
|
|
||||||
|
echo "Executing: cargo fmt --check"
|
||||||
|
cargo fmt --check
|
||||||
|
|
||||||
echo "Executing: make lint"
|
echo "Executing: make lint"
|
||||||
make lint
|
make lint
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,10 @@
|
|||||||
|
[tool.commitizen]
|
||||||
|
name = "cz_conventional_commits"
|
||||||
|
tag_format = "v$version"
|
||||||
|
version_scheme = "semver"
|
||||||
|
version_provider = "cargo"
|
||||||
|
update_changelog_on_bump = true
|
||||||
|
major_version_zero = true
|
||||||
|
|
||||||
|
[tool.commitizen.hooks]
|
||||||
|
pre-commit = "git add Cargo.toml Cargo.lock"
|
||||||
@@ -11,8 +11,6 @@ name: Check
|
|||||||
env:
|
env:
|
||||||
CARGO_TERM_COLOR: always
|
CARGO_TERM_COLOR: always
|
||||||
|
|
||||||
# ensure that the workflow is only triggered once per PR, subsequent pushes to the PR will cancel
|
|
||||||
# and restart the workflow. See https://docs.github.com/en/actions/using-jobs/using-concurrency
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
|
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
|
||||||
cancel-in-progress: true
|
cancel-in-progress: true
|
||||||
@@ -24,14 +22,18 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
- name: Install Rust stable
|
- name: Install Rust stable
|
||||||
uses: dtolnay/rust-toolchain@stable
|
uses: dtolnay/rust-toolchain@stable
|
||||||
with:
|
with:
|
||||||
components: rustfmt
|
components: rustfmt
|
||||||
|
|
||||||
- name: Run cargo fmt
|
- name: Run cargo fmt
|
||||||
run: cargo fmt -- --check
|
run: cargo fmt -- --check
|
||||||
|
|
||||||
- name: Cache Cargo dependencies
|
- name: Cache Cargo dependencies
|
||||||
uses: Swatinem/rust-cache@v2
|
uses: Swatinem/rust-cache@v2
|
||||||
|
|
||||||
clippy:
|
clippy:
|
||||||
name: ${{ matrix.toolchain }} / clippy
|
name: ${{ matrix.toolchain }} / clippy
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
@@ -45,12 +47,15 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
- name: Install Rust stable
|
- name: Install Rust stable
|
||||||
uses: dtolnay/rust-toolchain@stable
|
uses: dtolnay/rust-toolchain@stable
|
||||||
with:
|
with:
|
||||||
components: clippy
|
components: clippy
|
||||||
|
|
||||||
- name: Run clippy action
|
- name: Run clippy action
|
||||||
uses: clechasseur/rs-clippy-check@v3
|
uses: clechasseur/rs-clippy-check@v3
|
||||||
|
|
||||||
- name: Cache Cargo dependencies
|
- name: Cache Cargo dependencies
|
||||||
uses: Swatinem/rust-cache@v2
|
uses: Swatinem/rust-cache@v2
|
||||||
doc:
|
doc:
|
||||||
@@ -61,21 +66,25 @@ jobs:
|
|||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
|
|
||||||
- name: Install Rust nightly
|
- name: Install Rust nightly
|
||||||
uses: dtolnay/rust-toolchain@nightly
|
uses: dtolnay/rust-toolchain@nightly
|
||||||
|
|
||||||
- name: Run cargo doc
|
- name: Run cargo doc
|
||||||
run: cargo doc --no-deps --all-features
|
run: cargo doc --no-deps --all-features
|
||||||
env:
|
env:
|
||||||
RUSTDOCFLAGS: --cfg docsrs
|
RUSTDOCFLAGS: --cfg docsrs
|
||||||
msrv:
|
msrv:
|
||||||
# check that we can build using the minimal rust version that is specified by this crate
|
# check that we can build using the minimal rust version that is specified by this crate
|
||||||
name: 1.82.0 / check
|
name: 1.85.0 / check
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
- name: Install 1.82.0
|
|
||||||
|
- name: Install 1.85.0
|
||||||
uses: dtolnay/rust-toolchain@master
|
uses: dtolnay/rust-toolchain@master
|
||||||
with:
|
with:
|
||||||
toolchain: 1.82.0
|
toolchain: 1.85.0
|
||||||
- name: cargo +1.82.0 check
|
|
||||||
|
- name: cargo +1.85.0 check
|
||||||
run: cargo check
|
run: cargo check
|
||||||
|
|||||||
@@ -1,32 +1,505 @@
|
|||||||
# Adapted from https://github.com/joshka/github-workflows/blob/main/.github/workflows/rust-release-plz.yml
|
name: Create release
|
||||||
# Thanks to joshka for permission to use this template!
|
|
||||||
|
|
||||||
name: Create Release PR and Publish Release
|
|
||||||
|
|
||||||
permissions:
|
permissions:
|
||||||
pull-requests: write
|
pull-requests: write
|
||||||
contents: write
|
contents: write
|
||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
workflow_dispatch:
|
||||||
branches:
|
inputs:
|
||||||
- main
|
bump_type:
|
||||||
|
description: "Specify the type of version bump"
|
||||||
|
required: true
|
||||||
|
default: "patch"
|
||||||
|
type: choice
|
||||||
|
options:
|
||||||
|
- patch
|
||||||
|
- minor
|
||||||
|
- major
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
release-plz:
|
bump-version:
|
||||||
# see https://release-plz.ieni.dev/docs/github
|
name: bump-version
|
||||||
# for more information
|
|
||||||
name: Release-plz
|
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
|
- name: Configure SSH for Git
|
||||||
|
run: |
|
||||||
|
mkdir -p ~/.ssh
|
||||||
|
echo "${{ secrets.RELEASE_BOT_SSH_KEY }}" > ~/.ssh/id_ed25519
|
||||||
|
chmod 600 ~/.ssh/id_ed25519
|
||||||
|
ssh-keyscan -H github.com >> ~/.ssh/known_hosts
|
||||||
|
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v3
|
||||||
|
with:
|
||||||
|
ssh-key: ${{ secrets.RELEASE_BOT_SSH_KEY }}
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Set up Python
|
||||||
|
uses: actions/setup-python@v4
|
||||||
|
with:
|
||||||
|
python-version: "3.10"
|
||||||
|
|
||||||
|
- name: Install Commitizen
|
||||||
|
run: |
|
||||||
|
python -m pip install --upgrade pip
|
||||||
|
pip install commitizen
|
||||||
|
npm install -g conventional-changelog-cli
|
||||||
|
|
||||||
|
- name: Configure Git user
|
||||||
|
run: |
|
||||||
|
git config user.name "github-actions[bot]"
|
||||||
|
git config user.email "github-actions[bot]@users.noreply.github.com"
|
||||||
|
|
||||||
|
- name: Bump version with Commitizen
|
||||||
|
run: |
|
||||||
|
cz bump --yes --increment ${{ github.event.inputs.bump_type }}
|
||||||
|
|
||||||
|
- name: Amend commit message to include '[skip ci]'
|
||||||
|
run: |
|
||||||
|
git commit --amend --no-edit -m "$(git log -1 --pretty=%B) [skip ci]"
|
||||||
|
|
||||||
|
- name: Install Rust stable
|
||||||
|
uses: dtolnay/rust-toolchain@stable
|
||||||
|
|
||||||
|
- name: Update the Cargo.lock
|
||||||
|
run: |
|
||||||
|
cargo update
|
||||||
|
git add Cargo.lock
|
||||||
|
git commit -m "chore: Bump the version in Cargo.lock"
|
||||||
|
|
||||||
|
- name: Get the new version tag
|
||||||
|
id: version
|
||||||
|
run: |
|
||||||
|
mkdir -p artifacts
|
||||||
|
NEW_TAG=$(cz version --project)
|
||||||
|
echo "New version: $NEW_TAG"
|
||||||
|
echo "version=$NEW_TAG" >> $GITHUB_ENV
|
||||||
|
echo "$NEW_TAG" > artifacts/release-version
|
||||||
|
|
||||||
|
- name: Get the previous version tag
|
||||||
|
id: prev_version
|
||||||
|
run: |
|
||||||
|
PREV_TAG=$(git describe --tags --abbrev=0 ${GITHUB_SHA}^)
|
||||||
|
echo "Previous tag: $PREV_TAG"
|
||||||
|
echo "prev_version=$PREV_TAG" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Generate changelog for the version bump
|
||||||
|
id: changelog
|
||||||
|
run: |
|
||||||
|
changelog=$(conventional-changelog -p angular -i CHANGELOG.md -s --from ${{ env.prev_version }} --to ${{ env.version }})
|
||||||
|
echo "$changelog" > artifacts/changelog.md
|
||||||
|
echo "changelog_body=$(cat artifacts/changelog.md)" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Push changes
|
||||||
|
env:
|
||||||
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
run: |
|
||||||
|
git push origin --follow-tags
|
||||||
|
|
||||||
|
- name: Upload artifacts
|
||||||
|
uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
path: artifacts
|
||||||
|
|
||||||
|
build-release-artifacts:
|
||||||
|
name: build-release
|
||||||
|
needs: [bump-version]
|
||||||
|
runs-on: ${{ matrix.job.os }}
|
||||||
|
env:
|
||||||
|
RUST_BACKTRACE: 1
|
||||||
|
strategy:
|
||||||
|
fail-fast: true
|
||||||
|
matrix:
|
||||||
|
# prettier-ignore
|
||||||
|
job:
|
||||||
|
- { name: "macOS-arm64", os: "macOS-latest", target: "aarch64-apple-darwin", artifact_suffix: "macos-arm64", use-cross: true }
|
||||||
|
- { name: "macOS-amd64", os: "macOS-latest", target: "x86_64-apple-darwin", artifact_suffix: "macos" }
|
||||||
|
- { name: "windows-amd64", os: "windows-latest", target: "x86_64-pc-windows-msvc", artifact_suffix: "windows" }
|
||||||
|
- { name: "windows-aarch64", os: "windows-latest", target: "aarch64-pc-windows-msvc", artifact_suffix: "windows-aarch64", use-cross: true }
|
||||||
|
- { name: "linux-gnu", os: "ubuntu-latest", target: "x86_64-unknown-linux-gnu", artifact_suffix: "linux" }
|
||||||
|
- { name: "linux-musl", os: "ubuntu-latest", target: "x86_64-unknown-linux-musl", artifact_suffix: "linux-musl", use-cross: true, }
|
||||||
|
- { name: "linux-aarch64-gnu", os: "ubuntu-latest", target: "aarch64-unknown-linux-gnu", artifact_suffix: "aarch64-gnu", use-cross: true, test-bin: "--bin managarr" }
|
||||||
|
- { name: "linux-aarch64-musl", os: "ubuntu-latest", target: "aarch64-unknown-linux-musl", artifact_suffix: "aarch64-musl", use-cross: true, test-bin: "--bin managarr" }
|
||||||
|
- { name: "linux-arm-gnu", os: "ubuntu-latest", target: "arm-unknown-linux-gnueabi", artifact_suffix: "armv6-gnu", use-cross: true, test-bin: "--bin managarr" }
|
||||||
|
- { name: "linux-arm-musl", os: "ubuntu-latest", target: "arm-unknown-linux-musleabihf", artifact_suffix: "armv6-musl", use-cross: true, test-bin: "--bin managarr" }
|
||||||
|
- { name: "linux-armv7-gnu", os: "ubuntu-latest", target: "armv7-unknown-linux-gnueabihf", artifact_suffix: "armv7-gnu", use-cross: true, test-bin: "--bin managarr" }
|
||||||
|
- { name: "linux-armv7-musl", os: "ubuntu-latest", target: "armv7-unknown-linux-musleabihf", artifact_suffix: "armv7-musl", use-cross: true, test-bin: "--bin managarr" }
|
||||||
|
rust: [stable]
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Check if actor is repository owner
|
||||||
|
if: ${{ github.actor != github.repository_owner }}
|
||||||
|
run: |
|
||||||
|
echo "You are not authorized to run this workflow."
|
||||||
|
exit 1
|
||||||
|
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 1
|
||||||
|
|
||||||
|
- name: Ensure repository is up-to-date
|
||||||
|
run: |
|
||||||
|
git fetch --all
|
||||||
|
git pull
|
||||||
|
|
||||||
|
- uses: actions/cache@v3
|
||||||
|
name: Cache Cargo registry
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/registry
|
||||||
|
key: ${{ runner.os }}-cargo-registry-${{ hashFiles('Cargo.lock') }}
|
||||||
|
|
||||||
|
- uses: actions/cache@v3
|
||||||
|
if: startsWith(matrix.job.name, 'linux-')
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/bin
|
||||||
|
key: ${{ runner.os }}-cargo-bin-${{ hashFiles('.github/workflows/release.yml') }}
|
||||||
|
|
||||||
|
- uses: dtolnay/rust-toolchain@stable
|
||||||
|
name: Set Rust toolchain
|
||||||
|
with:
|
||||||
|
targets: ${{ matrix.job.target }}
|
||||||
|
|
||||||
|
- uses: taiki-e/setup-cross-toolchain-action@v1
|
||||||
|
with:
|
||||||
|
# NB: sets CARGO_BUILD_TARGET evar - do not need --target flag in build
|
||||||
|
target: ${{ matrix.job.target }}
|
||||||
|
|
||||||
|
- uses: taiki-e/install-action@cross
|
||||||
|
if: ${{ matrix.job.use-cross }}
|
||||||
|
|
||||||
|
- name: Installing needed Ubuntu dependencies
|
||||||
|
if: matrix.job.os == 'ubuntu-latest'
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
sudo apt-get -y update
|
||||||
|
case ${{ matrix.job.target }} in
|
||||||
|
arm*-linux-*) sudo apt-get -y install gcc-arm-linux-gnueabihf ;;
|
||||||
|
aarch64-*-linux-*) sudo apt-get -y install gcc-aarch64-linux-gnu ;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
- name: Build
|
||||||
|
run: cargo build --release --verbose --target=${{ matrix.job.target }} --locked
|
||||||
|
|
||||||
|
- name: Verify file
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
file target/${{ matrix.job.target }}/release/managarr
|
||||||
|
|
||||||
|
- name: Test
|
||||||
|
if: matrix.job.target != 'aarch64-apple-darwin' && matrix.job.target != 'aarch64-pc-windows-msvc'
|
||||||
|
run: cargo test --release --verbose --target=${{ matrix.job.target }} ${{ matrix.job.test-bin }}
|
||||||
|
|
||||||
|
- name: Packaging final binary (Windows)
|
||||||
|
if: matrix.job.os == 'windows-latest'
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
cd target/${{ matrix.job.target }}/release
|
||||||
|
BINARY_NAME=managarr.exe
|
||||||
|
if [ "${{ matrix.job.target }}" != "aarch64-pc-windows-msvc" ]; then
|
||||||
|
# strip the binary
|
||||||
|
strip $BINARY_NAME
|
||||||
|
fi
|
||||||
|
RELEASE_NAME=managarr-${{ matrix.job.artifact_suffix }}
|
||||||
|
mkdir -p artifacts
|
||||||
|
tar czvf $RELEASE_NAME.tar.gz $BINARY_NAME
|
||||||
|
# create sha checksum files
|
||||||
|
certutil -hashfile $RELEASE_NAME.tar.gz sha256 | grep -E [A-Fa-f0-9]{64} > $RELEASE_NAME.sha256
|
||||||
|
echo "RELEASE_NAME=$RELEASE_NAME" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Packaging final binary (macOS and Linux)
|
||||||
|
if: matrix.job.os != 'windows-latest'
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
# set the right strip executable
|
||||||
|
STRIP="strip";
|
||||||
|
case ${{ matrix.job.target }} in
|
||||||
|
arm*-linux-*) STRIP="arm-linux-gnueabihf-strip" ;;
|
||||||
|
aarch64-*-linux-*) STRIP="aarch64-linux-gnu-strip" ;;
|
||||||
|
esac;
|
||||||
|
cd target/${{ matrix.job.target }}/release
|
||||||
|
BINARY_NAME=managarr
|
||||||
|
# strip the binary
|
||||||
|
"$STRIP" "$BINARY_NAME"
|
||||||
|
RELEASE_NAME=managarr-${{ matrix.job.artifact_suffix }}
|
||||||
|
tar czvf $RELEASE_NAME.tar.gz $BINARY_NAME
|
||||||
|
# create sha checksum files
|
||||||
|
shasum -a 256 $RELEASE_NAME.tar.gz > $RELEASE_NAME.sha256
|
||||||
|
echo "RELEASE_NAME=$RELEASE_NAME" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Add artifacts
|
||||||
|
run: |
|
||||||
|
mkdir -p artifacts
|
||||||
|
cp target/${{ matrix.job.target }}/release/${{ env.RELEASE_NAME }}.tar.gz artifacts/
|
||||||
|
cp target/${{ matrix.job.target }}/release/${{ env.RELEASE_NAME }}.sha256 artifacts/
|
||||||
|
|
||||||
|
- name: Upload artifacts
|
||||||
|
uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
name: artifacts-${{ env.RELEASE_NAME }}
|
||||||
|
path: artifacts
|
||||||
|
overwrite: true
|
||||||
|
|
||||||
|
publish-github-release:
|
||||||
|
name: publish-github-release
|
||||||
|
needs: [build-release-artifacts]
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v3
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Download all artifacts
|
||||||
|
uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
path: artifacts
|
||||||
|
merge-multiple: true
|
||||||
|
|
||||||
|
- name: Ensure repository is up-to-date
|
||||||
|
run: |
|
||||||
|
git fetch --all
|
||||||
|
git pull
|
||||||
|
|
||||||
|
- name: Set environment variables
|
||||||
|
run: |
|
||||||
|
release_version="$(cat ./artifacts/release-version)"
|
||||||
|
echo "RELEASE_VERSION=$release_version" >> $GITHUB_ENV
|
||||||
|
changelog_body="$(cat ./artifacts/changelog.md)"
|
||||||
|
echo "changelog_body=$(cat artifacts/changelog.md)" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Create a GitHub Release
|
||||||
|
uses: softprops/action-gh-release@v1
|
||||||
|
env:
|
||||||
|
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
with:
|
||||||
|
files: |
|
||||||
|
artifacts/managarr-macos-arm64.tar.gz
|
||||||
|
artifacts/managarr-macos-arm64.sha256
|
||||||
|
artifacts/managarr-macos.tar.gz
|
||||||
|
artifacts/managarr-macos.sha256
|
||||||
|
artifacts/managarr-windows.tar.gz
|
||||||
|
artifacts/managarr-windows.sha256
|
||||||
|
artifacts/managarr-windows-aarch64.tar.gz
|
||||||
|
artifacts/managarr-windows-aarch64.sha256
|
||||||
|
artifacts/managarr-linux.tar.gz
|
||||||
|
artifacts/managarr-linux.sha256
|
||||||
|
artifacts/managarr-linux-musl.tar.gz
|
||||||
|
artifacts/managarr-linux-musl.sha256
|
||||||
|
artifacts/managarr-aarch64-gnu.tar.gz
|
||||||
|
artifacts/managarr-aarch64-gnu.sha256
|
||||||
|
artifacts/managarr-aarch64-musl.tar.gz
|
||||||
|
artifacts/managarr-aarch64-musl.sha256
|
||||||
|
artifacts/managarr-armv6-gnu.tar.gz
|
||||||
|
artifacts/managarr-armv6-gnu.sha256
|
||||||
|
artifacts/managarr-armv6-musl.tar.gz
|
||||||
|
artifacts/managarr-armv6-musl.sha256
|
||||||
|
artifacts/managarr-armv7-gnu.tar.gz
|
||||||
|
artifacts/managarr-armv7-gnu.sha256
|
||||||
|
artifacts/managarr-armv7-musl.tar.gz
|
||||||
|
artifacts/managarr-armv7-musl.sha256
|
||||||
|
tag_name: v${{ env.RELEASE_VERSION }}
|
||||||
|
name: "v${{ env.RELEASE_VERSION }}"
|
||||||
|
body: ${{ env.changelog_body }}
|
||||||
|
draft: false
|
||||||
|
prerelease: false
|
||||||
|
|
||||||
|
- name: Upload artifacts
|
||||||
|
uses: actions/upload-artifact@v4
|
||||||
|
with:
|
||||||
|
path: artifacts
|
||||||
|
overwrite: true
|
||||||
|
|
||||||
|
publish-chocolatey-package:
|
||||||
|
needs: [publish-github-release]
|
||||||
|
name: Publish Chocolatey Package
|
||||||
|
runs-on: windows-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 1
|
||||||
|
|
||||||
|
- name: Get release artifacts
|
||||||
|
uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
path: artifacts
|
||||||
|
merge-multiple: true
|
||||||
|
|
||||||
|
- name: Set release assets and version
|
||||||
|
shell: pwsh
|
||||||
|
run: |
|
||||||
|
# Read the first column from the SHA256 file
|
||||||
|
$windows_sha = Get-Content ./artifacts/managarr-windows.sha256 | ForEach-Object { $_.Split(' ')[0] }
|
||||||
|
Add-Content -Path $env:GITHUB_ENV -Value "WINDOWS_SHA=$windows_sha"
|
||||||
|
|
||||||
|
# Read the release version from the release-version file
|
||||||
|
$release_version = Get-Content ./artifacts/release-version
|
||||||
|
Add-Content -Path $env:GITHUB_ENV -Value "RELEASE_VERSION=$release_version"
|
||||||
|
|
||||||
|
- name: Validate release environment variables
|
||||||
|
run: |
|
||||||
|
echo "Release SHA windows: ${{ env.WINDOWS_SHA }}"
|
||||||
|
echo "Release version: ${{ env.RELEASE_VERSION }}"
|
||||||
|
|
||||||
|
- name: Package and Publish package to Chocolatey
|
||||||
|
run: |
|
||||||
|
mkdir ./deployment/chocolatey/tools
|
||||||
|
# Run packaging script
|
||||||
|
python "./deployment/chocolatey/packager.py" ${{ env.RELEASE_VERSION }} "./deployment/chocolatey/managarr.nuspec.template" "./deployment/chocolatey/managarr.nuspec" ${{ env.WINDOWS_SHA }}
|
||||||
|
python "./deployment/chocolatey/packager.py" ${{ env.RELEASE_VERSION }} "./deployment/chocolatey/chocolateyinstall.ps1.template" "./deployment/chocolatey/tools/chocolateyinstall.ps1" ${{ env.WINDOWS_SHA }}
|
||||||
|
|
||||||
|
# Publish to Chocolatey
|
||||||
|
cd ./deployment/chocolatey
|
||||||
|
choco pack
|
||||||
|
echo y | choco install managarr -dv -s .
|
||||||
|
$version = managarr --version
|
||||||
|
$version = $version -replace " ", "."
|
||||||
|
choco push $version.nupkg -s https://push.chocolatey.org/ --api-key ${{ secrets.CHOCOLATEY_API_KEY }};
|
||||||
|
|
||||||
|
publish-homebrew-formula:
|
||||||
|
needs: [publish-github-release]
|
||||||
|
name: Update Homebrew formulas
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 1
|
||||||
|
|
||||||
|
- name: Get release artifacts
|
||||||
|
uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
path: artifacts
|
||||||
|
merge-multiple: true
|
||||||
|
|
||||||
|
- name: Set release assets and version
|
||||||
|
shell: bash
|
||||||
|
run: |
|
||||||
|
# Set environment variables
|
||||||
|
macos_sha="$(cat ./artifacts/managarr-macos.sha256 | awk '{print $1}')"
|
||||||
|
echo "MACOS_SHA=$macos_sha" >> $GITHUB_ENV
|
||||||
|
macos_sha_arm="$(cat ./artifacts/managarr-macos-arm64.sha256 | awk '{print $1}')"
|
||||||
|
echo "MACOS_SHA_ARM=$macos_sha_arm" >> $GITHUB_ENV
|
||||||
|
linux_sha="$(cat ./artifacts/managarr-linux-musl.sha256 | awk '{print $1}')"
|
||||||
|
echo "LINUX_SHA=$linux_sha" >> $GITHUB_ENV
|
||||||
|
release_version="$(cat ./artifacts/release-version)"
|
||||||
|
echo "RELEASE_VERSION=$release_version" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Validate release environment variables
|
||||||
|
run: |
|
||||||
|
echo "Release SHA macos: ${{ env.MACOS_SHA }}"
|
||||||
|
echo "Release SHA macos-arm: ${{ env.MACOS_SHA_ARM }}"
|
||||||
|
echo "Release SHA linux musl: ${{ env.LINUX_SHA }}"
|
||||||
|
echo "Release version: ${{ env.RELEASE_VERSION }}"
|
||||||
|
|
||||||
|
- name: Execute Homebrew packaging script
|
||||||
|
run: |
|
||||||
|
# run packaging script
|
||||||
|
python "./deployment/homebrew/packager.py" ${{ env.RELEASE_VERSION }} "./deployment/homebrew/managarr.rb.template" "./managarr.rb" ${{ env.MACOS_SHA }} ${{ env.MACOS_SHA_ARM }} ${{ env.LINUX_SHA }}
|
||||||
|
|
||||||
|
- name: Push changes to Homebrew tap
|
||||||
|
env:
|
||||||
|
TOKEN: ${{ secrets.MANAGARR_GITHUB_TOKEN }}
|
||||||
|
run: |
|
||||||
|
# push to Git
|
||||||
|
git config --global user.name "Dark-Alex-17"
|
||||||
|
git config --global user.email "alex.j.tusa@gmail.com"
|
||||||
|
git clone https://Dark-Alex-17:${{ secrets.MANAGARR_GITHUB_TOKEN }}@github.com/Dark-Alex-17/homebrew-managarr.git
|
||||||
|
rm homebrew-managarr/Formula/managarr.rb
|
||||||
|
cp managarr.rb homebrew-managarr/Formula
|
||||||
|
cd homebrew-managarr
|
||||||
|
git add .
|
||||||
|
git diff-index --quiet HEAD || git commit -am "Update formula for Managarr release ${{ env.RELEASE_VERSION }}"
|
||||||
|
git push https://$TOKEN@github.com/Dark-Alex-17/homebrew-managarr.git
|
||||||
|
|
||||||
|
publish-docker-image:
|
||||||
|
needs: [publish-github-release]
|
||||||
|
name: Publishing Docker image to Docker Hub
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Checkout repository
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 1
|
||||||
|
|
||||||
|
- name: Get release artifacts
|
||||||
|
uses: actions/download-artifact@v4
|
||||||
|
with:
|
||||||
|
path: artifacts
|
||||||
|
merge-multiple: true
|
||||||
|
|
||||||
|
- name: Ensure repository is up-to-date
|
||||||
|
run: |
|
||||||
|
git fetch --all
|
||||||
|
git pull
|
||||||
|
|
||||||
|
- name: Set version variable
|
||||||
|
run: |
|
||||||
|
version="$(cat artifacts/release-version)"
|
||||||
|
echo "version=$version" >> $GITHUB_ENV
|
||||||
|
|
||||||
|
- name: Validate release environment variables
|
||||||
|
run: |
|
||||||
|
echo "Release version: ${{ env.version }}"
|
||||||
|
|
||||||
|
- name: Set up QEMU
|
||||||
|
uses: docker/setup-qemu-action@v3
|
||||||
|
|
||||||
|
- name: Set up Docker Buildx
|
||||||
|
uses: docker/setup-buildx-action@v3
|
||||||
|
|
||||||
|
- name: Login to Docker Hub
|
||||||
|
uses: docker/login-action@v3
|
||||||
|
with:
|
||||||
|
username: ${{ secrets.DOCKER_USERNAME }}
|
||||||
|
password: ${{ secrets.DOCKER_PASSWORD }}
|
||||||
|
|
||||||
|
- name: Push to Docker Hub
|
||||||
|
uses: docker/build-push-action@v5
|
||||||
|
with:
|
||||||
|
context: .
|
||||||
|
file: Dockerfile
|
||||||
|
platforms: linux/amd64,linux/arm64
|
||||||
|
push: true
|
||||||
|
tags: darkalex17/managarr:latest, darkalex17/managarr:${{ env.version }}
|
||||||
|
|
||||||
|
publish-crate:
|
||||||
|
needs: publish-github-release
|
||||||
|
name: Publish Crate
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- name: Check if actor is repository owner
|
||||||
|
if: ${{ github.actor != github.repository_owner }}
|
||||||
|
run: |
|
||||||
|
echo "You are not authorized to run this workflow."
|
||||||
|
exit 1
|
||||||
|
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
fetch-depth: 0
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Ensure repository is up-to-date
|
||||||
|
run: |
|
||||||
|
git fetch --all
|
||||||
|
git pull
|
||||||
|
|
||||||
|
- uses: actions/cache@v3
|
||||||
|
name: Cache Cargo registry
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/registry
|
||||||
|
key: ${{ runner.os }}-cargo-registry-${{ hashFiles('Cargo.lock') }}
|
||||||
|
|
||||||
|
- uses: actions/cache@v3
|
||||||
|
with:
|
||||||
|
path: ~/.cargo/bin
|
||||||
|
key: ${{ runner.os }}-cargo-bin-${{ hashFiles('.github/workflows/release.yml') }}
|
||||||
|
|
||||||
- name: Install Rust stable
|
- name: Install Rust stable
|
||||||
uses: dtolnay/rust-toolchain@stable
|
uses: dtolnay/rust-toolchain@stable
|
||||||
- name: Run release-plz
|
|
||||||
uses: MarcoIeni/release-plz-action@v0.5
|
- uses: katyo/publish-crates@v2
|
||||||
env:
|
with:
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
registry-token: ${{ secrets.CARGO_REGISTRY_TOKEN }}
|
||||||
CARGO_REGISTRY_TOKEN: ${{ secrets.CARGO_REGISTRY_TOKEN }}
|
|
||||||
|
|||||||
@@ -34,55 +34,66 @@ jobs:
|
|||||||
toolchain: [stable, beta]
|
toolchain: [stable, beta]
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
|
|
||||||
- name: Install ${{ matrix.toolchain }}
|
- name: Install ${{ matrix.toolchain }}
|
||||||
uses: dtolnay/rust-toolchain@master
|
uses: dtolnay/rust-toolchain@master
|
||||||
with:
|
with:
|
||||||
toolchain: ${{ matrix.toolchain }}
|
toolchain: ${{ matrix.toolchain }}
|
||||||
|
|
||||||
# enable this ci template to run regardless of whether the lockfile is checked in or not
|
# enable this ci template to run regardless of whether the lockfile is checked in or not
|
||||||
- name: cargo generate-lockfile
|
- name: cargo generate-lockfile
|
||||||
if: hashFiles('Cargo.lock') == ''
|
if: hashFiles('Cargo.lock') == ''
|
||||||
run: cargo generate-lockfile
|
run: cargo generate-lockfile
|
||||||
|
|
||||||
- name: cargo test --locked
|
- name: cargo test --locked
|
||||||
run: cargo test --locked --all-features --all-targets
|
run: cargo test --locked --all-features --all-targets
|
||||||
minimal-versions:
|
|
||||||
# This action chooses the oldest version of the dependencies permitted by Cargo.toml to ensure
|
# minimal-versions:
|
||||||
# that this crate is compatible with the minimal version that this crate and its dependencies
|
# # This action chooses the oldest version of the dependencies permitted by Cargo.toml to ensure
|
||||||
# require. This will pickup issues where this create relies on functionality that was introduced
|
# # that this crate is compatible with the minimal version that this crate and its dependencies
|
||||||
# later than the actual version specified (e.g., when we choose just a major version, but a
|
# # require. This will pickup issues where this create relies on functionality that was introduced
|
||||||
# method was added after this version).
|
# # later than the actual version specified (e.g., when we choose just a major version, but a
|
||||||
#
|
# # method was added after this version).
|
||||||
# This particular check can be difficult to get to succeed as often transitive dependencies may
|
# #
|
||||||
# be incorrectly specified (e.g., a dependency specifies 1.0 but really requires 1.1.5). There
|
# # This particular check can be difficult to get to succeed as often transitive dependencies may
|
||||||
# is an alternative flag available -Zdirect-minimal-versions that uses the minimal versions for
|
# # be incorrectly specified (e.g., a dependency specifies 1.0 but really requires 1.1.5). There
|
||||||
# direct dependencies of this crate, while selecting the maximal versions for the transitive
|
# # is an alternative flag available -Zdirect-minimal-versions that uses the minimal versions for
|
||||||
# dependencies. Alternatively, you can add a line in your Cargo.toml to artificially increase
|
# # direct dependencies of this crate, while selecting the maximal versions for the transitive
|
||||||
# the minimal dependency, which you do with e.g.:
|
# # dependencies. Alternatively, you can add a line in your Cargo.toml to artificially increase
|
||||||
# ```toml
|
# # the minimal dependency, which you do with e.g.:
|
||||||
# # for minimal-versions
|
# # ```toml
|
||||||
# [target.'cfg(any())'.dependencies]
|
# # # for minimal-versions
|
||||||
# openssl = { version = "0.10.55", optional = true } # needed to allow foo to build with -Zminimal-versions
|
# # [target.'cfg(any())'.dependencies]
|
||||||
# ```
|
# # openssl = { version = "0.10.55", optional = true } # needed to allow foo to build with -Zminimal-versions
|
||||||
# The optional = true is necessary in case that dependency isn't otherwise transitively required
|
# # ```
|
||||||
# by your library, and the target bit is so that this dependency edge never actually affects
|
# # The optional = true is necessary in case that dependency isn't otherwise transitively required
|
||||||
# Cargo build order. See also
|
# # by your library, and the target bit is so that this dependency edge never actually affects
|
||||||
# https://github.com/jonhoo/fantoccini/blob/fde336472b712bc7ebf5b4e772023a7ba71b2262/Cargo.toml#L47-L49.
|
# # Cargo build order. See also
|
||||||
# This action is run on ubuntu with the stable toolchain, as it is not expected to fail
|
# # https://github.com/jonhoo/fantoccini/blob/fde336472b712bc7ebf5b4e772023a7ba71b2262/Cargo.toml#L47-L49.
|
||||||
runs-on: ubuntu-latest
|
# # This action is run on ubuntu with the stable toolchain, as it is not expected to fail
|
||||||
name: ubuntu / stable / minimal-versions
|
# runs-on: ubuntu-latest
|
||||||
steps:
|
# name: ubuntu / stable / minimal-versions
|
||||||
- uses: actions/checkout@v4
|
# steps:
|
||||||
- name: Install Rust stable
|
# - uses: actions/checkout@v4
|
||||||
uses: dtolnay/rust-toolchain@stable
|
|
||||||
- name: Install nightly for -Zdirect-minimal-versions
|
# - name: Install Rust stable
|
||||||
uses: dtolnay/rust-toolchain@nightly
|
# uses: dtolnay/rust-toolchain@stable
|
||||||
- name: rustup default stable
|
|
||||||
run: rustup default stable
|
# - name: Install nightly for -Zdirect-minimal-versions
|
||||||
- name: cargo update -Zdirect-minimal-versions
|
# uses: dtolnay/rust-toolchain@nightly
|
||||||
run: cargo +nightly update -Zdirect-minimal-versions
|
|
||||||
- name: cargo test
|
# - name: rustup default stable
|
||||||
run: cargo test --locked --all-features --all-targets
|
# run: rustup default stable
|
||||||
- name: Cache Cargo dependencies
|
|
||||||
uses: Swatinem/rust-cache@v2
|
# - name: cargo update -Zdirect-minimal-versions
|
||||||
|
# run: cargo +nightly update -Zdirect-minimal-versions
|
||||||
|
|
||||||
|
# - name: cargo test
|
||||||
|
# run: cargo test --locked --all-features --all-targets
|
||||||
|
|
||||||
|
# - name: Cache Cargo dependencies
|
||||||
|
# uses: Swatinem/rust-cache@v2
|
||||||
|
|
||||||
os-check:
|
os-check:
|
||||||
# run cargo test on mac and windows
|
# run cargo test on mac and windows
|
||||||
runs-on: ${{ matrix.os }}
|
runs-on: ${{ matrix.os }}
|
||||||
@@ -100,15 +111,20 @@ jobs:
|
|||||||
# if: runner.os == 'Windows'
|
# if: runner.os == 'Windows'
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
- name: Install Rust stable
|
- name: Install Rust stable
|
||||||
uses: dtolnay/rust-toolchain@stable
|
uses: dtolnay/rust-toolchain@stable
|
||||||
|
|
||||||
- name: cargo generate-lockfile
|
- name: cargo generate-lockfile
|
||||||
if: hashFiles('Cargo.lock') == ''
|
if: hashFiles('Cargo.lock') == ''
|
||||||
run: cargo generate-lockfile
|
run: cargo generate-lockfile
|
||||||
|
|
||||||
- name: cargo test
|
- name: cargo test
|
||||||
run: cargo test --locked --all-features --all-targets
|
run: cargo test --locked --all-features --all-targets
|
||||||
|
|
||||||
- name: Cache Cargo dependencies
|
- name: Cache Cargo dependencies
|
||||||
uses: Swatinem/rust-cache@v2
|
uses: Swatinem/rust-cache@v2
|
||||||
|
|
||||||
coverage:
|
coverage:
|
||||||
# use llvm-cov to build and collect coverage and outputs in a format that
|
# use llvm-cov to build and collect coverage and outputs in a format that
|
||||||
# is compatible with codecov.io
|
# is compatible with codecov.io
|
||||||
@@ -136,21 +152,28 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
- name: Install Rust stable
|
- name: Install Rust stable
|
||||||
uses: dtolnay/rust-toolchain@stable
|
uses: dtolnay/rust-toolchain@stable
|
||||||
with:
|
with:
|
||||||
components: llvm-tools-preview
|
components: llvm-tools-preview
|
||||||
|
|
||||||
- name: cargo install cargo-llvm-cov
|
- name: cargo install cargo-llvm-cov
|
||||||
uses: taiki-e/install-action@cargo-llvm-cov
|
uses: taiki-e/install-action@cargo-llvm-cov
|
||||||
|
|
||||||
- name: cargo generate-lockfile
|
- name: cargo generate-lockfile
|
||||||
if: hashFiles('Cargo.lock') == ''
|
if: hashFiles('Cargo.lock') == ''
|
||||||
run: cargo generate-lockfile
|
run: cargo generate-lockfile
|
||||||
|
|
||||||
- name: cargo llvm-cov
|
- name: cargo llvm-cov
|
||||||
run: cargo llvm-cov --locked --all-features --lcov --output-path lcov.info
|
run: cargo llvm-cov --locked --all-features --lcov --output-path lcov.info
|
||||||
|
|
||||||
- name: Record Rust version
|
- name: Record Rust version
|
||||||
run: echo "RUST=$(rustc --version)" >> "$GITHUB_ENV"
|
run: echo "RUST=$(rustc --version)" >> "$GITHUB_ENV"
|
||||||
|
|
||||||
- name: Cache Cargo dependencies
|
- name: Cache Cargo dependencies
|
||||||
uses: Swatinem/rust-cache@v2
|
uses: Swatinem/rust-cache@v2
|
||||||
|
|
||||||
- name: Upload to codecov.io
|
- name: Upload to codecov.io
|
||||||
uses: codecov/codecov-action@v4
|
uses: codecov/codecov-action@v4
|
||||||
with:
|
with:
|
||||||
|
|||||||
@@ -0,0 +1,8 @@
|
|||||||
|
repos:
|
||||||
|
- hooks:
|
||||||
|
- id: commitizen
|
||||||
|
- id: commitizen-branch
|
||||||
|
stages:
|
||||||
|
- pre-push
|
||||||
|
repo: https://github.com/commitizen-tools/commitizen
|
||||||
|
rev: v3.30.0
|
||||||
@@ -5,7 +5,376 @@ All notable changes to this project will be documented in this file.
|
|||||||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
||||||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
||||||
|
|
||||||
## [Unreleased]
|
## v0.5.1 (2025-03-01)
|
||||||
|
|
||||||
|
### Feat
|
||||||
|
|
||||||
|
- CLI Support for multiple Servarr instances
|
||||||
|
- Support for multiple servarr definitions - no tests [skip ci]
|
||||||
|
- Support for loading Servarr API tokens from a file
|
||||||
|
- Tweaked the implementation for environment variables in the config a bit
|
||||||
|
- var interpolation
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- Updated openssl to 0.10.70 to mitigate CVE-2025-24898
|
||||||
|
- Addressed rustfmt complaints
|
||||||
|
- Corrected typo in the managarr.nuspec.template
|
||||||
|
|
||||||
|
### Refactor
|
||||||
|
|
||||||
|
- Updated dependencies
|
||||||
|
- Addressed Cargo fmt complaints
|
||||||
|
- Added a debug line for logging to output the config used when starting Managarr
|
||||||
|
- Updated the 2018 idiom lint to the 2021_compatibility lint
|
||||||
|
- Removed unnecessary clones in the networking module to speed up network request handling
|
||||||
|
- Corrected some clone instead of copy behaviors in the command line handlers
|
||||||
|
- Removed unnecessary clone from stateful table
|
||||||
|
- Removed unnecessary clone call from extract_and_add_tag_ids_vec method
|
||||||
|
- Reduced the number of clones necessary when building modal structs
|
||||||
|
- Refactored a handful of Option calls to use take instead
|
||||||
|
- Renamed KeyEventHandler::with to KeyEventHandler::new to keep with Rust best practices and conventions
|
||||||
|
|
||||||
|
## v0.4.2 (2024-12-21)
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- Revert failed release [skip ci]
|
||||||
|
- **sonarr**: Pass the series ID alongside all UpdateAndScan events when publishing to the networking channel
|
||||||
|
- **sonarr**: pass the series ID alongside all TriggerAutomaticSeriesSearch events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the series ID and season number alongside all TriggerAutomaticSeasonSearch events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the episode ID alongside all TriggerAutomaticEpisodeSearch events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the episode ID alongside all ToggleEpisodeMonitoring events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the series ID and season number alongside all toggle season monitoring events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the indexer ID directly alongside all TestIndexer events when publishing to the networking channel
|
||||||
|
- **sonarr**: Provide the task name directly alongside all StartTask events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the search query directly to the networking channel when searching for a new series
|
||||||
|
- **sonarr**: Pass the series ID alongside all GetSeriesHistory events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the series ID alongside all GetSeriesDetails events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass series ID and season number alongside all ManualSeasonSearch events when publishing to the networking channel
|
||||||
|
- **sonarr**: Provide the series ID and season number alongside all GetSeasonHistory events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the episode ID alongside all ManualEpisodeSearch events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass events alongside all GetLogs events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the episode ID alongside all GetEpisodeHistory events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass series ID alongside all GetEpisodeFiles events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass series ID alognside all GetEpisodes events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the episode ID alongside all GetEpisodeDetails events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass history events alongside all GetHistory events when publishing to the networking channel
|
||||||
|
- **sonarr**: Construct and pass edit series parameters alongside all EditSeries events when publishing to the networking channel
|
||||||
|
- **sonarr**: Construct and pass edit indexer parameters alongside all EditIndexer events when publishing to the networking channel
|
||||||
|
- **sonarr**: Construct and pass edit all indexer settings alongside all EditAllIndexerSettings events when publishing to the networking channel
|
||||||
|
- **sonarr**: Construct and pass delete series params alongside all DeleteSeries events when publishing to the networking channel
|
||||||
|
- **sonarr**: Corrected a bug that would cause a crash if a user spams the ESC key while searching for a new series and the search results are still loading
|
||||||
|
- **sonarr**: Pass the root folder ID alongside all DeleteRootFolder events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the indexer ID alongside all DeleteIndexer events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the episode file ID alongside all DeleteEpisodeFile events when publishing to the networking channel
|
||||||
|
- **sonarr**: Pass the download ID alongside all DeleteDownload events published to the networking channel
|
||||||
|
- **sonarr**: Pass the blocklist item ID alongside the DeleteBlocklistItem event when publishing to the networking channel
|
||||||
|
- **sonarr**: Construct and pass the add series body alongside AddSeries events when publishing to the networking channel
|
||||||
|
- **sonarr**: Construct and pass the AddRootFolderBody alongside all AddRootFolder events when publishing to the networking channel
|
||||||
|
- **radarr**: Pass the movie ID alongside all UpdateAndScan events published to the networking channel
|
||||||
|
- **radarr**: Provide the movie ID alongside all TriggerAutomaticMovieSearch events when publishing to the networking channel
|
||||||
|
- **radarr**: Pass in the indexer id with all TestIndexer events when publishing to the networking channel
|
||||||
|
- **radarr**: Pass in the task name alongside the StartTask event when publishing to the networking channel
|
||||||
|
- **radarr**: Pass in the search query for the SearchNewMovie event when publishing to the networking channel
|
||||||
|
- **radarr**: Pass in the movie ID alongside the GetReleases event when publishing to the networking channel
|
||||||
|
- **radarr**: Pass in the movie ID alongside the GetMovieHistory event when publishing to the networking channel
|
||||||
|
- **radarr**: Pass the movie ID in alongside the GetMovieDetaisl event when publishing to the networking channel
|
||||||
|
- **radarr**: Provide the movie id alongside the GetMovieCredits event when publishing to the networking channel
|
||||||
|
- **radarr**: Pass the number of log events to fetch in with the GetLogs event when publishing to the networking channel
|
||||||
|
- **radarr**: Construct and pass the edit movie parameters alongside the EditMovie event when publishing to the networking channel
|
||||||
|
- **radarr**: Construct and pass params when publishing the EditIndexer event to the networking channel
|
||||||
|
- **radarr**: Construct and pass edit collection parameters alongside the EditCollection event when publishing to the networking channel
|
||||||
|
- **radarr**: Build and pass the edit indexer settings body with the EditAllIndexerSettings event when publishing to the networking channel
|
||||||
|
- **radarr**: Send the parameters alongside the DownloadRelease event when publishing to the networking channel
|
||||||
|
- **radarr**: Pass the root folder ID in with the DeleteRootFolder event when publishing to the networking channel
|
||||||
|
- Pass the delete movie params in with the DeleteMovie event when publishing to the networking channel
|
||||||
|
- Pass the indexer ID in with the DeleteIndexer event when sending to the networking channel
|
||||||
|
- Pass the download ID directly in the DeleteDownload event when publishing into the networking channel
|
||||||
|
- Blocklist Item ID passed in the DeleteBlocklistItem event when sent to the networking channel
|
||||||
|
- AddRootFolderBody now constructed prior to AddRootFolder event being sent down the network channel
|
||||||
|
- Cancel all requests when switching Servarr tabs to both improve performance and fix issue #15
|
||||||
|
- **add_movie_handler_tests**: Added in a forgotten test for the build_add_movie_body function
|
||||||
|
- Missing tagged version of docker builds in release flow
|
||||||
|
- AddMovie Radarr event is now populated in the dispatch thread before being sent to the network thread
|
||||||
|
- dynamically load servarrs in UI based on what configs are provided
|
||||||
|
|
||||||
|
## v0.4.1 (2024-12-14)
|
||||||
|
|
||||||
|
### Feat
|
||||||
|
|
||||||
|
- **docs**: Updated the README with new screeshots for the Sonarr release
|
||||||
|
- **handler**: Support for toggling the monitoring status of a specified episode in the Sonarr UI
|
||||||
|
- **handlers**: Support for toggling the monitoring status of a season in the Sonarr UI
|
||||||
|
- **keybindings**: Added a new keybinding for toggling the monitoring of a highlighted table item
|
||||||
|
- **cli**: Support for toggling monitoring on a specific episode in Sonarr
|
||||||
|
- **network**: Support for toggling the monitoring status of an episode in Sonarr
|
||||||
|
- **cli**: Support for toggling monitoring for a specific season in Sonarr
|
||||||
|
- **network**: Support for toggling monitoring/unmonitoring a season
|
||||||
|
- **handlers**: Support for the episode details popup
|
||||||
|
- **ui**: Support for the episode details UI
|
||||||
|
- **handler**: Full handler support for the Season details UI in Sonarr
|
||||||
|
- **ui**: Sonarr support for viewing season details
|
||||||
|
- **cli**: Sonarr support for fetching a list of all episode files for a given series ID
|
||||||
|
- **app**: Dispatch support for Season Details to fetch both the current downloads as well as the episode files to match qualities to them
|
||||||
|
- **network**: Support for fetching all episode files for a given series
|
||||||
|
- **app**: Model and modal support for the season and episode details popups
|
||||||
|
- **cli**: Sonarr support for fetching season history events
|
||||||
|
- **network**: Sonarr support for fetching season history
|
||||||
|
- **ui**: Sonarr support for the series details popup
|
||||||
|
- **ui**: Sonarr support for editing a series from within the series details popup
|
||||||
|
- **ui**: Sonarr Series details UI is now available
|
||||||
|
- **ui**: Full Sonarr system tab support
|
||||||
|
- **handler**: System handler support for Sonarr
|
||||||
|
- **ui**: Full Sonarr support for the indexer tab
|
||||||
|
- **ui**: Support for modifying the indexer priority in Radarr
|
||||||
|
- **handler**: Full indexer tab handler support
|
||||||
|
- **ui**: Root folder tab support
|
||||||
|
- **handlers**: Support for root folder actions
|
||||||
|
- **ui**: History tab support
|
||||||
|
- **handler**: History tab support
|
||||||
|
- **ui**: Blocklist UI support
|
||||||
|
- **handler**: Wired in the blocklist handler to the main handlers
|
||||||
|
- **handler**: Blocklist handler support
|
||||||
|
- **ui**: Downloads tab support
|
||||||
|
- **handler**: Download tab support
|
||||||
|
- **ui**: Edit series support
|
||||||
|
- **handler**: Edit series support
|
||||||
|
- **ui**: Add series support Sonarr
|
||||||
|
- **handler**: Add series support for Sonarr
|
||||||
|
- **ui**: Delete a series
|
||||||
|
- **handler**: Support for deleting a series in Sonarr
|
||||||
|
- **ui**: Support for the Series table
|
||||||
|
- **handlers**: Sonarr key support for the Series table
|
||||||
|
- **models**: Added the necessary contextual help and tabs for the Sonarr UI
|
||||||
|
- **ui**: Initial UI support for switching to Sonarr tabs
|
||||||
|
- **app**: Dispatch support for all relevant Sonarr blocks
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- **blocklist_handler**: Fixed a breaking change between Sonarr v3 and v4
|
||||||
|
- **style**: Addressed linter complaints on formatting
|
||||||
|
- Implemented a handful of fixes that are breaking changes between Sonarr v3 and v4
|
||||||
|
- **handler_tests**: Fixed all delegation tests to have initial conditions set properly
|
||||||
|
- **ui**: Fixed a bug that requires a minimum height for all popups so all error messages and other simple popups appear
|
||||||
|
- **handler**: Fixed a bug in the history handler that wouldn't reset the filter or search if a user hit 'esc' on the History tab
|
||||||
|
- **ui**: Fix the System Details Tasks popup to be navigable in both Sonarr and Radarr
|
||||||
|
- **ui**: Fixed a potential rare bug in the UI where the application would panic if the height of the downloads window is 0.
|
||||||
|
|
||||||
|
### Refactor
|
||||||
|
|
||||||
|
- **network**: Changed the toggle episode monitoring handler to simply return empty since the response is always empty from Sonarr
|
||||||
|
- **ui**: Tweaked some of the color schemes in the series table
|
||||||
|
- Fixed a couple of typos in some test function names
|
||||||
|
- **handlers**: Refactored the handlers to all use the handle_table_events macro when appropriate and created tests for the macro so tests don't have to be duplicated across each handler
|
||||||
|
- **ui**: Simplified the popup delegation so all future UI is easier to implement
|
||||||
|
- **indexers_handler**: Use the new handle_table_events macro
|
||||||
|
- **root_folders_handler**: Use the new handle_table_events macro
|
||||||
|
- **blocklist_handler**: Use the new handle_table_events macro
|
||||||
|
- **downloads_handler**: Use the new handle_table_events macro
|
||||||
|
- **collection_details_handler**: use the new handle_table_events macro
|
||||||
|
- **collections_handler**: Use the new handle_table_events macro
|
||||||
|
- **movie_details_handler**: Use the new handle_table_events macro
|
||||||
|
- **library_handler**: Radarr use the new handle_table_events macro
|
||||||
|
- **indexers_handler**: Use the new handle_table_events macro
|
||||||
|
- **indexers_handler**: Use the new handle_table_events macro
|
||||||
|
- **root_folder_handler**: Use the new handle_table_events macro
|
||||||
|
- **history_handler**: Use the new handle_table_event macro
|
||||||
|
- **blocklist_handler**: Use the new handle_table_events macro
|
||||||
|
- **downloads_handler**: Use the new handle_table_events macro
|
||||||
|
- **series_details_handler**: Use the new handle_table_events macro
|
||||||
|
- **handler**: Created a macro to handle all table key events to reduce code duplication and make future implementations faster; Only refactored the Sonarr library to use it thus far
|
||||||
|
- **ui**: all table search and filter functionality is now available directly through the ManagarrTable widget to make life easier moving forward
|
||||||
|
- **keys**: Created a auto search key instead of reusing the existing search key to make things easier
|
||||||
|
- **BlockSelectionState**: Refactored so selection of blocks in 2x2 grids is more intuitive and added left() and right() methods to aid this effort.
|
||||||
|
|
||||||
|
### Perf
|
||||||
|
|
||||||
|
- Improved performance by optimizing API calls to only refresh when the tick prompts a refresh. All UI is now significantly faster
|
||||||
|
|
||||||
|
## v0.3.7 (2024-11-26)
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- **ci**: Forgot to also pull in the most recent changes [skip ci]
|
||||||
|
|
||||||
|
## v0.3.6 (2024-11-26)
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- **ci**: Ensure the Release Crate job fetches the most recent commit before publishing the crate [skip ci]
|
||||||
|
|
||||||
|
## v0.3.4 (2024-11-26)
|
||||||
|
|
||||||
|
## v0.3.3 (2024-11-26)
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- **ci**: Properly prefix version tags with 'v' [skip ci]
|
||||||
|
- **ci**: Bump the version in the Cargo.lock file and commit it as well when releasing [skip ci]
|
||||||
|
|
||||||
|
## v0.3.2 (2024-11-26)
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- **ci**: Updated the Cargo.lock file [skip ci]
|
||||||
|
- **ci**: Use a different GitHub action to release the crate to Crates.io [skip ci]
|
||||||
|
- **ci**: Don't manually push the tags and let Commitizen do it [skip ci]
|
||||||
|
|
||||||
|
## v0.3.1 (2024-11-26)
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- **ci**: Don't manually push the tags and let Commitizen do it [skip ci]
|
||||||
|
- **ci**: Fixed a typo in the version creation on GitHub [skip ci]
|
||||||
|
|
||||||
|
## v0.3.0 (2024-11-26)
|
||||||
|
|
||||||
|
### Feat
|
||||||
|
|
||||||
|
- **cli**: Support for editing a sonarr series
|
||||||
|
- **models**: Added the ActiveSonarrBlocks for editing a series
|
||||||
|
- **network**: Support for editing a series in Sonarr
|
||||||
|
- **models**: Created the EditSeriesModal
|
||||||
|
- **cli**: Support for editing Sonarr indexers
|
||||||
|
- **network**: Support for editing a sonarr indexer
|
||||||
|
- **cli**: Support for deleting an episode file from disk
|
||||||
|
- **network**: Support for deleting an episode file from disk in Sonarr
|
||||||
|
- **cli**: Support for editing all indexer settings in Sonarr
|
||||||
|
- **models**: Added the ActiveSonarrBlocks for editing all indexer settings
|
||||||
|
- **network**: Support for editing all sonarr indexer settings
|
||||||
|
- **cli**: Support for searching for new series to add to Sonarr
|
||||||
|
- **network**: Support for searching for new series
|
||||||
|
- **cli**: Support for adding a series to Sonarr
|
||||||
|
- **cli**: Support for adding a series to Sonarr
|
||||||
|
- **network**: Support for adding a new series to Sonarr
|
||||||
|
- **cli**: Support for fetching all sonarr language profiles
|
||||||
|
- **network**: Support for fetching all Sonarr language profiles
|
||||||
|
- **cli**: Support for deleting a series from Sonarr
|
||||||
|
- **network**: Support for deleting a series from Sonarr
|
||||||
|
- **cli**: Support for downloading an episode release in Sonarr
|
||||||
|
- **cli**: Support for downloading a season release in Sonarr
|
||||||
|
- **cli**: Support for downloading a Series release in Sonarr
|
||||||
|
- **network**: Support for downloading releases from Sonarr
|
||||||
|
- **cli**: Support for refreshing Sonarr downloads
|
||||||
|
- **network**: Support for updating Sonarr downloads
|
||||||
|
- **cli**: Support for refreshing a specific series in Sonarr
|
||||||
|
- **network**: Support for updating and scanning a series in Sonarr
|
||||||
|
- **cli**: Support for refreshing all Sonarr series data
|
||||||
|
- **network**: Support for updating all series in Sonarr
|
||||||
|
- **cli**: Support for triggering an automatic episode search in Sonarr
|
||||||
|
- **cli**: Support for triggering an automatic season search in Sonarr
|
||||||
|
- **cli**: Support for triggering an automatic series search in Sonarr
|
||||||
|
- **network**: Support for triggering an automatic episode search in Sonarr
|
||||||
|
- **network**: Support for triggering an automatic season search in Sonarr
|
||||||
|
- **network**: Support for triggering an automatic series search in Sonarr
|
||||||
|
- **cli**: Support for testing all Sonarr indexers at once
|
||||||
|
- **network**: Support for testing all Sonarr indexers at once
|
||||||
|
- **cli**: Support for testing an individual Sonarr indexer
|
||||||
|
- **network**: Added the ability to test an individual indexer in Sonarr
|
||||||
|
- **cli**: Support for starting a Sonarr task
|
||||||
|
- **network**: Support for starting a Sonarr task
|
||||||
|
- **cli**: Support for listing Sonarr updates
|
||||||
|
- **network**: Support for fetching Sonarr updates
|
||||||
|
- **cli**: Support for listing all Sonarr tasks
|
||||||
|
- **network**: Support for fetching all Sonarr tasks
|
||||||
|
- **cli**: Support for marking a Sonarr history item as 'failed'
|
||||||
|
- **network**: Support for marking a Sonarr history item as failed
|
||||||
|
- **cli**: Support for listing the available disk space for all provisioned root folders in both Radarr and Sonarr
|
||||||
|
- **network**: Support for listing disk space on a Sonarr instance
|
||||||
|
- **cli**: Support for listing all Sonarr tags
|
||||||
|
- **cli**: Support for adding a root folder to Sonarr
|
||||||
|
- **cli**: CLI support for adding a tag to Sonarr
|
||||||
|
- **network**: Support for fetching and listing all Sonarr tags
|
||||||
|
- **network**: Support for deleting tags from Sonarr
|
||||||
|
- **network**: Support for adding tags to Sonarr
|
||||||
|
- **network**: Support for adding a root folder to Sonarr
|
||||||
|
- **cli**: Support for deleting a root folder from Sonarr
|
||||||
|
- **network**: Support for deleting a Sonarr root folder
|
||||||
|
- **cli**: Support for fetching all Sonarr root folders
|
||||||
|
- **network**: Support for fetching all Sonarr root folders
|
||||||
|
- **cli**: Support for deleting a Sonarr indexer
|
||||||
|
- **network**: Support for deleting an indexer from Sonarr
|
||||||
|
- **cli**: Support for deleting a download from Sonarr
|
||||||
|
- **network**: Support for deleting a download from Sonarr
|
||||||
|
- **cli**: Support for fetching episode history events from Sonarr
|
||||||
|
- **network**: Support for fetching episode history
|
||||||
|
- **cli**: Added a spinner to the CLI for long running commands like fetching releases
|
||||||
|
- **cli**: Support for fetching history for a given series ID
|
||||||
|
- **network**: Support for fetching Sonarr series history for a given series ID
|
||||||
|
- **cli**: Support for fetching all Sonarr history events
|
||||||
|
- **network**: Support to fetch all Sonarr history events
|
||||||
|
- **models**: Added an additional History tab to the mocked tabs for viewing all Sonarr history at once
|
||||||
|
- **models**: Stubbed out the necessary ActiveSonarrBlocks for the UI mockup
|
||||||
|
- **cli**: Added support for manually searching for episode releases in Sonarr
|
||||||
|
- **network**: Added support for fetching episode releases in Sonarr
|
||||||
|
- **cli**: Added CLI support for fetching series details in Sonarr
|
||||||
|
- **network**: Added support for fetching series details for a given series ID in Sonarr
|
||||||
|
- **cli**: Added support for manually searching for season releases for Sonarr
|
||||||
|
- **network**: Added support for fetching season releases for Sonarr
|
||||||
|
- **cli**: Added support for listing Sonarr queued events
|
||||||
|
- **network**: Added support for fetching Sonarr queued events
|
||||||
|
- **cli**: Added CLI support for fetching all indexer settings for Sonarr
|
||||||
|
- **network**: Added netwwork support for fetching all indexer settings for Sonarr
|
||||||
|
- **cli**: Added Sonarr support for fetching host and security configs
|
||||||
|
- **network**: Added network support for fetching host and security configs from Sonarr
|
||||||
|
- **cli**: Added CLI support for listing Sonarr indexers
|
||||||
|
- **network**: Added the GetIndexers network call for Sonarr
|
||||||
|
- **cli**: Added sonarr support for listing downloads, listing quality profiles, and fetching detailed information about an episode
|
||||||
|
- **network**: Added get quality profiles and get episode details events for Sonarr
|
||||||
|
- **cli**: Sonarr CLI support for fetching all episodes for a given series
|
||||||
|
- **sonarr_network**: Added support for fetching episodes for a specified series to the network events
|
||||||
|
- **models**: Added the Episode model to Sonarr models
|
||||||
|
- **models**: Created the StatefulTree struct for displaying seasons and episodes (and any other structured data) for the UI.
|
||||||
|
- **sonarr**: Added CLI support for listing Sonarr logs
|
||||||
|
- **sonarr**: Added the ability to fetch Sonarr logs
|
||||||
|
- **sonarr**: Added blocklist commands (List, Clear, Delete)
|
||||||
|
- Added initial Sonarr CLI support and the initial network handler setup for the TUI
|
||||||
|
- Added a new command to the main managarr CLI: tail-logs, to enable users to tail the Managarr logs without needing to know where the log file itself is located
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- Reverted to old version to fix release [skip ci]
|
||||||
|
- **minimal-versions**: Addressed concerns with the minimal-versions CI checks
|
||||||
|
- **lint**: Addressed linter complaints
|
||||||
|
- **cli**: Corrected some copy/paste typos
|
||||||
|
- **network**: Force sonarr to save edits to indexers
|
||||||
|
- **network**: Made the overview field nullable in the Sonarr series model
|
||||||
|
- **network**: Added filtering for full seasons specifically in the UI when performing a manual full season search and added a message to the CLI that noes to only try to download a full season if that release includes 'fullSeason: true'
|
||||||
|
- **network**: Not all Sonarr tasks return the lastDuration field and was causing a crash
|
||||||
|
- **network**: Fixed an issue with dynamic typing in responses from Sonarr for history items
|
||||||
|
- **config**: The CLI panics if the servarr you specify has no config defined
|
||||||
|
- Imported a missing macro in the panic hook
|
||||||
|
|
||||||
|
### Refactor
|
||||||
|
|
||||||
|
- **cli**: the trigger-automatic-search commands now all have their own dedicated subcommand to keep things cleaner. Now they look like 'trigger-automatic-search episode/series/season' and their corresponding flags
|
||||||
|
- **cli**: Added an additional delegation test to ensure manual-search commands are delegated to the manual-search command handler
|
||||||
|
- **cli**: Moved the manual-season-search and manual-episode-search commands into their own dedicated handler so the commands can now be manual-search episode or manual-search season
|
||||||
|
|
||||||
|
## v0.2.2 (2024-11-06)
|
||||||
|
|
||||||
|
### Fix
|
||||||
|
|
||||||
|
- **handler**: Fixed a bug in the movie details handler that would allow key events to be processed before the data was finished loading
|
||||||
|
- **ui**: Fixed a bug that would freeze all user input while background network requests were running
|
||||||
|
- **radarr_ui**: Fixed a race condition bug in the movie details UI that would panic if the user changes tabs too quickly
|
||||||
|
|
||||||
|
### Perf
|
||||||
|
|
||||||
|
- **network**: Improved performance and reactiveness of the UI by speeding up network requests and clearing the channel whenever a request is cancelled/the UI is routing
|
||||||
|
|
||||||
|
## v0.2.1 (2024-11-06)
|
||||||
|
|
||||||
|
## [0.2.1](https://github.com/Dark-Alex-17/managarr/compare/v0.2.0...v0.2.1) - 2024-11-06
|
||||||
|
|
||||||
|
### Other
|
||||||
|
|
||||||
|
- Removed the need for use_ssl to indicate SSL usage; instead just use the ssl_cert_path
|
||||||
|
- Applied bug fix to the downloads tab as well as the context [skip ci]
|
||||||
|
- Updated the README to not include the GitHub downloads badge since all binary releases are on crates.io [skip ci]
|
||||||
|
- Set all releases as manually triggered instead of automatic [skip ci]
|
||||||
|
- Updated dockerfile to no longer use the --disable-terminal-size-checks flag [skip ci]
|
||||||
|
|
||||||
## [0.1.5](https://github.com/Dark-Alex-17/managarr/compare/v0.1.4...v0.1.5) - 2024-11-03
|
## [0.1.5](https://github.com/Dark-Alex-17/managarr/compare/v0.1.4...v0.1.5) - 2024-11-03
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,24 @@
|
|||||||
# Contributing
|
# Contributing
|
||||||
Contributors are very welcome! **No contribution is too small and all contributions are valued.**
|
Contributors are very welcome! **No contribution is too small and all contributions are valued.**
|
||||||
|
|
||||||
|
## License and Attribution
|
||||||
|
|
||||||
|
#### _If you plan on contributing to the base project, no attribution is needed!_ Feel free to proceed to the [next steps](CONTRIBUTING.md#Rust).
|
||||||
|
|
||||||
|
Otherwise, below are key points to understand from the [Managarr License, Version 1.0](LICENSE):
|
||||||
|
1. **Non-Commercial Use**:
|
||||||
|
- Managarr is licensed solely for non-commercial purposes. Any commercial use of Managarr (e.g., selling or offering as a paid service) requires separate permission.
|
||||||
|
|
||||||
|
2. **Attribution when Forking and Redistributing Without Contributing back to Main Project**:
|
||||||
|
- **If you fork the project and distribute it separately** (e.g., publish or _publicly_ host it independently from the original project), you are required to provide attribution.
|
||||||
|
- You may credit the original author by using any of the following phrasing:
|
||||||
|
- "Thanks to Alexander J. Clarke (Dark-Alex-17) for creating the original Managarr project!"
|
||||||
|
- "Forked from the Managarr project, created by Alexander J. Clarke (Dark-Alex-17)"
|
||||||
|
- "This software is based on the original Managarr project by Alexander J. Clarke (Dark-Alex-17)"
|
||||||
|
- "Inspired by Alexander J. Clarke (Dark-Alex-17)'s Managarr project"
|
||||||
|
- If changes are made to the base Managarr project, please note those modifications and provide the new attribution accordingly.
|
||||||
|
|
||||||
|
## Rust
|
||||||
You'll need to have the stable Rust toolchain installed in order to develop Managarr.
|
You'll need to have the stable Rust toolchain installed in order to develop Managarr.
|
||||||
|
|
||||||
The Rust toolchain (stable) can be installed via rustup using the following command:
|
The Rust toolchain (stable) can be installed via rustup using the following command:
|
||||||
@@ -11,6 +29,37 @@ curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh
|
|||||||
|
|
||||||
This will install `rustup`, `rustc` and `cargo`. For more information, refer to the [official Rust installation documentation](https://www.rust-lang.org/tools/install).
|
This will install `rustup`, `rustc` and `cargo`. For more information, refer to the [official Rust installation documentation](https://www.rust-lang.org/tools/install).
|
||||||
|
|
||||||
|
## Commitizen
|
||||||
|
[Commitizen](https://github.com/commitizen-tools/commitizen?tab=readme-ov-file) is a nifty tool that helps us write better commit messages. It ensures that our
|
||||||
|
commits have a consistent style and makes it easier to generate CHANGELOGS. Additionally,
|
||||||
|
Commitizen is used to run pre-commit checks to enforce style constraints.
|
||||||
|
|
||||||
|
To install `commitizen` and the `pre-commit` prerequisite, run the following command:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
python3 -m pip install commitizen pre-commit
|
||||||
|
```
|
||||||
|
|
||||||
|
### Commitizen Quick Guide
|
||||||
|
To see an example commit to get an idea for the Commitizen style, run:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
cz example
|
||||||
|
```
|
||||||
|
|
||||||
|
To see the allowed types of commits and their descriptions, run:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
cz info
|
||||||
|
```
|
||||||
|
|
||||||
|
If you'd like to create a commit using Commitizen with an interactive prompt to help you get
|
||||||
|
comfortable with the style, use:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
cz commit
|
||||||
|
```
|
||||||
|
|
||||||
## Setup workspace
|
## Setup workspace
|
||||||
|
|
||||||
1. Clone this repo
|
1. Clone this repo
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "managarr"
|
name = "managarr"
|
||||||
version = "0.2.0"
|
version = "0.5.1"
|
||||||
authors = ["Alex Clarke <alex.j.tusa@gmail.com>"]
|
authors = ["Alex Clarke <alex.j.tusa@gmail.com>"]
|
||||||
description = "A TUI and CLI to manage your Servarrs"
|
description = "A TUI and CLI to manage your Servarrs"
|
||||||
keywords = ["managarr", "ratatui", "dashboard", "servarr", "tui"]
|
keywords = ["managarr", "ratatui", "dashboard", "servarr", "tui"]
|
||||||
@@ -10,25 +10,28 @@ homepage = "https://github.com/Dark-Alex-17/managarr"
|
|||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
license = "MIT"
|
license = "MIT"
|
||||||
rust-version = "1.82.0"
|
rust-version = "1.85.0"
|
||||||
exclude = [".github", "CONTRIBUTING.md", "*.log", "tags"]
|
exclude = [".github", "CONTRIBUTING.md", "*.log", "tags"]
|
||||||
|
|
||||||
|
[workspace]
|
||||||
|
members = ["proc_macros/enum_display_style_derive", "proc_macros/validate_theme_derive"]
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
anyhow = "1.0.68"
|
anyhow = "1.0.68"
|
||||||
backtrace = "0.3.67"
|
backtrace = "0.3.74"
|
||||||
bimap = { version = "0.6.3", features = ["serde"] }
|
bimap = { version = "0.6.3", features = ["serde"] }
|
||||||
chrono = { version = "0.4.38", features = ["serde"] }
|
chrono = { version = "0.4.38", features = ["serde"] }
|
||||||
confy = { version = "0.6.0", default-features = false, features = [
|
confy = { version = "0.6.0", default-features = false, features = [
|
||||||
"yaml_conf",
|
"yaml_conf",
|
||||||
] }
|
] }
|
||||||
crossterm = "0.27.0"
|
crossterm = "0.28.1"
|
||||||
derivative = "2.2.0"
|
derivative = "2.2.0"
|
||||||
human-panic = "1.1.3"
|
human-panic = "2.0.2"
|
||||||
indoc = "2.0.0"
|
indoc = "2.0.0"
|
||||||
log = "0.4.17"
|
log = "0.4.17"
|
||||||
log4rs = { version = "1.2.0", features = ["file_appender"] }
|
log4rs = { version = "1.2.0", features = ["file_appender"] }
|
||||||
regex = "1.11.1"
|
regex = "1.11.1"
|
||||||
reqwest = { version = "0.11.14", features = ["json"] }
|
reqwest = { version = "0.12.9", features = ["json"] }
|
||||||
serde_yaml = "0.9.16"
|
serde_yaml = "0.9.16"
|
||||||
serde_json = "1.0.91"
|
serde_json = "1.0.91"
|
||||||
serde = { version = "1.0.214", features = ["derive"] }
|
serde = { version = "1.0.214", features = ["derive"] }
|
||||||
@@ -36,22 +39,40 @@ strum = { version = "0.26.3", features = ["derive"] }
|
|||||||
strum_macros = "0.26.4"
|
strum_macros = "0.26.4"
|
||||||
tokio = { version = "1.36.0", features = ["full"] }
|
tokio = { version = "1.36.0", features = ["full"] }
|
||||||
tokio-util = "0.7.8"
|
tokio-util = "0.7.8"
|
||||||
ratatui = { version = "0.28.0", features = ["all-widgets"] }
|
ratatui = { version = "0.29.0", features = [
|
||||||
|
"all-widgets",
|
||||||
|
"unstable-widget-ref",
|
||||||
|
] }
|
||||||
urlencoding = "2.1.2"
|
urlencoding = "2.1.2"
|
||||||
clap = { version = "4.5.20", features = ["derive", "cargo", "env"] }
|
clap = { version = "4.5.20", features = [
|
||||||
|
"derive",
|
||||||
|
"cargo",
|
||||||
|
"env",
|
||||||
|
"wrap_help",
|
||||||
|
] }
|
||||||
clap_complete = "4.5.33"
|
clap_complete = "4.5.33"
|
||||||
itertools = "0.13.0"
|
itertools = "0.14.0"
|
||||||
ctrlc = "3.4.5"
|
ctrlc = "3.4.5"
|
||||||
colored = "2.1.0"
|
colored = "3.0.0"
|
||||||
async-trait = "0.1.83"
|
async-trait = "0.1.83"
|
||||||
dirs-next = "2.0.0"
|
dirs-next = "2.0.0"
|
||||||
|
managarr-tree-widget = "0.24.0"
|
||||||
|
indicatif = "0.17.9"
|
||||||
|
derive_setters = "0.1.6"
|
||||||
|
deunicode = "1.6.0"
|
||||||
|
paste = "1.0.15"
|
||||||
|
openssl = { version = "0.10.70", features = ["vendored"] }
|
||||||
|
veil = "0.2.0"
|
||||||
|
validate_theme_derive = { path = "proc_macros/validate_theme_derive" }
|
||||||
|
enum_display_style_derive = { path = "proc_macros/enum_display_style_derive" }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
assert_cmd = "2.0.16"
|
assert_cmd = "2.0.16"
|
||||||
mockall = "0.13.0"
|
mockall = "0.13.0"
|
||||||
mockito = "1.0.0"
|
mockito = "1.0.0"
|
||||||
pretty_assertions = "1.3.0"
|
pretty_assertions = "1.3.0"
|
||||||
rstest = "0.18.2"
|
rstest = "0.25.0"
|
||||||
|
serial_test = "3.2.0"
|
||||||
|
|
||||||
[dev-dependencies.cargo-husky]
|
[dev-dependencies.cargo-husky]
|
||||||
version = "1"
|
version = "1"
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
FROM clux/muslrust:stable AS builder
|
FROM rust:1.85 AS builder
|
||||||
WORKDIR /usr/src
|
WORKDIR /usr/src
|
||||||
|
|
||||||
# Download and compile Rust dependencies in an empty project and cache as a separate Docker layer
|
# Download and compile Rust dependencies in an empty project and cache as a separate Docker layer
|
||||||
@@ -6,21 +6,21 @@ RUN USER=root cargo new --bin managarr-temp
|
|||||||
|
|
||||||
WORKDIR /usr/src/managarr-temp
|
WORKDIR /usr/src/managarr-temp
|
||||||
COPY Cargo.* .
|
COPY Cargo.* .
|
||||||
RUN cargo build --release --target x86_64-unknown-linux-musl
|
RUN cargo build --release
|
||||||
# remove src from empty project
|
# remove src from empty project
|
||||||
RUN rm -r src
|
RUN rm -r src
|
||||||
COPY src ./src
|
COPY src ./src
|
||||||
# remove previous deps
|
# remove previous deps
|
||||||
RUN rm ./target/x86_64-unknown-linux-musl/release/deps/managarr*
|
RUN rm ./target/release/deps/managarr*
|
||||||
|
|
||||||
RUN --mount=type=cache,target=/volume/target \
|
RUN --mount=type=cache,target=/volume/target \
|
||||||
--mount=type=cache,target=/root/.cargo/registry \
|
--mount=type=cache,target=/root/.cargo/registry \
|
||||||
cargo build --release --target x86_64-unknown-linux-musl --bin managarr
|
cargo build --release --bin managarr
|
||||||
RUN mv target/x86_64-unknown-linux-musl/release/managarr .
|
RUN mv target/release/managarr .
|
||||||
|
|
||||||
FROM debian:stable-slim
|
FROM debian:stable-slim
|
||||||
|
|
||||||
# Copy the compiled binary from the builder container
|
# Copy the compiled binary from the builder container
|
||||||
COPY --from=builder --chown=nonroot:nonroot /usr/src/managarr-temp/managarr /usr/local/bin
|
COPY --from=builder --chown=nonroot:nonroot /usr/src/managarr-temp/managarr /usr/local/bin
|
||||||
|
|
||||||
ENTRYPOINT [ "/usr/local/bin/managarr", "--disable-terminal-size-checks" ]
|
ENTRYPOINT [ "/usr/local/bin/managarr" ]
|
||||||
|
|||||||
@@ -1,16 +1,53 @@
|
|||||||
MIT License
|
Managarr License
|
||||||
|
Version 1.0, 2025
|
||||||
|
|
||||||
Copyright (c) 2023 Alexander J. Clarke
|
Copyright (c) 2025 Alexander J. Clarke (Dark-Alex-17)
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
of this software and associated documentation files (the "Software"), to deal
|
of this software and associated documentation files (the "Software"), to use,
|
||||||
in the Software without restriction, including without limitation the rights
|
copy, modify, merge, publish, and distribute the Software solely for
|
||||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
non-commercial purposes, subject to the following conditions:
|
||||||
copies of the Software, and to permit persons to whom the Software is
|
|
||||||
furnished to do so, subject to the following conditions:
|
|
||||||
|
|
||||||
The above copyright notice and this permission notice shall be included in all
|
1. Attribution:
|
||||||
copies or substantial portions of the Software.
|
- The above copyright notice, this permission notice, and a prominent notice stating
|
||||||
|
that the Software is part of the "Managarr" project shall be included in all copies or
|
||||||
|
substantial portions of the Software **when the Software is forked and redistributed separately** from the original project.
|
||||||
|
- If you fork the software and **distribute it separately** without merging it back into the original base project (the Managarr repository), you must provide attribution to the original author.
|
||||||
|
You may use any of the following forms of attribution:
|
||||||
|
- "Thanks to Alexander J. Clarke (Dark-Alex-17) for creating the original Managarr project!"
|
||||||
|
- "Forked from the Managarr project, created by Alexander J. Clarke (Dark-Alex-17)"
|
||||||
|
- "This software is based on the original Managarr project by Alexander J. Clarke (Dark-Alex-17)"
|
||||||
|
- "Inspired by Alexander J. Clarke (Dark-Alex-17)'s Managarr project"
|
||||||
|
- If you modify the software, the attribution must also note that changes were made and describe those modifications, if feasible.
|
||||||
|
|
||||||
|
2. Non-Commercial Use Only:
|
||||||
|
The use of this Software for commercial purposes, including but not limited
|
||||||
|
to sale, licensing, or use in any product or service for monetary
|
||||||
|
compensation, is strictly prohibited without prior written permission from
|
||||||
|
Alexander J. Clarke (Dark-Alex-17).
|
||||||
|
|
||||||
|
For avoidance of doubt:
|
||||||
|
- **Allowed:** Private use, educational purposes, research, or any usage
|
||||||
|
that does not directly generate revenue.
|
||||||
|
- **Prohibited:** Selling, sublicensing, or incorporating the Software into
|
||||||
|
commercial products or services.
|
||||||
|
|
||||||
|
3. Modifications and Derivatives:
|
||||||
|
- Any modifications or derivative works based on this Software must clearly
|
||||||
|
document all changes made and prominently credit the original "Managarr"
|
||||||
|
project as described in section 1.
|
||||||
|
- Derivative works must retain this license, the original copyright notice,
|
||||||
|
and all terms and conditions described herein. This applies to the entire
|
||||||
|
derivative work, even if combined with other software.
|
||||||
|
|
||||||
|
4. Warranty Disclaimer:
|
||||||
|
This Software is provided "as is," without warranty of any kind, express
|
||||||
|
or implied, including but not limited to the warranties of merchantability,
|
||||||
|
fitness for a particular purpose, and noninfringement. In no event shall the
|
||||||
|
authors or copyright holders be liable for any claim, damages, or other
|
||||||
|
liability, whether in an action of contract, tort, or otherwise, arising
|
||||||
|
from, out of, or in connection with the Software or the use or other
|
||||||
|
dealings in the Software.
|
||||||
|
|
||||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ default: run
|
|||||||
.PHONY: test test-cov build run lint lint-fix fmt analyze sonar release delete-tag
|
.PHONY: test test-cov build run lint lint-fix fmt analyze sonar release delete-tag
|
||||||
|
|
||||||
test:
|
test:
|
||||||
@cargo test
|
@cargo test --all
|
||||||
|
|
||||||
## Run all tests with coverage - `cargo install cargo-tarpaulin`
|
## Run all tests with coverage - `cargo install cargo-tarpaulin`
|
||||||
test-cov:
|
test-cov:
|
||||||
|
|||||||
@@ -1,24 +1,26 @@
|
|||||||
# managarr - A TUI and CLI to manage your Servarrs
|
# managarr - A TUI and CLI to manage your Servarrs
|
||||||
|
|
||||||

|

|
||||||

|

|
||||||

|
|
||||||

|

|
||||||

|

|
||||||
[](https://crates.io/crates/managarr)
|
[](https://crates.io/crates/managarr)
|
||||||

|

|
||||||
[](https://codecov.io/gh/Dark-Alex-17/managarr)
|
[](https://codecov.io/gh/Dark-Alex-17/managarr)
|
||||||
[](https://github.com/Dark-Alex-17/managarr/releases)
|
|
||||||

|

|
||||||
|
[](https://github.com/Dark-Alex-17/managarr/releases)
|
||||||
|

|
||||||
|
[](https://matrix.to/#/#managarr:matrix.org)
|
||||||
|
|
||||||
|
|
||||||
Managarr is a TUI and CLI to help you manage your HTPC (Home Theater PC). Built with 🤎 in Rust!
|
Managarr is a TUI and CLI to help you manage your HTPC (Home Theater PC). Built with 🤎 in Rust!
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
## What Servarrs are supported?
|
## What Servarrs are supported?
|
||||||
|
|
||||||
- [x]  [Radarr](https://wiki.servarr.com/radarr)
|
- [x]  [Radarr](https://wiki.servarr.com/radarr)
|
||||||
- [ ]  [Sonarr](https://wiki.servarr.com/en/sonarr)
|
- [x]  [Sonarr](https://wiki.servarr.com/en/sonarr)
|
||||||
- [ ]  [Readarr](https://wiki.servarr.com/en/readarr)
|
- [ ]  [Readarr](https://wiki.servarr.com/en/readarr)
|
||||||
- [ ]  [Lidarr](https://wiki.servarr.com/en/lidarr)
|
- [ ]  [Lidarr](https://wiki.servarr.com/en/lidarr)
|
||||||
- [ ]  [Prowlarr](https://wiki.servarr.com/en/prowlarr)
|
- [ ]  [Prowlarr](https://wiki.servarr.com/en/prowlarr)
|
||||||
@@ -34,7 +36,10 @@ Simply run the following command to start a demo:
|
|||||||
curl https://raw.githubusercontent.com/Dark-Alex-17/managarr-demo/main/managarr-demo.sh > /tmp/managarr-demo.sh && bash /tmp/managarr-demo.sh
|
curl https://raw.githubusercontent.com/Dark-Alex-17/managarr-demo/main/managarr-demo.sh > /tmp/managarr-demo.sh && bash /tmp/managarr-demo.sh
|
||||||
```
|
```
|
||||||
|
|
||||||
|
Alternatively, you can try out the demo container without downloading anything by visiting the [Managarr Demo site](https://managarr-demo.alexjclarke.com).
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
### Cargo
|
### Cargo
|
||||||
If you have Cargo installed, then you can install Managarr from Crates.io:
|
If you have Cargo installed, then you can install Managarr from Crates.io:
|
||||||
|
|
||||||
@@ -48,34 +53,134 @@ cargo install --locked managarr
|
|||||||
### Docker
|
### Docker
|
||||||
Run Managarr as a docker container by mounting your `config.yml` file to `/root/.config/managarr/config.yml`. For example:
|
Run Managarr as a docker container by mounting your `config.yml` file to `/root/.config/managarr/config.yml`. For example:
|
||||||
```shell
|
```shell
|
||||||
docker run --rm -it -v ~/.config/managarr:/root/.config/managarr darkalex17/managarr
|
docker run --rm -it -v /home/aclarke/.config/managarr/config.yml:/root/.config/managarr/config.yml darkalex17/managarr:latest
|
||||||
```
|
```
|
||||||
|
|
||||||
You can also clone this repo and run `make docker` to build a docker image locally and run it using the above command.
|
You can also clone this repo and run `make docker` to build a docker image locally and run it using the above command.
|
||||||
|
|
||||||
|
Please note that you will need to create and popular your configuration file first before starting the container. Otherwise, the container will fail to start.
|
||||||
|
|
||||||
|
**Note:** If you run into errors using relative file paths when mounting the volume with the configuration file, try using an absolute path.
|
||||||
|
|
||||||
|
### Homebrew (Mac and Linux)
|
||||||
|
To install Managarr from Homebrew, install the Managarr tap and then you'll be able to install Managarr:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
brew tap Dark-Alex-17/managarr
|
||||||
|
brew install managarr
|
||||||
|
|
||||||
|
# If you need to be more specific, use the following:
|
||||||
|
brew install Dark-Alex-17/managarr/managarr
|
||||||
|
```
|
||||||
|
|
||||||
|
To upgrade to a newer version of Managarr:
|
||||||
|
```shell
|
||||||
|
brew upgrade managarr
|
||||||
|
```
|
||||||
|
|
||||||
|
### Nix (Externally Maintained)
|
||||||
|
To install Managarr on NixOS, you can use the following command:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
nix-env --install managarr
|
||||||
|
|
||||||
|
# Alternatively, for non-NixOS users, you can spawn a temporary shell with Managarr available like so:
|
||||||
|
nix-shell -p managarr
|
||||||
|
```
|
||||||
|
|
||||||
|
### Chocolatey (Windows)
|
||||||
|
The Managarr Chocolatey package is located [here](https://community.chocolatey.org/packages/managarr). Please note that validation
|
||||||
|
of Chocolatey packages take quite some time, and thus the package may not be available immediately after a new release.
|
||||||
|
|
||||||
|
```powershell
|
||||||
|
choco install managarr
|
||||||
|
|
||||||
|
# Some newer releases may require a version number, so you can specify it like so:
|
||||||
|
choco install managarr --version=0.5.0
|
||||||
|
```
|
||||||
|
|
||||||
|
To upgrade to the latest and greatest version of Managarr:
|
||||||
|
```powershell
|
||||||
|
choco upgrade managarr
|
||||||
|
|
||||||
|
# To upgrade to a specific version:
|
||||||
|
choco upgrade managarr --version=0.5.0
|
||||||
|
```
|
||||||
|
|
||||||
|
### Manual
|
||||||
|
Binaries are available on the [releases](https://github.com/Dark-Alex-17/managarr/releases) page for the following platforms:
|
||||||
|
|
||||||
|
| Platform | Architecture(s) |
|
||||||
|
|----------------|----------------------------|
|
||||||
|
| macOS | x86_64, arm64 |
|
||||||
|
| Linux GNU/MUSL | x86_64,armv6,armv7,aarch64 |
|
||||||
|
| Windows | x86_64,aarch64 |
|
||||||
|
|
||||||
|
#### Windows Instructions
|
||||||
|
To use a binary from the releases page on Windows, do the following:
|
||||||
|
|
||||||
|
1. Download the latest binary [binary](https://github.com/Dark-Alex-17/managarr/releases) for your OS.
|
||||||
|
2. Use 7-Zip or TarTool to unpack the Tar file.
|
||||||
|
3. Run the executable `managarr.exe`!
|
||||||
|
|
||||||
|
#### Linux/MacOS Instructions
|
||||||
|
To use a binary from the releases page on Linux/MacOS, do the following:
|
||||||
|
|
||||||
|
1. Download the latest binary [binary](https://github.com/Dark-Alex-17/managarr/releases) for your OS.
|
||||||
|
2. `cd` to the directory where you downloaded the binary.
|
||||||
|
3. Extract the binary with `tar -C /usr/local/bin -xzf managarr-<arch>.tar.gz` (NB: This may require `sudo`)
|
||||||
|
4. Now you can run `managarr`!
|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
Key:
|
||||||
|
|
||||||
|
| Symbol | Status |
|
||||||
|
|--------|-----------|
|
||||||
|
| ✅ | Supported |
|
||||||
|
| ❌ | Missing |
|
||||||
|
| 🕒 | Planned |
|
||||||
|
| 🚫 | Won't Add |
|
||||||
|
|
||||||
### Radarr
|
### Radarr
|
||||||
|
|
||||||
- [x] View your library, downloads, collections, and blocklist
|
| TUI | CLI | Feature |
|
||||||
- [x] View details of a specific movie including description, history, downloaded file info, or the credits
|
|-----|-----|----------------------------------------------------------------------------------------------------------------|
|
||||||
- [x] View details of any collection and the movies in them
|
| ✅ | ✅ | View your library, downloads, collections, and blocklist |
|
||||||
- [x] View your host and security configs from the CLI to programmatically fetch the API token, among other settings
|
| ✅ | ✅ | View details of a specific movie including description, history, downloaded file info, or the credits |
|
||||||
- [x] Search your library or collections
|
| ✅ | ✅ | View details of any collection and the movies in them |
|
||||||
- [x] Add movies to your library
|
| 🚫 | ✅ | View your host and security configs from the CLI to programmatically fetch the API token, among other settings |
|
||||||
- [x] Delete movies, downloads, and indexers
|
| ✅ | ✅ | Search your library or collections |
|
||||||
- [x] Trigger automatic searches for movies
|
| ✅ | ✅ | Add movies to your library |
|
||||||
- [x] Trigger refresh and disk scan for movies, downloads, and collections
|
| ✅ | ✅ | Delete movies, downloads, and indexers |
|
||||||
- [x] Manually search for movies
|
| ✅ | ✅ | Trigger automatic searches for movies |
|
||||||
- [x] Edit your movies, collections, and indexers
|
| ✅ | ✅ | Trigger refresh and disk scan for movies, downloads, and collections |
|
||||||
- [x] Manage your tags
|
| ✅ | ✅ | Manually search for movies |
|
||||||
- [x] Manage your root folders
|
| ✅ | ✅ | Edit your movies, collections, and indexers |
|
||||||
- [x] Manage your blocklist
|
| ✅ | ✅ | Manage your tags |
|
||||||
- [x] View and browse logs, tasks, events queues, and updates
|
| ✅ | ✅ | Manage your root folders |
|
||||||
- [x] Manually trigger scheduled tasks
|
| ✅ | ✅ | Manage your blocklist |
|
||||||
|
| ✅ | ✅ | View and browse logs, tasks, events queues, and updates |
|
||||||
|
| ✅ | ✅ | Manually trigger scheduled tasks |
|
||||||
|
|
||||||
### Sonarr
|
### Sonarr
|
||||||
- [ ] Support for Sonarr
|
|
||||||
|
| TUI | CLI | Feature |
|
||||||
|
|-----|-----|--------------------------------------------------------------------------------------------------------------------|
|
||||||
|
| ✅ | ✅ | View your library, downloads, blocklist, episodes |
|
||||||
|
| ✅ | ✅ | View details of a specific series, or episode including description, history, downloaded file info, or the credits |
|
||||||
|
| 🚫 | ✅ | View your host and security configs from the CLI to programmatically fetch the API token, among other settings |
|
||||||
|
| ✅ | ✅ | Search your library |
|
||||||
|
| ✅ | ✅ | Add series to your library |
|
||||||
|
| ✅ | ✅ | Delete series, downloads, indexers, root folders, and episode files |
|
||||||
|
| ✅ | ✅ | Trigger automatic searches for series, seasons, or episodes |
|
||||||
|
| ✅ | ✅ | Trigger refresh and disk scan for series and downloads |
|
||||||
|
| ✅ | ✅ | Manually search for series, seasons, or episodes |
|
||||||
|
| ✅ | ✅ | Edit your series and indexers |
|
||||||
|
| ✅ | ✅ | Manage your tags |
|
||||||
|
| ✅ | ✅ | Manage your root folders |
|
||||||
|
| ✅ | ✅ | Manage your blocklist |
|
||||||
|
| ✅ | ✅ | View and browse logs, tasks, events queues, and updates |
|
||||||
|
| ✅ | ✅ | Manually trigger scheduled tasks |
|
||||||
|
|
||||||
### Readarr
|
### Readarr
|
||||||
|
|
||||||
@@ -101,19 +206,29 @@ You can also clone this repo and run `make docker` to build a docker image local
|
|||||||
|
|
||||||
- [ ] Support for Tautulli
|
- [ ] Support for Tautulli
|
||||||
|
|
||||||
|
### Themes
|
||||||
|
Managarr ships with a few themes out of the box. Here's a few examples:
|
||||||
|
|
||||||
|

|
||||||
|

|
||||||
|

|
||||||
|
|
||||||
|
You can also create your own custom themes as well. To learn more about what themes are built-in to Managarr and how
|
||||||
|
to create your own custom themes, check out the [Themes README](themes/README.md).
|
||||||
|
|
||||||
### The Managarr CLI
|
### The Managarr CLI
|
||||||
Managarr can be used in one of two ways: As a TUI, or as a CLI for managing your Servarrs.
|
Managarr can be used in one of two ways: As a TUI, or as a CLI for managing your Servarrs.
|
||||||
|
|
||||||
All management features available in the TUI are also available in the CLI. However, the CLI is
|
All management features available in the TUI are also available in the CLI. However, the CLI is
|
||||||
equipped with additional features to allow for more advanced usage and automation.
|
equipped with additional features to allow for more advanced usage and automation.
|
||||||
|
|
||||||
The CLI can be helpful for automating tasks or for use in scripts. For example, you can use the CLI to trigger a search for a movie, or to add a movie to your library.
|
The CLI can be helpful for automating tasks or for use in scripts. For example, you can use the CLI to trigger a search for a movie, or to add a movie to your Radarr library.
|
||||||
|
|
||||||
To see all available commands, simply run `managarr --help`:
|
To see all available commands, simply run `managarr --help`:
|
||||||
|
|
||||||
```shell
|
```shell
|
||||||
$ managarr --help
|
$ managarr --help
|
||||||
managarr 0.1.5
|
managarr 0.5.1
|
||||||
Alex Clarke <alex.j.tusa@gmail.com>
|
Alex Clarke <alex.j.tusa@gmail.com>
|
||||||
|
|
||||||
A TUI and CLI to manage your Servarrs
|
A TUI and CLI to manage your Servarrs
|
||||||
@@ -122,43 +237,55 @@ Usage: managarr [OPTIONS] [COMMAND]
|
|||||||
|
|
||||||
Commands:
|
Commands:
|
||||||
radarr Commands for manging your Radarr instance
|
radarr Commands for manging your Radarr instance
|
||||||
|
sonarr Commands for manging your Sonarr instance
|
||||||
completions Generate shell completions for the Managarr CLI
|
completions Generate shell completions for the Managarr CLI
|
||||||
|
tail-logs Tail Managarr logs
|
||||||
help Print this message or the help of the given subcommand(s)
|
help Print this message or the help of the given subcommand(s)
|
||||||
|
|
||||||
Options:
|
Options:
|
||||||
--config <CONFIG> The Managarr configuration file to use
|
--disable-spinner Disable the spinner (can sometimes make parsing output challenging) [env: MANAGARR_DISABLE_SPINNER=]
|
||||||
-h, --help Print help
|
--config-file <CONFIG_FILE> The Managarr configuration file to use [env: MANAGARR_CONFIG_FILE=]
|
||||||
-V, --version Print version
|
--themes-file <THEMES_FILE> The Managarr themes file to use [env: MANAGARR_THEMES_FILE=]
|
||||||
|
--theme <THEME> The name of the Managarr theme to use [env: MANAGARR_THEME=]
|
||||||
|
--servarr-name <SERVARR_NAME> For multi-instance configurations, you need to specify the name of the instance configuration that you want to use.
|
||||||
|
This is useful when you have multiple instances of the same Servarr defined in your config file.
|
||||||
|
By default, if left empty, the first configured Servarr instance listed in the config file will be used.
|
||||||
|
-h, --help Print help
|
||||||
|
-V, --version Print version
|
||||||
```
|
```
|
||||||
|
|
||||||
All subcommands also have detailed help menus to show you how to use them. For example, to see all available commands for Radarr, you would run:
|
All subcommands also have detailed help menus to show you how to use them. For example, to see all available commands for Sonarr, you would run:
|
||||||
|
|
||||||
```shell
|
```shell
|
||||||
$ managarr radarr --help
|
$ managarr sonarr --help
|
||||||
Commands for manging your Radarr instance
|
Commands for manging your Sonarr instance
|
||||||
|
|
||||||
Usage: managarr radarr [OPTIONS] <COMMAND>
|
Usage: managarr sonarr [OPTIONS] <COMMAND>
|
||||||
|
|
||||||
Commands:
|
Commands:
|
||||||
add Commands to add or create new resources within your Radarr instance
|
add Commands to add or create new resources within your Sonarr instance
|
||||||
delete Commands to delete resources from your Radarr instance
|
delete Commands to delete resources from your Sonarr instance
|
||||||
edit Commands to edit resources in your Radarr instance
|
edit Commands to edit resources in your Sonarr instance
|
||||||
get Commands to fetch details of the resources in your Radarr instance
|
get Commands to fetch details of the resources in your Sonarr instance
|
||||||
list Commands to list attributes from your Radarr instance
|
download Commands to download releases in your Sonarr instance
|
||||||
refresh Commands to refresh the data in your Radarr instance
|
list Commands to list attributes from your Sonarr instance
|
||||||
clear-blocklist Clear the blocklist
|
refresh Commands to refresh the data in your Sonarr instance
|
||||||
download-release Manually download the given release for the specified movie ID
|
manual-search Commands to manually search for releases
|
||||||
manual-search Trigger a manual search of releases for the movie with the given ID
|
trigger-automatic-search Commands to trigger automatic searches for releases of different resources in your Sonarr instance
|
||||||
search-new-movie Search for a new film to add to Radarr
|
clear-blocklist Clear the blocklist
|
||||||
start-task Start the specified Radarr task
|
mark-history-item-as-failed Mark the Sonarr history item with the given ID as 'failed'
|
||||||
test-indexer Test the indexer with the given ID. Note that a successful test returns an empty JSON body; i.e. '{}'
|
search-new-series Search for a new series to add to Sonarr
|
||||||
test-all-indexers Test all indexers
|
start-task Start the specified Sonarr task
|
||||||
trigger-automatic-search Trigger an automatic search for the movie with the specified ID
|
test-indexer Test the indexer with the given ID. Note that a successful test returns an empty JSON body; i.e. '{}'
|
||||||
help Print this message or the help of the given subcommand(s)
|
test-all-indexers Test all Sonarr indexers
|
||||||
|
toggle-episode-monitoring Toggle monitoring for the specified episode
|
||||||
|
toggle-season-monitoring Toggle monitoring for the specified season that corresponds to the specified series ID
|
||||||
|
help Print this message or the help of the given subcommand(s)
|
||||||
|
|
||||||
Options:
|
Options:
|
||||||
--config <CONFIG> The Managarr configuration file to use
|
--disable-spinner Disable the spinner (can sometimes make parsing output challenging) [env: MANAGARR_DISABLE_SPINNER=]
|
||||||
-h, --help Print help
|
--config <CONFIG> The Managarr configuration file to use [env: MANAGARR_CONFIG_FILE=]
|
||||||
|
-h, --help Print help
|
||||||
```
|
```
|
||||||
|
|
||||||
**Pro Tip:** The CLI is even more powerful and useful when used in conjunction with the `jq` CLI tool. This allows you to parse the JSON response from the Managarr CLI and use it in your scripts; For example, to extract the `movieId` of the movie "Ad Astra", you would run:
|
**Pro Tip:** The CLI is even more powerful and useful when used in conjunction with the `jq` CLI tool. This allows you to parse the JSON response from the Managarr CLI and use it in your scripts; For example, to extract the `movieId` of the movie "Ad Astra", you would run:
|
||||||
@@ -172,7 +299,7 @@ $ managarr radarr list movies | jq '.[] | select(.title == "Ad Astra") | .id'
|
|||||||
Managarr assumes reasonable defaults to connect to each service (i.e. Radarr is on localhost:7878),
|
Managarr assumes reasonable defaults to connect to each service (i.e. Radarr is on localhost:7878),
|
||||||
but all servers will require you to input the API token.
|
but all servers will require you to input the API token.
|
||||||
|
|
||||||
The configuration file is located somewhere different for each OS
|
The configuration file is located somewhere different for each OS.
|
||||||
|
|
||||||
### Linux
|
### Linux
|
||||||
```
|
```
|
||||||
@@ -195,73 +322,127 @@ where you may have more than one instance of a given Servarr running. Thus, you
|
|||||||
config file using the `--config` flag:
|
config file using the `--config` flag:
|
||||||
|
|
||||||
```shell
|
```shell
|
||||||
managarr --config /path/to/config.yml
|
managarr --config-file /path/to/config.yml
|
||||||
```
|
```
|
||||||
|
|
||||||
### Example Configuration:
|
### Example Configuration:
|
||||||
```yaml
|
```yaml
|
||||||
|
theme: default
|
||||||
radarr:
|
radarr:
|
||||||
host: 127.0.0.1
|
- host: 192.168.0.78
|
||||||
port: 7878
|
port: 7878
|
||||||
api_token: someApiToken1234567890
|
api_token: someApiToken1234567890
|
||||||
use_ssl: true
|
ssl_cert_path: /path/to/radarr.crt # Required to enable SSL
|
||||||
ssl_cert_path: /path/to/radarr.crt
|
|
||||||
sonarr:
|
sonarr:
|
||||||
host: 127.0.0.1
|
- uri: http://htpc.local/sonarr # Example of using the 'uri' key instead of 'host' and 'port'
|
||||||
port: 8989
|
api_token: someApiToken1234567890
|
||||||
api_token: someApiToken1234567890
|
|
||||||
|
- name: Anime Sonarr # An example of a custom name for a secondary Sonarr instance
|
||||||
|
host: 192.168.0.89
|
||||||
|
port: 8989
|
||||||
|
api_token: someApiToken1234567890
|
||||||
readarr:
|
readarr:
|
||||||
host: 127.0.0.1
|
- host: 192.168.0.87
|
||||||
port: 8787
|
port: 8787
|
||||||
api_token: someApiToken1234567890
|
api_token_file: /root/.config/readarr_api_token # Example of loading the API token from a file instead of hardcoding it in the configuration file
|
||||||
use_ssl: false
|
|
||||||
lidarr:
|
lidarr:
|
||||||
host: 127.0.0.1
|
- host: 192.168.0.86
|
||||||
port: 8686
|
port: 8686
|
||||||
api_token: someApiToken1234567890
|
api_token: ${MY_LIDARR_API_TOKEN} # Example of configuring using environment variables
|
||||||
use_ssl: false
|
|
||||||
whisparr:
|
whisparr:
|
||||||
host: 127.0.0.1
|
- host: 192.168.0.69
|
||||||
port: 6969
|
port: 6969
|
||||||
api_token: someApiToken1234567890
|
api_token: someApiToken1234567890
|
||||||
use_ssl: false
|
ssl_cert_path: /path/to/whisparr.crt
|
||||||
bazarr:
|
bazarr:
|
||||||
host: 127.0.0.1
|
- host: 192.168.0.67
|
||||||
port: 6767
|
port: 6767
|
||||||
api_token: someApiToken1234567890
|
api_token: someApiToken1234567890
|
||||||
use_ssl: false
|
|
||||||
prowlarr:
|
prowlarr:
|
||||||
host: 127.0.0.1
|
- host: 192.168.0.96
|
||||||
port: 9696
|
port: 9696
|
||||||
api_token: someApiToken1234567890
|
api_token: someApiToken1234567890
|
||||||
use_ssl: false
|
|
||||||
tautulli:
|
tautulli:
|
||||||
host: 127.0.0.1
|
- host: 192.168.0.81
|
||||||
port: 8181
|
port: 8181
|
||||||
api_token: someApiToken1234567890
|
api_token: someApiToken1234567890
|
||||||
use_ssl: false
|
```
|
||||||
|
|
||||||
|
### Example Multi-Instance Configuration:
|
||||||
|
```yaml
|
||||||
|
theme: default
|
||||||
|
radarr:
|
||||||
|
- host: 192.168.0.78 # No name specified, so this instance's name will default to 'Radarr 1'
|
||||||
|
port: 7878
|
||||||
|
api_token: someApiToken1234567890
|
||||||
|
ssl_cert_path: /path/to/radarr.crt
|
||||||
|
|
||||||
|
- name: International Movies
|
||||||
|
host: 192.168.0.79
|
||||||
|
port: 7878
|
||||||
|
api_token: someApiToken1234567890
|
||||||
|
sonarr:
|
||||||
|
- name: Anime
|
||||||
|
weight: 1 # This instance will be the first tab in the TUI
|
||||||
|
uri: http://htpc.local/sonarr
|
||||||
|
api_token: someApiToken1234567890
|
||||||
|
|
||||||
|
- name: TV Shows
|
||||||
|
weight: 2 # This instance will be the second tab in the TUI
|
||||||
|
host: 192.168.0.89
|
||||||
|
port: 8989
|
||||||
|
api_token: someApiToken1234567890
|
||||||
|
```
|
||||||
|
|
||||||
|
In this configuration, you can see that we have multiple instances of Radarr and Sonarr configured. The `weight` key is
|
||||||
|
used to specify the order in which the tabs will appear in the TUI. The lower the weight, the further to the left the
|
||||||
|
tab will appear. If no weight is specified, then tabs will be ordered in the order they appear in the configuration
|
||||||
|
file.
|
||||||
|
|
||||||
|
When no `name` is specified for a Servarr instance, the name will default to the name of the Servarr with a number
|
||||||
|
appended to it. For example, if you have two Radarr instances and neither has a name, they will be named `Radarr 1` and
|
||||||
|
`Radarr 2`, respectively.
|
||||||
|
|
||||||
|
In this example configuration, the tabs in the TUI would appear as follows:
|
||||||
|
|
||||||
|
`Anime | TV Shows | Radarr 1 | International Movies`
|
||||||
|
|
||||||
|
### Specify Which Servarr Instance to Use in the CLI
|
||||||
|
If you have multiple instances of the same Servarr running, you can specify which instance you want to use by using the `--servarr-name` flag:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
managarr --servarr-name "International Movies"
|
||||||
```
|
```
|
||||||
|
|
||||||
## Environment Variables
|
## Environment Variables
|
||||||
Managarr supports using environment variables on startup so you don't have to always specify certain flags:
|
Managarr supports using environment variables on startup so you don't have to always specify certain flags:
|
||||||
|
|
||||||
| Variable | Description | Equivalent Flag |
|
| Variable | Description | Equivalent Flag |
|
||||||
| --------------------------------------- | -------------------------------- | -------------------------------- |
|
|-----------------------------------------|--------------------------------------------------------------------------------|----------------------------------|
|
||||||
| `MANAGARR_CONFIG_FILE` | Set the path to the config file | `--config` |
|
| `MANAGARR_CONFIG_FILE` | Set the path to the config file | `--config` |
|
||||||
|
| `MANAGARR_DISABLE_SPINNER` | Disable the CLI spinner (this can be useful when scripting and parsing output) | `--disable-spinner` |
|
||||||
|
|
||||||
## Track My Progress for the Beta release (With Sonarr Support!)
|
## Track What I'm Currently Working On
|
||||||
Progress for the beta release can be followed on my [Wekan Board](https://wekan.alexjclarke.com/b/dHoGjBb44MHM9HSv4/managarr)
|
To see what feature(s) I'm currently working on, check out my [Wekan Board](https://wekan.alexjclarke.com/b/dHoGjBb44MHM9HSv4/managarr).
|
||||||
with all items tagged `Beta`.
|
|
||||||
|
|
||||||
## Screenshots
|
## Screenshots
|
||||||
|
|
||||||

|
### Radarr
|
||||||

|

|
||||||

|

|
||||||

|

|
||||||

|

|
||||||

|

|
||||||

|
|
||||||
|
### Sonarr
|
||||||
|

|
||||||
|

|
||||||
|

|
||||||
|

|
||||||
|
|
||||||
|
### General
|
||||||
|

|
||||||
|

|
||||||
|
|
||||||
## Dependencies
|
## Dependencies
|
||||||
* [ratatui](https://github.com/tui-rs-revival/ratatui)
|
* [ratatui](https://github.com/tui-rs-revival/ratatui)
|
||||||
@@ -273,7 +454,7 @@ with all items tagged `Beta`.
|
|||||||
|
|
||||||
## Servarr Requirements
|
## Servarr Requirements
|
||||||
* [Radarr >= 5.3.6.8612](https://radarr.video/docs/api/)
|
* [Radarr >= 5.3.6.8612](https://radarr.video/docs/api/)
|
||||||
* [Sonarr >= v3](https://sonarr.tv/docs/api/)
|
* [Sonarr >= v4](https://sonarr.tv/docs/api/)
|
||||||
* [Readarr v1](https://readarr.com/docs/api/)
|
* [Readarr v1](https://readarr.com/docs/api/)
|
||||||
* [Lidarr v1](https://lidarr.audio/docs/api/)
|
* [Lidarr v1](https://lidarr.audio/docs/api/)
|
||||||
* [Whisparr >= v3](https://whisparr.com/docs/api/)
|
* [Whisparr >= v3](https://whisparr.com/docs/api/)
|
||||||
|
|||||||
@@ -1,5 +1,14 @@
|
|||||||
coverage:
|
coverage:
|
||||||
range: "80..100"
|
range: "80..100"
|
||||||
|
status:
|
||||||
|
project:
|
||||||
|
default:
|
||||||
|
threshold: 0
|
||||||
|
target: 80%
|
||||||
|
patch:
|
||||||
|
default:
|
||||||
|
threshold: 0
|
||||||
|
target: 80%
|
||||||
|
|
||||||
ignore:
|
ignore:
|
||||||
- "**/*_tests.rs"
|
- "**/*_tests.rs"
|
||||||
|
|||||||
@@ -0,0 +1,20 @@
|
|||||||
|
$ErrorActionPreference = 'Stop';
|
||||||
|
|
||||||
|
$PackageName = 'managarr'
|
||||||
|
$toolsDir = "$(Split-Path -parent $MyInvocation.MyCommand.Definition)"
|
||||||
|
$url64 = 'https://github.com/Dark-Alex-17/managarr/releases/download/v$version/managarr-windows.tar.gz'
|
||||||
|
$checksum64 = '$hash_64'
|
||||||
|
|
||||||
|
$packageArgs = @{
|
||||||
|
packageName = $packageName
|
||||||
|
softwareName = $packageName
|
||||||
|
unzipLocation = $toolsDir
|
||||||
|
fileType = 'exe'
|
||||||
|
url = $url64
|
||||||
|
checksum = $checksum64
|
||||||
|
checksumType = 'sha256'
|
||||||
|
|
||||||
|
}
|
||||||
|
Install-ChocolateyZipPackage @packageArgs
|
||||||
|
$File = Get-ChildItem -File -Path $env:ChocolateyInstall\lib\$packageName\tools\ -Filter *.tar
|
||||||
|
Get-ChocolateyUnzip -fileFullPath $File.FullName -destination $env:ChocolateyInstall\lib\$packageName\tools\
|
||||||
@@ -0,0 +1,53 @@
|
|||||||
|
<?xml version="1.0" encoding="utf-8"?>
|
||||||
|
<!-- Read this before creating packages: https://chocolatey.org/docs/create-packages -->
|
||||||
|
<!-- It is especially important to read the above link to understand additional requirements when publishing packages to the community feed aka dot org (https://chocolatey.org/packages). -->
|
||||||
|
|
||||||
|
<!-- Test your packages in a test environment: https://github.com/chocolatey/chocolatey-test-environment -->
|
||||||
|
|
||||||
|
<!--
|
||||||
|
This is a nuspec. It mostly adheres to https://docs.nuget.org/create/Nuspec-Reference. Chocolatey uses a special version of NuGet.Core that allows us to do more than was initially possible. As such there are certain things to be aware of:
|
||||||
|
|
||||||
|
* the package xmlns schema url may cause issues with nuget.exe
|
||||||
|
* Any of the following elements can ONLY be used by choco tools - projectSourceUrl, docsUrl, mailingListUrl, bugTrackerUrl, packageSourceUrl, provides, conflicts, replaces
|
||||||
|
* nuget.exe can still install packages with those elements but they are ignored. Any authoring tools or commands will error on those elements
|
||||||
|
-->
|
||||||
|
|
||||||
|
<!-- You can embed software files directly into packages, as long as you are not bound by distribution rights. -->
|
||||||
|
<!-- * If you are an organization making private packages, you probably have no issues here -->
|
||||||
|
<!-- * If you are releasing to the community feed, you need to consider distribution rights. -->
|
||||||
|
<!-- Do not remove this test for UTF-8: if “Ω” doesn’t appear as greek uppercase omega letter enclosed in quotation marks, you should use an editor that supports UTF-8, not this one. -->
|
||||||
|
<package xmlns="http://schemas.microsoft.com/packaging/2015/06/nuspec.xsd">
|
||||||
|
<metadata>
|
||||||
|
<!-- == PACKAGE SPECIFIC SECTION == -->
|
||||||
|
<id>managarr</id>
|
||||||
|
<version>$version</version>
|
||||||
|
|
||||||
|
<!-- == SOFTWARE SPECIFIC SECTION == -->
|
||||||
|
<!-- This section is about the software itself -->
|
||||||
|
<title>Managarr</title>
|
||||||
|
<authors>Alex Clarke</authors>
|
||||||
|
<projectUrl>https://github.com/Dark-Alex-17/managarr</projectUrl>
|
||||||
|
<licenseUrl>https://github.com/Dark-Alex-17/managarr/blob/main/LICENSE</licenseUrl>
|
||||||
|
<requireLicenseAcceptance>true</requireLicenseAcceptance>
|
||||||
|
<projectSourceUrl>https://github.com/Dark-Alex-17/managarr</projectSourceUrl>
|
||||||
|
<docsUrl>https://github.com/Dark-Alex-17/managarr/blob/main/README.md</docsUrl>
|
||||||
|
<bugTrackerUrl>https://github.com/Dark-Alex-17/managarr/issues</bugTrackerUrl>
|
||||||
|
<tags>cli cross-platform terminal servarr tui sonarr radarr rust</tags>
|
||||||
|
<summary>A TUI and CLI for managing *arr servers.</summary>
|
||||||
|
<description>
|
||||||
|
A TUI and CLI for managing *arr servers. Built with love in Rust!
|
||||||
|
|
||||||
|
**Usage**
|
||||||
|
To use, run `managarr` in a terminal.
|
||||||
|
|
||||||
|
For more [documentation and usage](https://github.com/Dark-Alex-17/managarr/blob/main/README.md), see the [official repo](https://github.com/Dark-Alex-17/managarr).
|
||||||
|
|
||||||
|
</description>
|
||||||
|
<releaseNotes>https://github.com/Dark-Alex-17/managarr/releases/tag/v$version/</releaseNotes>
|
||||||
|
</metadata>
|
||||||
|
<files>
|
||||||
|
<!-- this section controls what actually gets packaged into the Chocolatey package -->
|
||||||
|
<file src="tools\**" target="tools" />
|
||||||
|
<!--Building from Linux? You may need this instead: <file src="tools/**" target="tools" />-->
|
||||||
|
</files>
|
||||||
|
</package>
|
||||||
@@ -0,0 +1,28 @@
|
|||||||
|
import hashlib
|
||||||
|
import sys
|
||||||
|
from string import Template
|
||||||
|
|
||||||
|
sys.stdout.reconfigure(encoding='utf-8')
|
||||||
|
args = sys.argv
|
||||||
|
version = args[1].replace("v", "")
|
||||||
|
template_file_path = args[2]
|
||||||
|
generated_file_path = args[3]
|
||||||
|
|
||||||
|
# Deployment files
|
||||||
|
hash_64 = args[4].strip()
|
||||||
|
|
||||||
|
print("Generating formula")
|
||||||
|
print(" VERSION: %s" % version)
|
||||||
|
print(" TEMPLATE PATH: %s" % template_file_path)
|
||||||
|
print(" SAVING AT: %s" % generated_file_path)
|
||||||
|
print(" HASH: %s" % hash_64)
|
||||||
|
|
||||||
|
with open(template_file_path, "r", encoding="utf-8") as template_file:
|
||||||
|
template = Template(template_file.read())
|
||||||
|
substitute = template.safe_substitute(version=version, hash_64=hash_64)
|
||||||
|
print("\n================== Generated package file ==================\n")
|
||||||
|
print(substitute)
|
||||||
|
print("\n============================================================\n")
|
||||||
|
|
||||||
|
with open(generated_file_path, "w", encoding="utf-8") as generated_file:
|
||||||
|
generated_file.write(substitute)
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
# Documentation: https://docs.brew.sh/Formula-Cookbook
|
||||||
|
# https://rubydoc.brew.sh/Formula
|
||||||
|
class Managarr < Formula
|
||||||
|
desc "A fast and simple dashboard for Kubernetes written in Rust"
|
||||||
|
homepage "https://github.com/Dark-Alex-17/managarr"
|
||||||
|
if OS.mac? and Hardware::CPU.arm?
|
||||||
|
url "https://github.com/Dark-Alex-17/managarr/releases/download/v$version/managarr-macos-arm64.tar.gz"
|
||||||
|
sha256 "$hash_mac_arm"
|
||||||
|
elsif OS.mac? and Hardware::CPU.intel?
|
||||||
|
url "https://github.com/Dark-Alex-17/managarr/releases/download/v$version/managarr-macos.tar.gz"
|
||||||
|
sha256 "$hash_mac"
|
||||||
|
else
|
||||||
|
url "https://github.com/Dark-Alex-17/managarr/releases/download/v$version/managarr-linux-musl.tar.gz"
|
||||||
|
sha256 "$hash_linux"
|
||||||
|
end
|
||||||
|
version "$version"
|
||||||
|
license "MIT"
|
||||||
|
|
||||||
|
def install
|
||||||
|
bin.install "managarr"
|
||||||
|
ohai "You're done! Run with \"managarr\""
|
||||||
|
ohai "For runtime flags, see \"managarr --help\""
|
||||||
|
end
|
||||||
|
end
|
||||||
@@ -0,0 +1,31 @@
|
|||||||
|
import hashlib
|
||||||
|
import sys
|
||||||
|
from string import Template
|
||||||
|
|
||||||
|
args = sys.argv
|
||||||
|
version = args[1]
|
||||||
|
template_file_path = args[2]
|
||||||
|
generated_file_path = args[3]
|
||||||
|
|
||||||
|
# Deployment files
|
||||||
|
hash_mac = args[4].strip()
|
||||||
|
hash_mac_arm = args[5].strip()
|
||||||
|
hash_linux = args[6].strip()
|
||||||
|
|
||||||
|
print("Generating formula")
|
||||||
|
print(" VERSION: %s" % version)
|
||||||
|
print(" TEMPLATE PATH: %s" % template_file_path)
|
||||||
|
print(" SAVING AT: %s" % generated_file_path)
|
||||||
|
print(" MAC HASH: %s" % hash_mac)
|
||||||
|
print(" MAC ARM HASH: %s" % hash_mac_arm)
|
||||||
|
print(" LINUX HASH: %s" % hash_linux)
|
||||||
|
|
||||||
|
with open(template_file_path, "r") as template_file:
|
||||||
|
template = Template(template_file.read())
|
||||||
|
substitute = template.safe_substitute(version=version, hash_mac=hash_mac, hash_mac_arm=hash_mac_arm, hash_linux=hash_linux)
|
||||||
|
print("\n================== Generated package file ==================\n")
|
||||||
|
print(substitute)
|
||||||
|
print("\n============================================================\n")
|
||||||
|
|
||||||
|
with open(generated_file_path, "w") as generated_file:
|
||||||
|
generated_file.write(substitute)
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
{
|
||||||
|
"name": "managarr",
|
||||||
|
"lockfileVersion": 3,
|
||||||
|
"requires": true,
|
||||||
|
"packages": {}
|
||||||
|
}
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
{}
|
||||||
@@ -0,0 +1,12 @@
|
|||||||
|
[package]
|
||||||
|
name = "enum_display_style_derive"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[lib]
|
||||||
|
proc-macro = true
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
quote = "1.0.39"
|
||||||
|
syn = "2.0.99"
|
||||||
|
darling = "0.20.10"
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
mod macro_models;
|
||||||
|
|
||||||
|
use crate::macro_models::DisplayStyleArgs;
|
||||||
|
use darling::FromVariant;
|
||||||
|
use quote::quote;
|
||||||
|
use syn::{Data, DeriveInput, parse_macro_input};
|
||||||
|
|
||||||
|
/// Derive macro for generating a `to_display_str` method for an enum.
|
||||||
|
///
|
||||||
|
/// # Example
|
||||||
|
///
|
||||||
|
/// Using default values for the display style:
|
||||||
|
///
|
||||||
|
/// ```
|
||||||
|
/// use enum_display_style_derive::EnumDisplayStyle;
|
||||||
|
///
|
||||||
|
/// #[derive(EnumDisplayStyle)]
|
||||||
|
/// enum Weekend {
|
||||||
|
/// Saturday,
|
||||||
|
/// Sunday,
|
||||||
|
/// }
|
||||||
|
///
|
||||||
|
/// assert_eq!(Weekend::Saturday.to_display_str(), "Saturday");
|
||||||
|
/// assert_eq!(Weekend::Sunday.to_display_str(), "Sunday");
|
||||||
|
/// ```
|
||||||
|
///
|
||||||
|
/// Using custom values for the display style:
|
||||||
|
///
|
||||||
|
/// ```
|
||||||
|
/// use enum_display_style_derive::EnumDisplayStyle;
|
||||||
|
///
|
||||||
|
/// #[derive(EnumDisplayStyle)]
|
||||||
|
/// enum MonitorStatus {
|
||||||
|
/// #[display_style(name = "Monitor Transactions")]
|
||||||
|
/// Active,
|
||||||
|
/// #[display_style(name = "Don't Monitor Transactions")]
|
||||||
|
/// None,
|
||||||
|
/// }
|
||||||
|
///
|
||||||
|
/// assert_eq!(MonitorStatus::Active.to_display_str(), "Monitor Transactions");
|
||||||
|
/// assert_eq!(MonitorStatus::None.to_display_str(), "Don't Monitor Transactions");
|
||||||
|
/// ```
|
||||||
|
#[proc_macro_derive(EnumDisplayStyle, attributes(display_style))]
|
||||||
|
pub fn enum_display_style_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
|
||||||
|
let input = parse_macro_input!(input as DeriveInput);
|
||||||
|
let enum_name = &input.ident;
|
||||||
|
|
||||||
|
let mut match_arms = Vec::new();
|
||||||
|
|
||||||
|
if let Data::Enum(data_enum) = &input.data {
|
||||||
|
let variants = &data_enum.variants;
|
||||||
|
|
||||||
|
for variant in variants {
|
||||||
|
let variant_ident = &variant.ident;
|
||||||
|
let variant_display_name = DisplayStyleArgs::from_variant(variant)
|
||||||
|
.unwrap()
|
||||||
|
.name
|
||||||
|
.unwrap_or_else(|| variant_ident.to_string());
|
||||||
|
|
||||||
|
match_arms.push(quote! {
|
||||||
|
#enum_name::#variant_ident => #variant_display_name,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
quote! {
|
||||||
|
impl<'a> #enum_name {
|
||||||
|
pub fn to_display_str(self) -> &'a str {
|
||||||
|
match self {
|
||||||
|
#(#match_arms)*
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.into()
|
||||||
|
}
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
use darling::FromVariant;
|
||||||
|
|
||||||
|
#[derive(Debug, FromVariant)]
|
||||||
|
#[darling(attributes(display_style))]
|
||||||
|
pub struct DisplayStyleArgs {
|
||||||
|
pub name: Option<String>,
|
||||||
|
}
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
[package]
|
||||||
|
name = "validate_theme_derive"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[lib]
|
||||||
|
proc-macro = true
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
quote = "1.0.39"
|
||||||
|
syn = "2.0.99"
|
||||||
|
|
||||||
|
[dev-dependencies]
|
||||||
|
log = "0.4.17"
|
||||||
@@ -0,0 +1,106 @@
|
|||||||
|
use proc_macro::TokenStream;
|
||||||
|
use quote::quote;
|
||||||
|
use syn::{Data, DeriveInput, Fields, parse_macro_input};
|
||||||
|
|
||||||
|
/// Derive macro for generating a `validate` method for a Theme struct.
|
||||||
|
/// The `validate` method ensures that all values with the `validate` attribute are not `None`.
|
||||||
|
/// Otherwise, an error message it output to both the log file and stdout and the program exits.
|
||||||
|
///
|
||||||
|
/// # Example
|
||||||
|
///
|
||||||
|
/// Valid themes pass through the program transitively without any messages being output.
|
||||||
|
///
|
||||||
|
/// ```
|
||||||
|
/// use validate_theme_derive::ValidateTheme;
|
||||||
|
///
|
||||||
|
/// #[derive(ValidateTheme, Default)]
|
||||||
|
/// struct Theme {
|
||||||
|
/// pub name: String,
|
||||||
|
/// #[validate]
|
||||||
|
/// pub good: Option<Style>,
|
||||||
|
/// #[validate]
|
||||||
|
/// pub bad: Option<Style>,
|
||||||
|
/// pub ugly: Option<Style>,
|
||||||
|
/// }
|
||||||
|
///
|
||||||
|
/// struct Style {
|
||||||
|
/// color: String,
|
||||||
|
/// }
|
||||||
|
///
|
||||||
|
/// let theme = Theme {
|
||||||
|
/// good: Some(Style { color: "Green".to_owned() }),
|
||||||
|
/// bad: Some(Style { color: "Red".to_owned() }),
|
||||||
|
/// ..Theme::default()
|
||||||
|
/// };
|
||||||
|
///
|
||||||
|
/// // Since only `good` and `bad` have the `validate` attribute, the `validate` method will only check those fields.
|
||||||
|
/// theme.validate();
|
||||||
|
/// // Since both `good` and `bad` have values, the program will not exit and no message is output.
|
||||||
|
/// ```
|
||||||
|
///
|
||||||
|
/// Invalid themes will output an error message to both the log file and stdout and the program will exit.
|
||||||
|
///
|
||||||
|
/// ```should_panic
|
||||||
|
/// use validate_theme_derive::ValidateTheme;
|
||||||
|
///
|
||||||
|
/// #[derive(ValidateTheme, Default)]
|
||||||
|
/// struct Theme {
|
||||||
|
/// pub name: String,
|
||||||
|
/// #[validate]
|
||||||
|
/// pub good: Option<Style>,
|
||||||
|
/// #[validate]
|
||||||
|
/// pub bad: Option<Style>,
|
||||||
|
/// pub ugly: Option<Style>,
|
||||||
|
/// }
|
||||||
|
///
|
||||||
|
/// struct Style {
|
||||||
|
/// color: String,
|
||||||
|
/// }
|
||||||
|
///
|
||||||
|
/// let theme = Theme {
|
||||||
|
/// bad: Some(Style { color: "Red".to_owned() }),
|
||||||
|
/// ..Theme::default()
|
||||||
|
/// };
|
||||||
|
///
|
||||||
|
/// // Since `good` has the `validate` attribute and since `good` is `None`, the `validate` method will output an error message and exit the program.
|
||||||
|
/// theme.validate();
|
||||||
|
/// ```
|
||||||
|
#[proc_macro_derive(ValidateTheme, attributes(validate))]
|
||||||
|
pub fn derive_validate_theme(input: TokenStream) -> TokenStream {
|
||||||
|
let input = parse_macro_input!(input as DeriveInput);
|
||||||
|
let struct_name = &input.ident;
|
||||||
|
|
||||||
|
let mut validation_checks = Vec::new();
|
||||||
|
|
||||||
|
if let Data::Struct(data_struct) = &input.data {
|
||||||
|
if let Fields::Named(fields) = &data_struct.fields {
|
||||||
|
for field in &fields.named {
|
||||||
|
let field_name = &field.ident;
|
||||||
|
|
||||||
|
let has_validate_attr = field
|
||||||
|
.attrs
|
||||||
|
.iter()
|
||||||
|
.any(|attr| attr.path().is_ident("validate"));
|
||||||
|
|
||||||
|
if has_validate_attr {
|
||||||
|
validation_checks.push(quote! {
|
||||||
|
if self.#field_name.is_none() {
|
||||||
|
log::error!("{} is missing a color value.", stringify!(#field_name));
|
||||||
|
eprintln!("{} is missing a color value.", stringify!(#field_name));
|
||||||
|
std::process::exit(1);
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
quote! {
|
||||||
|
impl #struct_name {
|
||||||
|
pub fn validate(&self) {
|
||||||
|
#(#validation_checks)*
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
.into()
|
||||||
|
}
|
||||||
|
Before Width: | Height: | Size: 382 KiB |
|
Before Width: | Height: | Size: 152 KiB After Width: | Height: | Size: 152 KiB |
|
Before Width: | Height: | Size: 154 KiB After Width: | Height: | Size: 154 KiB |
|
Before Width: | Height: | Size: 142 KiB After Width: | Height: | Size: 142 KiB |
|
Before Width: | Height: | Size: 374 KiB After Width: | Height: | Size: 374 KiB |
|
After Width: | Height: | Size: 340 KiB |
|
Before Width: | Height: | Size: 176 KiB After Width: | Height: | Size: 176 KiB |
|
Before Width: | Height: | Size: 236 KiB After Width: | Height: | Size: 236 KiB |
|
After Width: | Height: | Size: 241 KiB |
|
After Width: | Height: | Size: 305 KiB |
|
After Width: | Height: | Size: 196 KiB |
|
After Width: | Height: | Size: 124 KiB |
|
After Width: | Height: | Size: 220 KiB |
@@ -2,44 +2,96 @@
|
|||||||
mod tests {
|
mod tests {
|
||||||
use anyhow::anyhow;
|
use anyhow::anyhow;
|
||||||
use pretty_assertions::{assert_eq, assert_str_eq};
|
use pretty_assertions::{assert_eq, assert_str_eq};
|
||||||
|
use serial_test::serial;
|
||||||
use tokio::sync::mpsc;
|
use tokio::sync::mpsc;
|
||||||
|
|
||||||
use crate::app::context_clues::{build_context_clue_string, SERVARR_CONTEXT_CLUES};
|
use crate::app::context_clues::{build_context_clue_string, SERVARR_CONTEXT_CLUES};
|
||||||
use crate::app::{App, Data, RadarrConfig, DEFAULT_ROUTE};
|
use crate::app::{interpolate_env_vars, App, AppConfig, Data, ServarrConfig};
|
||||||
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, RadarrData};
|
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, RadarrData};
|
||||||
use crate::models::{HorizontallyScrollableText, Route, TabRoute};
|
use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, SonarrData};
|
||||||
|
use crate::models::{HorizontallyScrollableText, TabRoute};
|
||||||
use crate::network::radarr_network::RadarrEvent;
|
use crate::network::radarr_network::RadarrEvent;
|
||||||
use crate::network::NetworkEvent;
|
use crate::network::NetworkEvent;
|
||||||
|
use tokio_util::sync::CancellationToken;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_app_default() {
|
fn test_app_new() {
|
||||||
let app = App::default();
|
let radarr_config_1 = ServarrConfig {
|
||||||
|
name: Some("Radarr Test".to_owned()),
|
||||||
|
..ServarrConfig::default()
|
||||||
|
};
|
||||||
|
let radarr_config_2 = ServarrConfig {
|
||||||
|
weight: Some(3),
|
||||||
|
..ServarrConfig::default()
|
||||||
|
};
|
||||||
|
let sonarr_config_1 = ServarrConfig {
|
||||||
|
name: Some("Sonarr Test".to_owned()),
|
||||||
|
weight: Some(1),
|
||||||
|
..ServarrConfig::default()
|
||||||
|
};
|
||||||
|
let sonarr_config_2 = ServarrConfig::default();
|
||||||
|
let config = AppConfig {
|
||||||
|
theme: None,
|
||||||
|
radarr: Some(vec![radarr_config_1.clone(), radarr_config_2.clone()]),
|
||||||
|
sonarr: Some(vec![sonarr_config_1.clone(), sonarr_config_2.clone()]),
|
||||||
|
};
|
||||||
|
let expected_tab_routes = vec![
|
||||||
|
TabRoute {
|
||||||
|
title: "Sonarr Test".to_owned(),
|
||||||
|
route: ActiveSonarrBlock::default().into(),
|
||||||
|
help: format!(
|
||||||
|
"<↑↓> scroll | ←→ change tab | {} ",
|
||||||
|
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
||||||
|
),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(sonarr_config_1),
|
||||||
|
},
|
||||||
|
TabRoute {
|
||||||
|
title: "Radarr 1".to_owned(),
|
||||||
|
route: ActiveRadarrBlock::default().into(),
|
||||||
|
help: format!(
|
||||||
|
"<↑↓> scroll | ←→ change tab | {} ",
|
||||||
|
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
||||||
|
),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(radarr_config_2),
|
||||||
|
},
|
||||||
|
TabRoute {
|
||||||
|
title: "Radarr Test".to_owned(),
|
||||||
|
route: ActiveRadarrBlock::default().into(),
|
||||||
|
help: format!(
|
||||||
|
"<↑↓> scroll | ←→ change tab | {} ",
|
||||||
|
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
||||||
|
),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(radarr_config_1),
|
||||||
|
},
|
||||||
|
TabRoute {
|
||||||
|
title: "Sonarr 1".to_owned(),
|
||||||
|
route: ActiveSonarrBlock::default().into(),
|
||||||
|
help: format!(
|
||||||
|
"<↑↓> scroll | ←→ change tab | {} ",
|
||||||
|
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
||||||
|
),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(sonarr_config_2),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
assert_eq!(app.navigation_stack, vec![DEFAULT_ROUTE]);
|
let app = App::new(
|
||||||
assert!(app.network_tx.is_none());
|
mpsc::channel::<NetworkEvent>(500).0,
|
||||||
|
config,
|
||||||
|
CancellationToken::new(),
|
||||||
|
);
|
||||||
|
|
||||||
|
assert!(app.navigation_stack.is_empty());
|
||||||
|
assert_eq!(app.get_current_route(), ActiveSonarrBlock::default().into());
|
||||||
|
assert!(app.network_tx.is_some());
|
||||||
assert!(!app.cancellation_token.is_cancelled());
|
assert!(!app.cancellation_token.is_cancelled());
|
||||||
|
assert!(app.is_first_render);
|
||||||
assert_eq!(app.error, HorizontallyScrollableText::default());
|
assert_eq!(app.error, HorizontallyScrollableText::default());
|
||||||
assert_eq!(app.server_tabs.index, 0);
|
assert_eq!(app.server_tabs.index, 0);
|
||||||
assert_eq!(
|
assert_eq!(app.server_tabs.tabs, expected_tab_routes);
|
||||||
app.server_tabs.tabs,
|
|
||||||
vec![
|
|
||||||
TabRoute {
|
|
||||||
title: "Radarr",
|
|
||||||
route: ActiveRadarrBlock::Movies.into(),
|
|
||||||
help: format!(
|
|
||||||
"<↑↓> scroll | ←→ change tab | {} ",
|
|
||||||
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
|
||||||
),
|
|
||||||
contextual_help: None,
|
|
||||||
},
|
|
||||||
TabRoute {
|
|
||||||
title: "Sonarr",
|
|
||||||
route: Route::Sonarr,
|
|
||||||
help: format!("{} ", build_context_clue_string(&SERVARR_CONTEXT_CLUES)),
|
|
||||||
contextual_help: None,
|
|
||||||
},
|
|
||||||
]
|
|
||||||
);
|
|
||||||
assert_eq!(app.tick_until_poll, 400);
|
assert_eq!(app.tick_until_poll, 400);
|
||||||
assert_eq!(app.ticks_until_scroll, 4);
|
assert_eq!(app.ticks_until_scroll, 4);
|
||||||
assert_eq!(app.tick_count, 0);
|
assert_eq!(app.tick_count, 0);
|
||||||
@@ -47,20 +99,39 @@ mod tests {
|
|||||||
assert!(!app.is_routing);
|
assert!(!app.is_routing);
|
||||||
assert!(!app.should_refresh);
|
assert!(!app.should_refresh);
|
||||||
assert!(!app.should_ignore_quit_key);
|
assert!(!app.should_ignore_quit_key);
|
||||||
|
assert!(!app.cli_mode);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_app_default() {
|
||||||
|
let app = App::default();
|
||||||
|
|
||||||
|
assert!(app.navigation_stack.is_empty());
|
||||||
|
assert!(app.network_tx.is_none());
|
||||||
|
assert!(!app.cancellation_token.is_cancelled());
|
||||||
|
assert!(app.is_first_render);
|
||||||
|
assert_eq!(app.error, HorizontallyScrollableText::default());
|
||||||
|
assert_eq!(app.server_tabs.index, 0);
|
||||||
|
assert_eq!(app.tick_until_poll, 400);
|
||||||
|
assert_eq!(app.ticks_until_scroll, 4);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
assert!(!app.is_loading);
|
||||||
|
assert!(!app.is_routing);
|
||||||
|
assert!(!app.should_refresh);
|
||||||
|
assert!(!app.should_ignore_quit_key);
|
||||||
|
assert!(!app.cli_mode);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_navigation_stack_methods() {
|
fn test_navigation_stack_methods() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
|
let default_route = app.server_tabs.tabs.first().unwrap().route;
|
||||||
|
|
||||||
assert_eq!(app.get_current_route(), &DEFAULT_ROUTE);
|
assert_eq!(app.get_current_route(), default_route);
|
||||||
|
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Downloads.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Downloads.into());
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Downloads.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Downloads.into()
|
|
||||||
);
|
|
||||||
assert!(app.is_routing);
|
assert!(app.is_routing);
|
||||||
|
|
||||||
app.is_routing = false;
|
app.is_routing = false;
|
||||||
@@ -68,26 +139,30 @@ mod tests {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::Collections.into()
|
ActiveRadarrBlock::Collections.into()
|
||||||
);
|
);
|
||||||
assert!(app.is_routing);
|
assert!(app.is_routing);
|
||||||
|
|
||||||
app.is_routing = false;
|
app.is_routing = false;
|
||||||
app.pop_navigation_stack();
|
app.pop_navigation_stack();
|
||||||
|
|
||||||
assert_eq!(app.get_current_route(), &DEFAULT_ROUTE);
|
assert_eq!(app.get_current_route(), default_route);
|
||||||
assert!(app.is_routing);
|
assert!(app.is_routing);
|
||||||
|
|
||||||
app.is_routing = false;
|
app.is_routing = false;
|
||||||
app.pop_navigation_stack();
|
app.pop_navigation_stack();
|
||||||
|
|
||||||
assert_eq!(app.get_current_route(), &DEFAULT_ROUTE);
|
assert_eq!(app.get_current_route(), default_route);
|
||||||
assert!(app.is_routing);
|
assert!(app.is_routing);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_reset_cancellation_token() {
|
fn test_reset_cancellation_token() {
|
||||||
let mut app = App::default();
|
let mut app = App {
|
||||||
|
is_loading: true,
|
||||||
|
should_refresh: false,
|
||||||
|
..App::test_default()
|
||||||
|
};
|
||||||
app.cancellation_token.cancel();
|
app.cancellation_token.cancel();
|
||||||
|
|
||||||
assert!(app.cancellation_token.is_cancelled());
|
assert!(app.cancellation_token.is_cancelled());
|
||||||
@@ -96,13 +171,15 @@ mod tests {
|
|||||||
|
|
||||||
assert!(!app.cancellation_token.is_cancelled());
|
assert!(!app.cancellation_token.is_cancelled());
|
||||||
assert!(!new_token.is_cancelled());
|
assert!(!new_token.is_cancelled());
|
||||||
|
assert!(!app.is_loading);
|
||||||
|
assert!(app.should_refresh);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_reset_tick_count() {
|
fn test_reset_tick_count() {
|
||||||
let mut app = App {
|
let mut app = App {
|
||||||
tick_count: 2,
|
tick_count: 2,
|
||||||
..App::default()
|
..App::test_default()
|
||||||
};
|
};
|
||||||
|
|
||||||
app.reset_tick_count();
|
app.reset_tick_count();
|
||||||
@@ -112,28 +189,38 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_reset() {
|
fn test_reset() {
|
||||||
|
let radarr_data = RadarrData {
|
||||||
|
version: "test".into(),
|
||||||
|
..RadarrData::default()
|
||||||
|
};
|
||||||
|
let sonarr_data = SonarrData {
|
||||||
|
version: "test".into(),
|
||||||
|
..SonarrData::default()
|
||||||
|
};
|
||||||
|
let data = Data {
|
||||||
|
radarr_data,
|
||||||
|
sonarr_data,
|
||||||
|
};
|
||||||
let mut app = App {
|
let mut app = App {
|
||||||
tick_count: 2,
|
tick_count: 2,
|
||||||
error: "Test error".to_owned().into(),
|
error: "Test error".to_owned().into(),
|
||||||
data: Data {
|
is_first_render: false,
|
||||||
radarr_data: RadarrData {
|
data,
|
||||||
version: "test".to_owned(),
|
..App::test_default()
|
||||||
..RadarrData::default()
|
|
||||||
},
|
|
||||||
},
|
|
||||||
..App::default()
|
|
||||||
};
|
};
|
||||||
|
|
||||||
app.reset();
|
app.reset();
|
||||||
|
|
||||||
assert_eq!(app.tick_count, 0);
|
assert_eq!(app.tick_count, 0);
|
||||||
assert_eq!(app.error, HorizontallyScrollableText::default());
|
assert_eq!(app.error, HorizontallyScrollableText::default());
|
||||||
|
assert!(app.is_first_render);
|
||||||
assert!(app.data.radarr_data.version.is_empty());
|
assert!(app.data.radarr_data.version.is_empty());
|
||||||
|
assert!(app.data.sonarr_data.version.is_empty());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_handle_error() {
|
fn test_handle_error() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
let test_string = "Testing";
|
let test_string = "Testing";
|
||||||
|
|
||||||
app.handle_error(anyhow!(test_string));
|
app.handle_error(anyhow!(test_string));
|
||||||
@@ -145,6 +232,29 @@ mod tests {
|
|||||||
assert_eq!(app.error.text, test_string);
|
assert_eq!(app.error.text, test_string);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_network_event() {
|
||||||
|
let (sync_network_tx, mut sync_network_rx) = mpsc::channel::<NetworkEvent>(500);
|
||||||
|
|
||||||
|
let mut app = App {
|
||||||
|
tick_until_poll: 2,
|
||||||
|
network_tx: Some(sync_network_tx),
|
||||||
|
..App::test_default()
|
||||||
|
};
|
||||||
|
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_network_event(RadarrEvent::GetStatus.into())
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
RadarrEvent::GetStatus.into()
|
||||||
|
);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_on_tick_first_render() {
|
async fn test_on_tick_first_render() {
|
||||||
let (sync_network_tx, mut sync_network_rx) = mpsc::channel::<NetworkEvent>(500);
|
let (sync_network_tx, mut sync_network_rx) = mpsc::channel::<NetworkEvent>(500);
|
||||||
@@ -152,12 +262,14 @@ mod tests {
|
|||||||
let mut app = App {
|
let mut app = App {
|
||||||
tick_until_poll: 2,
|
tick_until_poll: 2,
|
||||||
network_tx: Some(sync_network_tx),
|
network_tx: Some(sync_network_tx),
|
||||||
..App::default()
|
is_first_render: true,
|
||||||
|
..App::test_default()
|
||||||
};
|
};
|
||||||
|
|
||||||
assert_eq!(app.tick_count, 0);
|
assert_eq!(app.tick_count, 0);
|
||||||
|
|
||||||
app.on_tick(true).await;
|
app.on_tick().await;
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
sync_network_rx.recv().await.unwrap(),
|
sync_network_rx.recv().await.unwrap(),
|
||||||
RadarrEvent::GetQualityProfiles.into()
|
RadarrEvent::GetQualityProfiles.into()
|
||||||
@@ -172,7 +284,11 @@ mod tests {
|
|||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
sync_network_rx.recv().await.unwrap(),
|
sync_network_rx.recv().await.unwrap(),
|
||||||
RadarrEvent::GetOverview.into()
|
RadarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
RadarrEvent::GetDiskSpace.into()
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
sync_network_rx.recv().await.unwrap(),
|
sync_network_rx.recv().await.unwrap(),
|
||||||
@@ -180,11 +296,15 @@ mod tests {
|
|||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
sync_network_rx.recv().await.unwrap(),
|
sync_network_rx.recv().await.unwrap(),
|
||||||
RadarrEvent::GetMovies.into()
|
RadarrEvent::GetQualityProfiles.into()
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
sync_network_rx.recv().await.unwrap(),
|
sync_network_rx.recv().await.unwrap(),
|
||||||
RadarrEvent::GetDownloads.into()
|
RadarrEvent::GetTags.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
RadarrEvent::GetMovies.into()
|
||||||
);
|
);
|
||||||
assert!(!app.is_routing);
|
assert!(!app.is_routing);
|
||||||
assert!(!app.should_refresh);
|
assert!(!app.should_refresh);
|
||||||
@@ -197,10 +317,10 @@ mod tests {
|
|||||||
tick_until_poll: 2,
|
tick_until_poll: 2,
|
||||||
tick_count: 2,
|
tick_count: 2,
|
||||||
is_routing: true,
|
is_routing: true,
|
||||||
..App::default()
|
..App::test_default()
|
||||||
};
|
};
|
||||||
|
|
||||||
app.on_tick(false).await;
|
app.on_tick().await;
|
||||||
assert!(!app.is_routing);
|
assert!(!app.is_routing);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -210,21 +330,202 @@ mod tests {
|
|||||||
tick_until_poll: 2,
|
tick_until_poll: 2,
|
||||||
tick_count: 2,
|
tick_count: 2,
|
||||||
should_refresh: true,
|
should_refresh: true,
|
||||||
..App::default()
|
..App::test_default()
|
||||||
};
|
};
|
||||||
|
|
||||||
app.on_tick(false).await;
|
app.on_tick().await;
|
||||||
assert!(!app.should_refresh);
|
assert!(!app.should_refresh);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_radarr_config_default() {
|
fn test_app_config_default() {
|
||||||
let radarr_config = RadarrConfig::default();
|
let app_config = AppConfig::default();
|
||||||
|
|
||||||
assert_str_eq!(radarr_config.host, "localhost");
|
assert!(app_config.radarr.is_none());
|
||||||
assert_eq!(radarr_config.port, Some(7878));
|
assert!(app_config.sonarr.is_none());
|
||||||
assert!(radarr_config.api_token.is_empty());
|
}
|
||||||
assert!(!radarr_config.use_ssl);
|
|
||||||
assert_eq!(radarr_config.ssl_cert_path, None);
|
#[test]
|
||||||
|
fn test_servarr_config_default() {
|
||||||
|
let servarr_config = ServarrConfig::default();
|
||||||
|
|
||||||
|
assert_eq!(servarr_config.name, None);
|
||||||
|
assert_eq!(servarr_config.host, Some("localhost".to_string()));
|
||||||
|
assert_eq!(servarr_config.port, None);
|
||||||
|
assert_eq!(servarr_config.uri, None);
|
||||||
|
assert_eq!(servarr_config.weight, None);
|
||||||
|
assert_eq!(servarr_config.api_token, Some(String::new()));
|
||||||
|
assert_eq!(servarr_config.api_token_file, None);
|
||||||
|
assert_eq!(servarr_config.ssl_cert_path, None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[serial]
|
||||||
|
fn test_deserialize_optional_env_var_is_present() {
|
||||||
|
unsafe { std::env::set_var("TEST_VAR_DESERIALIZE_OPTION", "localhost") };
|
||||||
|
let yaml_data = r#"
|
||||||
|
host: ${TEST_VAR_DESERIALIZE_OPTION}
|
||||||
|
api_token: "test123"
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let config: ServarrConfig = serde_yaml::from_str(yaml_data).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(config.host, Some("localhost".to_string()));
|
||||||
|
unsafe { std::env::remove_var("TEST_VAR_DESERIALIZE_OPTION") };
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[serial]
|
||||||
|
fn test_deserialize_optional_env_var_does_not_overwrite_non_env_value() {
|
||||||
|
unsafe { std::env::set_var("TEST_VAR_DESERIALIZE_OPTION_NO_OVERWRITE", "localhost") };
|
||||||
|
let yaml_data = r#"
|
||||||
|
host: www.example.com
|
||||||
|
api_token: "test123"
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let config: ServarrConfig = serde_yaml::from_str(yaml_data).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(config.host, Some("www.example.com".to_string()));
|
||||||
|
unsafe { std::env::remove_var("TEST_VAR_DESERIALIZE_OPTION_NO_OVERWRITE") };
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_deserialize_optional_env_var_empty() {
|
||||||
|
let yaml_data = r#"
|
||||||
|
api_token: "test123"
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let config: ServarrConfig = serde_yaml::from_str(yaml_data).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(config.port, None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[serial]
|
||||||
|
fn test_deserialize_optional_u16_env_var_is_present() {
|
||||||
|
unsafe { std::env::set_var("TEST_VAR_DESERIALIZE_OPTION_U16", "1") };
|
||||||
|
let yaml_data = r#"
|
||||||
|
port: ${TEST_VAR_DESERIALIZE_OPTION_U16}
|
||||||
|
api_token: "test123"
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let config: ServarrConfig = serde_yaml::from_str(yaml_data).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(config.port, Some(1));
|
||||||
|
unsafe { std::env::remove_var("TEST_VAR_DESERIALIZE_OPTION_U16") };
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[serial]
|
||||||
|
fn test_deserialize_optional_u16_env_var_does_not_overwrite_non_env_value() {
|
||||||
|
unsafe { std::env::set_var("TEST_VAR_DESERIALIZE_OPTION_U16_UNUSED", "1") };
|
||||||
|
let yaml_data = r#"
|
||||||
|
port: 1234
|
||||||
|
api_token: "test123"
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let config: ServarrConfig = serde_yaml::from_str(yaml_data).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(config.port, Some(1234));
|
||||||
|
unsafe { std::env::remove_var("TEST_VAR_DESERIALIZE_OPTION_U16_UNUSED") };
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_deserialize_optional_u16_env_var_invalid_number() {
|
||||||
|
let yaml_data = r#"
|
||||||
|
port: "hi"
|
||||||
|
api_token: "test123"
|
||||||
|
"#;
|
||||||
|
let result: Result<ServarrConfig, _> = serde_yaml::from_str(yaml_data);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
let err = result.unwrap_err().to_string();
|
||||||
|
assert!(err.contains("invalid digit found in string"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_deserialize_optional_u16_env_var_empty() {
|
||||||
|
let yaml_data = r#"
|
||||||
|
api_token: "test123"
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let config: ServarrConfig = serde_yaml::from_str(yaml_data).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(config.port, None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[serial]
|
||||||
|
fn test_interpolate_env_vars() {
|
||||||
|
unsafe { std::env::set_var("TEST_VAR_INTERPOLATION", "testing") };
|
||||||
|
|
||||||
|
let var = interpolate_env_vars("${TEST_VAR_INTERPOLATION}");
|
||||||
|
|
||||||
|
assert_str_eq!(var, "testing");
|
||||||
|
unsafe { std::env::remove_var("TEST_VAR_INTERPOLATION") };
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_interpolate_env_vars_defaults_to_original_string_if_not_in_yaml_interpolation_format() {
|
||||||
|
let var = interpolate_env_vars("TEST_VAR_INTERPOLATION_NON_YAML");
|
||||||
|
|
||||||
|
assert_str_eq!(var, "TEST_VAR_INTERPOLATION_NON_YAML");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
#[serial]
|
||||||
|
fn test_interpolate_env_vars_scrubs_all_unnecessary_characters() {
|
||||||
|
unsafe {
|
||||||
|
std::env::set_var(
|
||||||
|
"TEST_VAR_INTERPOLATION_UNNECESSARY_CHARACTERS",
|
||||||
|
r#"""
|
||||||
|
`"'https://dontdo:this@testing.com/query?test=%20query#results'"` {([\|$!])}
|
||||||
|
"""#,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
|
||||||
|
let var = interpolate_env_vars("${TEST_VAR_INTERPOLATION_UNNECESSARY_CHARACTERS}");
|
||||||
|
|
||||||
|
assert_str_eq!(
|
||||||
|
var,
|
||||||
|
"https://dontdo:this@testing.com/query?test=%20query#results"
|
||||||
|
);
|
||||||
|
unsafe { std::env::remove_var("TEST_VAR_INTERPOLATION_UNNECESSARY_CHARACTERS") };
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_interpolate_env_vars_scrubs_all_unnecessary_characters_from_non_environment_variable() {
|
||||||
|
let var = interpolate_env_vars("https://dontdo:this@testing.com/query?test=%20query#results");
|
||||||
|
|
||||||
|
assert_str_eq!(
|
||||||
|
var,
|
||||||
|
"https://dontdo:this@testing.com/query?test=%20query#results"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_servarr_config_redacted_debug() {
|
||||||
|
let name = "Servarr".to_owned();
|
||||||
|
let host = "localhost".to_owned();
|
||||||
|
let port = 1234;
|
||||||
|
let uri = "http://localhost:1234".to_owned();
|
||||||
|
let weight = 100;
|
||||||
|
let api_token = "thisisatest".to_owned();
|
||||||
|
let api_token_file = "/root/.config/api_token".to_owned();
|
||||||
|
let ssl_cert_path = "/some/path".to_owned();
|
||||||
|
let expected_str = format!("ServarrConfig {{ name: Some(\"{}\"), host: Some(\"{}\"), port: Some({}), uri: Some(\"{}\"), weight: Some({}), api_token: Some(\"***********\"), api_token_file: Some(\"{}\"), ssl_cert_path: Some(\"{}\") }}",
|
||||||
|
name, host, port, uri, weight, api_token_file, ssl_cert_path);
|
||||||
|
let servarr_config = ServarrConfig {
|
||||||
|
name: Some(name),
|
||||||
|
host: Some(host),
|
||||||
|
port: Some(port),
|
||||||
|
uri: Some(uri),
|
||||||
|
weight: Some(weight),
|
||||||
|
api_token: Some(api_token),
|
||||||
|
api_token_file: Some(api_token_file),
|
||||||
|
ssl_cert_path: Some(ssl_cert_path),
|
||||||
|
};
|
||||||
|
|
||||||
|
assert_str_eq!(format!("{servarr_config:?}"), expected_str);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,10 +14,77 @@ pub fn build_context_clue_string(context_clues: &[(KeyBinding, &str)]) -> String
|
|||||||
.join(" | ")
|
.join(" | ")
|
||||||
}
|
}
|
||||||
|
|
||||||
pub static SERVARR_CONTEXT_CLUES: [ContextClue; 2] = [
|
pub static SERVARR_CONTEXT_CLUES: [ContextClue; 3] = [
|
||||||
(DEFAULT_KEYBINDINGS.tab, "change servarr"),
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.next_servarr,
|
||||||
|
DEFAULT_KEYBINDINGS.next_servarr.desc,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.previous_servarr,
|
||||||
|
DEFAULT_KEYBINDINGS.previous_servarr.desc,
|
||||||
|
),
|
||||||
(DEFAULT_KEYBINDINGS.quit, DEFAULT_KEYBINDINGS.quit.desc),
|
(DEFAULT_KEYBINDINGS.quit, DEFAULT_KEYBINDINGS.quit.desc),
|
||||||
];
|
];
|
||||||
|
|
||||||
pub static BARE_POPUP_CONTEXT_CLUES: [ContextClue; 1] =
|
pub static BARE_POPUP_CONTEXT_CLUES: [ContextClue; 1] =
|
||||||
[(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc)];
|
[(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc)];
|
||||||
|
|
||||||
|
pub static BLOCKLIST_CONTEXT_CLUES: [ContextClue; 5] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "details"),
|
||||||
|
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.clear, "clear blocklist"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static CONFIRMATION_PROMPT_CONTEXT_CLUES: [ContextClue; 2] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.confirm, "submit"),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, "cancel"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static DOWNLOADS_CONTEXT_CLUES: [ContextClue; 3] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.update, "update downloads"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static ROOT_FOLDERS_CONTEXT_CLUES: [ContextClue; 3] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.add, DEFAULT_KEYBINDINGS.add.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static INDEXERS_CONTEXT_CLUES: [ContextClue; 6] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "edit indexer"),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.settings,
|
||||||
|
DEFAULT_KEYBINDINGS.settings.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.test, "test indexer"),
|
||||||
|
(DEFAULT_KEYBINDINGS.test_all, "test all indexers"),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SYSTEM_CONTEXT_CLUES: [ContextClue; 5] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.tasks, "open tasks"),
|
||||||
|
(DEFAULT_KEYBINDINGS.events, "open events"),
|
||||||
|
(DEFAULT_KEYBINDINGS.logs, "open logs"),
|
||||||
|
(DEFAULT_KEYBINDINGS.update, "open updates"),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
];
|
||||||
|
|||||||
@@ -2,7 +2,11 @@
|
|||||||
mod test {
|
mod test {
|
||||||
use pretty_assertions::{assert_eq, assert_str_eq};
|
use pretty_assertions::{assert_eq, assert_str_eq};
|
||||||
|
|
||||||
use crate::app::context_clues::{BARE_POPUP_CONTEXT_CLUES, SERVARR_CONTEXT_CLUES};
|
use crate::app::context_clues::{
|
||||||
|
BARE_POPUP_CONTEXT_CLUES, BLOCKLIST_CONTEXT_CLUES, CONFIRMATION_PROMPT_CONTEXT_CLUES,
|
||||||
|
DOWNLOADS_CONTEXT_CLUES, INDEXERS_CONTEXT_CLUES, ROOT_FOLDERS_CONTEXT_CLUES,
|
||||||
|
SERVARR_CONTEXT_CLUES, SYSTEM_CONTEXT_CLUES,
|
||||||
|
};
|
||||||
use crate::app::{context_clues::build_context_clue_string, key_binding::DEFAULT_KEYBINDINGS};
|
use crate::app::{context_clues::build_context_clue_string, key_binding::DEFAULT_KEYBINDINGS};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -24,8 +28,13 @@ mod test {
|
|||||||
|
|
||||||
let (key_binding, description) = servarr_context_clues_iter.next().unwrap();
|
let (key_binding, description) = servarr_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.tab);
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.next_servarr);
|
||||||
assert_str_eq!(*description, "change servarr");
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.next_servarr.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = servarr_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.previous_servarr);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.previous_servarr.desc);
|
||||||
|
|
||||||
let (key_binding, description) = servarr_context_clues_iter.next().unwrap();
|
let (key_binding, description) = servarr_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
@@ -44,4 +53,160 @@ mod test {
|
|||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
assert_eq!(bare_popup_context_clues_iter.next(), None);
|
assert_eq!(bare_popup_context_clues_iter.next(), None);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_downloads_context_clues() {
|
||||||
|
let mut downloads_context_clues_iter = DOWNLOADS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = downloads_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = downloads_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = downloads_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.update);
|
||||||
|
assert_str_eq!(*description, "update downloads");
|
||||||
|
assert_eq!(downloads_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_blocklist_context_clues() {
|
||||||
|
let mut blocklist_context_clues_iter = BLOCKLIST_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "details");
|
||||||
|
|
||||||
|
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.clear);
|
||||||
|
assert_str_eq!(*description, "clear blocklist");
|
||||||
|
assert_eq!(blocklist_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_confirmation_prompt_context_clues() {
|
||||||
|
let mut confirmation_prompt_context_clues_iter = CONFIRMATION_PROMPT_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = confirmation_prompt_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.confirm);
|
||||||
|
assert_str_eq!(*description, "submit");
|
||||||
|
|
||||||
|
let (key_binding, description) = confirmation_prompt_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, "cancel");
|
||||||
|
assert_eq!(confirmation_prompt_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_root_folders_context_clues() {
|
||||||
|
let mut root_folders_context_clues_iter = ROOT_FOLDERS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = root_folders_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.add);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.add.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = root_folders_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = root_folders_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
assert_eq!(root_folders_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_indexers_context_clues() {
|
||||||
|
let mut indexers_context_clues_iter = INDEXERS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "edit indexer");
|
||||||
|
|
||||||
|
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.settings);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.settings.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.test);
|
||||||
|
assert_str_eq!(*description, "test indexer");
|
||||||
|
|
||||||
|
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.test_all);
|
||||||
|
assert_str_eq!(*description, "test all indexers");
|
||||||
|
|
||||||
|
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
assert_eq!(indexers_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_system_context_clues() {
|
||||||
|
let mut system_context_clues_iter = SYSTEM_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.tasks);
|
||||||
|
assert_str_eq!(*description, "open tasks");
|
||||||
|
|
||||||
|
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.events);
|
||||||
|
assert_str_eq!(*description, "open events");
|
||||||
|
|
||||||
|
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.logs);
|
||||||
|
assert_str_eq!(*description, "open logs");
|
||||||
|
|
||||||
|
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.update);
|
||||||
|
assert_str_eq!(*description, "open updates");
|
||||||
|
|
||||||
|
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
assert_eq!(system_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,8 +15,11 @@ generate_keybindings! {
|
|||||||
left,
|
left,
|
||||||
right,
|
right,
|
||||||
backspace,
|
backspace,
|
||||||
|
next_servarr,
|
||||||
|
previous_servarr,
|
||||||
clear,
|
clear,
|
||||||
search,
|
search,
|
||||||
|
auto_search,
|
||||||
settings,
|
settings,
|
||||||
filter,
|
filter,
|
||||||
sort,
|
sort,
|
||||||
@@ -25,12 +28,12 @@ generate_keybindings! {
|
|||||||
tasks,
|
tasks,
|
||||||
test,
|
test,
|
||||||
test_all,
|
test_all,
|
||||||
|
toggle_monitoring,
|
||||||
refresh,
|
refresh,
|
||||||
update,
|
update,
|
||||||
events,
|
events,
|
||||||
home,
|
home,
|
||||||
end,
|
end,
|
||||||
tab,
|
|
||||||
delete,
|
delete,
|
||||||
submit,
|
submit,
|
||||||
confirm,
|
confirm,
|
||||||
@@ -41,116 +44,188 @@ generate_keybindings! {
|
|||||||
#[derive(Clone, Copy, Eq, PartialEq, Debug)]
|
#[derive(Clone, Copy, Eq, PartialEq, Debug)]
|
||||||
pub struct KeyBinding {
|
pub struct KeyBinding {
|
||||||
pub key: Key,
|
pub key: Key,
|
||||||
|
pub alt: Option<Key>,
|
||||||
pub desc: &'static str,
|
pub desc: &'static str,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub const DEFAULT_KEYBINDINGS: KeyBindings = KeyBindings {
|
pub const DEFAULT_KEYBINDINGS: KeyBindings = KeyBindings {
|
||||||
add: KeyBinding {
|
add: KeyBinding {
|
||||||
key: Key::Char('a'),
|
key: Key::Char('a'),
|
||||||
|
alt: None,
|
||||||
desc: "add",
|
desc: "add",
|
||||||
},
|
},
|
||||||
up: KeyBinding {
|
up: KeyBinding {
|
||||||
key: Key::Up,
|
key: Key::Up,
|
||||||
|
alt: Some(Key::Char('k')),
|
||||||
desc: "up",
|
desc: "up",
|
||||||
},
|
},
|
||||||
down: KeyBinding {
|
down: KeyBinding {
|
||||||
key: Key::Down,
|
key: Key::Down,
|
||||||
|
alt: Some(Key::Char('j')),
|
||||||
desc: "down",
|
desc: "down",
|
||||||
},
|
},
|
||||||
left: KeyBinding {
|
left: KeyBinding {
|
||||||
key: Key::Left,
|
key: Key::Left,
|
||||||
|
alt: Some(Key::Char('h')),
|
||||||
desc: "left",
|
desc: "left",
|
||||||
},
|
},
|
||||||
right: KeyBinding {
|
right: KeyBinding {
|
||||||
key: Key::Right,
|
key: Key::Right,
|
||||||
|
alt: Some(Key::Char('l')),
|
||||||
desc: "right",
|
desc: "right",
|
||||||
},
|
},
|
||||||
backspace: KeyBinding {
|
backspace: KeyBinding {
|
||||||
key: Key::Backspace,
|
key: Key::Backspace,
|
||||||
|
alt: None,
|
||||||
desc: "backspace",
|
desc: "backspace",
|
||||||
},
|
},
|
||||||
|
next_servarr: KeyBinding {
|
||||||
|
key: Key::Tab,
|
||||||
|
alt: None,
|
||||||
|
desc: "next servarr",
|
||||||
|
},
|
||||||
|
previous_servarr: KeyBinding {
|
||||||
|
key: Key::BackTab,
|
||||||
|
alt: None,
|
||||||
|
desc: "previous servarr",
|
||||||
|
},
|
||||||
clear: KeyBinding {
|
clear: KeyBinding {
|
||||||
key: Key::Char('c'),
|
key: Key::Char('c'),
|
||||||
|
alt: None,
|
||||||
desc: "clear",
|
desc: "clear",
|
||||||
},
|
},
|
||||||
|
auto_search: KeyBinding {
|
||||||
|
key: Key::Char('S'),
|
||||||
|
alt: None,
|
||||||
|
desc: "auto search",
|
||||||
|
},
|
||||||
search: KeyBinding {
|
search: KeyBinding {
|
||||||
key: Key::Char('s'),
|
key: Key::Char('s'),
|
||||||
|
alt: None,
|
||||||
desc: "search",
|
desc: "search",
|
||||||
},
|
},
|
||||||
settings: KeyBinding {
|
settings: KeyBinding {
|
||||||
key: Key::Char('s'),
|
key: Key::Char('S'),
|
||||||
|
alt: None,
|
||||||
desc: "settings",
|
desc: "settings",
|
||||||
},
|
},
|
||||||
filter: KeyBinding {
|
filter: KeyBinding {
|
||||||
key: Key::Char('f'),
|
key: Key::Char('f'),
|
||||||
|
alt: None,
|
||||||
desc: "filter",
|
desc: "filter",
|
||||||
},
|
},
|
||||||
sort: KeyBinding {
|
sort: KeyBinding {
|
||||||
key: Key::Char('o'),
|
key: Key::Char('o'),
|
||||||
|
alt: None,
|
||||||
desc: "sort",
|
desc: "sort",
|
||||||
},
|
},
|
||||||
edit: KeyBinding {
|
edit: KeyBinding {
|
||||||
key: Key::Char('e'),
|
key: Key::Char('e'),
|
||||||
|
alt: None,
|
||||||
desc: "edit",
|
desc: "edit",
|
||||||
},
|
},
|
||||||
events: KeyBinding {
|
events: KeyBinding {
|
||||||
key: Key::Char('e'),
|
key: Key::Char('e'),
|
||||||
|
alt: None,
|
||||||
desc: "events",
|
desc: "events",
|
||||||
},
|
},
|
||||||
logs: KeyBinding {
|
logs: KeyBinding {
|
||||||
key: Key::Char('l'),
|
key: Key::Char('L'),
|
||||||
|
alt: None,
|
||||||
desc: "logs",
|
desc: "logs",
|
||||||
},
|
},
|
||||||
tasks: KeyBinding {
|
tasks: KeyBinding {
|
||||||
key: Key::Char('t'),
|
key: Key::Char('t'),
|
||||||
|
alt: None,
|
||||||
desc: "tasks",
|
desc: "tasks",
|
||||||
},
|
},
|
||||||
test: KeyBinding {
|
test: KeyBinding {
|
||||||
key: Key::Char('t'),
|
key: Key::Char('t'),
|
||||||
|
alt: None,
|
||||||
desc: "test",
|
desc: "test",
|
||||||
},
|
},
|
||||||
test_all: KeyBinding {
|
test_all: KeyBinding {
|
||||||
key: Key::Char('T'),
|
key: Key::Char('T'),
|
||||||
|
alt: None,
|
||||||
desc: "test all",
|
desc: "test all",
|
||||||
},
|
},
|
||||||
|
toggle_monitoring: KeyBinding {
|
||||||
|
key: Key::Char('m'),
|
||||||
|
alt: None,
|
||||||
|
desc: "toggle monitoring",
|
||||||
|
},
|
||||||
refresh: KeyBinding {
|
refresh: KeyBinding {
|
||||||
key: Key::Ctrl('r'),
|
key: Key::Ctrl('r'),
|
||||||
|
alt: None,
|
||||||
desc: "refresh",
|
desc: "refresh",
|
||||||
},
|
},
|
||||||
update: KeyBinding {
|
update: KeyBinding {
|
||||||
key: Key::Char('u'),
|
key: Key::Char('u'),
|
||||||
|
alt: None,
|
||||||
desc: "update",
|
desc: "update",
|
||||||
},
|
},
|
||||||
home: KeyBinding {
|
home: KeyBinding {
|
||||||
key: Key::Home,
|
key: Key::Home,
|
||||||
|
alt: None,
|
||||||
desc: "home",
|
desc: "home",
|
||||||
},
|
},
|
||||||
end: KeyBinding {
|
end: KeyBinding {
|
||||||
key: Key::End,
|
key: Key::End,
|
||||||
|
alt: None,
|
||||||
desc: "end",
|
desc: "end",
|
||||||
},
|
},
|
||||||
tab: KeyBinding {
|
|
||||||
key: Key::Tab,
|
|
||||||
desc: "tab",
|
|
||||||
},
|
|
||||||
delete: KeyBinding {
|
delete: KeyBinding {
|
||||||
key: Key::Delete,
|
key: Key::Delete,
|
||||||
|
alt: None,
|
||||||
desc: "delete",
|
desc: "delete",
|
||||||
},
|
},
|
||||||
submit: KeyBinding {
|
submit: KeyBinding {
|
||||||
key: Key::Enter,
|
key: Key::Enter,
|
||||||
|
alt: None,
|
||||||
desc: "submit",
|
desc: "submit",
|
||||||
},
|
},
|
||||||
confirm: KeyBinding {
|
confirm: KeyBinding {
|
||||||
key: Key::Ctrl('s'),
|
key: Key::Ctrl('s'),
|
||||||
|
alt: None,
|
||||||
desc: "submit",
|
desc: "submit",
|
||||||
},
|
},
|
||||||
quit: KeyBinding {
|
quit: KeyBinding {
|
||||||
key: Key::Char('q'),
|
key: Key::Char('q'),
|
||||||
|
alt: None,
|
||||||
desc: "quit",
|
desc: "quit",
|
||||||
},
|
},
|
||||||
esc: KeyBinding {
|
esc: KeyBinding {
|
||||||
key: Key::Esc,
|
key: Key::Esc,
|
||||||
|
alt: None,
|
||||||
desc: "close",
|
desc: "close",
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! matches_key {
|
||||||
|
($binding:ident, $key:expr) => {
|
||||||
|
$crate::app::key_binding::DEFAULT_KEYBINDINGS.$binding.key == $key
|
||||||
|
|| ($crate::app::key_binding::DEFAULT_KEYBINDINGS
|
||||||
|
.$binding
|
||||||
|
.alt
|
||||||
|
.is_some()
|
||||||
|
&& $crate::app::key_binding::DEFAULT_KEYBINDINGS
|
||||||
|
.$binding
|
||||||
|
.alt
|
||||||
|
.unwrap()
|
||||||
|
== $key)
|
||||||
|
};
|
||||||
|
($binding:ident, $key:expr, $ignore_alt_navigation:expr) => {
|
||||||
|
$crate::app::key_binding::DEFAULT_KEYBINDINGS.$binding.key == $key
|
||||||
|
|| !$ignore_alt_navigation
|
||||||
|
&& ($crate::app::key_binding::DEFAULT_KEYBINDINGS
|
||||||
|
.$binding
|
||||||
|
.alt
|
||||||
|
.is_some()
|
||||||
|
&& $crate::app::key_binding::DEFAULT_KEYBINDINGS
|
||||||
|
.$binding
|
||||||
|
.alt
|
||||||
|
.unwrap()
|
||||||
|
== $key)
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|||||||
@@ -13,22 +13,25 @@ mod test {
|
|||||||
#[case(DEFAULT_KEYBINDINGS.left, Key::Left, "left")]
|
#[case(DEFAULT_KEYBINDINGS.left, Key::Left, "left")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.right, Key::Right, "right")]
|
#[case(DEFAULT_KEYBINDINGS.right, Key::Right, "right")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.backspace, Key::Backspace, "backspace")]
|
#[case(DEFAULT_KEYBINDINGS.backspace, Key::Backspace, "backspace")]
|
||||||
|
#[case(DEFAULT_KEYBINDINGS.next_servarr, Key::Tab, "next servarr")]
|
||||||
|
#[case(DEFAULT_KEYBINDINGS.previous_servarr, Key::BackTab, "previous servarr")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.clear, Key::Char('c'), "clear")]
|
#[case(DEFAULT_KEYBINDINGS.clear, Key::Char('c'), "clear")]
|
||||||
|
#[case(DEFAULT_KEYBINDINGS.auto_search, Key::Char('S'), "auto search")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.search, Key::Char('s'), "search")]
|
#[case(DEFAULT_KEYBINDINGS.search, Key::Char('s'), "search")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.settings, Key::Char('s'), "settings")]
|
#[case(DEFAULT_KEYBINDINGS.settings, Key::Char('S'), "settings")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.filter, Key::Char('f'), "filter")]
|
#[case(DEFAULT_KEYBINDINGS.filter, Key::Char('f'), "filter")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.sort, Key::Char('o'), "sort")]
|
#[case(DEFAULT_KEYBINDINGS.sort, Key::Char('o'), "sort")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.edit, Key::Char('e'), "edit")]
|
#[case(DEFAULT_KEYBINDINGS.edit, Key::Char('e'), "edit")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.events, Key::Char('e'), "events")]
|
#[case(DEFAULT_KEYBINDINGS.events, Key::Char('e'), "events")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.logs, Key::Char('l'), "logs")]
|
#[case(DEFAULT_KEYBINDINGS.logs, Key::Char('L'), "logs")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.tasks, Key::Char('t'), "tasks")]
|
#[case(DEFAULT_KEYBINDINGS.tasks, Key::Char('t'), "tasks")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.test, Key::Char('t'), "test")]
|
#[case(DEFAULT_KEYBINDINGS.test, Key::Char('t'), "test")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.test_all, Key::Char('T'), "test all")]
|
#[case(DEFAULT_KEYBINDINGS.test_all, Key::Char('T'), "test all")]
|
||||||
|
#[case(DEFAULT_KEYBINDINGS.toggle_monitoring, Key::Char('m'), "toggle monitoring")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.refresh, Key::Ctrl('r'), "refresh")]
|
#[case(DEFAULT_KEYBINDINGS.refresh, Key::Ctrl('r'), "refresh")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.update, Key::Char('u'), "update")]
|
#[case(DEFAULT_KEYBINDINGS.update, Key::Char('u'), "update")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.home, Key::Home, "home")]
|
#[case(DEFAULT_KEYBINDINGS.home, Key::Home, "home")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.end, Key::End, "end")]
|
#[case(DEFAULT_KEYBINDINGS.end, Key::End, "end")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.tab, Key::Tab, "tab")]
|
|
||||||
#[case(DEFAULT_KEYBINDINGS.delete, Key::Delete, "delete")]
|
#[case(DEFAULT_KEYBINDINGS.delete, Key::Delete, "delete")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.submit, Key::Enter, "submit")]
|
#[case(DEFAULT_KEYBINDINGS.submit, Key::Enter, "submit")]
|
||||||
#[case(DEFAULT_KEYBINDINGS.confirm, Key::Ctrl('s'), "submit")]
|
#[case(DEFAULT_KEYBINDINGS.confirm, Key::Ctrl('s'), "submit")]
|
||||||
|
|||||||
@@ -1,11 +1,19 @@
|
|||||||
use anyhow::anyhow;
|
use anyhow::{anyhow, Error, Result};
|
||||||
|
use colored::Colorize;
|
||||||
|
use itertools::Itertools;
|
||||||
use log::{debug, error};
|
use log::{debug, error};
|
||||||
|
use regex::Regex;
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
use std::path::PathBuf;
|
||||||
|
use std::{fs, process};
|
||||||
use tokio::sync::mpsc::Sender;
|
use tokio::sync::mpsc::Sender;
|
||||||
use tokio_util::sync::CancellationToken;
|
use tokio_util::sync::CancellationToken;
|
||||||
|
use veil::Redact;
|
||||||
|
|
||||||
use crate::app::context_clues::{build_context_clue_string, SERVARR_CONTEXT_CLUES};
|
use crate::app::context_clues::{build_context_clue_string, SERVARR_CONTEXT_CLUES};
|
||||||
|
use crate::cli::Command;
|
||||||
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, RadarrData};
|
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, RadarrData};
|
||||||
|
use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, SonarrData};
|
||||||
use crate::models::{HorizontallyScrollableText, Route, TabRoute, TabState};
|
use crate::models::{HorizontallyScrollableText, Route, TabRoute, TabState};
|
||||||
use crate::network::NetworkEvent;
|
use crate::network::NetworkEvent;
|
||||||
|
|
||||||
@@ -16,13 +24,13 @@ pub mod context_clues;
|
|||||||
pub mod key_binding;
|
pub mod key_binding;
|
||||||
mod key_binding_tests;
|
mod key_binding_tests;
|
||||||
pub mod radarr;
|
pub mod radarr;
|
||||||
|
pub mod sonarr;
|
||||||
const DEFAULT_ROUTE: Route = Route::Radarr(ActiveRadarrBlock::Movies, None);
|
|
||||||
|
|
||||||
pub struct App<'a> {
|
pub struct App<'a> {
|
||||||
navigation_stack: Vec<Route>,
|
navigation_stack: Vec<Route>,
|
||||||
network_tx: Option<Sender<NetworkEvent>>,
|
network_tx: Option<Sender<NetworkEvent>>,
|
||||||
cancellation_token: CancellationToken,
|
pub cancellation_token: CancellationToken,
|
||||||
|
pub is_first_render: bool,
|
||||||
pub server_tabs: TabState,
|
pub server_tabs: TabState,
|
||||||
pub error: HorizontallyScrollableText,
|
pub error: HorizontallyScrollableText,
|
||||||
pub tick_until_poll: u64,
|
pub tick_until_poll: u64,
|
||||||
@@ -32,20 +40,89 @@ pub struct App<'a> {
|
|||||||
pub is_loading: bool,
|
pub is_loading: bool,
|
||||||
pub should_refresh: bool,
|
pub should_refresh: bool,
|
||||||
pub should_ignore_quit_key: bool,
|
pub should_ignore_quit_key: bool,
|
||||||
pub config: AppConfig,
|
pub cli_mode: bool,
|
||||||
pub data: Data<'a>,
|
pub data: Data<'a>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a> App<'a> {
|
impl App<'_> {
|
||||||
pub fn new(
|
pub fn new(
|
||||||
network_tx: Sender<NetworkEvent>,
|
network_tx: Sender<NetworkEvent>,
|
||||||
config: AppConfig,
|
config: AppConfig,
|
||||||
cancellation_token: CancellationToken,
|
cancellation_token: CancellationToken,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
|
let mut server_tabs = Vec::new();
|
||||||
|
let help = format!(
|
||||||
|
"<↑↓> scroll | ←→ change tab | {} ",
|
||||||
|
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Some(radarr_configs) = config.radarr {
|
||||||
|
let mut idx = 0;
|
||||||
|
for radarr_config in radarr_configs {
|
||||||
|
let name = if let Some(name) = radarr_config.name.clone() {
|
||||||
|
name
|
||||||
|
} else {
|
||||||
|
idx += 1;
|
||||||
|
format!("Radarr {}", idx)
|
||||||
|
};
|
||||||
|
|
||||||
|
server_tabs.push(TabRoute {
|
||||||
|
title: name,
|
||||||
|
route: ActiveRadarrBlock::Movies.into(),
|
||||||
|
help: help.clone(),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(radarr_config),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(sonarr_configs) = config.sonarr {
|
||||||
|
let mut idx = 0;
|
||||||
|
|
||||||
|
for sonarr_config in sonarr_configs {
|
||||||
|
let name = if let Some(name) = sonarr_config.name.clone() {
|
||||||
|
name
|
||||||
|
} else {
|
||||||
|
idx += 1;
|
||||||
|
format!("Sonarr {}", idx)
|
||||||
|
};
|
||||||
|
|
||||||
|
server_tabs.push(TabRoute {
|
||||||
|
title: name,
|
||||||
|
route: ActiveSonarrBlock::Series.into(),
|
||||||
|
help: help.clone(),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(sonarr_config),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let weight_sorted_tabs = server_tabs
|
||||||
|
.into_iter()
|
||||||
|
.sorted_by(|tab1, tab2| {
|
||||||
|
Ord::cmp(
|
||||||
|
tab1
|
||||||
|
.config
|
||||||
|
.as_ref()
|
||||||
|
.unwrap()
|
||||||
|
.weight
|
||||||
|
.as_ref()
|
||||||
|
.unwrap_or(&1000),
|
||||||
|
tab2
|
||||||
|
.config
|
||||||
|
.as_ref()
|
||||||
|
.unwrap()
|
||||||
|
.weight
|
||||||
|
.as_ref()
|
||||||
|
.unwrap_or(&1000),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
App {
|
App {
|
||||||
network_tx: Some(network_tx),
|
network_tx: Some(network_tx),
|
||||||
config,
|
|
||||||
cancellation_token,
|
cancellation_token,
|
||||||
|
server_tabs: TabState::new(weight_sorted_tabs),
|
||||||
..App::default()
|
..App::default()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -53,7 +130,10 @@ impl<'a> App<'a> {
|
|||||||
pub async fn dispatch_network_event(&mut self, action: NetworkEvent) {
|
pub async fn dispatch_network_event(&mut self, action: NetworkEvent) {
|
||||||
debug!("Dispatching network event: {action:?}");
|
debug!("Dispatching network event: {action:?}");
|
||||||
|
|
||||||
self.is_loading = true;
|
if !self.should_refresh {
|
||||||
|
self.is_loading = true;
|
||||||
|
}
|
||||||
|
|
||||||
if let Some(network_tx) = &self.network_tx {
|
if let Some(network_tx) = &self.network_tx {
|
||||||
if let Err(e) = network_tx.send(action).await {
|
if let Err(e) = network_tx.send(action).await {
|
||||||
self.is_loading = false;
|
self.is_loading = false;
|
||||||
@@ -67,26 +147,26 @@ impl<'a> App<'a> {
|
|||||||
self.tick_count = 0;
|
self.tick_count = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Allowing this code for now since we'll eventually be implementing additional Servarr support and we'll need it then
|
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
pub fn reset(&mut self) {
|
pub fn reset(&mut self) {
|
||||||
self.reset_tick_count();
|
self.reset_tick_count();
|
||||||
self.error = HorizontallyScrollableText::default();
|
self.error = HorizontallyScrollableText::default();
|
||||||
|
self.is_first_render = true;
|
||||||
self.data = Data::default();
|
self.data = Data::default();
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn handle_error(&mut self, error: anyhow::Error) {
|
pub fn handle_error(&mut self, error: Error) {
|
||||||
if self.error.text.is_empty() {
|
if self.error.text.is_empty() {
|
||||||
self.error = error.to_string().into();
|
self.error = error.to_string().into();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn on_tick(&mut self, is_first_render: bool) {
|
pub async fn on_tick(&mut self) {
|
||||||
if self.tick_count % self.tick_until_poll == 0 || self.is_routing || self.should_refresh {
|
if self.tick_count % self.tick_until_poll == 0 || self.is_routing || self.should_refresh {
|
||||||
if let Route::Radarr(active_radarr_block, _) = self.get_current_route() {
|
match self.get_current_route() {
|
||||||
self
|
Route::Radarr(active_radarr_block, _) => self.radarr_on_tick(active_radarr_block).await,
|
||||||
.radarr_on_tick(*active_radarr_block, is_first_render)
|
Route::Sonarr(active_sonarr_block, _) => self.sonarr_on_tick(active_sonarr_block).await,
|
||||||
.await;
|
_ => (),
|
||||||
}
|
}
|
||||||
|
|
||||||
self.is_routing = false;
|
self.is_routing = false;
|
||||||
@@ -103,13 +183,15 @@ impl<'a> App<'a> {
|
|||||||
|
|
||||||
pub fn pop_navigation_stack(&mut self) {
|
pub fn pop_navigation_stack(&mut self) {
|
||||||
self.is_routing = true;
|
self.is_routing = true;
|
||||||
if self.navigation_stack.len() > 1 {
|
if !self.navigation_stack.is_empty() {
|
||||||
self.navigation_stack.pop();
|
self.navigation_stack.pop();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn reset_cancellation_token(&mut self) -> CancellationToken {
|
pub fn reset_cancellation_token(&mut self) -> CancellationToken {
|
||||||
self.cancellation_token = CancellationToken::new();
|
self.cancellation_token = CancellationToken::new();
|
||||||
|
self.should_refresh = true;
|
||||||
|
self.is_loading = false;
|
||||||
|
|
||||||
self.cancellation_token.clone()
|
self.cancellation_token.clone()
|
||||||
}
|
}
|
||||||
@@ -119,35 +201,23 @@ impl<'a> App<'a> {
|
|||||||
self.push_navigation_stack(route);
|
self.push_navigation_stack(route);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_current_route(&self) -> &Route {
|
pub fn get_current_route(&self) -> Route {
|
||||||
self.navigation_stack.last().unwrap_or(&DEFAULT_ROUTE)
|
*self
|
||||||
|
.navigation_stack
|
||||||
|
.last()
|
||||||
|
.unwrap_or(&self.server_tabs.tabs.first().unwrap().route)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a> Default for App<'a> {
|
impl Default for App<'_> {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
App {
|
App {
|
||||||
navigation_stack: vec![DEFAULT_ROUTE],
|
navigation_stack: Vec::new(),
|
||||||
network_tx: None,
|
network_tx: None,
|
||||||
cancellation_token: CancellationToken::new(),
|
cancellation_token: CancellationToken::new(),
|
||||||
error: HorizontallyScrollableText::default(),
|
error: HorizontallyScrollableText::default(),
|
||||||
server_tabs: TabState::new(vec![
|
is_first_render: true,
|
||||||
TabRoute {
|
server_tabs: TabState::new(Vec::new()),
|
||||||
title: "Radarr",
|
|
||||||
route: ActiveRadarrBlock::Movies.into(),
|
|
||||||
help: format!(
|
|
||||||
"<↑↓> scroll | ←→ change tab | {} ",
|
|
||||||
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
|
||||||
),
|
|
||||||
contextual_help: None,
|
|
||||||
},
|
|
||||||
TabRoute {
|
|
||||||
title: "Sonarr",
|
|
||||||
route: Route::Sonarr,
|
|
||||||
help: format!("{} ", build_context_clue_string(&SERVARR_CONTEXT_CLUES)),
|
|
||||||
contextual_help: None,
|
|
||||||
},
|
|
||||||
]),
|
|
||||||
tick_until_poll: 400,
|
tick_until_poll: 400,
|
||||||
ticks_until_scroll: 4,
|
ticks_until_scroll: 4,
|
||||||
tick_count: 0,
|
tick_count: 0,
|
||||||
@@ -155,40 +225,228 @@ impl<'a> Default for App<'a> {
|
|||||||
is_routing: false,
|
is_routing: false,
|
||||||
should_refresh: false,
|
should_refresh: false,
|
||||||
should_ignore_quit_key: false,
|
should_ignore_quit_key: false,
|
||||||
config: AppConfig::default(),
|
cli_mode: false,
|
||||||
data: Data::default(),
|
data: Data::default(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
impl App<'_> {
|
||||||
|
pub fn test_default() -> Self {
|
||||||
|
App {
|
||||||
|
server_tabs: TabState::new(vec![
|
||||||
|
TabRoute {
|
||||||
|
title: "Radarr".to_owned(),
|
||||||
|
route: ActiveRadarrBlock::Movies.into(),
|
||||||
|
help: format!(
|
||||||
|
"<↑↓> scroll | ←→ change tab | {} ",
|
||||||
|
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
||||||
|
),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(ServarrConfig::default()),
|
||||||
|
},
|
||||||
|
TabRoute {
|
||||||
|
title: "Sonarr".to_owned(),
|
||||||
|
route: ActiveSonarrBlock::Series.into(),
|
||||||
|
help: format!(
|
||||||
|
"<↑↓> scroll | ←→ change tab | {} ",
|
||||||
|
build_context_clue_string(&SERVARR_CONTEXT_CLUES)
|
||||||
|
),
|
||||||
|
contextual_help: None,
|
||||||
|
config: Some(ServarrConfig::default()),
|
||||||
|
},
|
||||||
|
]),
|
||||||
|
..App::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub struct Data<'a> {
|
pub struct Data<'a> {
|
||||||
pub radarr_data: RadarrData<'a>,
|
pub radarr_data: RadarrData<'a>,
|
||||||
|
pub sonarr_data: SonarrData<'a>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize, Serialize, Default)]
|
#[derive(Debug, Deserialize, Serialize, Default, Clone)]
|
||||||
pub struct AppConfig {
|
pub struct AppConfig {
|
||||||
pub radarr: RadarrConfig,
|
pub theme: Option<String>,
|
||||||
|
pub radarr: Option<Vec<ServarrConfig>>,
|
||||||
|
pub sonarr: Option<Vec<ServarrConfig>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize, Serialize)]
|
impl AppConfig {
|
||||||
pub struct RadarrConfig {
|
pub fn validate(&self) {
|
||||||
pub host: String,
|
if self.radarr.is_none() && self.sonarr.is_none() {
|
||||||
|
log_and_print_error(
|
||||||
|
"No Servarr configuration provided in the specified configuration file".to_owned(),
|
||||||
|
);
|
||||||
|
process::exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(radarr_configs) = &self.radarr {
|
||||||
|
radarr_configs.iter().for_each(|config| config.validate());
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(sonarr_configs) = &self.sonarr {
|
||||||
|
sonarr_configs.iter().for_each(|config| config.validate());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn verify_config_present_for_cli(&self, command: &Command) {
|
||||||
|
let msg = |servarr: &str| {
|
||||||
|
log_and_print_error(format!(
|
||||||
|
"{} configuration missing; Unable to run any {} commands.",
|
||||||
|
servarr, servarr
|
||||||
|
))
|
||||||
|
};
|
||||||
|
match command {
|
||||||
|
Command::Radarr(_) if self.radarr.is_none() => {
|
||||||
|
msg("Radarr");
|
||||||
|
process::exit(1);
|
||||||
|
}
|
||||||
|
Command::Sonarr(_) if self.sonarr.is_none() => {
|
||||||
|
msg("Sonarr");
|
||||||
|
process::exit(1);
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn post_process_initialization(&mut self) {
|
||||||
|
if let Some(radarr_configs) = self.radarr.as_mut() {
|
||||||
|
for radarr_config in radarr_configs {
|
||||||
|
radarr_config.post_process_initialization();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(sonarr_configs) = self.sonarr.as_mut() {
|
||||||
|
for sonarr_config in sonarr_configs {
|
||||||
|
sonarr_config.post_process_initialization();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Redact, Deserialize, Serialize, Clone, PartialEq, Eq)]
|
||||||
|
pub struct ServarrConfig {
|
||||||
|
#[serde(default, deserialize_with = "deserialize_optional_env_var")]
|
||||||
|
pub name: Option<String>,
|
||||||
|
#[serde(default, deserialize_with = "deserialize_optional_env_var")]
|
||||||
|
pub host: Option<String>,
|
||||||
|
#[serde(default, deserialize_with = "deserialize_u16_env_var")]
|
||||||
pub port: Option<u16>,
|
pub port: Option<u16>,
|
||||||
pub api_token: String,
|
#[serde(default, deserialize_with = "deserialize_optional_env_var")]
|
||||||
#[serde(default)]
|
pub uri: Option<String>,
|
||||||
pub use_ssl: bool,
|
#[serde(default, deserialize_with = "deserialize_u16_env_var")]
|
||||||
|
pub weight: Option<u16>,
|
||||||
|
#[serde(default, deserialize_with = "deserialize_optional_env_var")]
|
||||||
|
#[redact]
|
||||||
|
pub api_token: Option<String>,
|
||||||
|
#[serde(default, deserialize_with = "deserialize_optional_env_var")]
|
||||||
|
pub api_token_file: Option<String>,
|
||||||
|
#[serde(default, deserialize_with = "deserialize_optional_env_var")]
|
||||||
pub ssl_cert_path: Option<String>,
|
pub ssl_cert_path: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for RadarrConfig {
|
impl ServarrConfig {
|
||||||
|
fn validate(&self) {
|
||||||
|
if self.host.is_none() && self.uri.is_none() {
|
||||||
|
log_and_print_error("'host' or 'uri' is required for configuration".to_owned());
|
||||||
|
process::exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.api_token_file.is_none() && self.api_token.is_none() {
|
||||||
|
log_and_print_error(
|
||||||
|
"'api_token' or 'api_token_path' is required for configuration".to_owned(),
|
||||||
|
);
|
||||||
|
process::exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn post_process_initialization(&mut self) {
|
||||||
|
if let Some(api_token_file) = self.api_token_file.as_ref() {
|
||||||
|
if !PathBuf::from(api_token_file).exists() {
|
||||||
|
log_and_print_error(format!(
|
||||||
|
"The specified {} API token file does not exist",
|
||||||
|
api_token_file
|
||||||
|
));
|
||||||
|
process::exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
let api_token = fs::read_to_string(api_token_file)
|
||||||
|
.map_err(|e| anyhow!(e))
|
||||||
|
.unwrap();
|
||||||
|
self.api_token = Some(api_token.trim().to_owned());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for ServarrConfig {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
RadarrConfig {
|
ServarrConfig {
|
||||||
host: "localhost".to_string(),
|
name: None,
|
||||||
port: Some(7878),
|
host: Some("localhost".to_string()),
|
||||||
api_token: "".to_string(),
|
port: None,
|
||||||
use_ssl: false,
|
uri: None,
|
||||||
|
weight: None,
|
||||||
|
api_token: Some(String::new()),
|
||||||
|
api_token_file: None,
|
||||||
ssl_cert_path: None,
|
ssl_cert_path: None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn log_and_print_error(error: String) {
|
||||||
|
error!("{}", error);
|
||||||
|
eprintln!("error: {}", error.red());
|
||||||
|
}
|
||||||
|
|
||||||
|
fn deserialize_optional_env_var<'de, D>(deserializer: D) -> Result<Option<String>, D::Error>
|
||||||
|
where
|
||||||
|
D: serde::Deserializer<'de>,
|
||||||
|
{
|
||||||
|
let s: Option<String> = Option::deserialize(deserializer)?;
|
||||||
|
match s {
|
||||||
|
Some(value) => {
|
||||||
|
let interpolated = interpolate_env_vars(&value);
|
||||||
|
Ok(Some(interpolated))
|
||||||
|
}
|
||||||
|
None => Ok(None),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn deserialize_u16_env_var<'de, D>(deserializer: D) -> Result<Option<u16>, D::Error>
|
||||||
|
where
|
||||||
|
D: serde::Deserializer<'de>,
|
||||||
|
{
|
||||||
|
let s: Option<String> = Option::deserialize(deserializer)?;
|
||||||
|
match s {
|
||||||
|
Some(value) => {
|
||||||
|
let interpolated = interpolate_env_vars(&value);
|
||||||
|
interpolated
|
||||||
|
.parse::<u16>()
|
||||||
|
.map(Some)
|
||||||
|
.map_err(serde::de::Error::custom)
|
||||||
|
}
|
||||||
|
None => Ok(None),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn interpolate_env_vars(s: &str) -> String {
|
||||||
|
let result = s.to_string();
|
||||||
|
let scrubbing_regex = Regex::new(r#"[\s\{\}!\$^\(\)\[\]\\\|`'"]+"#).unwrap();
|
||||||
|
let var_regex = Regex::new(r"\$\{(.*?)\}").unwrap();
|
||||||
|
|
||||||
|
var_regex
|
||||||
|
.replace_all(s, |caps: ®ex::Captures<'_>| {
|
||||||
|
if let Some(mat) = caps.get(1) {
|
||||||
|
if let Ok(value) = std::env::var(mat.as_str()) {
|
||||||
|
return scrubbing_regex.replace_all(&value, "").to_string();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
scrubbing_regex.replace_all(&result, "").to_string()
|
||||||
|
})
|
||||||
|
.to_string()
|
||||||
|
}
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ pub mod radarr_context_clues;
|
|||||||
#[path = "radarr_tests.rs"]
|
#[path = "radarr_tests.rs"]
|
||||||
mod radarr_tests;
|
mod radarr_tests;
|
||||||
|
|
||||||
impl<'a> App<'a> {
|
impl App<'_> {
|
||||||
pub(super) async fn dispatch_by_radarr_block(&mut self, active_radarr_block: &ActiveRadarrBlock) {
|
pub(super) async fn dispatch_by_radarr_block(&mut self, active_radarr_block: &ActiveRadarrBlock) {
|
||||||
match active_radarr_block {
|
match active_radarr_block {
|
||||||
ActiveRadarrBlock::Blocklist => {
|
ActiveRadarrBlock::Blocklist => {
|
||||||
@@ -17,11 +17,23 @@ impl<'a> App<'a> {
|
|||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::Collections => {
|
ActiveRadarrBlock::Collections => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetQualityProfiles.into())
|
||||||
|
.await;
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetCollections.into())
|
.dispatch_network_event(RadarrEvent::GetCollections.into())
|
||||||
.await;
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetMovies.into())
|
||||||
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::CollectionDetails => {
|
ActiveRadarrBlock::CollectionDetails => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetQualityProfiles.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetTags.into())
|
||||||
|
.await;
|
||||||
self.is_loading = true;
|
self.is_loading = true;
|
||||||
self.populate_movie_collection_table().await;
|
self.populate_movie_collection_table().await;
|
||||||
self.is_loading = false;
|
self.is_loading = false;
|
||||||
@@ -37,6 +49,12 @@ impl<'a> App<'a> {
|
|||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::Movies => {
|
ActiveRadarrBlock::Movies => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetQualityProfiles.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetTags.into())
|
||||||
|
.await;
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetMovies.into())
|
.dispatch_network_event(RadarrEvent::GetMovies.into())
|
||||||
.await;
|
.await;
|
||||||
@@ -45,6 +63,9 @@ impl<'a> App<'a> {
|
|||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::Indexers => {
|
ActiveRadarrBlock::Indexers => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetTags.into())
|
||||||
|
.await;
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetIndexers.into())
|
.dispatch_network_event(RadarrEvent::GetIndexers.into())
|
||||||
.await;
|
.await;
|
||||||
@@ -56,7 +77,9 @@ impl<'a> App<'a> {
|
|||||||
}
|
}
|
||||||
ActiveRadarrBlock::TestIndexer => {
|
ActiveRadarrBlock::TestIndexer => {
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::TestIndexer(None).into())
|
.dispatch_network_event(
|
||||||
|
RadarrEvent::TestIndexer(self.extract_radarr_indexer_id().await).into(),
|
||||||
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::TestAllIndexers => {
|
ActiveRadarrBlock::TestAllIndexers => {
|
||||||
@@ -72,7 +95,7 @@ impl<'a> App<'a> {
|
|||||||
.dispatch_network_event(RadarrEvent::GetQueuedEvents.into())
|
.dispatch_network_event(RadarrEvent::GetQueuedEvents.into())
|
||||||
.await;
|
.await;
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetLogs(None).into())
|
.dispatch_network_event(RadarrEvent::GetLogs(500).into())
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::SystemUpdates => {
|
ActiveRadarrBlock::SystemUpdates => {
|
||||||
@@ -82,17 +105,23 @@ impl<'a> App<'a> {
|
|||||||
}
|
}
|
||||||
ActiveRadarrBlock::AddMovieSearchResults => {
|
ActiveRadarrBlock::AddMovieSearchResults => {
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::SearchNewMovie(None).into())
|
.dispatch_network_event(
|
||||||
|
RadarrEvent::SearchNewMovie(self.extract_movie_search_query().await).into(),
|
||||||
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::MovieDetails | ActiveRadarrBlock::FileInfo => {
|
ActiveRadarrBlock::MovieDetails | ActiveRadarrBlock::FileInfo => {
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetMovieDetails(None).into())
|
.dispatch_network_event(
|
||||||
|
RadarrEvent::GetMovieDetails(self.extract_movie_id().await).into(),
|
||||||
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::MovieHistory => {
|
ActiveRadarrBlock::MovieHistory => {
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetMovieHistory(None).into())
|
.dispatch_network_event(
|
||||||
|
RadarrEvent::GetMovieHistory(self.extract_movie_id().await).into(),
|
||||||
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::Cast | ActiveRadarrBlock::Crew => {
|
ActiveRadarrBlock::Cast | ActiveRadarrBlock::Crew => {
|
||||||
@@ -102,7 +131,9 @@ impl<'a> App<'a> {
|
|||||||
|| movie_details_modal.movie_crew.items.is_empty() =>
|
|| movie_details_modal.movie_crew.items.is_empty() =>
|
||||||
{
|
{
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetMovieCredits(None).into())
|
.dispatch_network_event(
|
||||||
|
RadarrEvent::GetMovieCredits(self.extract_movie_id().await).into(),
|
||||||
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
_ => (),
|
_ => (),
|
||||||
@@ -111,7 +142,7 @@ impl<'a> App<'a> {
|
|||||||
ActiveRadarrBlock::ManualSearch => match self.data.radarr_data.movie_details_modal.as_ref() {
|
ActiveRadarrBlock::ManualSearch => match self.data.radarr_data.movie_details_modal.as_ref() {
|
||||||
Some(movie_details_modal) if movie_details_modal.movie_releases.items.is_empty() => {
|
Some(movie_details_modal) if movie_details_modal.movie_releases.items.is_empty() => {
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetReleases(None).into())
|
.dispatch_network_event(RadarrEvent::GetReleases(self.extract_movie_id().await).into())
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
_ => (),
|
_ => (),
|
||||||
@@ -119,66 +150,47 @@ impl<'a> App<'a> {
|
|||||||
_ => (),
|
_ => (),
|
||||||
}
|
}
|
||||||
|
|
||||||
self.check_for_prompt_action().await;
|
self.check_for_radarr_prompt_action().await;
|
||||||
self.reset_tick_count();
|
self.reset_tick_count();
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn check_for_prompt_action(&mut self) {
|
async fn check_for_radarr_prompt_action(&mut self) {
|
||||||
if self.data.radarr_data.prompt_confirm {
|
if self.data.radarr_data.prompt_confirm {
|
||||||
self.data.radarr_data.prompt_confirm = false;
|
self.data.radarr_data.prompt_confirm = false;
|
||||||
if let Some(radarr_event) = &self.data.radarr_data.prompt_confirm_action {
|
if let Some(radarr_event) = self.data.radarr_data.prompt_confirm_action.take() {
|
||||||
self
|
self.dispatch_network_event(radarr_event.into()).await;
|
||||||
.dispatch_network_event(radarr_event.clone().into())
|
|
||||||
.await;
|
|
||||||
self.should_refresh = true;
|
self.should_refresh = true;
|
||||||
self.data.radarr_data.prompt_confirm_action = None;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(super) async fn radarr_on_tick(
|
pub(super) async fn radarr_on_tick(&mut self, active_radarr_block: ActiveRadarrBlock) {
|
||||||
&mut self,
|
if self.is_first_render {
|
||||||
active_radarr_block: ActiveRadarrBlock,
|
self.refresh_radarr_metadata().await;
|
||||||
is_first_render: bool,
|
|
||||||
) {
|
|
||||||
if is_first_render {
|
|
||||||
self
|
|
||||||
.dispatch_network_event(RadarrEvent::GetQualityProfiles.into())
|
|
||||||
.await;
|
|
||||||
self
|
|
||||||
.dispatch_network_event(RadarrEvent::GetTags.into())
|
|
||||||
.await;
|
|
||||||
self
|
|
||||||
.dispatch_network_event(RadarrEvent::GetRootFolders.into())
|
|
||||||
.await;
|
|
||||||
self
|
|
||||||
.dispatch_network_event(RadarrEvent::GetOverview.into())
|
|
||||||
.await;
|
|
||||||
self
|
|
||||||
.dispatch_network_event(RadarrEvent::GetStatus.into())
|
|
||||||
.await;
|
|
||||||
self.dispatch_by_radarr_block(&active_radarr_block).await;
|
self.dispatch_by_radarr_block(&active_radarr_block).await;
|
||||||
|
self.is_first_render = false;
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if self.should_refresh {
|
if self.should_refresh {
|
||||||
self.dispatch_by_radarr_block(&active_radarr_block).await;
|
self.dispatch_by_radarr_block(&active_radarr_block).await;
|
||||||
|
self.refresh_radarr_metadata().await;
|
||||||
}
|
}
|
||||||
|
|
||||||
if self.is_routing {
|
if self.is_routing {
|
||||||
if self.is_loading && !self.should_refresh {
|
if !self.should_refresh {
|
||||||
self.cancellation_token.cancel();
|
self.cancellation_token.cancel();
|
||||||
|
} else {
|
||||||
|
self.dispatch_by_radarr_block(&active_radarr_block).await;
|
||||||
}
|
}
|
||||||
|
|
||||||
self.dispatch_by_radarr_block(&active_radarr_block).await;
|
|
||||||
self.refresh_metadata().await;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if self.tick_count % self.tick_until_poll == 0 {
|
if self.tick_count % self.tick_until_poll == 0 {
|
||||||
self.refresh_metadata().await;
|
self.refresh_radarr_metadata().await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn refresh_metadata(&mut self) {
|
async fn refresh_radarr_metadata(&mut self) {
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetQualityProfiles.into())
|
.dispatch_network_event(RadarrEvent::GetQualityProfiles.into())
|
||||||
.await;
|
.await;
|
||||||
@@ -191,6 +203,12 @@ impl<'a> App<'a> {
|
|||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::GetDownloads.into())
|
.dispatch_network_event(RadarrEvent::GetDownloads.into())
|
||||||
.await;
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetDiskSpace.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(RadarrEvent::GetStatus.into())
|
||||||
|
.await;
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn populate_movie_collection_table(&mut self) {
|
async fn populate_movie_collection_table(&mut self) {
|
||||||
@@ -208,4 +226,23 @@ impl<'a> App<'a> {
|
|||||||
.collection_movies
|
.collection_movies
|
||||||
.set_items(collection_movies);
|
.set_items(collection_movies);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn extract_movie_id(&self) -> i64 {
|
||||||
|
self.data.radarr_data.movies.current_selection().id
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn extract_movie_search_query(&self) -> String {
|
||||||
|
self
|
||||||
|
.data
|
||||||
|
.radarr_data
|
||||||
|
.add_movie_search
|
||||||
|
.as_ref()
|
||||||
|
.expect("Add movie search is empty")
|
||||||
|
.text
|
||||||
|
.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn extract_radarr_indexer_id(&self) -> i64 {
|
||||||
|
self.data.radarr_data.indexers.current_selection().id
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -35,60 +35,6 @@ pub static COLLECTIONS_CONTEXT_CLUES: [ContextClue; 8] = [
|
|||||||
(DEFAULT_KEYBINDINGS.esc, "cancel filter"),
|
(DEFAULT_KEYBINDINGS.esc, "cancel filter"),
|
||||||
];
|
];
|
||||||
|
|
||||||
pub static DOWNLOADS_CONTEXT_CLUES: [ContextClue; 2] = [
|
|
||||||
(
|
|
||||||
DEFAULT_KEYBINDINGS.refresh,
|
|
||||||
DEFAULT_KEYBINDINGS.refresh.desc,
|
|
||||||
),
|
|
||||||
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
|
||||||
];
|
|
||||||
|
|
||||||
pub static BLOCKLIST_CONTEXT_CLUES: [ContextClue; 5] = [
|
|
||||||
(
|
|
||||||
DEFAULT_KEYBINDINGS.refresh,
|
|
||||||
DEFAULT_KEYBINDINGS.refresh.desc,
|
|
||||||
),
|
|
||||||
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
|
||||||
(DEFAULT_KEYBINDINGS.submit, "details"),
|
|
||||||
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
|
||||||
(DEFAULT_KEYBINDINGS.clear, "clear blocklist"),
|
|
||||||
];
|
|
||||||
|
|
||||||
pub static ROOT_FOLDERS_CONTEXT_CLUES: [ContextClue; 3] = [
|
|
||||||
(DEFAULT_KEYBINDINGS.add, DEFAULT_KEYBINDINGS.add.desc),
|
|
||||||
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
|
||||||
(
|
|
||||||
DEFAULT_KEYBINDINGS.refresh,
|
|
||||||
DEFAULT_KEYBINDINGS.refresh.desc,
|
|
||||||
),
|
|
||||||
];
|
|
||||||
|
|
||||||
pub static INDEXERS_CONTEXT_CLUES: [ContextClue; 6] = [
|
|
||||||
(DEFAULT_KEYBINDINGS.submit, "edit indexer"),
|
|
||||||
(
|
|
||||||
DEFAULT_KEYBINDINGS.settings,
|
|
||||||
DEFAULT_KEYBINDINGS.settings.desc,
|
|
||||||
),
|
|
||||||
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
|
||||||
(DEFAULT_KEYBINDINGS.test, "test indexer"),
|
|
||||||
(DEFAULT_KEYBINDINGS.test_all, "test all indexers"),
|
|
||||||
(
|
|
||||||
DEFAULT_KEYBINDINGS.refresh,
|
|
||||||
DEFAULT_KEYBINDINGS.refresh.desc,
|
|
||||||
),
|
|
||||||
];
|
|
||||||
|
|
||||||
pub static SYSTEM_CONTEXT_CLUES: [ContextClue; 5] = [
|
|
||||||
(DEFAULT_KEYBINDINGS.tasks, "open tasks"),
|
|
||||||
(DEFAULT_KEYBINDINGS.events, "open events"),
|
|
||||||
(DEFAULT_KEYBINDINGS.logs, "open logs"),
|
|
||||||
(DEFAULT_KEYBINDINGS.update, "open updates"),
|
|
||||||
(
|
|
||||||
DEFAULT_KEYBINDINGS.refresh,
|
|
||||||
DEFAULT_KEYBINDINGS.refresh.desc,
|
|
||||||
),
|
|
||||||
];
|
|
||||||
|
|
||||||
pub static MOVIE_DETAILS_CONTEXT_CLUES: [ContextClue; 5] = [
|
pub static MOVIE_DETAILS_CONTEXT_CLUES: [ContextClue; 5] = [
|
||||||
(
|
(
|
||||||
DEFAULT_KEYBINDINGS.refresh,
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
@@ -96,7 +42,10 @@ pub static MOVIE_DETAILS_CONTEXT_CLUES: [ContextClue; 5] = [
|
|||||||
),
|
),
|
||||||
(DEFAULT_KEYBINDINGS.update, DEFAULT_KEYBINDINGS.update.desc),
|
(DEFAULT_KEYBINDINGS.update, DEFAULT_KEYBINDINGS.update.desc),
|
||||||
(DEFAULT_KEYBINDINGS.edit, DEFAULT_KEYBINDINGS.edit.desc),
|
(DEFAULT_KEYBINDINGS.edit, DEFAULT_KEYBINDINGS.edit.desc),
|
||||||
(DEFAULT_KEYBINDINGS.search, "auto search"),
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
];
|
];
|
||||||
|
|
||||||
@@ -108,7 +57,10 @@ pub static MANUAL_MOVIE_SEARCH_CONTEXT_CLUES: [ContextClue; 6] = [
|
|||||||
(DEFAULT_KEYBINDINGS.update, DEFAULT_KEYBINDINGS.update.desc),
|
(DEFAULT_KEYBINDINGS.update, DEFAULT_KEYBINDINGS.update.desc),
|
||||||
(DEFAULT_KEYBINDINGS.edit, DEFAULT_KEYBINDINGS.edit.desc),
|
(DEFAULT_KEYBINDINGS.edit, DEFAULT_KEYBINDINGS.edit.desc),
|
||||||
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
(DEFAULT_KEYBINDINGS.search, "auto search"),
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
];
|
];
|
||||||
|
|
||||||
@@ -120,17 +72,13 @@ pub static ADD_MOVIE_SEARCH_RESULTS_CONTEXT_CLUES: [ContextClue; 2] = [
|
|||||||
(DEFAULT_KEYBINDINGS.esc, "edit search"),
|
(DEFAULT_KEYBINDINGS.esc, "edit search"),
|
||||||
];
|
];
|
||||||
|
|
||||||
pub static CONFIRMATION_PROMPT_CONTEXT_CLUES: [ContextClue; 2] = [
|
|
||||||
(DEFAULT_KEYBINDINGS.confirm, "submit"),
|
|
||||||
(DEFAULT_KEYBINDINGS.esc, "cancel"),
|
|
||||||
];
|
|
||||||
|
|
||||||
pub static SYSTEM_TASKS_CONTEXT_CLUES: [ContextClue; 2] = [
|
pub static SYSTEM_TASKS_CONTEXT_CLUES: [ContextClue; 2] = [
|
||||||
(DEFAULT_KEYBINDINGS.submit, "start task"),
|
(DEFAULT_KEYBINDINGS.submit, "start task"),
|
||||||
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
];
|
];
|
||||||
|
|
||||||
pub static COLLECTION_DETAILS_CONTEXT_CLUES: [ContextClue; 2] = [
|
pub static COLLECTION_DETAILS_CONTEXT_CLUES: [ContextClue; 3] = [
|
||||||
(DEFAULT_KEYBINDINGS.submit, "show overview/add movie"),
|
(DEFAULT_KEYBINDINGS.submit, "show overview/add movie"),
|
||||||
|
(DEFAULT_KEYBINDINGS.edit, "edit collection"),
|
||||||
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
];
|
];
|
||||||
|
|||||||
@@ -4,11 +4,10 @@ mod tests {
|
|||||||
|
|
||||||
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
||||||
use crate::app::radarr::radarr_context_clues::{
|
use crate::app::radarr::radarr_context_clues::{
|
||||||
ADD_MOVIE_SEARCH_RESULTS_CONTEXT_CLUES, BLOCKLIST_CONTEXT_CLUES, COLLECTIONS_CONTEXT_CLUES,
|
ADD_MOVIE_SEARCH_RESULTS_CONTEXT_CLUES, COLLECTIONS_CONTEXT_CLUES,
|
||||||
COLLECTION_DETAILS_CONTEXT_CLUES, CONFIRMATION_PROMPT_CONTEXT_CLUES, DOWNLOADS_CONTEXT_CLUES,
|
COLLECTION_DETAILS_CONTEXT_CLUES, LIBRARY_CONTEXT_CLUES,
|
||||||
INDEXERS_CONTEXT_CLUES, LIBRARY_CONTEXT_CLUES, MANUAL_MOVIE_SEARCH_CONTEXTUAL_CONTEXT_CLUES,
|
MANUAL_MOVIE_SEARCH_CONTEXTUAL_CONTEXT_CLUES, MANUAL_MOVIE_SEARCH_CONTEXT_CLUES,
|
||||||
MANUAL_MOVIE_SEARCH_CONTEXT_CLUES, MOVIE_DETAILS_CONTEXT_CLUES, ROOT_FOLDERS_CONTEXT_CLUES,
|
MOVIE_DETAILS_CONTEXT_CLUES, SYSTEM_TASKS_CONTEXT_CLUES,
|
||||||
SYSTEM_CONTEXT_CLUES, SYSTEM_TASKS_CONTEXT_CLUES,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -113,141 +112,6 @@ mod tests {
|
|||||||
assert_eq!(collections_context_clues.next(), None);
|
assert_eq!(collections_context_clues.next(), None);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_downloads_context_clues() {
|
|
||||||
let mut downloads_context_clues_iter = DOWNLOADS_CONTEXT_CLUES.iter();
|
|
||||||
|
|
||||||
let (key_binding, description) = downloads_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = downloads_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
|
||||||
assert_eq!(downloads_context_clues_iter.next(), None);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_blocklist_context_clues() {
|
|
||||||
let mut blocklist_context_clues_iter = BLOCKLIST_CONTEXT_CLUES.iter();
|
|
||||||
|
|
||||||
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
|
||||||
assert_str_eq!(*description, "details");
|
|
||||||
|
|
||||||
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = blocklist_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.clear);
|
|
||||||
assert_str_eq!(*description, "clear blocklist");
|
|
||||||
assert_eq!(blocklist_context_clues_iter.next(), None);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_root_folders_context_clues() {
|
|
||||||
let mut root_folders_context_clues_iter = ROOT_FOLDERS_CONTEXT_CLUES.iter();
|
|
||||||
|
|
||||||
let (key_binding, description) = root_folders_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.add);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.add.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = root_folders_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = root_folders_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
|
||||||
assert_eq!(root_folders_context_clues_iter.next(), None);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_indexers_context_clues() {
|
|
||||||
let mut indexers_context_clues_iter = INDEXERS_CONTEXT_CLUES.iter();
|
|
||||||
|
|
||||||
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
|
||||||
assert_str_eq!(*description, "edit indexer");
|
|
||||||
|
|
||||||
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.settings);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.settings.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
|
||||||
|
|
||||||
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.test);
|
|
||||||
assert_str_eq!(*description, "test indexer");
|
|
||||||
|
|
||||||
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.test_all);
|
|
||||||
assert_str_eq!(*description, "test all indexers");
|
|
||||||
|
|
||||||
let (key_binding, description) = indexers_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
|
||||||
assert_eq!(indexers_context_clues_iter.next(), None);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_system_context_clues() {
|
|
||||||
let mut system_context_clues_iter = SYSTEM_CONTEXT_CLUES.iter();
|
|
||||||
|
|
||||||
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.tasks);
|
|
||||||
assert_str_eq!(*description, "open tasks");
|
|
||||||
|
|
||||||
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.events);
|
|
||||||
assert_str_eq!(*description, "open events");
|
|
||||||
|
|
||||||
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.logs);
|
|
||||||
assert_str_eq!(*description, "open logs");
|
|
||||||
|
|
||||||
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.update);
|
|
||||||
assert_str_eq!(*description, "open updates");
|
|
||||||
|
|
||||||
let (key_binding, description) = system_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
|
||||||
assert_eq!(system_context_clues_iter.next(), None);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_movie_details_context_clues() {
|
fn test_movie_details_context_clues() {
|
||||||
let mut movie_details_context_clues_iter = MOVIE_DETAILS_CONTEXT_CLUES.iter();
|
let mut movie_details_context_clues_iter = MOVIE_DETAILS_CONTEXT_CLUES.iter();
|
||||||
@@ -269,8 +133,8 @@ mod tests {
|
|||||||
|
|
||||||
let (key_binding, description) = movie_details_context_clues_iter.next().unwrap();
|
let (key_binding, description) = movie_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
assert_str_eq!(*description, "auto search");
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
let (key_binding, description) = movie_details_context_clues_iter.next().unwrap();
|
let (key_binding, description) = movie_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
@@ -305,8 +169,8 @@ mod tests {
|
|||||||
|
|
||||||
let (key_binding, description) = manual_movie_search_context_clues_iter.next().unwrap();
|
let (key_binding, description) = manual_movie_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
assert_str_eq!(*description, "auto search");
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
let (key_binding, description) = manual_movie_search_context_clues_iter.next().unwrap();
|
let (key_binding, description) = manual_movie_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
@@ -349,22 +213,6 @@ mod tests {
|
|||||||
assert_eq!(add_movie_search_results_context_clues_iter.next(), None);
|
assert_eq!(add_movie_search_results_context_clues_iter.next(), None);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_confirmation_prompt_context_clues() {
|
|
||||||
let mut confirmation_prompt_context_clues_iter = CONFIRMATION_PROMPT_CONTEXT_CLUES.iter();
|
|
||||||
|
|
||||||
let (key_binding, description) = confirmation_prompt_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.confirm);
|
|
||||||
assert_str_eq!(*description, "submit");
|
|
||||||
|
|
||||||
let (key_binding, description) = confirmation_prompt_context_clues_iter.next().unwrap();
|
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
|
||||||
assert_str_eq!(*description, "cancel");
|
|
||||||
assert_eq!(confirmation_prompt_context_clues_iter.next(), None);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_system_tasks_context_clues() {
|
fn test_system_tasks_context_clues() {
|
||||||
let mut system_tasks_context_clues_iter = SYSTEM_TASKS_CONTEXT_CLUES.iter();
|
let mut system_tasks_context_clues_iter = SYSTEM_TASKS_CONTEXT_CLUES.iter();
|
||||||
@@ -392,6 +240,11 @@ mod tests {
|
|||||||
|
|
||||||
let (key_binding, description) = collection_details_context_clues_iter.next().unwrap();
|
let (key_binding, description) = collection_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.edit);
|
||||||
|
assert_str_eq!(*description, "edit collection");
|
||||||
|
|
||||||
|
let (key_binding, description) = collection_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
assert_eq!(collection_details_context_clues_iter.next(), None);
|
assert_eq!(collection_details_context_clues_iter.next(), None);
|
||||||
|
|||||||
@@ -0,0 +1,306 @@
|
|||||||
|
use crate::{
|
||||||
|
models::servarr_data::sonarr::sonarr_data::ActiveSonarrBlock,
|
||||||
|
network::sonarr_network::SonarrEvent,
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::App;
|
||||||
|
|
||||||
|
pub mod sonarr_context_clues;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "sonarr_tests.rs"]
|
||||||
|
mod sonarr_tests;
|
||||||
|
|
||||||
|
impl App<'_> {
|
||||||
|
pub(super) async fn dispatch_by_sonarr_block(&mut self, active_sonarr_block: &ActiveSonarrBlock) {
|
||||||
|
match active_sonarr_block {
|
||||||
|
ActiveSonarrBlock::Series => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetQualityProfiles.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetLanguageProfiles.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetTags.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::ListSeries.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::SeriesDetails => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::ListSeries.into())
|
||||||
|
.await;
|
||||||
|
self.is_loading = true;
|
||||||
|
self.populate_seasons_table().await;
|
||||||
|
self.is_loading = false;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::SeriesHistory => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::GetSeriesHistory(self.extract_series_id().await).into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::SeasonDetails => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetEpisodes(self.extract_series_id().await).into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::GetEpisodeFiles(self.extract_series_id().await).into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetDownloads.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::SeasonHistory => {
|
||||||
|
if !self.data.sonarr_data.seasons.is_empty() {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::GetSeasonHistory(self.extract_series_id_season_number_tuple().await)
|
||||||
|
.into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::ManualSeasonSearch => {
|
||||||
|
match self.data.sonarr_data.season_details_modal.as_ref() {
|
||||||
|
Some(season_details_modal) if season_details_modal.season_releases.is_empty() => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::GetSeasonReleases(self.extract_series_id_season_number_tuple().await)
|
||||||
|
.into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::EpisodeDetails | ActiveSonarrBlock::EpisodeFile => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::GetEpisodeDetails(self.extract_episode_id().await).into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::EpisodeHistory => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::GetEpisodeHistory(self.extract_episode_id().await).into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::ManualEpisodeSearch => {
|
||||||
|
if let Some(season_details_modal) = self.data.sonarr_data.season_details_modal.as_ref() {
|
||||||
|
if let Some(episode_details_modal) = season_details_modal.episode_details_modal.as_ref() {
|
||||||
|
if episode_details_modal.episode_releases.is_empty() {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::GetEpisodeReleases(self.extract_episode_id().await).into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::Downloads => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetDownloads.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::Blocklist => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::ListSeries.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetBlocklist.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::History => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetHistory(500).into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::RootFolders => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetRootFolders.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::Indexers => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetTags.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetIndexers.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::AllIndexerSettingsPrompt => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetAllIndexerSettings.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::TestIndexer => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::TestIndexer(self.extract_sonarr_indexer_id().await).into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::TestAllIndexers => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::TestAllIndexers.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::System => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetTasks.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetQueuedEvents.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetLogs(500).into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::AddSeriesSearchResults => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(
|
||||||
|
SonarrEvent::SearchNewSeries(self.extract_add_new_series_search_query().await).into(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
ActiveSonarrBlock::SystemUpdates => {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetUpdates.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
|
||||||
|
self.check_for_sonarr_prompt_action().await;
|
||||||
|
self.reset_tick_count();
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn check_for_sonarr_prompt_action(&mut self) {
|
||||||
|
if self.data.sonarr_data.prompt_confirm {
|
||||||
|
self.data.sonarr_data.prompt_confirm = false;
|
||||||
|
if let Some(sonarr_event) = self.data.sonarr_data.prompt_confirm_action.take() {
|
||||||
|
self.dispatch_network_event(sonarr_event.into()).await;
|
||||||
|
self.should_refresh = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) async fn sonarr_on_tick(&mut self, active_sonarr_block: ActiveSonarrBlock) {
|
||||||
|
if self.is_first_render {
|
||||||
|
self.refresh_sonarr_metadata().await;
|
||||||
|
self.dispatch_by_sonarr_block(&active_sonarr_block).await;
|
||||||
|
self.is_first_render = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.should_refresh {
|
||||||
|
self.dispatch_by_sonarr_block(&active_sonarr_block).await;
|
||||||
|
self.refresh_sonarr_metadata().await;
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.is_routing {
|
||||||
|
if !self.should_refresh {
|
||||||
|
self.cancellation_token.cancel();
|
||||||
|
} else {
|
||||||
|
self.dispatch_by_sonarr_block(&active_sonarr_block).await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.tick_count % self.tick_until_poll == 0 {
|
||||||
|
self.refresh_sonarr_metadata().await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn refresh_sonarr_metadata(&mut self) {
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetQualityProfiles.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetLanguageProfiles.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetTags.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetRootFolders.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetDownloads.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetDiskSpace.into())
|
||||||
|
.await;
|
||||||
|
self
|
||||||
|
.dispatch_network_event(SonarrEvent::GetStatus.into())
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn populate_seasons_table(&mut self) {
|
||||||
|
let seasons = self
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.series
|
||||||
|
.current_selection()
|
||||||
|
.clone()
|
||||||
|
.seasons
|
||||||
|
.unwrap_or_default()
|
||||||
|
.into_iter()
|
||||||
|
.map(|mut season| {
|
||||||
|
season.title = Some(format!("Season {}", season.season_number));
|
||||||
|
season
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
self.data.sonarr_data.seasons.set_items(seasons);
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn extract_episode_id(&self) -> i64 {
|
||||||
|
self
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.season_details_modal
|
||||||
|
.as_ref()
|
||||||
|
.expect("Season details have not been loaded")
|
||||||
|
.episodes
|
||||||
|
.current_selection()
|
||||||
|
.id
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn extract_series_id(&self) -> i64 {
|
||||||
|
self.data.sonarr_data.series.current_selection().id
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn extract_series_id_season_number_tuple(&self) -> (i64, i64) {
|
||||||
|
let series_id = self.data.sonarr_data.series.current_selection().id;
|
||||||
|
let season_number = self
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.seasons
|
||||||
|
.current_selection()
|
||||||
|
.season_number;
|
||||||
|
(series_id, season_number)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn extract_add_new_series_search_query(&self) -> String {
|
||||||
|
self
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.add_series_search
|
||||||
|
.as_ref()
|
||||||
|
.expect("Add series search is empty")
|
||||||
|
.text
|
||||||
|
.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn extract_sonarr_indexer_id(&self) -> i64 {
|
||||||
|
self.data.sonarr_data.indexers.current_selection().id
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,159 @@
|
|||||||
|
use crate::app::{context_clues::ContextClue, key_binding::DEFAULT_KEYBINDINGS};
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "sonarr_context_clues_tests.rs"]
|
||||||
|
mod sonarr_context_clues_tests;
|
||||||
|
|
||||||
|
pub static ADD_SERIES_SEARCH_RESULTS_CONTEXT_CLUES: [ContextClue; 2] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "details"),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, "edit search"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SERIES_CONTEXT_CLUES: [ContextClue; 10] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.add, DEFAULT_KEYBINDINGS.add.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.edit, DEFAULT_KEYBINDINGS.edit.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.delete, DEFAULT_KEYBINDINGS.delete.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.search, DEFAULT_KEYBINDINGS.search.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.filter, DEFAULT_KEYBINDINGS.filter.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.update, "update all"),
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "details"),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, "cancel filter"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SERIES_DETAILS_CONTEXT_CLUES: [ContextClue; 8] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.edit, DEFAULT_KEYBINDINGS.edit.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.toggle_monitoring,
|
||||||
|
DEFAULT_KEYBINDINGS.toggle_monitoring.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "season details"),
|
||||||
|
(DEFAULT_KEYBINDINGS.search, DEFAULT_KEYBINDINGS.search.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.update, DEFAULT_KEYBINDINGS.update.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static HISTORY_CONTEXT_CLUES: [ContextClue; 6] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "details"),
|
||||||
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.search, DEFAULT_KEYBINDINGS.search.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.filter, DEFAULT_KEYBINDINGS.filter.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, "cancel filter"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SERIES_HISTORY_CONTEXT_CLUES: [ContextClue; 9] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.edit, DEFAULT_KEYBINDINGS.edit.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "details"),
|
||||||
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.search, DEFAULT_KEYBINDINGS.search.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.filter, DEFAULT_KEYBINDINGS.filter.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.update, DEFAULT_KEYBINDINGS.update.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, "cancel filter/close"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SEASON_DETAILS_CONTEXTUAL_CONTEXT_CLUES: [ContextClue; 2] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "episode details"),
|
||||||
|
(DEFAULT_KEYBINDINGS.delete, "delete episode"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SEASON_DETAILS_CONTEXT_CLUES: [ContextClue; 5] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.toggle_monitoring,
|
||||||
|
DEFAULT_KEYBINDINGS.toggle_monitoring.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.search, DEFAULT_KEYBINDINGS.search.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SEASON_HISTORY_CONTEXT_CLUES: [ContextClue; 6] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.search, DEFAULT_KEYBINDINGS.search.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.filter, DEFAULT_KEYBINDINGS.filter.desc),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, "cancel filter/close"),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static MANUAL_SEASON_SEARCH_CONTEXT_CLUES: [ContextClue; 4] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static MANUAL_EPISODE_SEARCH_CONTEXT_CLUES: [ContextClue; 4] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.sort, DEFAULT_KEYBINDINGS.sort.desc),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static DETAILS_CONTEXTUAL_CONTEXT_CLUES: [ContextClue; 1] =
|
||||||
|
[(DEFAULT_KEYBINDINGS.submit, "details")];
|
||||||
|
|
||||||
|
pub static EPISODE_DETAILS_CONTEXT_CLUES: [ContextClue; 3] = [
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.refresh,
|
||||||
|
DEFAULT_KEYBINDINGS.refresh.desc,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search,
|
||||||
|
DEFAULT_KEYBINDINGS.auto_search.desc,
|
||||||
|
),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static SYSTEM_TASKS_CONTEXT_CLUES: [ContextClue; 2] = [
|
||||||
|
(DEFAULT_KEYBINDINGS.submit, "start task"),
|
||||||
|
(DEFAULT_KEYBINDINGS.esc, DEFAULT_KEYBINDINGS.esc.desc),
|
||||||
|
];
|
||||||
@@ -0,0 +1,402 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use pretty_assertions::{assert_eq, assert_str_eq};
|
||||||
|
|
||||||
|
use crate::app::{
|
||||||
|
key_binding::DEFAULT_KEYBINDINGS,
|
||||||
|
sonarr::sonarr_context_clues::{
|
||||||
|
ADD_SERIES_SEARCH_RESULTS_CONTEXT_CLUES, DETAILS_CONTEXTUAL_CONTEXT_CLUES,
|
||||||
|
EPISODE_DETAILS_CONTEXT_CLUES, HISTORY_CONTEXT_CLUES, MANUAL_EPISODE_SEARCH_CONTEXT_CLUES,
|
||||||
|
MANUAL_SEASON_SEARCH_CONTEXT_CLUES, SEASON_DETAILS_CONTEXTUAL_CONTEXT_CLUES,
|
||||||
|
SEASON_DETAILS_CONTEXT_CLUES, SEASON_HISTORY_CONTEXT_CLUES, SERIES_CONTEXT_CLUES,
|
||||||
|
SERIES_DETAILS_CONTEXT_CLUES, SERIES_HISTORY_CONTEXT_CLUES, SYSTEM_TASKS_CONTEXT_CLUES,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_search_results_context_clues() {
|
||||||
|
let mut add_series_search_results_context_clues_iter =
|
||||||
|
ADD_SERIES_SEARCH_RESULTS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = add_series_search_results_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "details");
|
||||||
|
|
||||||
|
let (key_binding, description) = add_series_search_results_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, "edit search");
|
||||||
|
assert_eq!(add_series_search_results_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_series_context_clues() {
|
||||||
|
let mut series_context_clues_iter = SERIES_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.add);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.add.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.edit);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.edit.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.delete.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.filter);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.filter.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.update);
|
||||||
|
assert_str_eq!(*description, "update all");
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "details");
|
||||||
|
|
||||||
|
let (key_binding, description) = series_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, "cancel filter");
|
||||||
|
assert_eq!(series_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_series_history_context_clues() {
|
||||||
|
let mut series_history_context_clues_iter = SERIES_HISTORY_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.edit);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.edit.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "details");
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.filter);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.filter.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.update);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.update.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, "cancel filter/close");
|
||||||
|
assert_eq!(series_history_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_history_context_clues() {
|
||||||
|
let mut history_context_clues_iter = HISTORY_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "details");
|
||||||
|
|
||||||
|
let (key_binding, description) = history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.filter);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.filter.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, "cancel filter");
|
||||||
|
assert_eq!(history_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_series_details_context_clues() {
|
||||||
|
let mut series_details_context_clues_iter = SERIES_DETAILS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.edit);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.edit.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.toggle_monitoring);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.toggle_monitoring.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "season details");
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.update);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.update.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = series_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
|
assert_eq!(series_details_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_season_details_context_clues() {
|
||||||
|
let mut season_details_context_clues_iter = SEASON_DETAILS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = season_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.toggle_monitoring);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.toggle_monitoring.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
|
assert_eq!(season_details_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_season_details_contextual_context_clues() {
|
||||||
|
let mut season_details_contextual_context_clues_iter =
|
||||||
|
SEASON_DETAILS_CONTEXTUAL_CONTEXT_CLUES.iter();
|
||||||
|
let (key_binding, description) = season_details_contextual_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "episode details");
|
||||||
|
|
||||||
|
let (key_binding, description) = season_details_contextual_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.delete);
|
||||||
|
assert_str_eq!(*description, "delete episode");
|
||||||
|
assert_eq!(season_details_contextual_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_season_history_context_clues() {
|
||||||
|
let mut season_history_context_clues_iter = SEASON_HISTORY_CONTEXT_CLUES.iter();
|
||||||
|
let (key_binding, description) = season_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.filter);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.filter.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = season_history_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, "cancel filter/close");
|
||||||
|
assert_eq!(season_history_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_manual_season_search_context_clues() {
|
||||||
|
let mut manual_season_search_context_clues_iter = MANUAL_SEASON_SEARCH_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_season_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_season_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_season_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_season_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
|
assert_eq!(manual_season_search_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_manual_episode_search_context_clues() {
|
||||||
|
let mut manual_episode_search_context_clues_iter = MANUAL_EPISODE_SEARCH_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_episode_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_episode_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_episode_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.sort);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.sort.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = manual_episode_search_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
|
assert_eq!(manual_episode_search_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn details_contextual_context_clues() {
|
||||||
|
let mut manual_search_contextual_context_clues_iter = DETAILS_CONTEXTUAL_CONTEXT_CLUES.iter();
|
||||||
|
let (key_binding, description) = manual_search_contextual_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "details");
|
||||||
|
assert_eq!(manual_search_contextual_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_episode_details_context_clues() {
|
||||||
|
let mut episode_details_context_clues_iter = EPISODE_DETAILS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = episode_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.refresh);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.refresh.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = episode_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.auto_search);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.auto_search.desc);
|
||||||
|
|
||||||
|
let (key_binding, description) = episode_details_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
|
assert_eq!(episode_details_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_system_tasks_context_clues() {
|
||||||
|
let mut system_tasks_context_clues_iter = SYSTEM_TASKS_CONTEXT_CLUES.iter();
|
||||||
|
|
||||||
|
let (key_binding, description) = system_tasks_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.submit);
|
||||||
|
assert_str_eq!(*description, "start task");
|
||||||
|
|
||||||
|
let (key_binding, description) = system_tasks_context_clues_iter.next().unwrap();
|
||||||
|
|
||||||
|
assert_eq!(*key_binding, DEFAULT_KEYBINDINGS.esc);
|
||||||
|
assert_str_eq!(*description, DEFAULT_KEYBINDINGS.esc.desc);
|
||||||
|
assert_eq!(system_tasks_context_clues_iter.next(), None);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,881 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
mod sonarr_tests {
|
||||||
|
use pretty_assertions::{assert_eq, assert_str_eq};
|
||||||
|
use tokio::sync::mpsc;
|
||||||
|
|
||||||
|
use crate::models::servarr_data::sonarr::sonarr_data::sonarr_test_utils::utils::create_test_sonarr_data;
|
||||||
|
use crate::models::servarr_models::Indexer;
|
||||||
|
use crate::models::sonarr_models::Episode;
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
models::{
|
||||||
|
servarr_data::sonarr::{
|
||||||
|
modals::{EpisodeDetailsModal, SeasonDetailsModal},
|
||||||
|
sonarr_data::ActiveSonarrBlock,
|
||||||
|
},
|
||||||
|
sonarr_models::{Season, Series, SonarrRelease},
|
||||||
|
},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_blocklist_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::Blocklist)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::ListSeries.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetBlocklist.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_series_history_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
id: 1,
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::SeriesHistory)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetSeriesHistory(1).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_series_details_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
seasons: Some(vec![Season::default()]),
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::SeriesDetails)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(!app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::ListSeries.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.seasons.items.is_empty());
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_season_details_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
id: 1,
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::SeasonDetails)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetEpisodes(1).into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetEpisodeFiles(1).into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_season_history_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
id: 1,
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
app.data.sonarr_data.seasons.set_items(vec![Season {
|
||||||
|
season_number: 1,
|
||||||
|
..Season::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::SeasonHistory)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetSeasonHistory((1, 1)).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_season_history_block_no_op_when_seasons_table_is_empty() {
|
||||||
|
let (mut app, _) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
id: 1,
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::SeasonHistory)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(!app.is_loading);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_manual_season_search_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.season_details_modal = Some(SeasonDetailsModal::default());
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
id: 1,
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
app.data.sonarr_data.seasons.set_items(vec![Season {
|
||||||
|
season_number: 1,
|
||||||
|
..Season::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::ManualSeasonSearch)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetSeasonReleases((1, 1)).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_manual_season_search_block_is_loading() {
|
||||||
|
let mut app = App {
|
||||||
|
is_loading: true,
|
||||||
|
..App::test_default()
|
||||||
|
};
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::ManualSeasonSearch)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_manual_season_search_block_season_releases_non_empty() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
let mut season_details_modal = SeasonDetailsModal::default();
|
||||||
|
season_details_modal
|
||||||
|
.season_releases
|
||||||
|
.set_items(vec![SonarrRelease::default()]);
|
||||||
|
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::ManualSeasonSearch)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(!app.is_loading);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_episode_details_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data = create_test_sonarr_data();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::EpisodeDetails)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetEpisodeDetails(0).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_episode_file_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data = create_test_sonarr_data();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::EpisodeFile)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetEpisodeDetails(0).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_episode_history_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
let mut season_details_modal = SeasonDetailsModal::default();
|
||||||
|
season_details_modal.episodes.set_items(vec![Episode {
|
||||||
|
id: 1,
|
||||||
|
..Episode::default()
|
||||||
|
}]);
|
||||||
|
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::EpisodeHistory)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetEpisodeHistory(1).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_manual_episode_search_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
let mut season_details_modal = SeasonDetailsModal {
|
||||||
|
episode_details_modal: Some(EpisodeDetailsModal::default()),
|
||||||
|
..SeasonDetailsModal::default()
|
||||||
|
};
|
||||||
|
season_details_modal.episodes.set_items(vec![Episode {
|
||||||
|
id: 1,
|
||||||
|
..Episode::default()
|
||||||
|
}]);
|
||||||
|
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::ManualEpisodeSearch)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetEpisodeReleases(1).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_manual_episode_search_block_is_loading() {
|
||||||
|
let mut app = App {
|
||||||
|
is_loading: true,
|
||||||
|
..App::test_default()
|
||||||
|
};
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::ManualEpisodeSearch)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_manual_episode_search_block_episode_releases_non_empty() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
let mut episode_details_modal = EpisodeDetailsModal::default();
|
||||||
|
episode_details_modal
|
||||||
|
.episode_releases
|
||||||
|
.set_items(vec![SonarrRelease::default()]);
|
||||||
|
let season_details_modal = SeasonDetailsModal {
|
||||||
|
episode_details_modal: Some(episode_details_modal),
|
||||||
|
..SeasonDetailsModal::default()
|
||||||
|
};
|
||||||
|
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::ManualEpisodeSearch)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(!app.is_loading);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_history_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::History)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetHistory(500).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_downloads_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::Downloads)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_root_folders_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::RootFolders)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetRootFolders.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_series_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::Series)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetQualityProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetLanguageProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetTags.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::ListSeries.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_indexers_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::Indexers)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetTags.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetIndexers.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_all_indexer_settings_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::AllIndexerSettingsPrompt)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetAllIndexerSettings.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_test_indexer_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.indexers.set_items(vec![Indexer {
|
||||||
|
id: 1,
|
||||||
|
..Indexer::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::TestIndexer)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::TestIndexer(1).into()
|
||||||
|
);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_test_all_indexers_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::TestAllIndexers)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::TestAllIndexers.into()
|
||||||
|
);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_system_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::System)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetTasks.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetQueuedEvents.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetLogs(500).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_system_updates_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::SystemUpdates)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetUpdates.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_dispatch_by_add_series_search_results_block() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.add_series_search = Some("test search".into());
|
||||||
|
|
||||||
|
app
|
||||||
|
.dispatch_by_sonarr_block(&ActiveSonarrBlock::AddSeriesSearchResults)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::SearchNewSeries("test search".into()).into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(app.tick_count, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_check_for_sonarr_prompt_action_no_prompt_confirm() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.sonarr_data.prompt_confirm = false;
|
||||||
|
|
||||||
|
app.check_for_sonarr_prompt_action().await;
|
||||||
|
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert!(!app.should_refresh);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_check_for_sonarr_prompt_action() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.sonarr_data.prompt_confirm_action = Some(SonarrEvent::GetStatus);
|
||||||
|
|
||||||
|
app.check_for_sonarr_prompt_action().await;
|
||||||
|
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetStatus.into()
|
||||||
|
);
|
||||||
|
assert!(app.should_refresh);
|
||||||
|
assert_eq!(app.data.sonarr_data.prompt_confirm_action, None);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_refresh_metadata() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.is_routing = true;
|
||||||
|
|
||||||
|
app.refresh_sonarr_metadata().await;
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetQualityProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetLanguageProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetTags.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetRootFolders.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDiskSpace.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetStatus.into()
|
||||||
|
);
|
||||||
|
assert!(app.is_loading);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_on_tick_first_render() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.is_first_render = true;
|
||||||
|
|
||||||
|
app.sonarr_on_tick(ActiveSonarrBlock::Downloads).await;
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetQualityProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetLanguageProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetTags.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetRootFolders.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDiskSpace.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetStatus.into()
|
||||||
|
);
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert!(!app.is_first_render);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_on_tick_routing() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.is_routing = true;
|
||||||
|
app.should_refresh = true;
|
||||||
|
|
||||||
|
app.sonarr_on_tick(ActiveSonarrBlock::Downloads).await;
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_on_tick_routing_while_long_request_is_running_should_cancel_request() {
|
||||||
|
let (mut app, _) = construct_app_unit();
|
||||||
|
app.is_routing = true;
|
||||||
|
app.should_refresh = false;
|
||||||
|
|
||||||
|
app.sonarr_on_tick(ActiveSonarrBlock::Downloads).await;
|
||||||
|
|
||||||
|
assert!(app.cancellation_token.is_cancelled());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_on_tick_should_refresh() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.should_refresh = true;
|
||||||
|
|
||||||
|
app.sonarr_on_tick(ActiveSonarrBlock::Downloads).await;
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert!(app.should_refresh);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_on_tick_should_refresh_does_not_cancel_prompt_requests() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.is_loading = true;
|
||||||
|
app.is_routing = true;
|
||||||
|
app.should_refresh = true;
|
||||||
|
|
||||||
|
app.sonarr_on_tick(ActiveSonarrBlock::Downloads).await;
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert!(app.is_loading);
|
||||||
|
assert!(app.should_refresh);
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
assert!(!app.cancellation_token.is_cancelled());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_on_tick_network_tick_frequency() {
|
||||||
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.tick_count = 2;
|
||||||
|
app.tick_until_poll = 2;
|
||||||
|
|
||||||
|
app.sonarr_on_tick(ActiveSonarrBlock::Downloads).await;
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetQualityProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetLanguageProfiles.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetTags.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetRootFolders.into()
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
sync_network_rx.recv().await.unwrap(),
|
||||||
|
SonarrEvent::GetDownloads.into()
|
||||||
|
);
|
||||||
|
assert!(app.is_loading);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_populate_seasons_table_unfiltered() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
seasons: Some(vec![Season::default()]),
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app.populate_seasons_table().await;
|
||||||
|
|
||||||
|
assert!(!app.data.sonarr_data.seasons.items.is_empty());
|
||||||
|
assert_str_eq!(
|
||||||
|
app.data.sonarr_data.seasons.items[0]
|
||||||
|
.title
|
||||||
|
.as_ref()
|
||||||
|
.unwrap(),
|
||||||
|
"Season 0"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_populate_seasons_table_filtered() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.sonarr_data.series.set_filtered_items(vec![Series {
|
||||||
|
seasons: Some(vec![Season::default()]),
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
app.populate_seasons_table().await;
|
||||||
|
|
||||||
|
assert!(!app.data.sonarr_data.seasons.items.is_empty());
|
||||||
|
assert_str_eq!(
|
||||||
|
app.data.sonarr_data.seasons.items[0]
|
||||||
|
.title
|
||||||
|
.as_ref()
|
||||||
|
.unwrap(),
|
||||||
|
"Season 0"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_extract_episode_id() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
let mut season_details_modal = SeasonDetailsModal::default();
|
||||||
|
season_details_modal.episodes.set_items(vec![Episode {
|
||||||
|
id: 1,
|
||||||
|
..Episode::default()
|
||||||
|
}]);
|
||||||
|
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
|
||||||
|
|
||||||
|
assert_eq!(app.extract_episode_id().await, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
#[should_panic(expected = "Season details have not been loaded")]
|
||||||
|
async fn test_extract_episode_id_requires_season_details_modal_to_be_some() {
|
||||||
|
let app = App::test_default();
|
||||||
|
|
||||||
|
assert_eq!(app.extract_episode_id().await, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_extract_series_id() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
id: 1,
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
assert_eq!(app.extract_series_id().await, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_extract_series_id_season_number_tuple() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.sonarr_data.series.set_items(vec![Series {
|
||||||
|
id: 1,
|
||||||
|
..Series::default()
|
||||||
|
}]);
|
||||||
|
app.data.sonarr_data.seasons.set_items(vec![Season {
|
||||||
|
season_number: 1,
|
||||||
|
..Season::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
assert_eq!(app.extract_series_id_season_number_tuple().await, (1, 1));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_extract_add_new_series_search_query() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.sonarr_data.add_series_search = Some("test search".into());
|
||||||
|
|
||||||
|
assert_str_eq!(
|
||||||
|
app.extract_add_new_series_search_query().await,
|
||||||
|
"test search"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
#[should_panic(expected = "Add series search is empty")]
|
||||||
|
async fn test_extract_add_new_series_search_query_panics_when_the_query_is_not_set() {
|
||||||
|
let app = App::test_default();
|
||||||
|
|
||||||
|
app.extract_add_new_series_search_query().await;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_extract_sonarr_indexer_id() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.sonarr_data.indexers.set_items(vec![Indexer {
|
||||||
|
id: 1,
|
||||||
|
..Indexer::default()
|
||||||
|
}]);
|
||||||
|
|
||||||
|
assert_eq!(app.extract_sonarr_indexer_id().await, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn construct_app_unit<'a>() -> (App<'a>, mpsc::Receiver<NetworkEvent>) {
|
||||||
|
let (sync_network_tx, sync_network_rx) = mpsc::channel::<NetworkEvent>(500);
|
||||||
|
let mut app = App {
|
||||||
|
network_tx: Some(sync_network_tx),
|
||||||
|
tick_count: 1,
|
||||||
|
is_first_render: false,
|
||||||
|
..App::test_default()
|
||||||
|
};
|
||||||
|
app.data.sonarr_data.prompt_confirm = true;
|
||||||
|
|
||||||
|
(app, sync_network_rx)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -10,19 +10,28 @@ mod tests {
|
|||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
app::App,
|
app::App,
|
||||||
cli::{handle_command, mutex_flags_or_option, radarr::RadarrCommand},
|
cli::{handle_command, mutex_flags_or_option, radarr::RadarrCommand, sonarr::SonarrCommand},
|
||||||
models::{
|
models::{
|
||||||
radarr_models::{BlocklistItem, BlocklistResponse, RadarrSerdeable},
|
radarr_models::{
|
||||||
|
BlocklistItem as RadarrBlocklistItem, BlocklistResponse as RadarrBlocklistResponse,
|
||||||
|
RadarrSerdeable,
|
||||||
|
},
|
||||||
|
sonarr_models::{
|
||||||
|
BlocklistItem as SonarrBlocklistItem, BlocklistResponse as SonarrBlocklistResponse,
|
||||||
|
SonarrSerdeable,
|
||||||
|
},
|
||||||
Serdeable,
|
Serdeable,
|
||||||
},
|
},
|
||||||
network::{radarr_network::RadarrEvent, MockNetworkTrait, NetworkEvent},
|
network::{
|
||||||
|
radarr_network::RadarrEvent, sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent,
|
||||||
|
},
|
||||||
Cli,
|
Cli,
|
||||||
};
|
};
|
||||||
use pretty_assertions::assert_eq;
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
#[test]
|
#[rstest]
|
||||||
fn test_radarr_subcommand_requires_subcommand() {
|
fn test_servarr_subcommand_requires_subcommand(#[values("radarr", "sonarr")] subcommand: &str) {
|
||||||
let result = Cli::command().try_get_matches_from(["managarr", "radarr"]);
|
let result = Cli::command().try_get_matches_from(["managarr", subcommand]);
|
||||||
|
|
||||||
assert!(result.is_err());
|
assert!(result.is_err());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
@@ -39,6 +48,13 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_subcommand_delegates_to_sonarr() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "series"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_completions_requires_argument() {
|
fn test_completions_requires_argument() {
|
||||||
let result = Cli::command().try_get_matches_from(["managarr", "completions"]);
|
let result = Cli::command().try_get_matches_from(["managarr", "completions"]);
|
||||||
@@ -106,8 +122,8 @@ mod tests {
|
|||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
Ok(Serdeable::Radarr(RadarrSerdeable::BlocklistResponse(
|
Ok(Serdeable::Radarr(RadarrSerdeable::BlocklistResponse(
|
||||||
BlocklistResponse {
|
RadarrBlocklistResponse {
|
||||||
records: vec![BlocklistItem::default()],
|
records: vec![RadarrBlocklistItem::default()],
|
||||||
},
|
},
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
@@ -120,10 +136,41 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let claer_blocklist_command = RadarrCommand::ClearBlocklist.into();
|
let clear_blocklist_command = RadarrCommand::ClearBlocklist.into();
|
||||||
|
|
||||||
let result = handle_command(&app_arc, claer_blocklist_command, &mut mock_network).await;
|
let result = handle_command(&app_arc, clear_blocklist_command, &mut mock_network).await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_cli_handler_delegates_sonarr_commands_to_the_sonarr_cli_handler() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::GetBlocklist.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::BlocklistResponse(
|
||||||
|
SonarrBlocklistResponse {
|
||||||
|
records: vec![SonarrBlocklistItem::default()],
|
||||||
|
},
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::ClearBlocklist.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let clear_blocklist_command = SonarrCommand::ClearBlocklist.into();
|
||||||
|
|
||||||
|
let result = handle_command(&app_arc, clear_blocklist_command, &mut mock_network).await;
|
||||||
|
|
||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,11 +4,13 @@ use anyhow::Result;
|
|||||||
use clap::{command, Subcommand};
|
use clap::{command, Subcommand};
|
||||||
use clap_complete::Shell;
|
use clap_complete::Shell;
|
||||||
use radarr::{RadarrCliHandler, RadarrCommand};
|
use radarr::{RadarrCliHandler, RadarrCommand};
|
||||||
|
use sonarr::{SonarrCliHandler, SonarrCommand};
|
||||||
use tokio::sync::Mutex;
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
use crate::{app::App, network::NetworkTrait};
|
use crate::{app::App, network::NetworkTrait};
|
||||||
|
|
||||||
pub mod radarr;
|
pub mod radarr;
|
||||||
|
pub mod sonarr;
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[path = "cli_tests.rs"]
|
#[path = "cli_tests.rs"]
|
||||||
@@ -19,6 +21,9 @@ pub enum Command {
|
|||||||
#[command(subcommand, about = "Commands for manging your Radarr instance")]
|
#[command(subcommand, about = "Commands for manging your Radarr instance")]
|
||||||
Radarr(RadarrCommand),
|
Radarr(RadarrCommand),
|
||||||
|
|
||||||
|
#[command(subcommand, about = "Commands for manging your Sonarr instance")]
|
||||||
|
Sonarr(SonarrCommand),
|
||||||
|
|
||||||
#[command(
|
#[command(
|
||||||
arg_required_else_help = true,
|
arg_required_else_help = true,
|
||||||
about = "Generate shell completions for the Managarr CLI"
|
about = "Generate shell completions for the Managarr CLI"
|
||||||
@@ -27,24 +32,39 @@ pub enum Command {
|
|||||||
#[arg(value_enum)]
|
#[arg(value_enum)]
|
||||||
shell: Shell,
|
shell: Shell,
|
||||||
},
|
},
|
||||||
|
|
||||||
|
#[command(about = "Tail Managarr logs")]
|
||||||
|
TailLogs {
|
||||||
|
#[arg(long, help = "Disable colored log output")]
|
||||||
|
no_color: bool,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait CliCommandHandler<'a, 'b, T: Into<Command>> {
|
pub trait CliCommandHandler<'a, 'b, T: Into<Command>> {
|
||||||
fn with(app: &'a Arc<Mutex<App<'b>>>, command: T, network: &'a mut dyn NetworkTrait) -> Self;
|
fn with(app: &'a Arc<Mutex<App<'b>>>, command: T, network: &'a mut dyn NetworkTrait) -> Self;
|
||||||
async fn handle(self) -> Result<()>;
|
async fn handle(self) -> Result<String>;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) async fn handle_command(
|
pub(crate) async fn handle_command(
|
||||||
app: &Arc<Mutex<App<'_>>>,
|
app: &Arc<Mutex<App<'_>>>,
|
||||||
command: Command,
|
command: Command,
|
||||||
network: &mut dyn NetworkTrait,
|
network: &mut dyn NetworkTrait,
|
||||||
) -> Result<()> {
|
) -> Result<String> {
|
||||||
if let Command::Radarr(radarr_command) = command {
|
let result = match command {
|
||||||
RadarrCliHandler::with(app, radarr_command, network)
|
Command::Radarr(radarr_command) => {
|
||||||
.handle()
|
RadarrCliHandler::with(app, radarr_command, network)
|
||||||
.await?
|
.handle()
|
||||||
}
|
.await?
|
||||||
Ok(())
|
}
|
||||||
|
Command::Sonarr(sonarr_command) => {
|
||||||
|
SonarrCliHandler::with(app, sonarr_command, network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
_ => String::new(),
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
@@ -68,16 +88,3 @@ pub fn mutex_flags_or_default(positive: bool, negative: bool, default_value: boo
|
|||||||
default_value
|
default_value
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[macro_export]
|
|
||||||
macro_rules! execute_network_event {
|
|
||||||
($self:ident, $event:expr) => {
|
|
||||||
let resp = $self.network.handle_network_event($event.into()).await?;
|
|
||||||
let json = serde_json::to_string_pretty(&resp)?;
|
|
||||||
println!("{}", json);
|
|
||||||
};
|
|
||||||
($self:ident, $event:expr, $happy_output:expr) => {
|
|
||||||
$self.network.handle_network_event($event.into()).await?;
|
|
||||||
println!("{}", $happy_output);
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -4,16 +4,15 @@ use anyhow::Result;
|
|||||||
use clap::{arg, command, ArgAction, Subcommand};
|
use clap::{arg, command, ArgAction, Subcommand};
|
||||||
use tokio::sync::Mutex;
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use super::RadarrCommand;
|
||||||
|
use crate::models::servarr_models::AddRootFolderBody;
|
||||||
use crate::{
|
use crate::{
|
||||||
app::App,
|
app::App,
|
||||||
cli::{CliCommandHandler, Command},
|
cli::{CliCommandHandler, Command},
|
||||||
execute_network_event,
|
models::radarr_models::{AddMovieBody, AddMovieOptions, MinimumAvailability, MovieMonitor},
|
||||||
models::radarr_models::{AddMovieBody, AddOptions, MinimumAvailability, Monitor},
|
|
||||||
network::{radarr_network::RadarrEvent, NetworkTrait},
|
network::{radarr_network::RadarrEvent, NetworkTrait},
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::RadarrCommand;
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[path = "add_command_handler_tests.rs"]
|
#[path = "add_command_handler_tests.rs"]
|
||||||
mod add_command_handler_tests;
|
mod add_command_handler_tests;
|
||||||
@@ -47,7 +46,7 @@ pub enum RadarrAddCommand {
|
|||||||
default_value_t = MinimumAvailability::default()
|
default_value_t = MinimumAvailability::default()
|
||||||
)]
|
)]
|
||||||
minimum_availability: MinimumAvailability,
|
minimum_availability: MinimumAvailability,
|
||||||
#[arg(long, help = "Should Radarr monitor this film")]
|
#[arg(long, help = "Disable monitoring for this film")]
|
||||||
disable_monitoring: bool,
|
disable_monitoring: bool,
|
||||||
#[arg(
|
#[arg(
|
||||||
long,
|
long,
|
||||||
@@ -60,9 +59,9 @@ pub enum RadarrAddCommand {
|
|||||||
long,
|
long,
|
||||||
help = "What Radarr should monitor",
|
help = "What Radarr should monitor",
|
||||||
value_enum,
|
value_enum,
|
||||||
default_value_t = Monitor::default()
|
default_value_t = MovieMonitor::default()
|
||||||
)]
|
)]
|
||||||
monitor: Monitor,
|
monitor: MovieMonitor,
|
||||||
#[arg(
|
#[arg(
|
||||||
long,
|
long,
|
||||||
help = "Tell Radarr to not start a search for this film once it's added to your library"
|
help = "Tell Radarr to not start a search for this film once it's added to your library"
|
||||||
@@ -106,8 +105,8 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrAddCommand> for RadarrAddCommandHan
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(self) -> Result<()> {
|
async fn handle(self) -> Result<String> {
|
||||||
match self.command {
|
let result = match self.command {
|
||||||
RadarrAddCommand::Movie {
|
RadarrAddCommand::Movie {
|
||||||
tmdb_id,
|
tmdb_id,
|
||||||
root_folder_path,
|
root_folder_path,
|
||||||
@@ -126,24 +125,37 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrAddCommand> for RadarrAddCommandHan
|
|||||||
minimum_availability: minimum_availability.to_string(),
|
minimum_availability: minimum_availability.to_string(),
|
||||||
monitored: !disable_monitoring,
|
monitored: !disable_monitoring,
|
||||||
tags,
|
tags,
|
||||||
add_options: AddOptions {
|
tag_input_string: None,
|
||||||
|
add_options: AddMovieOptions {
|
||||||
monitor: monitor.to_string(),
|
monitor: monitor.to_string(),
|
||||||
search_for_movie: !no_search_for_movie,
|
search_for_movie: !no_search_for_movie,
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
execute_network_event!(self, RadarrEvent::AddMovie(Some(body)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::AddMovie(body).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrAddCommand::RootFolder { root_folder_path } => {
|
RadarrAddCommand::RootFolder { root_folder_path } => {
|
||||||
execute_network_event!(
|
let add_root_folder_body = AddRootFolderBody {
|
||||||
self,
|
path: root_folder_path,
|
||||||
RadarrEvent::AddRootFolder(Some(root_folder_path.clone()))
|
};
|
||||||
);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::AddRootFolder(add_root_folder_body).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrAddCommand::Tag { name } => {
|
RadarrAddCommand::Tag { name } => {
|
||||||
execute_network_event!(self, RadarrEvent::AddTag(name.clone()));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::AddTag(name).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,9 +7,10 @@ mod tests {
|
|||||||
radarr::{add_command_handler::RadarrAddCommand, RadarrCommand},
|
radarr::{add_command_handler::RadarrAddCommand, RadarrCommand},
|
||||||
Command,
|
Command,
|
||||||
},
|
},
|
||||||
models::radarr_models::{MinimumAvailability, Monitor},
|
models::radarr_models::{MinimumAvailability, MovieMonitor},
|
||||||
Cli,
|
Cli,
|
||||||
};
|
};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_radarr_add_command_from() {
|
fn test_radarr_add_command_from() {
|
||||||
@@ -111,6 +112,8 @@ mod tests {
|
|||||||
"/test",
|
"/test",
|
||||||
"--quality-profile-id",
|
"--quality-profile-id",
|
||||||
"1",
|
"1",
|
||||||
|
"--tmdb-id",
|
||||||
|
"1",
|
||||||
flag,
|
flag,
|
||||||
]);
|
]);
|
||||||
|
|
||||||
@@ -187,7 +190,7 @@ mod tests {
|
|||||||
minimum_availability: MinimumAvailability::default(),
|
minimum_availability: MinimumAvailability::default(),
|
||||||
disable_monitoring: false,
|
disable_monitoring: false,
|
||||||
tag: vec![],
|
tag: vec![],
|
||||||
monitor: Monitor::default(),
|
monitor: MovieMonitor::default(),
|
||||||
no_search_for_movie: false,
|
no_search_for_movie: false,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -219,7 +222,7 @@ mod tests {
|
|||||||
minimum_availability: MinimumAvailability::default(),
|
minimum_availability: MinimumAvailability::default(),
|
||||||
disable_monitoring: false,
|
disable_monitoring: false,
|
||||||
tag: vec![1, 2],
|
tag: vec![1, 2],
|
||||||
monitor: Monitor::default(),
|
monitor: MovieMonitor::default(),
|
||||||
no_search_for_movie: false,
|
no_search_for_movie: false,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -255,7 +258,7 @@ mod tests {
|
|||||||
minimum_availability: MinimumAvailability::Released,
|
minimum_availability: MinimumAvailability::Released,
|
||||||
disable_monitoring: true,
|
disable_monitoring: true,
|
||||||
tag: vec![1, 2],
|
tag: vec![1, 2],
|
||||||
monitor: Monitor::MovieAndCollection,
|
monitor: MovieMonitor::MovieAndCollection,
|
||||||
no_search_for_movie: true,
|
no_search_for_movie: true,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -356,7 +359,7 @@ mod tests {
|
|||||||
app::App,
|
app::App,
|
||||||
cli::{radarr::add_command_handler::RadarrAddCommandHandler, CliCommandHandler},
|
cli::{radarr::add_command_handler::RadarrAddCommandHandler, CliCommandHandler},
|
||||||
models::{
|
models::{
|
||||||
radarr_models::{AddMovieBody, AddOptions, RadarrSerdeable},
|
radarr_models::{AddMovieBody, AddMovieOptions, RadarrSerdeable},
|
||||||
Serdeable,
|
Serdeable,
|
||||||
},
|
},
|
||||||
network::{radarr_network::RadarrEvent, MockNetworkTrait, NetworkEvent},
|
network::{radarr_network::RadarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
@@ -365,6 +368,7 @@ mod tests {
|
|||||||
use super::*;
|
use super::*;
|
||||||
use mockall::predicate::eq;
|
use mockall::predicate::eq;
|
||||||
|
|
||||||
|
use crate::models::servarr_models::AddRootFolderBody;
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
use tokio::sync::Mutex;
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
@@ -378,7 +382,8 @@ mod tests {
|
|||||||
minimum_availability: "released".to_owned(),
|
minimum_availability: "released".to_owned(),
|
||||||
monitored: false,
|
monitored: false,
|
||||||
tags: vec![1, 2],
|
tags: vec![1, 2],
|
||||||
add_options: AddOptions {
|
tag_input_string: None,
|
||||||
|
add_options: AddMovieOptions {
|
||||||
monitor: "movieAndCollection".to_owned(),
|
monitor: "movieAndCollection".to_owned(),
|
||||||
search_for_movie: false,
|
search_for_movie: false,
|
||||||
},
|
},
|
||||||
@@ -387,7 +392,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::AddMovie(Some(expected_add_movie_body)).into(),
|
RadarrEvent::AddMovie(expected_add_movie_body).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -395,7 +400,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let add_movie_command = RadarrAddCommand::Movie {
|
let add_movie_command = RadarrAddCommand::Movie {
|
||||||
tmdb_id: 1,
|
tmdb_id: 1,
|
||||||
root_folder_path: "/test".to_owned(),
|
root_folder_path: "/test".to_owned(),
|
||||||
@@ -403,7 +408,7 @@ mod tests {
|
|||||||
minimum_availability: MinimumAvailability::Released,
|
minimum_availability: MinimumAvailability::Released,
|
||||||
disable_monitoring: true,
|
disable_monitoring: true,
|
||||||
tag: vec![1, 2],
|
tag: vec![1, 2],
|
||||||
monitor: Monitor::MovieAndCollection,
|
monitor: MovieMonitor::MovieAndCollection,
|
||||||
no_search_for_movie: true,
|
no_search_for_movie: true,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -417,11 +422,14 @@ mod tests {
|
|||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_handle_add_root_folder_command() {
|
async fn test_handle_add_root_folder_command() {
|
||||||
let expected_root_folder_path = "/nfs/test".to_owned();
|
let expected_root_folder_path = "/nfs/test".to_owned();
|
||||||
|
let expected_add_root_folder_body = AddRootFolderBody {
|
||||||
|
path: expected_root_folder_path.clone(),
|
||||||
|
};
|
||||||
let mut mock_network = MockNetworkTrait::new();
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::AddRootFolder(Some(expected_root_folder_path.clone())).into(),
|
RadarrEvent::AddRootFolder(expected_add_root_folder_body).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -429,7 +437,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let add_root_folder_command = RadarrAddCommand::RootFolder {
|
let add_root_folder_command = RadarrAddCommand::RootFolder {
|
||||||
root_folder_path: expected_root_folder_path,
|
root_folder_path: expected_root_folder_path,
|
||||||
};
|
};
|
||||||
@@ -457,7 +465,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let add_tag_command = RadarrAddCommand::Tag {
|
let add_tag_command = RadarrAddCommand::Tag {
|
||||||
name: expected_tag_name,
|
name: expected_tag_name,
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ use tokio::sync::Mutex;
|
|||||||
use crate::{
|
use crate::{
|
||||||
app::App,
|
app::App,
|
||||||
cli::{CliCommandHandler, Command},
|
cli::{CliCommandHandler, Command},
|
||||||
execute_network_event,
|
|
||||||
models::radarr_models::DeleteMovieParams,
|
models::radarr_models::DeleteMovieParams,
|
||||||
network::{radarr_network::RadarrEvent, NetworkTrait},
|
network::{radarr_network::RadarrEvent, NetworkTrait},
|
||||||
};
|
};
|
||||||
@@ -85,19 +84,28 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrDeleteCommand> for RadarrDeleteComm
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(self) -> Result<()> {
|
async fn handle(self) -> Result<String> {
|
||||||
match self.command {
|
let result = match self.command {
|
||||||
RadarrDeleteCommand::BlocklistItem { blocklist_item_id } => {
|
RadarrDeleteCommand::BlocklistItem { blocklist_item_id } => {
|
||||||
execute_network_event!(
|
let resp = self
|
||||||
self,
|
.network
|
||||||
RadarrEvent::DeleteBlocklistItem(Some(blocklist_item_id))
|
.handle_network_event(RadarrEvent::DeleteBlocklistItem(blocklist_item_id).into())
|
||||||
);
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrDeleteCommand::Download { download_id } => {
|
RadarrDeleteCommand::Download { download_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::DeleteDownload(Some(download_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::DeleteDownload(download_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrDeleteCommand::Indexer { indexer_id } => {
|
RadarrDeleteCommand::Indexer { indexer_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::DeleteIndexer(Some(indexer_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::DeleteIndexer(indexer_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrDeleteCommand::Movie {
|
RadarrDeleteCommand::Movie {
|
||||||
movie_id,
|
movie_id,
|
||||||
@@ -109,16 +117,28 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrDeleteCommand> for RadarrDeleteComm
|
|||||||
delete_movie_files: delete_files_from_disk,
|
delete_movie_files: delete_files_from_disk,
|
||||||
add_list_exclusion,
|
add_list_exclusion,
|
||||||
};
|
};
|
||||||
execute_network_event!(self, RadarrEvent::DeleteMovie(Some(delete_movie_params)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::DeleteMovie(delete_movie_params).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrDeleteCommand::RootFolder { root_folder_id } => {
|
RadarrDeleteCommand::RootFolder { root_folder_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::DeleteRootFolder(Some(root_folder_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::DeleteRootFolder(root_folder_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrDeleteCommand::Tag { tag_id } => {
|
RadarrDeleteCommand::Tag { tag_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::DeleteTag(tag_id));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::DeleteTag(tag_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ mod tests {
|
|||||||
Cli,
|
Cli,
|
||||||
};
|
};
|
||||||
use clap::{error::ErrorKind, CommandFactory, Parser};
|
use clap::{error::ErrorKind, CommandFactory, Parser};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_radarr_delete_command_from() {
|
fn test_radarr_delete_command_from() {
|
||||||
@@ -267,7 +268,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::DeleteBlocklistItem(Some(expected_blocklist_item_id)).into(),
|
RadarrEvent::DeleteBlocklistItem(expected_blocklist_item_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -275,7 +276,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let delete_blocklist_item_command = RadarrDeleteCommand::BlocklistItem {
|
let delete_blocklist_item_command = RadarrDeleteCommand::BlocklistItem {
|
||||||
blocklist_item_id: 1,
|
blocklist_item_id: 1,
|
||||||
};
|
};
|
||||||
@@ -298,7 +299,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::DeleteDownload(Some(expected_download_id)).into(),
|
RadarrEvent::DeleteDownload(expected_download_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -306,7 +307,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let delete_download_command = RadarrDeleteCommand::Download { download_id: 1 };
|
let delete_download_command = RadarrDeleteCommand::Download { download_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -324,7 +325,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::DeleteIndexer(Some(expected_indexer_id)).into(),
|
RadarrEvent::DeleteIndexer(expected_indexer_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -332,7 +333,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let delete_indexer_command = RadarrDeleteCommand::Indexer { indexer_id: 1 };
|
let delete_indexer_command = RadarrDeleteCommand::Indexer { indexer_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -354,7 +355,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::DeleteMovie(Some(expected_delete_movie_params)).into(),
|
RadarrEvent::DeleteMovie(expected_delete_movie_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -362,7 +363,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let delete_movie_command = RadarrDeleteCommand::Movie {
|
let delete_movie_command = RadarrDeleteCommand::Movie {
|
||||||
movie_id: 1,
|
movie_id: 1,
|
||||||
delete_files_from_disk: true,
|
delete_files_from_disk: true,
|
||||||
@@ -384,7 +385,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::DeleteRootFolder(Some(expected_root_folder_id)).into(),
|
RadarrEvent::DeleteRootFolder(expected_root_folder_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -392,7 +393,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let delete_root_folder_command = RadarrDeleteCommand::RootFolder { root_folder_id: 1 };
|
let delete_root_folder_command = RadarrDeleteCommand::RootFolder { root_folder_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -418,7 +419,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let delete_tag_command = RadarrDeleteCommand::Tag { tag_id: 1 };
|
let delete_tag_command = RadarrDeleteCommand::Tag { tag_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
|
|||||||
@@ -7,12 +7,11 @@ use tokio::sync::Mutex;
|
|||||||
use crate::{
|
use crate::{
|
||||||
app::App,
|
app::App,
|
||||||
cli::{mutex_flags_or_default, mutex_flags_or_option, CliCommandHandler, Command},
|
cli::{mutex_flags_or_default, mutex_flags_or_option, CliCommandHandler, Command},
|
||||||
execute_network_event,
|
|
||||||
models::{
|
models::{
|
||||||
radarr_models::{
|
radarr_models::{
|
||||||
EditCollectionParams, EditIndexerParams, EditMovieParams, IndexerSettings,
|
EditCollectionParams, EditMovieParams, IndexerSettings, MinimumAvailability, RadarrSerdeable,
|
||||||
MinimumAvailability, RadarrSerdeable,
|
|
||||||
},
|
},
|
||||||
|
servarr_models::EditIndexerParams,
|
||||||
Serdeable,
|
Serdeable,
|
||||||
},
|
},
|
||||||
network::{radarr_network::RadarrEvent, NetworkTrait},
|
network::{radarr_network::RadarrEvent, NetworkTrait},
|
||||||
@@ -339,8 +338,8 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrEditCommand> for RadarrEditCommandH
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(self) -> Result<()> {
|
async fn handle(self) -> Result<String> {
|
||||||
match self.command {
|
let result = match self.command {
|
||||||
RadarrEditCommand::AllIndexerSettings {
|
RadarrEditCommand::AllIndexerSettings {
|
||||||
allow_hardcoded_subs,
|
allow_hardcoded_subs,
|
||||||
disable_allow_hardcoded_subs,
|
disable_allow_hardcoded_subs,
|
||||||
@@ -380,20 +379,16 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrEditCommand> for RadarrEditCommandH
|
|||||||
rss_sync_interval: rss_sync_interval
|
rss_sync_interval: rss_sync_interval
|
||||||
.unwrap_or(previous_indexer_settings.rss_sync_interval),
|
.unwrap_or(previous_indexer_settings.rss_sync_interval),
|
||||||
whitelisted_hardcoded_subs: whitelisted_subtitle_tags
|
whitelisted_hardcoded_subs: whitelisted_subtitle_tags
|
||||||
.clone()
|
.unwrap_or(previous_indexer_settings.whitelisted_hardcoded_subs.text)
|
||||||
.unwrap_or_else(|| {
|
|
||||||
previous_indexer_settings
|
|
||||||
.whitelisted_hardcoded_subs
|
|
||||||
.text
|
|
||||||
.clone()
|
|
||||||
})
|
|
||||||
.into(),
|
.into(),
|
||||||
};
|
};
|
||||||
execute_network_event!(
|
self
|
||||||
self,
|
.network
|
||||||
RadarrEvent::EditAllIndexerSettings(Some(params)),
|
.handle_network_event(RadarrEvent::EditAllIndexerSettings(params).into())
|
||||||
"All indexer settings updated"
|
.await?;
|
||||||
);
|
"All indexer settings updated".to_owned()
|
||||||
|
} else {
|
||||||
|
String::new()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
RadarrEditCommand::Collection {
|
RadarrEditCommand::Collection {
|
||||||
@@ -417,11 +412,11 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrEditCommand> for RadarrEditCommandH
|
|||||||
root_folder_path,
|
root_folder_path,
|
||||||
search_on_add: search_on_add_value,
|
search_on_add: search_on_add_value,
|
||||||
};
|
};
|
||||||
execute_network_event!(
|
self
|
||||||
self,
|
.network
|
||||||
RadarrEvent::EditCollection(Some(edit_collection_params)),
|
.handle_network_event(RadarrEvent::EditCollection(edit_collection_params).into())
|
||||||
"Collection Updated"
|
.await?;
|
||||||
);
|
"Collection updated".to_owned()
|
||||||
}
|
}
|
||||||
RadarrEditCommand::Indexer {
|
RadarrEditCommand::Indexer {
|
||||||
indexer_id,
|
indexer_id,
|
||||||
@@ -454,15 +449,16 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrEditCommand> for RadarrEditCommandH
|
|||||||
api_key,
|
api_key,
|
||||||
seed_ratio,
|
seed_ratio,
|
||||||
tags: tag,
|
tags: tag,
|
||||||
|
tag_input_string: None,
|
||||||
priority,
|
priority,
|
||||||
clear_tags,
|
clear_tags,
|
||||||
};
|
};
|
||||||
|
|
||||||
execute_network_event!(
|
self
|
||||||
self,
|
.network
|
||||||
RadarrEvent::EditIndexer(Some(edit_indexer_params)),
|
.handle_network_event(RadarrEvent::EditIndexer(edit_indexer_params).into())
|
||||||
"Indexer updated"
|
.await?;
|
||||||
);
|
"Indexer updated".to_owned()
|
||||||
}
|
}
|
||||||
RadarrEditCommand::Movie {
|
RadarrEditCommand::Movie {
|
||||||
movie_id,
|
movie_id,
|
||||||
@@ -482,17 +478,18 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrEditCommand> for RadarrEditCommandH
|
|||||||
quality_profile_id,
|
quality_profile_id,
|
||||||
root_folder_path,
|
root_folder_path,
|
||||||
tags: tag,
|
tags: tag,
|
||||||
|
tag_input_string: None,
|
||||||
clear_tags,
|
clear_tags,
|
||||||
};
|
};
|
||||||
|
|
||||||
execute_network_event!(
|
self
|
||||||
self,
|
.network
|
||||||
RadarrEvent::EditMovie(Some(edit_movie_params)),
|
.handle_network_event(RadarrEvent::EditMovie(edit_movie_params).into())
|
||||||
"Movie updated"
|
.await?;
|
||||||
);
|
"Movie Updated".to_owned()
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ mod tests {
|
|||||||
Cli,
|
Cli,
|
||||||
};
|
};
|
||||||
use clap::{error::ErrorKind, CommandFactory, Parser};
|
use clap::{error::ErrorKind, CommandFactory, Parser};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_radarr_edit_command_from() {
|
fn test_radarr_edit_command_from() {
|
||||||
@@ -809,9 +810,10 @@ mod tests {
|
|||||||
},
|
},
|
||||||
models::{
|
models::{
|
||||||
radarr_models::{
|
radarr_models::{
|
||||||
EditCollectionParams, EditIndexerParams, EditMovieParams, IndexerSettings,
|
EditCollectionParams, EditMovieParams, IndexerSettings, MinimumAvailability,
|
||||||
MinimumAvailability, RadarrSerdeable,
|
RadarrSerdeable,
|
||||||
},
|
},
|
||||||
|
servarr_models::EditIndexerParams,
|
||||||
Serdeable,
|
Serdeable,
|
||||||
},
|
},
|
||||||
network::{radarr_network::RadarrEvent, MockNetworkTrait, NetworkEvent},
|
network::{radarr_network::RadarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
@@ -855,7 +857,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(),
|
RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -863,7 +865,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_all_indexer_settings_command = RadarrEditCommand::AllIndexerSettings {
|
let edit_all_indexer_settings_command = RadarrEditCommand::AllIndexerSettings {
|
||||||
allow_hardcoded_subs: true,
|
allow_hardcoded_subs: true,
|
||||||
disable_allow_hardcoded_subs: false,
|
disable_allow_hardcoded_subs: false,
|
||||||
@@ -926,7 +928,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(),
|
RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -934,7 +936,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_all_indexer_settings_command = RadarrEditCommand::AllIndexerSettings {
|
let edit_all_indexer_settings_command = RadarrEditCommand::AllIndexerSettings {
|
||||||
allow_hardcoded_subs: false,
|
allow_hardcoded_subs: false,
|
||||||
disable_allow_hardcoded_subs: true,
|
disable_allow_hardcoded_subs: true,
|
||||||
@@ -998,7 +1000,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(),
|
RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1006,7 +1008,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_all_indexer_settings_command = RadarrEditCommand::AllIndexerSettings {
|
let edit_all_indexer_settings_command = RadarrEditCommand::AllIndexerSettings {
|
||||||
allow_hardcoded_subs: false,
|
allow_hardcoded_subs: false,
|
||||||
disable_allow_hardcoded_subs: false,
|
disable_allow_hardcoded_subs: false,
|
||||||
@@ -1045,7 +1047,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditCollection(Some(expected_edit_collection_params)).into(),
|
RadarrEvent::EditCollection(expected_edit_collection_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1053,7 +1055,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_collection_command = RadarrEditCommand::Collection {
|
let edit_collection_command = RadarrEditCommand::Collection {
|
||||||
collection_id: 1,
|
collection_id: 1,
|
||||||
enable_monitoring: true,
|
enable_monitoring: true,
|
||||||
@@ -1087,7 +1089,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditCollection(Some(expected_edit_collection_params)).into(),
|
RadarrEvent::EditCollection(expected_edit_collection_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1095,7 +1097,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_collection_command = RadarrEditCommand::Collection {
|
let edit_collection_command = RadarrEditCommand::Collection {
|
||||||
collection_id: 1,
|
collection_id: 1,
|
||||||
enable_monitoring: false,
|
enable_monitoring: false,
|
||||||
@@ -1129,7 +1131,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditCollection(Some(expected_edit_collection_params)).into(),
|
RadarrEvent::EditCollection(expected_edit_collection_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1137,7 +1139,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_collection_command = RadarrEditCommand::Collection {
|
let edit_collection_command = RadarrEditCommand::Collection {
|
||||||
collection_id: 1,
|
collection_id: 1,
|
||||||
enable_monitoring: false,
|
enable_monitoring: false,
|
||||||
@@ -1169,6 +1171,7 @@ mod tests {
|
|||||||
api_key: Some("testKey".to_owned()),
|
api_key: Some("testKey".to_owned()),
|
||||||
seed_ratio: Some("1.2".to_owned()),
|
seed_ratio: Some("1.2".to_owned()),
|
||||||
tags: Some(vec![1, 2]),
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
priority: Some(25),
|
priority: Some(25),
|
||||||
clear_tags: false,
|
clear_tags: false,
|
||||||
};
|
};
|
||||||
@@ -1176,7 +1179,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditIndexer(Some(expected_edit_indexer_params)).into(),
|
RadarrEvent::EditIndexer(expected_edit_indexer_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1184,7 +1187,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_indexer_command = RadarrEditCommand::Indexer {
|
let edit_indexer_command = RadarrEditCommand::Indexer {
|
||||||
indexer_id: 1,
|
indexer_id: 1,
|
||||||
name: Some("Test".to_owned()),
|
name: Some("Test".to_owned()),
|
||||||
@@ -1222,6 +1225,7 @@ mod tests {
|
|||||||
api_key: Some("testKey".to_owned()),
|
api_key: Some("testKey".to_owned()),
|
||||||
seed_ratio: Some("1.2".to_owned()),
|
seed_ratio: Some("1.2".to_owned()),
|
||||||
tags: Some(vec![1, 2]),
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
priority: Some(25),
|
priority: Some(25),
|
||||||
clear_tags: false,
|
clear_tags: false,
|
||||||
};
|
};
|
||||||
@@ -1229,7 +1233,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditIndexer(Some(expected_edit_indexer_params)).into(),
|
RadarrEvent::EditIndexer(expected_edit_indexer_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1237,7 +1241,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_indexer_command = RadarrEditCommand::Indexer {
|
let edit_indexer_command = RadarrEditCommand::Indexer {
|
||||||
indexer_id: 1,
|
indexer_id: 1,
|
||||||
name: Some("Test".to_owned()),
|
name: Some("Test".to_owned()),
|
||||||
@@ -1275,6 +1279,7 @@ mod tests {
|
|||||||
api_key: Some("testKey".to_owned()),
|
api_key: Some("testKey".to_owned()),
|
||||||
seed_ratio: Some("1.2".to_owned()),
|
seed_ratio: Some("1.2".to_owned()),
|
||||||
tags: Some(vec![1, 2]),
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
priority: Some(25),
|
priority: Some(25),
|
||||||
clear_tags: false,
|
clear_tags: false,
|
||||||
};
|
};
|
||||||
@@ -1282,7 +1287,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditIndexer(Some(expected_edit_indexer_params)).into(),
|
RadarrEvent::EditIndexer(expected_edit_indexer_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1290,7 +1295,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_indexer_command = RadarrEditCommand::Indexer {
|
let edit_indexer_command = RadarrEditCommand::Indexer {
|
||||||
indexer_id: 1,
|
indexer_id: 1,
|
||||||
name: Some("Test".to_owned()),
|
name: Some("Test".to_owned()),
|
||||||
@@ -1325,13 +1330,14 @@ mod tests {
|
|||||||
quality_profile_id: Some(1),
|
quality_profile_id: Some(1),
|
||||||
root_folder_path: Some("/nfs/test".to_owned()),
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
tags: Some(vec![1, 2]),
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
clear_tags: false,
|
clear_tags: false,
|
||||||
};
|
};
|
||||||
let mut mock_network = MockNetworkTrait::new();
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditMovie(Some(expected_edit_movie_params)).into(),
|
RadarrEvent::EditMovie(expected_edit_movie_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1339,7 +1345,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_movie_command = RadarrEditCommand::Movie {
|
let edit_movie_command = RadarrEditCommand::Movie {
|
||||||
movie_id: 1,
|
movie_id: 1,
|
||||||
enable_monitoring: true,
|
enable_monitoring: true,
|
||||||
@@ -1367,13 +1373,14 @@ mod tests {
|
|||||||
quality_profile_id: Some(1),
|
quality_profile_id: Some(1),
|
||||||
root_folder_path: Some("/nfs/test".to_owned()),
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
tags: Some(vec![1, 2]),
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
clear_tags: false,
|
clear_tags: false,
|
||||||
};
|
};
|
||||||
let mut mock_network = MockNetworkTrait::new();
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditMovie(Some(expected_edit_movie_params)).into(),
|
RadarrEvent::EditMovie(expected_edit_movie_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1381,7 +1388,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_movie_command = RadarrEditCommand::Movie {
|
let edit_movie_command = RadarrEditCommand::Movie {
|
||||||
movie_id: 1,
|
movie_id: 1,
|
||||||
enable_monitoring: false,
|
enable_monitoring: false,
|
||||||
@@ -1409,13 +1416,14 @@ mod tests {
|
|||||||
quality_profile_id: Some(1),
|
quality_profile_id: Some(1),
|
||||||
root_folder_path: Some("/nfs/test".to_owned()),
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
tags: Some(vec![1, 2]),
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
clear_tags: false,
|
clear_tags: false,
|
||||||
};
|
};
|
||||||
let mut mock_network = MockNetworkTrait::new();
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditMovie(Some(expected_edit_movie_params)).into(),
|
RadarrEvent::EditMovie(expected_edit_movie_params).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -1423,7 +1431,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_movie_command = RadarrEditCommand::Movie {
|
let edit_movie_command = RadarrEditCommand::Movie {
|
||||||
movie_id: 1,
|
movie_id: 1,
|
||||||
enable_monitoring: false,
|
enable_monitoring: false,
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ use tokio::sync::Mutex;
|
|||||||
use crate::{
|
use crate::{
|
||||||
app::App,
|
app::App,
|
||||||
cli::{CliCommandHandler, Command},
|
cli::{CliCommandHandler, Command},
|
||||||
execute_network_event,
|
|
||||||
network::{radarr_network::RadarrEvent, NetworkTrait},
|
network::{radarr_network::RadarrEvent, NetworkTrait},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -72,28 +71,52 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrGetCommand> for RadarrGetCommandHan
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(self) -> Result<()> {
|
async fn handle(self) -> Result<String> {
|
||||||
match self.command {
|
let result = match self.command {
|
||||||
RadarrGetCommand::AllIndexerSettings => {
|
RadarrGetCommand::AllIndexerSettings => {
|
||||||
execute_network_event!(self, RadarrEvent::GetAllIndexerSettings);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetAllIndexerSettings.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrGetCommand::HostConfig => {
|
RadarrGetCommand::HostConfig => {
|
||||||
execute_network_event!(self, RadarrEvent::GetHostConfig);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetHostConfig.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrGetCommand::MovieDetails { movie_id } => {
|
RadarrGetCommand::MovieDetails { movie_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::GetMovieDetails(Some(movie_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetMovieDetails(movie_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrGetCommand::MovieHistory { movie_id } => {
|
RadarrGetCommand::MovieHistory { movie_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::GetMovieHistory(Some(movie_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetMovieHistory(movie_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrGetCommand::SecurityConfig => {
|
RadarrGetCommand::SecurityConfig => {
|
||||||
execute_network_event!(self, RadarrEvent::GetSecurityConfig);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetSecurityConfig.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrGetCommand::SystemStatus => {
|
RadarrGetCommand::SystemStatus => {
|
||||||
execute_network_event!(self, RadarrEvent::GetStatus);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetStatus.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod tests {
|
||||||
use clap::error::ErrorKind;
|
use clap::error::ErrorKind;
|
||||||
use clap::CommandFactory;
|
use clap::CommandFactory;
|
||||||
|
|
||||||
@@ -7,6 +7,7 @@ mod test {
|
|||||||
use crate::cli::radarr::RadarrCommand;
|
use crate::cli::radarr::RadarrCommand;
|
||||||
use crate::cli::Command;
|
use crate::cli::Command;
|
||||||
use crate::Cli;
|
use crate::Cli;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_radarr_get_command_from() {
|
fn test_radarr_get_command_from() {
|
||||||
@@ -137,7 +138,7 @@ mod test {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let get_all_indexer_settings_command = RadarrGetCommand::AllIndexerSettings;
|
let get_all_indexer_settings_command = RadarrGetCommand::AllIndexerSettings;
|
||||||
|
|
||||||
let result = RadarrGetCommandHandler::with(
|
let result = RadarrGetCommandHandler::with(
|
||||||
@@ -163,7 +164,7 @@ mod test {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let get_host_config_command = RadarrGetCommand::HostConfig;
|
let get_host_config_command = RadarrGetCommand::HostConfig;
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -181,7 +182,7 @@ mod test {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::GetMovieDetails(Some(expected_movie_id)).into(),
|
RadarrEvent::GetMovieDetails(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -189,7 +190,7 @@ mod test {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let get_movie_details_command = RadarrGetCommand::MovieDetails { movie_id: 1 };
|
let get_movie_details_command = RadarrGetCommand::MovieDetails { movie_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -207,7 +208,7 @@ mod test {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::GetMovieHistory(Some(expected_movie_id)).into(),
|
RadarrEvent::GetMovieHistory(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -215,7 +216,7 @@ mod test {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let get_movie_history_command = RadarrGetCommand::MovieHistory { movie_id: 1 };
|
let get_movie_history_command = RadarrGetCommand::MovieHistory { movie_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -238,7 +239,7 @@ mod test {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let get_security_config_command = RadarrGetCommand::SecurityConfig;
|
let get_security_config_command = RadarrGetCommand::SecurityConfig;
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -261,7 +262,7 @@ mod test {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let get_system_status_command = RadarrGetCommand::SystemStatus;
|
let get_system_status_command = RadarrGetCommand::SystemStatus;
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ use tokio::sync::Mutex;
|
|||||||
use crate::{
|
use crate::{
|
||||||
app::App,
|
app::App,
|
||||||
cli::{CliCommandHandler, Command},
|
cli::{CliCommandHandler, Command},
|
||||||
execute_network_event,
|
|
||||||
network::{radarr_network::RadarrEvent, NetworkTrait},
|
network::{radarr_network::RadarrEvent, NetworkTrait},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -25,6 +24,8 @@ pub enum RadarrListCommand {
|
|||||||
Collections,
|
Collections,
|
||||||
#[command(about = "List all active downloads in Radarr")]
|
#[command(about = "List all active downloads in Radarr")]
|
||||||
Downloads,
|
Downloads,
|
||||||
|
#[command(about = "List disk space details for all provisioned root folders in Radarr")]
|
||||||
|
DiskSpace,
|
||||||
#[command(about = "List all Radarr indexers")]
|
#[command(about = "List all Radarr indexers")]
|
||||||
Indexers,
|
Indexers,
|
||||||
#[command(about = "Fetch Radarr logs")]
|
#[command(about = "Fetch Radarr logs")]
|
||||||
@@ -56,7 +57,7 @@ pub enum RadarrListCommand {
|
|||||||
RootFolders,
|
RootFolders,
|
||||||
#[command(about = "List all Radarr tags")]
|
#[command(about = "List all Radarr tags")]
|
||||||
Tags,
|
Tags,
|
||||||
#[command(about = "List tasks")]
|
#[command(about = "List all Radarr tasks")]
|
||||||
Tasks,
|
Tasks,
|
||||||
#[command(about = "List all Radarr updates")]
|
#[command(about = "List all Radarr updates")]
|
||||||
Updates,
|
Updates,
|
||||||
@@ -87,19 +88,42 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrListCommand> for RadarrListCommandH
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(self) -> Result<()> {
|
async fn handle(self) -> Result<String> {
|
||||||
match self.command {
|
let result = match self.command {
|
||||||
RadarrListCommand::Blocklist => {
|
RadarrListCommand::Blocklist => {
|
||||||
execute_network_event!(self, RadarrEvent::GetBlocklist);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetBlocklist.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::Collections => {
|
RadarrListCommand::Collections => {
|
||||||
execute_network_event!(self, RadarrEvent::GetCollections);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetCollections.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::Downloads => {
|
RadarrListCommand::Downloads => {
|
||||||
execute_network_event!(self, RadarrEvent::GetDownloads);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetDownloads.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
RadarrListCommand::DiskSpace => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetDiskSpace.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::Indexers => {
|
RadarrListCommand::Indexers => {
|
||||||
execute_network_event!(self, RadarrEvent::GetIndexers);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetIndexers.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::Logs {
|
RadarrListCommand::Logs {
|
||||||
events,
|
events,
|
||||||
@@ -107,45 +131,75 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrListCommand> for RadarrListCommandH
|
|||||||
} => {
|
} => {
|
||||||
let logs = self
|
let logs = self
|
||||||
.network
|
.network
|
||||||
.handle_network_event(RadarrEvent::GetLogs(Some(events)).into())
|
.handle_network_event(RadarrEvent::GetLogs(events).into())
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
if output_in_log_format {
|
if output_in_log_format {
|
||||||
let log_lines = self.app.lock().await.data.radarr_data.logs.items.clone();
|
let log_lines = &self.app.lock().await.data.radarr_data.logs.items;
|
||||||
|
|
||||||
let json = serde_json::to_string_pretty(&log_lines)?;
|
serde_json::to_string_pretty(log_lines)?
|
||||||
println!("{}", json);
|
|
||||||
} else {
|
} else {
|
||||||
let json = serde_json::to_string_pretty(&logs)?;
|
serde_json::to_string_pretty(&logs)?
|
||||||
println!("{}", json);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
RadarrListCommand::Movies => {
|
RadarrListCommand::Movies => {
|
||||||
execute_network_event!(self, RadarrEvent::GetMovies);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetMovies.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::MovieCredits { movie_id } => {
|
RadarrListCommand::MovieCredits { movie_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::GetMovieCredits(Some(movie_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetMovieCredits(movie_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::QualityProfiles => {
|
RadarrListCommand::QualityProfiles => {
|
||||||
execute_network_event!(self, RadarrEvent::GetQualityProfiles);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetQualityProfiles.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::QueuedEvents => {
|
RadarrListCommand::QueuedEvents => {
|
||||||
execute_network_event!(self, RadarrEvent::GetQueuedEvents);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetQueuedEvents.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::RootFolders => {
|
RadarrListCommand::RootFolders => {
|
||||||
execute_network_event!(self, RadarrEvent::GetRootFolders);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetRootFolders.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::Tags => {
|
RadarrListCommand::Tags => {
|
||||||
execute_network_event!(self, RadarrEvent::GetTags);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetTags.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::Tasks => {
|
RadarrListCommand::Tasks => {
|
||||||
execute_network_event!(self, RadarrEvent::GetTasks);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetTasks.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrListCommand::Updates => {
|
RadarrListCommand::Updates => {
|
||||||
execute_network_event!(self, RadarrEvent::GetUpdates);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetUpdates.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ mod tests {
|
|||||||
use crate::cli::radarr::RadarrCommand;
|
use crate::cli::radarr::RadarrCommand;
|
||||||
use crate::cli::Command;
|
use crate::cli::Command;
|
||||||
use crate::Cli;
|
use crate::Cli;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_radarr_list_command_from() {
|
fn test_radarr_list_command_from() {
|
||||||
@@ -29,6 +30,7 @@ mod tests {
|
|||||||
"blocklist",
|
"blocklist",
|
||||||
"collections",
|
"collections",
|
||||||
"downloads",
|
"downloads",
|
||||||
|
"disk-space",
|
||||||
"indexers",
|
"indexers",
|
||||||
"movies",
|
"movies",
|
||||||
"quality-profiles",
|
"quality-profiles",
|
||||||
@@ -80,8 +82,8 @@ mod tests {
|
|||||||
|
|
||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
|
|
||||||
if let Some(Command::Radarr(RadarrCommand::List(refresh_command))) = result.unwrap().command {
|
if let Some(Command::Radarr(RadarrCommand::List(credits_command))) = result.unwrap().command {
|
||||||
assert_eq!(refresh_command, expected_args);
|
assert_eq!(credits_command, expected_args);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -121,6 +123,7 @@ mod tests {
|
|||||||
#[case(RadarrListCommand::Blocklist, RadarrEvent::GetBlocklist)]
|
#[case(RadarrListCommand::Blocklist, RadarrEvent::GetBlocklist)]
|
||||||
#[case(RadarrListCommand::Collections, RadarrEvent::GetCollections)]
|
#[case(RadarrListCommand::Collections, RadarrEvent::GetCollections)]
|
||||||
#[case(RadarrListCommand::Downloads, RadarrEvent::GetDownloads)]
|
#[case(RadarrListCommand::Downloads, RadarrEvent::GetDownloads)]
|
||||||
|
#[case(RadarrListCommand::DiskSpace, RadarrEvent::GetDiskSpace)]
|
||||||
#[case(RadarrListCommand::Indexers, RadarrEvent::GetIndexers)]
|
#[case(RadarrListCommand::Indexers, RadarrEvent::GetIndexers)]
|
||||||
#[case(RadarrListCommand::Movies, RadarrEvent::GetMovies)]
|
#[case(RadarrListCommand::Movies, RadarrEvent::GetMovies)]
|
||||||
#[case(RadarrListCommand::QualityProfiles, RadarrEvent::GetQualityProfiles)]
|
#[case(RadarrListCommand::QualityProfiles, RadarrEvent::GetQualityProfiles)]
|
||||||
@@ -130,7 +133,7 @@ mod tests {
|
|||||||
#[case(RadarrListCommand::Tasks, RadarrEvent::GetTasks)]
|
#[case(RadarrListCommand::Tasks, RadarrEvent::GetTasks)]
|
||||||
#[case(RadarrListCommand::Updates, RadarrEvent::GetUpdates)]
|
#[case(RadarrListCommand::Updates, RadarrEvent::GetUpdates)]
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_handle_list_blocklist_command(
|
async fn test_handle_list_command(
|
||||||
#[case] list_command: RadarrListCommand,
|
#[case] list_command: RadarrListCommand,
|
||||||
#[case] expected_radarr_event: RadarrEvent,
|
#[case] expected_radarr_event: RadarrEvent,
|
||||||
) {
|
) {
|
||||||
@@ -144,7 +147,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
|
||||||
let result = RadarrListCommandHandler::with(&app_arc, list_command, &mut mock_network)
|
let result = RadarrListCommandHandler::with(&app_arc, list_command, &mut mock_network)
|
||||||
.handle()
|
.handle()
|
||||||
@@ -160,7 +163,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::GetMovieCredits(Some(expected_movie_id)).into(),
|
RadarrEvent::GetMovieCredits(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -168,7 +171,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let list_movie_credits_command = RadarrListCommand::MovieCredits { movie_id: 1 };
|
let list_movie_credits_command = RadarrListCommand::MovieCredits { movie_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
@@ -186,7 +189,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::GetLogs(Some(expected_events)).into(),
|
RadarrEvent::GetLogs(expected_events).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -194,7 +197,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let list_logs_command = RadarrListCommand::Logs {
|
let list_logs_command = RadarrListCommand::Logs {
|
||||||
events: 1000,
|
events: 1000,
|
||||||
output_in_log_format: false,
|
output_in_log_format: false,
|
||||||
|
|||||||
@@ -12,8 +12,7 @@ use tokio::sync::Mutex;
|
|||||||
use crate::app::App;
|
use crate::app::App;
|
||||||
|
|
||||||
use crate::cli::CliCommandHandler;
|
use crate::cli::CliCommandHandler;
|
||||||
use crate::execute_network_event;
|
use crate::models::radarr_models::{RadarrReleaseDownloadBody, RadarrTaskName};
|
||||||
use crate::models::radarr_models::{ReleaseDownloadBody, TaskName};
|
|
||||||
use crate::network::radarr_network::RadarrEvent;
|
use crate::network::radarr_network::RadarrEvent;
|
||||||
use crate::network::NetworkTrait;
|
use crate::network::NetworkTrait;
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
@@ -86,7 +85,7 @@ pub enum RadarrCommand {
|
|||||||
ManualSearch {
|
ManualSearch {
|
||||||
#[arg(
|
#[arg(
|
||||||
long,
|
long,
|
||||||
help = "The Radarr ID of the movie whose releases you wish to fetch and list",
|
help = "The Radarr ID of the movie whose releases you wish to fetch",
|
||||||
required = true
|
required = true
|
||||||
)]
|
)]
|
||||||
movie_id: i64,
|
movie_id: i64,
|
||||||
@@ -108,7 +107,7 @@ pub enum RadarrCommand {
|
|||||||
value_enum,
|
value_enum,
|
||||||
required = true
|
required = true
|
||||||
)]
|
)]
|
||||||
task_name: TaskName,
|
task_name: RadarrTaskName,
|
||||||
},
|
},
|
||||||
#[command(
|
#[command(
|
||||||
about = "Test the indexer with the given ID. Note that a successful test returns an empty JSON body; i.e. '{}'"
|
about = "Test the indexer with the given ID. Note that a successful test returns an empty JSON body; i.e. '{}'"
|
||||||
@@ -117,7 +116,7 @@ pub enum RadarrCommand {
|
|||||||
#[arg(long, help = "The ID of the indexer to test", required = true)]
|
#[arg(long, help = "The ID of the indexer to test", required = true)]
|
||||||
indexer_id: i64,
|
indexer_id: i64,
|
||||||
},
|
},
|
||||||
#[command(about = "Test all indexers")]
|
#[command(about = "Test all Radarr indexers")]
|
||||||
TestAllIndexers,
|
TestAllIndexers,
|
||||||
#[command(about = "Trigger an automatic search for the movie with the specified ID")]
|
#[command(about = "Trigger an automatic search for the movie with the specified ID")]
|
||||||
TriggerAutomaticSearch {
|
TriggerAutomaticSearch {
|
||||||
@@ -155,8 +154,8 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrCommand> for RadarrCliHandler<'a, '
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(self) -> Result<()> {
|
async fn handle(self) -> Result<String> {
|
||||||
match self.command {
|
let result = match self.command {
|
||||||
RadarrCommand::Add(add_command) => {
|
RadarrCommand::Add(add_command) => {
|
||||||
RadarrAddCommandHandler::with(self.app, add_command, self.network)
|
RadarrAddCommandHandler::with(self.app, add_command, self.network)
|
||||||
.handle()
|
.handle()
|
||||||
@@ -192,41 +191,74 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrCommand> for RadarrCliHandler<'a, '
|
|||||||
.network
|
.network
|
||||||
.handle_network_event(RadarrEvent::GetBlocklist.into())
|
.handle_network_event(RadarrEvent::GetBlocklist.into())
|
||||||
.await?;
|
.await?;
|
||||||
execute_network_event!(self, RadarrEvent::ClearBlocklist);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::ClearBlocklist.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrCommand::DownloadRelease {
|
RadarrCommand::DownloadRelease {
|
||||||
guid,
|
guid,
|
||||||
indexer_id,
|
indexer_id,
|
||||||
movie_id,
|
movie_id,
|
||||||
} => {
|
} => {
|
||||||
let params = ReleaseDownloadBody {
|
let params = RadarrReleaseDownloadBody {
|
||||||
guid,
|
guid,
|
||||||
indexer_id,
|
indexer_id,
|
||||||
movie_id,
|
movie_id,
|
||||||
};
|
};
|
||||||
execute_network_event!(self, RadarrEvent::DownloadRelease(Some(params)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::DownloadRelease(params).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrCommand::ManualSearch { movie_id } => {
|
RadarrCommand::ManualSearch { movie_id } => {
|
||||||
println!("Searching for releases. This may take a minute...");
|
println!("Searching for releases. This may take a minute...");
|
||||||
execute_network_event!(self, RadarrEvent::GetReleases(Some(movie_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::GetReleases(movie_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrCommand::SearchNewMovie { query } => {
|
RadarrCommand::SearchNewMovie { query } => {
|
||||||
execute_network_event!(self, RadarrEvent::SearchNewMovie(Some(query)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::SearchNewMovie(query).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrCommand::StartTask { task_name } => {
|
RadarrCommand::StartTask { task_name } => {
|
||||||
execute_network_event!(self, RadarrEvent::StartTask(Some(task_name)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::StartTask(task_name).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrCommand::TestIndexer { indexer_id } => {
|
RadarrCommand::TestIndexer { indexer_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::TestIndexer(Some(indexer_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::TestIndexer(indexer_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrCommand::TestAllIndexers => {
|
RadarrCommand::TestAllIndexers => {
|
||||||
execute_network_event!(self, RadarrEvent::TestAllIndexers);
|
println!("Testing all Radarr indexers. This may take a minute...");
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::TestAllIndexers.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrCommand::TriggerAutomaticSearch { movie_id } => {
|
RadarrCommand::TriggerAutomaticSearch { movie_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::TriggerAutomaticSearch(Some(movie_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::TriggerAutomaticSearch(movie_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_download_release_requires_movie_id() {
|
fn test_download_release_requires_movie_id() {
|
||||||
let result = Cli::command().try_get_matches_from([
|
let result = Cli::command().try_get_matches_from([
|
||||||
"managarr",
|
"managarr",
|
||||||
@@ -50,7 +50,7 @@ mod tests {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_download_release_requires_guid() {
|
fn test_download_release_requires_guid() {
|
||||||
let result = Cli::command().try_get_matches_from([
|
let result = Cli::command().try_get_matches_from([
|
||||||
"managarr",
|
"managarr",
|
||||||
@@ -69,7 +69,7 @@ mod tests {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_download_release_requires_indexer_id() {
|
fn test_download_release_requires_indexer_id() {
|
||||||
let result = Cli::command().try_get_matches_from([
|
let result = Cli::command().try_get_matches_from([
|
||||||
"managarr",
|
"managarr",
|
||||||
@@ -105,7 +105,7 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_manual_search_requires_movie_id() {
|
fn test_manual_search_requires_movie_id() {
|
||||||
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "manual-search"]);
|
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "manual-search"]);
|
||||||
|
|
||||||
@@ -129,7 +129,7 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_search_new_movie_requires_query() {
|
fn test_search_new_movie_requires_query() {
|
||||||
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "search-new-movie"]);
|
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "search-new-movie"]);
|
||||||
|
|
||||||
@@ -153,7 +153,7 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_start_task_requires_task_name() {
|
fn test_start_task_requires_task_name() {
|
||||||
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "start-task"]);
|
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "start-task"]);
|
||||||
|
|
||||||
@@ -164,7 +164,7 @@ mod tests {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_start_task_task_name_validation() {
|
fn test_start_task_task_name_validation() {
|
||||||
let result = Cli::command().try_get_matches_from([
|
let result = Cli::command().try_get_matches_from([
|
||||||
"managarr",
|
"managarr",
|
||||||
@@ -191,7 +191,7 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_test_indexer_requires_indexer_id() {
|
fn test_test_indexer_requires_indexer_id() {
|
||||||
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "test-indexer"]);
|
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "test-indexer"]);
|
||||||
|
|
||||||
@@ -215,7 +215,7 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[test]
|
||||||
fn test_trigger_automatic_search_requires_movie_id() {
|
fn test_trigger_automatic_search_requires_movie_id() {
|
||||||
let result =
|
let result =
|
||||||
Cli::command().try_get_matches_from(["managarr", "radarr", "trigger-automatic-search"]);
|
Cli::command().try_get_matches_from(["managarr", "radarr", "trigger-automatic-search"]);
|
||||||
@@ -261,8 +261,8 @@ mod tests {
|
|||||||
},
|
},
|
||||||
models::{
|
models::{
|
||||||
radarr_models::{
|
radarr_models::{
|
||||||
BlocklistItem, BlocklistResponse, IndexerSettings, RadarrSerdeable, ReleaseDownloadBody,
|
BlocklistItem, BlocklistResponse, IndexerSettings, RadarrReleaseDownloadBody,
|
||||||
TaskName,
|
RadarrSerdeable, RadarrTaskName,
|
||||||
},
|
},
|
||||||
Serdeable,
|
Serdeable,
|
||||||
},
|
},
|
||||||
@@ -292,10 +292,10 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let claer_blocklist_command = RadarrCommand::ClearBlocklist;
|
let clear_blocklist_command = RadarrCommand::ClearBlocklist;
|
||||||
|
|
||||||
let result = RadarrCliHandler::with(&app_arc, claer_blocklist_command, &mut mock_network)
|
let result = RadarrCliHandler::with(&app_arc, clear_blocklist_command, &mut mock_network)
|
||||||
.handle()
|
.handle()
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
@@ -304,7 +304,7 @@ mod tests {
|
|||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_download_release_command() {
|
async fn test_download_release_command() {
|
||||||
let expected_release_download_body = ReleaseDownloadBody {
|
let expected_release_download_body = RadarrReleaseDownloadBody {
|
||||||
guid: "guid".to_owned(),
|
guid: "guid".to_owned(),
|
||||||
indexer_id: 1,
|
indexer_id: 1,
|
||||||
movie_id: 1,
|
movie_id: 1,
|
||||||
@@ -313,7 +313,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::DownloadRelease(Some(expected_release_download_body)).into(),
|
RadarrEvent::DownloadRelease(expected_release_download_body).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -321,7 +321,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let download_release_command = RadarrCommand::DownloadRelease {
|
let download_release_command = RadarrCommand::DownloadRelease {
|
||||||
guid: "guid".to_owned(),
|
guid: "guid".to_owned(),
|
||||||
indexer_id: 1,
|
indexer_id: 1,
|
||||||
@@ -342,7 +342,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::GetReleases(Some(expected_movie_id)).into(),
|
RadarrEvent::GetReleases(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -350,7 +350,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let manual_search_command = RadarrCommand::ManualSearch { movie_id: 1 };
|
let manual_search_command = RadarrCommand::ManualSearch { movie_id: 1 };
|
||||||
|
|
||||||
let result = RadarrCliHandler::with(&app_arc, manual_search_command, &mut mock_network)
|
let result = RadarrCliHandler::with(&app_arc, manual_search_command, &mut mock_network)
|
||||||
@@ -367,7 +367,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::SearchNewMovie(Some(expected_search_query)).into(),
|
RadarrEvent::SearchNewMovie(expected_search_query).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -375,7 +375,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let search_new_movie_command = RadarrCommand::SearchNewMovie {
|
let search_new_movie_command = RadarrCommand::SearchNewMovie {
|
||||||
query: "halo".to_owned(),
|
query: "halo".to_owned(),
|
||||||
};
|
};
|
||||||
@@ -389,12 +389,12 @@ mod tests {
|
|||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_start_task_command() {
|
async fn test_start_task_command() {
|
||||||
let expected_task_name = TaskName::ApplicationCheckUpdate;
|
let expected_task_name = RadarrTaskName::ApplicationCheckUpdate;
|
||||||
let mut mock_network = MockNetworkTrait::new();
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::StartTask(Some(expected_task_name)).into(),
|
RadarrEvent::StartTask(expected_task_name).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -402,9 +402,9 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let start_task_command = RadarrCommand::StartTask {
|
let start_task_command = RadarrCommand::StartTask {
|
||||||
task_name: TaskName::ApplicationCheckUpdate,
|
task_name: RadarrTaskName::ApplicationCheckUpdate,
|
||||||
};
|
};
|
||||||
|
|
||||||
let result = RadarrCliHandler::with(&app_arc, start_task_command, &mut mock_network)
|
let result = RadarrCliHandler::with(&app_arc, start_task_command, &mut mock_network)
|
||||||
@@ -421,7 +421,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::TestIndexer(Some(expected_indexer_id)).into(),
|
RadarrEvent::TestIndexer(expected_indexer_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -429,7 +429,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let test_indexer_command = RadarrCommand::TestIndexer { indexer_id: 1 };
|
let test_indexer_command = RadarrCommand::TestIndexer { indexer_id: 1 };
|
||||||
|
|
||||||
let result = RadarrCliHandler::with(&app_arc, test_indexer_command, &mut mock_network)
|
let result = RadarrCliHandler::with(&app_arc, test_indexer_command, &mut mock_network)
|
||||||
@@ -451,7 +451,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let test_all_indexers_command = RadarrCommand::TestAllIndexers;
|
let test_all_indexers_command = RadarrCommand::TestAllIndexers;
|
||||||
|
|
||||||
let result = RadarrCliHandler::with(&app_arc, test_all_indexers_command, &mut mock_network)
|
let result = RadarrCliHandler::with(&app_arc, test_all_indexers_command, &mut mock_network)
|
||||||
@@ -468,7 +468,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::TriggerAutomaticSearch(Some(expected_movie_id)).into(),
|
RadarrEvent::TriggerAutomaticSearch(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -476,7 +476,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let trigger_automatic_search_command = RadarrCommand::TriggerAutomaticSearch { movie_id: 1 };
|
let trigger_automatic_search_command = RadarrCommand::TriggerAutomaticSearch { movie_id: 1 };
|
||||||
|
|
||||||
let result = RadarrCliHandler::with(
|
let result = RadarrCliHandler::with(
|
||||||
@@ -505,7 +505,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let add_tag_command = RadarrCommand::Add(RadarrAddCommand::Tag {
|
let add_tag_command = RadarrCommand::Add(RadarrAddCommand::Tag {
|
||||||
name: expected_tag_name,
|
name: expected_tag_name,
|
||||||
});
|
});
|
||||||
@@ -524,7 +524,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::DeleteBlocklistItem(Some(expected_blocklist_item_id)).into(),
|
RadarrEvent::DeleteBlocklistItem(expected_blocklist_item_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -532,7 +532,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let delete_blocklist_item_command =
|
let delete_blocklist_item_command =
|
||||||
RadarrCommand::Delete(RadarrDeleteCommand::BlocklistItem {
|
RadarrCommand::Delete(RadarrDeleteCommand::BlocklistItem {
|
||||||
blocklist_item_id: 1,
|
blocklist_item_id: 1,
|
||||||
@@ -584,7 +584,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(),
|
RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -592,7 +592,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let edit_all_indexer_settings_command =
|
let edit_all_indexer_settings_command =
|
||||||
RadarrCommand::Edit(RadarrEditCommand::AllIndexerSettings {
|
RadarrCommand::Edit(RadarrEditCommand::AllIndexerSettings {
|
||||||
allow_hardcoded_subs: true,
|
allow_hardcoded_subs: true,
|
||||||
@@ -632,7 +632,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let get_all_indexer_settings_command =
|
let get_all_indexer_settings_command =
|
||||||
RadarrCommand::Get(RadarrGetCommand::AllIndexerSettings);
|
RadarrCommand::Get(RadarrGetCommand::AllIndexerSettings);
|
||||||
|
|
||||||
@@ -654,7 +654,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::GetMovieCredits(Some(expected_movie_id)).into(),
|
RadarrEvent::GetMovieCredits(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -662,7 +662,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let list_movie_credits_command =
|
let list_movie_credits_command =
|
||||||
RadarrCommand::List(RadarrListCommand::MovieCredits { movie_id: 1 });
|
RadarrCommand::List(RadarrListCommand::MovieCredits { movie_id: 1 });
|
||||||
|
|
||||||
@@ -680,7 +680,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::UpdateAndScan(Some(expected_movie_id)).into(),
|
RadarrEvent::UpdateAndScan(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -688,7 +688,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let refresh_movie_command =
|
let refresh_movie_command =
|
||||||
RadarrCommand::Refresh(RadarrRefreshCommand::Movie { movie_id: 1 });
|
RadarrCommand::Refresh(RadarrRefreshCommand::Movie { movie_id: 1 });
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ use tokio::sync::Mutex;
|
|||||||
use crate::{
|
use crate::{
|
||||||
app::App,
|
app::App,
|
||||||
cli::{CliCommandHandler, Command},
|
cli::{CliCommandHandler, Command},
|
||||||
execute_network_event,
|
|
||||||
network::{radarr_network::RadarrEvent, NetworkTrait},
|
network::{radarr_network::RadarrEvent, NetworkTrait},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -19,7 +18,7 @@ mod refresh_command_handler_tests;
|
|||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
pub enum RadarrRefreshCommand {
|
pub enum RadarrRefreshCommand {
|
||||||
#[command(about = "Refresh all movie data for all movies in your library")]
|
#[command(about = "Refresh all movie data for all movies in your Radarr library")]
|
||||||
AllMovies,
|
AllMovies,
|
||||||
#[command(about = "Refresh movie data and scan disk for the movie with the given ID")]
|
#[command(about = "Refresh movie data and scan disk for the movie with the given ID")]
|
||||||
Movie {
|
Movie {
|
||||||
@@ -63,22 +62,38 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrRefreshCommand>
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle(self) -> Result<()> {
|
async fn handle(self) -> Result<String> {
|
||||||
match self.command {
|
let result = match self.command {
|
||||||
RadarrRefreshCommand::AllMovies => {
|
RadarrRefreshCommand::AllMovies => {
|
||||||
execute_network_event!(self, RadarrEvent::UpdateAllMovies);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::UpdateAllMovies.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrRefreshCommand::Collections => {
|
RadarrRefreshCommand::Collections => {
|
||||||
execute_network_event!(self, RadarrEvent::UpdateCollections);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::UpdateCollections.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrRefreshCommand::Downloads => {
|
RadarrRefreshCommand::Downloads => {
|
||||||
execute_network_event!(self, RadarrEvent::UpdateDownloads);
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::UpdateDownloads.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
RadarrRefreshCommand::Movie { movie_id } => {
|
RadarrRefreshCommand::Movie { movie_id } => {
|
||||||
execute_network_event!(self, RadarrEvent::UpdateAndScan(Some(movie_id)));
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(RadarrEvent::UpdateAndScan(movie_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
Ok(())
|
Ok(result)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ mod tests {
|
|||||||
use crate::cli::radarr::RadarrCommand;
|
use crate::cli::radarr::RadarrCommand;
|
||||||
use crate::cli::Command;
|
use crate::cli::Command;
|
||||||
use crate::Cli;
|
use crate::Cli;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_radarr_refresh_command_from() {
|
fn test_radarr_refresh_command_from() {
|
||||||
@@ -81,7 +82,7 @@ mod tests {
|
|||||||
#[case(RadarrRefreshCommand::Collections, RadarrEvent::UpdateCollections)]
|
#[case(RadarrRefreshCommand::Collections, RadarrEvent::UpdateCollections)]
|
||||||
#[case(RadarrRefreshCommand::Downloads, RadarrEvent::UpdateDownloads)]
|
#[case(RadarrRefreshCommand::Downloads, RadarrEvent::UpdateDownloads)]
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_handle_list_blocklist_command(
|
async fn test_handle_refresh_command(
|
||||||
#[case] refresh_command: RadarrRefreshCommand,
|
#[case] refresh_command: RadarrRefreshCommand,
|
||||||
#[case] expected_radarr_event: RadarrEvent,
|
#[case] expected_radarr_event: RadarrEvent,
|
||||||
) {
|
) {
|
||||||
@@ -95,7 +96,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
|
||||||
let result = RadarrRefreshCommandHandler::with(&app_arc, refresh_command, &mut mock_network)
|
let result = RadarrRefreshCommandHandler::with(&app_arc, refresh_command, &mut mock_network)
|
||||||
.handle()
|
.handle()
|
||||||
@@ -111,7 +112,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::UpdateAndScan(Some(expected_movie_id)).into(),
|
RadarrEvent::UpdateAndScan(expected_movie_id).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
@@ -119,7 +120,7 @@ mod tests {
|
|||||||
json!({"testResponse": "response"}),
|
json!({"testResponse": "response"}),
|
||||||
)))
|
)))
|
||||||
});
|
});
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
let refresh_movie_command = RadarrRefreshCommand::Movie { movie_id: 1 };
|
let refresh_movie_command = RadarrRefreshCommand::Movie { movie_id: 1 };
|
||||||
|
|
||||||
let result =
|
let result =
|
||||||
|
|||||||
@@ -0,0 +1,176 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::{ArgAction, Subcommand};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
use crate::models::servarr_models::AddRootFolderBody;
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
models::sonarr_models::{AddSeriesBody, AddSeriesOptions, SeriesMonitor, SeriesType},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "add_command_handler_tests.rs"]
|
||||||
|
mod add_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrAddCommand {
|
||||||
|
#[command(about = "Add a new series to your Sonarr library")]
|
||||||
|
Series {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The TVDB ID of the series you wish to add to your library",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
tvdb_id: i64,
|
||||||
|
#[arg(long, help = "The title of the series", required = true)]
|
||||||
|
title: String,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The root folder path where all series data and metadata should live",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
root_folder_path: String,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the quality profile to use for this series",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
quality_profile_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the language profile to use for this series",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
language_profile_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The type of series",
|
||||||
|
value_enum,
|
||||||
|
default_value_t = SeriesType::default()
|
||||||
|
)]
|
||||||
|
series_type: SeriesType,
|
||||||
|
#[arg(long, help = "Disable monitoring for this series")]
|
||||||
|
disable_monitoring: bool,
|
||||||
|
#[arg(long, help = "Don't use season folders for this series")]
|
||||||
|
disable_season_folders: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Tag IDs to tag the series with",
|
||||||
|
value_parser,
|
||||||
|
action = ArgAction::Append
|
||||||
|
)]
|
||||||
|
tag: Vec<i64>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "What Sonarr should monitor",
|
||||||
|
value_enum,
|
||||||
|
default_value_t = SeriesMonitor::default()
|
||||||
|
)]
|
||||||
|
monitor: SeriesMonitor,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Tell Sonarr to not start a search for this series once it's added to your library"
|
||||||
|
)]
|
||||||
|
no_search_for_series: bool,
|
||||||
|
},
|
||||||
|
#[command(about = "Add a new root folder")]
|
||||||
|
RootFolder {
|
||||||
|
#[arg(long, help = "The path of the new root folder", required = true)]
|
||||||
|
root_folder_path: String,
|
||||||
|
},
|
||||||
|
#[command(about = "Add new tag")]
|
||||||
|
Tag {
|
||||||
|
#[arg(long, help = "The name of the tag to be added", required = true)]
|
||||||
|
name: String,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrAddCommand> for Command {
|
||||||
|
fn from(value: SonarrAddCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::Add(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrAddCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrAddCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrAddCommand> for SonarrAddCommandHandler<'a, 'b> {
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrAddCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrAddCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrAddCommand::Series {
|
||||||
|
tvdb_id,
|
||||||
|
title,
|
||||||
|
root_folder_path,
|
||||||
|
quality_profile_id,
|
||||||
|
language_profile_id,
|
||||||
|
series_type,
|
||||||
|
disable_monitoring,
|
||||||
|
disable_season_folders,
|
||||||
|
tag: tags,
|
||||||
|
monitor,
|
||||||
|
no_search_for_series,
|
||||||
|
} => {
|
||||||
|
let body = AddSeriesBody {
|
||||||
|
tvdb_id,
|
||||||
|
title,
|
||||||
|
monitored: !disable_monitoring,
|
||||||
|
root_folder_path,
|
||||||
|
quality_profile_id,
|
||||||
|
language_profile_id,
|
||||||
|
series_type: series_type.to_string(),
|
||||||
|
season_folder: !disable_season_folders,
|
||||||
|
tags,
|
||||||
|
tag_input_string: None,
|
||||||
|
add_options: AddSeriesOptions {
|
||||||
|
monitor: monitor.to_string(),
|
||||||
|
search_for_cutoff_unmet_episodes: !no_search_for_series,
|
||||||
|
search_for_missing_episodes: !no_search_for_series,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::AddSeries(body).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrAddCommand::RootFolder { root_folder_path } => {
|
||||||
|
let add_root_folder_body = AddRootFolderBody {
|
||||||
|
path: root_folder_path,
|
||||||
|
};
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::AddRootFolder(add_root_folder_body).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrAddCommand::Tag { name } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::AddTag(name).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,587 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use clap::{error::ErrorKind, CommandFactory, Parser};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
cli::{
|
||||||
|
sonarr::{add_command_handler::SonarrAddCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
},
|
||||||
|
Cli,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_add_command_from() {
|
||||||
|
let command = SonarrAddCommand::Tag {
|
||||||
|
name: String::new(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(SonarrCommand::Add(command)));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use crate::models::sonarr_models::{SeriesMonitor, SeriesType};
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
use rstest::rstest;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_root_folder_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "add", "root-folder"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_root_folder_success() {
|
||||||
|
let expected_args = SonarrAddCommand::RootFolder {
|
||||||
|
root_folder_path: "/nfs/test".to_owned(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"root-folder",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/nfs/test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Add(add_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(add_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "add", "series"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_requires_tvdb_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--root-folder-path",
|
||||||
|
"test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_requires_title() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--root-folder-path",
|
||||||
|
"test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_requires_root_folder_path() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_requires_quality_profile_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--root-folder-path",
|
||||||
|
"test",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_requires_language_profile_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--root-folder-path",
|
||||||
|
"test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_add_series_assert_argument_flags_require_args(
|
||||||
|
#[values("--series-type", "--tag", "--monitor")] flag: &str,
|
||||||
|
) {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
flag,
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_all_arguments_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_series_type_validation() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
"--series-type",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_monitor_validation() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--tvdb-id",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
"1",
|
||||||
|
"--monitor",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_defaults() {
|
||||||
|
let expected_args = SonarrAddCommand::Series {
|
||||||
|
tvdb_id: 1,
|
||||||
|
title: "test".to_owned(),
|
||||||
|
root_folder_path: "/test".to_owned(),
|
||||||
|
quality_profile_id: 1,
|
||||||
|
language_profile_id: 1,
|
||||||
|
series_type: SeriesType::default(),
|
||||||
|
disable_monitoring: false,
|
||||||
|
disable_season_folders: false,
|
||||||
|
tag: vec![],
|
||||||
|
monitor: SeriesMonitor::default(),
|
||||||
|
no_search_for_series: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Add(add_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(add_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_tags_is_repeatable() {
|
||||||
|
let expected_args = SonarrAddCommand::Series {
|
||||||
|
tvdb_id: 1,
|
||||||
|
title: "test".to_owned(),
|
||||||
|
root_folder_path: "/test".to_owned(),
|
||||||
|
quality_profile_id: 1,
|
||||||
|
language_profile_id: 1,
|
||||||
|
series_type: SeriesType::default(),
|
||||||
|
disable_monitoring: false,
|
||||||
|
disable_season_folders: false,
|
||||||
|
tag: vec![1, 2],
|
||||||
|
monitor: SeriesMonitor::default(),
|
||||||
|
no_search_for_series: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"2",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Add(add_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(add_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_all_args_defined() {
|
||||||
|
let expected_args = SonarrAddCommand::Series {
|
||||||
|
tvdb_id: 1,
|
||||||
|
title: "test".to_owned(),
|
||||||
|
root_folder_path: "/test".to_owned(),
|
||||||
|
quality_profile_id: 1,
|
||||||
|
language_profile_id: 1,
|
||||||
|
series_type: SeriesType::Anime,
|
||||||
|
disable_monitoring: true,
|
||||||
|
disable_season_folders: true,
|
||||||
|
tag: vec![1, 2],
|
||||||
|
monitor: SeriesMonitor::Future,
|
||||||
|
no_search_for_series: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"add",
|
||||||
|
"series",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/test",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--series-type",
|
||||||
|
"anime",
|
||||||
|
"--disable-monitoring",
|
||||||
|
"--disable-season-folders",
|
||||||
|
"--tvdb-id",
|
||||||
|
"1",
|
||||||
|
"--title",
|
||||||
|
"test",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"2",
|
||||||
|
"--monitor",
|
||||||
|
"future",
|
||||||
|
"--no-search-for-series",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Add(add_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(add_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_tag_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "add", "tag"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_tag_success() {
|
||||||
|
let expected_args = SonarrAddCommand::Tag {
|
||||||
|
name: "test".to_owned(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from(["managarr", "sonarr", "add", "tag", "--name", "test"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Add(add_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(add_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{sonarr::add_command_handler::SonarrAddCommandHandler, CliCommandHandler},
|
||||||
|
models::{
|
||||||
|
sonarr_models::{
|
||||||
|
AddSeriesBody, AddSeriesOptions, SeriesMonitor, SeriesType, SonarrSerdeable,
|
||||||
|
},
|
||||||
|
Serdeable,
|
||||||
|
},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
|
||||||
|
use crate::models::servarr_models::AddRootFolderBody;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_add_root_folder_command() {
|
||||||
|
let expected_root_folder_path = "/nfs/test".to_owned();
|
||||||
|
let expected_add_root_folder_body = AddRootFolderBody {
|
||||||
|
path: expected_root_folder_path.clone(),
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::AddRootFolder(expected_add_root_folder_body.clone()).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let add_root_folder_command = SonarrAddCommand::RootFolder {
|
||||||
|
root_folder_path: expected_root_folder_path,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrAddCommandHandler::with(&app_arc, add_root_folder_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_add_series_command() {
|
||||||
|
let expected_add_series_body = AddSeriesBody {
|
||||||
|
tvdb_id: 1,
|
||||||
|
title: "test".to_owned(),
|
||||||
|
root_folder_path: "/test".to_owned(),
|
||||||
|
quality_profile_id: 1,
|
||||||
|
language_profile_id: 1,
|
||||||
|
series_type: "anime".to_owned(),
|
||||||
|
monitored: false,
|
||||||
|
tags: vec![1, 2],
|
||||||
|
tag_input_string: None,
|
||||||
|
season_folder: false,
|
||||||
|
add_options: AddSeriesOptions {
|
||||||
|
monitor: "future".to_owned(),
|
||||||
|
search_for_cutoff_unmet_episodes: false,
|
||||||
|
search_for_missing_episodes: false,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::AddSeries(expected_add_series_body).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let add_series_command = SonarrAddCommand::Series {
|
||||||
|
tvdb_id: 1,
|
||||||
|
title: "test".to_owned(),
|
||||||
|
root_folder_path: "/test".to_owned(),
|
||||||
|
quality_profile_id: 1,
|
||||||
|
language_profile_id: 1,
|
||||||
|
series_type: SeriesType::Anime,
|
||||||
|
disable_monitoring: true,
|
||||||
|
disable_season_folders: true,
|
||||||
|
tag: vec![1, 2],
|
||||||
|
monitor: SeriesMonitor::Future,
|
||||||
|
no_search_for_series: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrAddCommandHandler::with(&app_arc, add_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_add_tag_command() {
|
||||||
|
let expected_tag_name = "test".to_owned();
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::AddTag(expected_tag_name.clone()).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let add_tag_command = SonarrAddCommand::Tag {
|
||||||
|
name: expected_tag_name,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrAddCommandHandler::with(&app_arc, add_tag_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,156 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::Subcommand;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
models::sonarr_models::DeleteSeriesParams,
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "delete_command_handler_tests.rs"]
|
||||||
|
mod delete_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrDeleteCommand {
|
||||||
|
#[command(about = "Delete the specified item from the Sonarr blocklist")]
|
||||||
|
BlocklistItem {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the blocklist item to remove from the blocklist",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
blocklist_item_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Delete the specified download")]
|
||||||
|
Download {
|
||||||
|
#[arg(long, help = "The ID of the download to delete", required = true)]
|
||||||
|
download_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Delete the specified episode file from disk")]
|
||||||
|
EpisodeFile {
|
||||||
|
#[arg(long, help = "The ID of the episode file to delete", required = true)]
|
||||||
|
episode_file_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Delete the indexer with the given ID")]
|
||||||
|
Indexer {
|
||||||
|
#[arg(long, help = "The ID of the indexer to delete", required = true)]
|
||||||
|
indexer_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Delete the root folder with the given ID")]
|
||||||
|
RootFolder {
|
||||||
|
#[arg(long, help = "The ID of the root folder to delete", required = true)]
|
||||||
|
root_folder_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Delete a series from your Sonarr library")]
|
||||||
|
Series {
|
||||||
|
#[arg(long, help = "The ID of the series to delete", required = true)]
|
||||||
|
series_id: i64,
|
||||||
|
#[arg(long, help = "Delete the series files from disk as well")]
|
||||||
|
delete_files_from_disk: bool,
|
||||||
|
#[arg(long, help = "Add a list exclusion for this series")]
|
||||||
|
add_list_exclusion: bool,
|
||||||
|
},
|
||||||
|
#[command(about = "Delete the tag with the specified ID")]
|
||||||
|
Tag {
|
||||||
|
#[arg(long, help = "The ID of the tag to delete", required = true)]
|
||||||
|
tag_id: i64,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrDeleteCommand> for Command {
|
||||||
|
fn from(value: SonarrDeleteCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::Delete(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrDeleteCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrDeleteCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrDeleteCommand> for SonarrDeleteCommandHandler<'a, 'b> {
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrDeleteCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrDeleteCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let resp = match self.command {
|
||||||
|
SonarrDeleteCommand::BlocklistItem { blocklist_item_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DeleteBlocklistItem(blocklist_item_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDeleteCommand::Download { download_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DeleteDownload(download_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDeleteCommand::EpisodeFile { episode_file_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DeleteEpisodeFile(episode_file_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDeleteCommand::Indexer { indexer_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DeleteIndexer(indexer_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDeleteCommand::RootFolder { root_folder_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DeleteRootFolder(root_folder_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDeleteCommand::Series {
|
||||||
|
series_id,
|
||||||
|
delete_files_from_disk,
|
||||||
|
add_list_exclusion,
|
||||||
|
} => {
|
||||||
|
let delete_series_params = DeleteSeriesParams {
|
||||||
|
id: series_id,
|
||||||
|
delete_series_files: delete_files_from_disk,
|
||||||
|
add_list_exclusion,
|
||||||
|
};
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DeleteSeries(delete_series_params).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDeleteCommand::Tag { tag_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DeleteTag(tag_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(resp)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,492 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
cli::{
|
||||||
|
sonarr::{delete_command_handler::SonarrDeleteCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
},
|
||||||
|
Cli,
|
||||||
|
};
|
||||||
|
use clap::{error::ErrorKind, CommandFactory, Parser};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_delete_command_from() {
|
||||||
|
let command = SonarrDeleteCommand::BlocklistItem {
|
||||||
|
blocklist_item_id: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(SonarrCommand::Delete(command)));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use super::*;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_blocklist_item_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "blocklist-item"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_blocklist_item_success() {
|
||||||
|
let expected_args = SonarrDeleteCommand::BlocklistItem {
|
||||||
|
blocklist_item_id: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"delete",
|
||||||
|
"blocklist-item",
|
||||||
|
"--blocklist-item-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_download_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "download"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_download_success() {
|
||||||
|
let expected_args = SonarrDeleteCommand::Download { download_id: 1 };
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"delete",
|
||||||
|
"download",
|
||||||
|
"--download-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_episode_file_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "episode-file"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_episode_file_success() {
|
||||||
|
let expected_args = SonarrDeleteCommand::EpisodeFile { episode_file_id: 1 };
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"delete",
|
||||||
|
"episode-file",
|
||||||
|
"--episode-file-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_indexer_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "indexer"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_indexer_success() {
|
||||||
|
let expected_args = SonarrDeleteCommand::Indexer { indexer_id: 1 };
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"delete",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_root_folder_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "root-folder"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_root_folder_success() {
|
||||||
|
let expected_args = SonarrDeleteCommand::RootFolder { root_folder_id: 1 };
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"delete",
|
||||||
|
"root-folder",
|
||||||
|
"--root-folder-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_series_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "series"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_series_defaults() {
|
||||||
|
let expected_args = SonarrDeleteCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
delete_files_from_disk: false,
|
||||||
|
add_list_exclusion: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
Cli::try_parse_from(["managarr", "sonarr", "delete", "series", "--series-id", "1"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_series_all_args_defined() {
|
||||||
|
let expected_args = SonarrDeleteCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
delete_files_from_disk: true,
|
||||||
|
add_list_exclusion: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"delete",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--delete-files-from-disk",
|
||||||
|
"--add-list-exclusion",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_tag_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "tag"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_delete_tag_success() {
|
||||||
|
let expected_args = SonarrDeleteCommand::Tag { tag_id: 1 };
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from(["managarr", "sonarr", "delete", "tag", "--tag-id", "1"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(delete_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{
|
||||||
|
sonarr::delete_command_handler::{SonarrDeleteCommand, SonarrDeleteCommandHandler},
|
||||||
|
CliCommandHandler,
|
||||||
|
},
|
||||||
|
models::{
|
||||||
|
sonarr_models::{DeleteSeriesParams, SonarrSerdeable},
|
||||||
|
Serdeable,
|
||||||
|
},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_delete_blocklist_item_command() {
|
||||||
|
let expected_blocklist_item_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteBlocklistItem(expected_blocklist_item_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_blocklist_item_command = SonarrDeleteCommand::BlocklistItem {
|
||||||
|
blocklist_item_id: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrDeleteCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
delete_blocklist_item_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_delete_download_command() {
|
||||||
|
let expected_download_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteDownload(expected_download_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_download_command = SonarrDeleteCommand::Download { download_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDeleteCommandHandler::with(&app_arc, delete_download_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_delete_episode_file_command() {
|
||||||
|
let expected_episode_file_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteEpisodeFile(expected_episode_file_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_episode_file_command = SonarrDeleteCommand::EpisodeFile { episode_file_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDeleteCommandHandler::with(&app_arc, delete_episode_file_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_delete_indexer_command() {
|
||||||
|
let expected_indexer_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteIndexer(expected_indexer_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_indexer_command = SonarrDeleteCommand::Indexer { indexer_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDeleteCommandHandler::with(&app_arc, delete_indexer_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_delete_root_folder_command() {
|
||||||
|
let expected_root_folder_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteRootFolder(expected_root_folder_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_root_folder_command = SonarrDeleteCommand::RootFolder { root_folder_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDeleteCommandHandler::with(&app_arc, delete_root_folder_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_delete_series_command() {
|
||||||
|
let expected_delete_series_params = DeleteSeriesParams {
|
||||||
|
id: 1,
|
||||||
|
delete_series_files: true,
|
||||||
|
add_list_exclusion: true,
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteSeries(expected_delete_series_params).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_series_command = SonarrDeleteCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
delete_files_from_disk: true,
|
||||||
|
add_list_exclusion: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDeleteCommandHandler::with(&app_arc, delete_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_delete_tag_command() {
|
||||||
|
let expected_tag_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteTag(expected_tag_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_tag_command = SonarrDeleteCommand::Tag { tag_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDeleteCommandHandler::with(&app_arc, delete_tag_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,169 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::Subcommand;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
models::sonarr_models::SonarrReleaseDownloadBody,
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "download_command_handler_tests.rs"]
|
||||||
|
mod download_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrDownloadCommand {
|
||||||
|
#[command(about = "Manually download the given series release for the specified series ID")]
|
||||||
|
Series {
|
||||||
|
#[arg(long, help = "The GUID of the release to download", required = true)]
|
||||||
|
guid: String,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The indexer ID to download the release from",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
indexer_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The series ID that the release is associated with",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
},
|
||||||
|
#[command(
|
||||||
|
about = "Manually download the given season release corresponding to the series specified with the series ID"
|
||||||
|
)]
|
||||||
|
Season {
|
||||||
|
#[arg(long, help = "The GUID of the release to download", required = true)]
|
||||||
|
guid: String,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The indexer ID to download the release from",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
indexer_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The series ID that the release is associated with",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The season number that the release corresponds to",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
season_number: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Manually download the given episode release for the specified episode ID")]
|
||||||
|
Episode {
|
||||||
|
#[arg(long, help = "The GUID of the release to download", required = true)]
|
||||||
|
guid: String,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The indexer ID to download the release from",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
indexer_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The episode ID that the release is associated with",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
episode_id: i64,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrDownloadCommand> for Command {
|
||||||
|
fn from(value: SonarrDownloadCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::Download(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrDownloadCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrDownloadCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrDownloadCommand>
|
||||||
|
for SonarrDownloadCommandHandler<'a, 'b>
|
||||||
|
{
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrDownloadCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrDownloadCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrDownloadCommand::Series {
|
||||||
|
guid,
|
||||||
|
indexer_id,
|
||||||
|
series_id,
|
||||||
|
} => {
|
||||||
|
let params = SonarrReleaseDownloadBody {
|
||||||
|
guid,
|
||||||
|
indexer_id,
|
||||||
|
series_id: Some(series_id),
|
||||||
|
..SonarrReleaseDownloadBody::default()
|
||||||
|
};
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DownloadRelease(params).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDownloadCommand::Season {
|
||||||
|
guid,
|
||||||
|
indexer_id,
|
||||||
|
series_id,
|
||||||
|
season_number,
|
||||||
|
} => {
|
||||||
|
let params = SonarrReleaseDownloadBody {
|
||||||
|
guid,
|
||||||
|
indexer_id,
|
||||||
|
series_id: Some(series_id),
|
||||||
|
season_number: Some(season_number),
|
||||||
|
..SonarrReleaseDownloadBody::default()
|
||||||
|
};
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DownloadRelease(params).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrDownloadCommand::Episode {
|
||||||
|
guid,
|
||||||
|
indexer_id,
|
||||||
|
episode_id,
|
||||||
|
} => {
|
||||||
|
let params = SonarrReleaseDownloadBody {
|
||||||
|
guid,
|
||||||
|
indexer_id,
|
||||||
|
episode_id: Some(episode_id),
|
||||||
|
..SonarrReleaseDownloadBody::default()
|
||||||
|
};
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::DownloadRelease(params).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,423 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::{
|
||||||
|
cli::{
|
||||||
|
sonarr::{download_command_handler::SonarrDownloadCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
},
|
||||||
|
Cli,
|
||||||
|
};
|
||||||
|
use clap::CommandFactory;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_download_command_from() {
|
||||||
|
let command = SonarrDownloadCommand::Series {
|
||||||
|
guid: "Test".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
series_id: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(SonarrCommand::Download(command)));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use super::*;
|
||||||
|
use clap::error::ErrorKind;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_series_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"series",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_series_requires_guid() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"series",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_series_requires_indexer_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"series",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_series_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"series",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_season_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"season",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_season_requires_season_number() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"season",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_season_requires_guid() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"season",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_season_requires_indexer_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"season",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_season_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"season",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_episode_requires_episode_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"episode",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_episode_requires_guid() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"episode",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_episode_requires_indexer_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"episode",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_download_episode_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"download",
|
||||||
|
"episode",
|
||||||
|
"--guid",
|
||||||
|
"1",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{
|
||||||
|
sonarr::download_command_handler::{SonarrDownloadCommand, SonarrDownloadCommandHandler},
|
||||||
|
CliCommandHandler,
|
||||||
|
},
|
||||||
|
models::{
|
||||||
|
sonarr_models::{SonarrReleaseDownloadBody, SonarrSerdeable},
|
||||||
|
Serdeable,
|
||||||
|
},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_download_series_release_command() {
|
||||||
|
let expected_release_download_body = SonarrReleaseDownloadBody {
|
||||||
|
guid: "guid".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
series_id: Some(1),
|
||||||
|
..SonarrReleaseDownloadBody::default()
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DownloadRelease(expected_release_download_body).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let download_release_command = SonarrDownloadCommand::Series {
|
||||||
|
guid: "guid".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
series_id: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDownloadCommandHandler::with(&app_arc, download_release_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_download_season_release_command() {
|
||||||
|
let expected_release_download_body = SonarrReleaseDownloadBody {
|
||||||
|
guid: "guid".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
series_id: Some(1),
|
||||||
|
season_number: Some(1),
|
||||||
|
..SonarrReleaseDownloadBody::default()
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DownloadRelease(expected_release_download_body).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let download_release_command = SonarrDownloadCommand::Season {
|
||||||
|
guid: "guid".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
series_id: 1,
|
||||||
|
season_number: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDownloadCommandHandler::with(&app_arc, download_release_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_download_episode_release_command() {
|
||||||
|
let expected_release_download_body = SonarrReleaseDownloadBody {
|
||||||
|
guid: "guid".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
episode_id: Some(1),
|
||||||
|
..SonarrReleaseDownloadBody::default()
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DownloadRelease(expected_release_download_body).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let download_release_command = SonarrDownloadCommand::Episode {
|
||||||
|
guid: "guid".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
episode_id: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrDownloadCommandHandler::with(&app_arc, download_release_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,365 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::{ArgAction, ArgGroup, Subcommand};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{mutex_flags_or_option, CliCommandHandler, Command},
|
||||||
|
models::{
|
||||||
|
servarr_models::EditIndexerParams,
|
||||||
|
sonarr_models::{EditSeriesParams, IndexerSettings, SeriesType, SonarrSerdeable},
|
||||||
|
Serdeable,
|
||||||
|
},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "edit_command_handler_tests.rs"]
|
||||||
|
mod edit_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrEditCommand {
|
||||||
|
#[command(
|
||||||
|
about = "Edit and indexer settings that apply to all indexers",
|
||||||
|
group(
|
||||||
|
ArgGroup::new("edit_settings")
|
||||||
|
.args([
|
||||||
|
"maximum_size",
|
||||||
|
"minimum_age",
|
||||||
|
"retention",
|
||||||
|
"rss_sync_interval",
|
||||||
|
]).required(true)
|
||||||
|
.multiple(true))
|
||||||
|
)]
|
||||||
|
AllIndexerSettings {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The maximum size for a release to be grabbed in MB. Set to zero to set to unlimited"
|
||||||
|
)]
|
||||||
|
maximum_size: Option<i64>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Usenet only: Minimum age in minutes of NZBs before they are grabbed. Use this to give new releases time to propagate to your usenet provider."
|
||||||
|
)]
|
||||||
|
minimum_age: Option<i64>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Usenet only: The retention time in days to retain releases. Set to zero to set for unlimited retention"
|
||||||
|
)]
|
||||||
|
retention: Option<i64>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The RSS sync interval in minutes. Set to zero to disable (this will stop all automatic release grabbing)"
|
||||||
|
)]
|
||||||
|
rss_sync_interval: Option<i64>,
|
||||||
|
},
|
||||||
|
#[command(
|
||||||
|
about = "Edit preferences for the specified indexer",
|
||||||
|
group(
|
||||||
|
ArgGroup::new("edit_indexer")
|
||||||
|
.args([
|
||||||
|
"name",
|
||||||
|
"enable_rss",
|
||||||
|
"disable_rss",
|
||||||
|
"enable_automatic_search",
|
||||||
|
"disable_automatic_search",
|
||||||
|
"enable_interactive_search",
|
||||||
|
"disable_automatic_search",
|
||||||
|
"url",
|
||||||
|
"api_key",
|
||||||
|
"seed_ratio",
|
||||||
|
"tag",
|
||||||
|
"priority",
|
||||||
|
"clear_tags"
|
||||||
|
]).required(true)
|
||||||
|
.multiple(true))
|
||||||
|
)]
|
||||||
|
Indexer {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the indexer whose settings you wish to edit",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
indexer_id: i64,
|
||||||
|
#[arg(long, help = "The name of the indexer")]
|
||||||
|
name: Option<String>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Indicate to Sonarr that this indexer should be used when Sonarr periodically looks for releases via RSS Sync",
|
||||||
|
conflicts_with = "disable_rss"
|
||||||
|
)]
|
||||||
|
enable_rss: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Disable using this indexer when Sonarr periodically looks for releases via RSS Sync",
|
||||||
|
conflicts_with = "enable_rss"
|
||||||
|
)]
|
||||||
|
disable_rss: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Indicate to Sonarr that this indexer should be used when automatic searches are performed via the UI or by Sonarr",
|
||||||
|
conflicts_with = "disable_automatic_search"
|
||||||
|
)]
|
||||||
|
enable_automatic_search: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Disable using this indexer whenever automatic searches are performed via the UI or by Sonarr",
|
||||||
|
conflicts_with = "enable_automatic_search"
|
||||||
|
)]
|
||||||
|
disable_automatic_search: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Indicate to Sonarr that this indexer should be used when an interactive search is used",
|
||||||
|
conflicts_with = "disable_interactive_search"
|
||||||
|
)]
|
||||||
|
enable_interactive_search: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Disable using this indexer whenever an interactive search is performed",
|
||||||
|
conflicts_with = "enable_interactive_search"
|
||||||
|
)]
|
||||||
|
disable_interactive_search: bool,
|
||||||
|
#[arg(long, help = "The URL of the indexer")]
|
||||||
|
url: Option<String>,
|
||||||
|
#[arg(long, help = "The API key used to access the indexer's API")]
|
||||||
|
api_key: Option<String>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ratio a torrent should reach before stopping; Empty uses the download client's default. Ratio should be at least 1.0 and follow the indexer's rules"
|
||||||
|
)]
|
||||||
|
seed_ratio: Option<String>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Only use this indexer for series with at least one matching tag ID. Leave blank to use with all series.",
|
||||||
|
value_parser,
|
||||||
|
action = ArgAction::Append,
|
||||||
|
conflicts_with = "clear_tags"
|
||||||
|
)]
|
||||||
|
tag: Option<Vec<i64>>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Indexer Priority from 1 (Highest) to 50 (Lowest). Default: 25. Used when grabbing releases as a tiebreaker for otherwise equal releases, Sonarr will still use all enabled indexers for RSS Sync and Searching"
|
||||||
|
)]
|
||||||
|
priority: Option<i64>,
|
||||||
|
#[arg(long, help = "Clear all tags on this indexer", conflicts_with = "tag")]
|
||||||
|
clear_tags: bool,
|
||||||
|
},
|
||||||
|
#[command(
|
||||||
|
about = "Edit preferences for the specified series",
|
||||||
|
group(
|
||||||
|
ArgGroup::new("edit_series")
|
||||||
|
.args([
|
||||||
|
"enable_monitoring",
|
||||||
|
"disable_monitoring",
|
||||||
|
"enable_season_folders",
|
||||||
|
"disable_season_folders",
|
||||||
|
"series_type",
|
||||||
|
"quality_profile_id",
|
||||||
|
"language_profile_id",
|
||||||
|
"root_folder_path",
|
||||||
|
"tag",
|
||||||
|
"clear_tags"
|
||||||
|
]).required(true)
|
||||||
|
.multiple(true))
|
||||||
|
)]
|
||||||
|
Series {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the series whose settings you want to edit",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Enable monitoring of this series in Sonarr so Sonarr will automatically download this series if it is available",
|
||||||
|
conflicts_with = "disable_monitoring"
|
||||||
|
)]
|
||||||
|
enable_monitoring: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Disable monitoring of this series so Sonarr does not automatically download the series if it is found to be available",
|
||||||
|
conflicts_with = "enable_monitoring"
|
||||||
|
)]
|
||||||
|
disable_monitoring: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The minimum availability to monitor for this film",
|
||||||
|
value_enum
|
||||||
|
)]
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Enable sorting episodes of this series into season folders",
|
||||||
|
conflicts_with = "disable_season_folders"
|
||||||
|
)]
|
||||||
|
enable_season_folders: bool,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Disable sorting episodes of this series into season folders",
|
||||||
|
conflicts_with = "enable_season_folders"
|
||||||
|
)]
|
||||||
|
disable_season_folders: bool,
|
||||||
|
#[arg(long, help = "The type of series", value_enum)]
|
||||||
|
series_type: Option<SeriesType>,
|
||||||
|
#[arg(long, help = "The ID of the quality profile to use for this series")]
|
||||||
|
quality_profile_id: Option<i64>,
|
||||||
|
#[arg(long, help = "The ID of the language profile to use for this series")]
|
||||||
|
language_profile_id: Option<i64>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The root folder path where all film data and metadata should live"
|
||||||
|
)]
|
||||||
|
root_folder_path: Option<String>,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Tag IDs to tag this series with",
|
||||||
|
value_parser,
|
||||||
|
action = ArgAction::Append,
|
||||||
|
conflicts_with = "clear_tags"
|
||||||
|
)]
|
||||||
|
tag: Option<Vec<i64>>,
|
||||||
|
#[arg(long, help = "Clear all tags on this series", conflicts_with = "tag")]
|
||||||
|
clear_tags: bool,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrEditCommand> for Command {
|
||||||
|
fn from(value: SonarrEditCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::Edit(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrEditCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrEditCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrEditCommand> for SonarrEditCommandHandler<'a, 'b> {
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrEditCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrEditCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrEditCommand::AllIndexerSettings {
|
||||||
|
maximum_size,
|
||||||
|
minimum_age,
|
||||||
|
retention,
|
||||||
|
rss_sync_interval,
|
||||||
|
} => {
|
||||||
|
if let Serdeable::Sonarr(SonarrSerdeable::IndexerSettings(previous_indexer_settings)) = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetAllIndexerSettings.into())
|
||||||
|
.await?
|
||||||
|
{
|
||||||
|
let params = IndexerSettings {
|
||||||
|
id: 1,
|
||||||
|
maximum_size: maximum_size.unwrap_or(previous_indexer_settings.maximum_size),
|
||||||
|
minimum_age: minimum_age.unwrap_or(previous_indexer_settings.minimum_age),
|
||||||
|
retention: retention.unwrap_or(previous_indexer_settings.retention),
|
||||||
|
rss_sync_interval: rss_sync_interval
|
||||||
|
.unwrap_or(previous_indexer_settings.rss_sync_interval),
|
||||||
|
};
|
||||||
|
self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::EditAllIndexerSettings(params).into())
|
||||||
|
.await?;
|
||||||
|
"All indexer settings updated".to_owned()
|
||||||
|
} else {
|
||||||
|
String::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
SonarrEditCommand::Indexer {
|
||||||
|
indexer_id,
|
||||||
|
name,
|
||||||
|
enable_rss,
|
||||||
|
disable_rss,
|
||||||
|
enable_automatic_search,
|
||||||
|
disable_automatic_search,
|
||||||
|
enable_interactive_search,
|
||||||
|
disable_interactive_search,
|
||||||
|
url,
|
||||||
|
api_key,
|
||||||
|
seed_ratio,
|
||||||
|
tag,
|
||||||
|
priority,
|
||||||
|
clear_tags,
|
||||||
|
} => {
|
||||||
|
let rss_value = mutex_flags_or_option(enable_rss, disable_rss);
|
||||||
|
let automatic_search_value =
|
||||||
|
mutex_flags_or_option(enable_automatic_search, disable_automatic_search);
|
||||||
|
let interactive_search_value =
|
||||||
|
mutex_flags_or_option(enable_interactive_search, disable_interactive_search);
|
||||||
|
let edit_indexer_params = EditIndexerParams {
|
||||||
|
indexer_id,
|
||||||
|
name,
|
||||||
|
enable_rss: rss_value,
|
||||||
|
enable_automatic_search: automatic_search_value,
|
||||||
|
enable_interactive_search: interactive_search_value,
|
||||||
|
url,
|
||||||
|
api_key,
|
||||||
|
seed_ratio,
|
||||||
|
tags: tag,
|
||||||
|
tag_input_string: None,
|
||||||
|
priority,
|
||||||
|
clear_tags,
|
||||||
|
};
|
||||||
|
|
||||||
|
self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::EditIndexer(edit_indexer_params).into())
|
||||||
|
.await?;
|
||||||
|
"Indexer updated".to_owned()
|
||||||
|
}
|
||||||
|
SonarrEditCommand::Series {
|
||||||
|
series_id,
|
||||||
|
enable_monitoring,
|
||||||
|
disable_monitoring,
|
||||||
|
enable_season_folders,
|
||||||
|
disable_season_folders,
|
||||||
|
series_type,
|
||||||
|
quality_profile_id,
|
||||||
|
language_profile_id,
|
||||||
|
root_folder_path,
|
||||||
|
tag,
|
||||||
|
clear_tags,
|
||||||
|
} => {
|
||||||
|
let monitored_value = mutex_flags_or_option(enable_monitoring, disable_monitoring);
|
||||||
|
let season_folders_value =
|
||||||
|
mutex_flags_or_option(enable_season_folders, disable_season_folders);
|
||||||
|
let edit_series_params = EditSeriesParams {
|
||||||
|
series_id,
|
||||||
|
monitored: monitored_value,
|
||||||
|
use_season_folders: season_folders_value,
|
||||||
|
series_type,
|
||||||
|
quality_profile_id,
|
||||||
|
language_profile_id,
|
||||||
|
root_folder_path,
|
||||||
|
tags: tag,
|
||||||
|
tag_input_string: None,
|
||||||
|
clear_tags,
|
||||||
|
};
|
||||||
|
|
||||||
|
self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::EditSeries(edit_series_params).into())
|
||||||
|
.await?;
|
||||||
|
"Series Updated".to_owned()
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,878 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::cli::{
|
||||||
|
sonarr::{edit_command_handler::SonarrEditCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_edit_command_from() {
|
||||||
|
let command = SonarrEditCommand::AllIndexerSettings {
|
||||||
|
maximum_size: None,
|
||||||
|
minimum_age: None,
|
||||||
|
retention: None,
|
||||||
|
rss_sync_interval: None,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(SonarrCommand::Edit(command)));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use crate::{models::sonarr_models::SeriesType, Cli};
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
use clap::{error::ErrorKind, CommandFactory, Parser};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
use rstest::rstest;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_all_indexer_settings_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "edit", "all-indexer-settings"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_edit_all_indexer_settings_assert_argument_flags_require_args(
|
||||||
|
#[values(
|
||||||
|
"--maximum-size",
|
||||||
|
"--minimum-age",
|
||||||
|
"--retention",
|
||||||
|
"--rss-sync-interval"
|
||||||
|
)]
|
||||||
|
flag: &str,
|
||||||
|
) {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"all-indexer-settings",
|
||||||
|
flag,
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_all_indexer_settings_only_requires_at_least_one_argument() {
|
||||||
|
let expected_args = SonarrEditCommand::AllIndexerSettings {
|
||||||
|
maximum_size: Some(1),
|
||||||
|
minimum_age: None,
|
||||||
|
retention: None,
|
||||||
|
rss_sync_interval: None,
|
||||||
|
};
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"all-indexer-settings",
|
||||||
|
"--maximum-size",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_all_indexer_settings_all_arguments_defined() {
|
||||||
|
let expected_args = SonarrEditCommand::AllIndexerSettings {
|
||||||
|
maximum_size: Some(1),
|
||||||
|
minimum_age: Some(1),
|
||||||
|
retention: Some(1),
|
||||||
|
rss_sync_interval: Some(1),
|
||||||
|
};
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"all-indexer-settings",
|
||||||
|
"--maximum-size",
|
||||||
|
"1",
|
||||||
|
"--minimum-age",
|
||||||
|
"1",
|
||||||
|
"--retention",
|
||||||
|
"1",
|
||||||
|
"--rss-sync-interval",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "edit", "indexer"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_with_indexer_id_still_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_rss_flags_conflict() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--enable-rss",
|
||||||
|
"--disable-rss",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::ArgumentConflict);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_automatic_search_flags_conflict() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--enable-automatic-search",
|
||||||
|
"--disable-automatic-search",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::ArgumentConflict);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_interactive_search_flags_conflict() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--enable-interactive-search",
|
||||||
|
"--disable-interactive-search",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::ArgumentConflict);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_tag_flags_conflict() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--clear-tags",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::ArgumentConflict);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_edit_indexer_assert_argument_flags_require_args(
|
||||||
|
#[values("--name", "--url", "--api-key", "--seed-ratio", "--tag", "--priority")] flag: &str,
|
||||||
|
) {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
flag,
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_only_requires_at_least_one_argument_plus_indexer_id() {
|
||||||
|
let expected_args = SonarrEditCommand::Indexer {
|
||||||
|
indexer_id: 1,
|
||||||
|
name: Some("Test".to_owned()),
|
||||||
|
enable_rss: false,
|
||||||
|
disable_rss: false,
|
||||||
|
enable_automatic_search: false,
|
||||||
|
disable_automatic_search: false,
|
||||||
|
enable_interactive_search: false,
|
||||||
|
disable_interactive_search: false,
|
||||||
|
url: None,
|
||||||
|
api_key: None,
|
||||||
|
seed_ratio: None,
|
||||||
|
tag: None,
|
||||||
|
priority: None,
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--name",
|
||||||
|
"Test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_tag_argument_is_repeatable() {
|
||||||
|
let expected_args = SonarrEditCommand::Indexer {
|
||||||
|
indexer_id: 1,
|
||||||
|
name: None,
|
||||||
|
enable_rss: false,
|
||||||
|
disable_rss: false,
|
||||||
|
enable_automatic_search: false,
|
||||||
|
disable_automatic_search: false,
|
||||||
|
enable_interactive_search: false,
|
||||||
|
disable_interactive_search: false,
|
||||||
|
url: None,
|
||||||
|
api_key: None,
|
||||||
|
seed_ratio: None,
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
priority: None,
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"2",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_indexer_all_arguments_defined() {
|
||||||
|
let expected_args = SonarrEditCommand::Indexer {
|
||||||
|
indexer_id: 1,
|
||||||
|
name: Some("Test".to_owned()),
|
||||||
|
enable_rss: true,
|
||||||
|
disable_rss: false,
|
||||||
|
enable_automatic_search: true,
|
||||||
|
disable_automatic_search: false,
|
||||||
|
enable_interactive_search: true,
|
||||||
|
disable_interactive_search: false,
|
||||||
|
url: Some("http://test.com".to_owned()),
|
||||||
|
api_key: Some("testKey".to_owned()),
|
||||||
|
seed_ratio: Some("1.2".to_owned()),
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
priority: Some(25),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
"--name",
|
||||||
|
"Test",
|
||||||
|
"--enable-rss",
|
||||||
|
"--enable-automatic-search",
|
||||||
|
"--enable-interactive-search",
|
||||||
|
"--url",
|
||||||
|
"http://test.com",
|
||||||
|
"--api-key",
|
||||||
|
"testKey",
|
||||||
|
"--seed-ratio",
|
||||||
|
"1.2",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"2",
|
||||||
|
"--priority",
|
||||||
|
"25",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "edit", "series"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_with_series_id_still_requires_arguments() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_monitoring_flags_conflict() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--enable-monitoring",
|
||||||
|
"--disable-monitoring",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::ArgumentConflict);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_season_folders_flags_conflict() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--enable-season-folders",
|
||||||
|
"--disable-season-folders",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::ArgumentConflict);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_tag_flags_conflict() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--clear-tags",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::ArgumentConflict);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_edit_series_assert_argument_flags_require_args(
|
||||||
|
#[values(
|
||||||
|
"--series-type",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"--language-profile-id",
|
||||||
|
"--root-folder-path",
|
||||||
|
"--tag"
|
||||||
|
)]
|
||||||
|
flag: &str,
|
||||||
|
) {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
flag,
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_series_type_validation() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--series-type",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_only_requires_at_least_one_argument_plus_series_id() {
|
||||||
|
let expected_args = SonarrEditCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
enable_monitoring: false,
|
||||||
|
disable_monitoring: false,
|
||||||
|
enable_season_folders: false,
|
||||||
|
disable_season_folders: false,
|
||||||
|
series_type: None,
|
||||||
|
quality_profile_id: None,
|
||||||
|
language_profile_id: None,
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tag: None,
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/nfs/test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_tag_argument_is_repeatable() {
|
||||||
|
let expected_args = SonarrEditCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
enable_monitoring: false,
|
||||||
|
disable_monitoring: false,
|
||||||
|
enable_season_folders: false,
|
||||||
|
disable_season_folders: false,
|
||||||
|
series_type: None,
|
||||||
|
quality_profile_id: None,
|
||||||
|
language_profile_id: None,
|
||||||
|
root_folder_path: None,
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"2",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_edit_series_all_arguments_defined() {
|
||||||
|
let expected_args = SonarrEditCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
enable_monitoring: true,
|
||||||
|
disable_monitoring: false,
|
||||||
|
enable_season_folders: true,
|
||||||
|
disable_season_folders: false,
|
||||||
|
series_type: Some(SeriesType::Anime),
|
||||||
|
quality_profile_id: Some(1),
|
||||||
|
language_profile_id: Some(1),
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"edit",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--enable-monitoring",
|
||||||
|
"--enable-season-folders",
|
||||||
|
"--series-type",
|
||||||
|
"anime",
|
||||||
|
"--quality-profile-id",
|
||||||
|
"1",
|
||||||
|
"--language-profile-id",
|
||||||
|
"1",
|
||||||
|
"--root-folder-path",
|
||||||
|
"/nfs/test",
|
||||||
|
"--tag",
|
||||||
|
"1",
|
||||||
|
"--tag",
|
||||||
|
"2",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Edit(edit_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(edit_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{
|
||||||
|
sonarr::edit_command_handler::{SonarrEditCommand, SonarrEditCommandHandler},
|
||||||
|
CliCommandHandler,
|
||||||
|
},
|
||||||
|
models::{
|
||||||
|
servarr_models::EditIndexerParams,
|
||||||
|
sonarr_models::{EditSeriesParams, IndexerSettings, SeriesType, SonarrSerdeable},
|
||||||
|
Serdeable,
|
||||||
|
},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_edit_all_indexer_settings_command() {
|
||||||
|
let expected_edit_all_indexer_settings = IndexerSettings {
|
||||||
|
id: 1,
|
||||||
|
maximum_size: 1,
|
||||||
|
minimum_age: 1,
|
||||||
|
retention: 1,
|
||||||
|
rss_sync_interval: 1,
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetAllIndexerSettings.into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::IndexerSettings(
|
||||||
|
IndexerSettings {
|
||||||
|
id: 1,
|
||||||
|
maximum_size: 2,
|
||||||
|
minimum_age: 2,
|
||||||
|
retention: 2,
|
||||||
|
rss_sync_interval: 2,
|
||||||
|
},
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let edit_all_indexer_settings_command = SonarrEditCommand::AllIndexerSettings {
|
||||||
|
maximum_size: Some(1),
|
||||||
|
minimum_age: Some(1),
|
||||||
|
retention: Some(1),
|
||||||
|
rss_sync_interval: Some(1),
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrEditCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
edit_all_indexer_settings_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_edit_indexer_command() {
|
||||||
|
let expected_edit_indexer_params = EditIndexerParams {
|
||||||
|
indexer_id: 1,
|
||||||
|
name: Some("Test".to_owned()),
|
||||||
|
enable_rss: Some(true),
|
||||||
|
enable_automatic_search: Some(true),
|
||||||
|
enable_interactive_search: Some(true),
|
||||||
|
url: Some("http://test.com".to_owned()),
|
||||||
|
api_key: Some("testKey".to_owned()),
|
||||||
|
seed_ratio: Some("1.2".to_owned()),
|
||||||
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
|
priority: Some(25),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::EditIndexer(expected_edit_indexer_params).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let edit_indexer_command = SonarrEditCommand::Indexer {
|
||||||
|
indexer_id: 1,
|
||||||
|
name: Some("Test".to_owned()),
|
||||||
|
enable_rss: true,
|
||||||
|
disable_rss: false,
|
||||||
|
enable_automatic_search: true,
|
||||||
|
disable_automatic_search: false,
|
||||||
|
enable_interactive_search: true,
|
||||||
|
disable_interactive_search: false,
|
||||||
|
url: Some("http://test.com".to_owned()),
|
||||||
|
api_key: Some("testKey".to_owned()),
|
||||||
|
seed_ratio: Some("1.2".to_owned()),
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
priority: Some(25),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrEditCommandHandler::with(&app_arc, edit_indexer_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_edit_series_command() {
|
||||||
|
let expected_edit_series_params = EditSeriesParams {
|
||||||
|
series_id: 1,
|
||||||
|
monitored: Some(true),
|
||||||
|
use_season_folders: Some(true),
|
||||||
|
series_type: Some(SeriesType::Anime),
|
||||||
|
quality_profile_id: Some(1),
|
||||||
|
language_profile_id: Some(1),
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::EditSeries(expected_edit_series_params).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let edit_series_command = SonarrEditCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
enable_monitoring: true,
|
||||||
|
disable_monitoring: false,
|
||||||
|
enable_season_folders: true,
|
||||||
|
disable_season_folders: false,
|
||||||
|
series_type: Some(SeriesType::Anime),
|
||||||
|
quality_profile_id: Some(1),
|
||||||
|
language_profile_id: Some(1),
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrEditCommandHandler::with(&app_arc, edit_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_edit_series_command_handles_disable_monitoring_flag_properly() {
|
||||||
|
let expected_edit_series_params = EditSeriesParams {
|
||||||
|
series_id: 1,
|
||||||
|
monitored: Some(false),
|
||||||
|
use_season_folders: Some(false),
|
||||||
|
series_type: Some(SeriesType::Anime),
|
||||||
|
quality_profile_id: Some(1),
|
||||||
|
language_profile_id: Some(1),
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::EditSeries(expected_edit_series_params).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let edit_series_command = SonarrEditCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
enable_monitoring: false,
|
||||||
|
disable_monitoring: true,
|
||||||
|
enable_season_folders: false,
|
||||||
|
disable_season_folders: true,
|
||||||
|
series_type: Some(SeriesType::Anime),
|
||||||
|
quality_profile_id: Some(1),
|
||||||
|
language_profile_id: Some(1),
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrEditCommandHandler::with(&app_arc, edit_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_edit_series_command_no_monitoring_boolean_flags_returns_none_value() {
|
||||||
|
let expected_edit_series_params = EditSeriesParams {
|
||||||
|
series_id: 1,
|
||||||
|
monitored: None,
|
||||||
|
use_season_folders: None,
|
||||||
|
series_type: Some(SeriesType::Anime),
|
||||||
|
quality_profile_id: Some(1),
|
||||||
|
language_profile_id: Some(1),
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tags: Some(vec![1, 2]),
|
||||||
|
tag_input_string: None,
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::EditSeries(expected_edit_series_params).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let edit_series_command = SonarrEditCommand::Series {
|
||||||
|
series_id: 1,
|
||||||
|
enable_monitoring: false,
|
||||||
|
disable_monitoring: false,
|
||||||
|
enable_season_folders: false,
|
||||||
|
disable_season_folders: false,
|
||||||
|
series_type: Some(SeriesType::Anime),
|
||||||
|
quality_profile_id: Some(1),
|
||||||
|
language_profile_id: Some(1),
|
||||||
|
root_folder_path: Some("/nfs/test".to_owned()),
|
||||||
|
tag: Some(vec![1, 2]),
|
||||||
|
clear_tags: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrEditCommandHandler::with(&app_arc, edit_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,122 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::Subcommand;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "get_command_handler_tests.rs"]
|
||||||
|
mod get_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrGetCommand {
|
||||||
|
#[command(about = "Get the shared settings for all indexers")]
|
||||||
|
AllIndexerSettings,
|
||||||
|
#[command(about = "Get detailed information for the episode with the given ID")]
|
||||||
|
EpisodeDetails {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the episode whose details you wish to fetch",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
episode_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Fetch the host config for your Sonarr instance")]
|
||||||
|
HostConfig,
|
||||||
|
#[command(about = "Fetch the security config for your Sonarr instance")]
|
||||||
|
SecurityConfig,
|
||||||
|
#[command(about = "Get detailed information for the series with the given ID")]
|
||||||
|
SeriesDetails {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series whose details you wish to fetch",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Get the system status")]
|
||||||
|
SystemStatus,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrGetCommand> for Command {
|
||||||
|
fn from(value: SonarrGetCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::Get(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrGetCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrGetCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrGetCommand> for SonarrGetCommandHandler<'a, 'b> {
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrGetCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrGetCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrGetCommand::AllIndexerSettings => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetAllIndexerSettings.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrGetCommand::EpisodeDetails { episode_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetEpisodeDetails(episode_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrGetCommand::HostConfig => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetHostConfig.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrGetCommand::SecurityConfig => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetSecurityConfig.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrGetCommand::SeriesDetails { series_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetSeriesDetails(series_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrGetCommand::SystemStatus => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetStatus.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,277 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::cli::{
|
||||||
|
sonarr::{get_command_handler::SonarrGetCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
};
|
||||||
|
use crate::Cli;
|
||||||
|
use clap::CommandFactory;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_get_command_from() {
|
||||||
|
let command = SonarrGetCommand::SystemStatus;
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(SonarrCommand::Get(command)));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use clap::error::ErrorKind;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_all_indexer_settings_has_no_arg_requirements() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "get", "all-indexer-settings"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_system_status_has_no_arg_requirements() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "get", "system-status"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_episode_details_requires_episode_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "get", "episode-details"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_episode_details_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"get",
|
||||||
|
"episode-details",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_get_host_config_has_no_arg_requirements() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "get", "host-config"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_get_security_config_has_no_arg_requirements() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "get", "security-config"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_series_details_requires_series_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "get", "series-details"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_series_details_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"get",
|
||||||
|
"series-details",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{
|
||||||
|
sonarr::get_command_handler::{SonarrGetCommand, SonarrGetCommandHandler},
|
||||||
|
CliCommandHandler,
|
||||||
|
},
|
||||||
|
models::{sonarr_models::SonarrSerdeable, Serdeable},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_get_all_indexer_settings_command() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetAllIndexerSettings.into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let get_all_indexer_settings_command = SonarrGetCommand::AllIndexerSettings;
|
||||||
|
|
||||||
|
let result = SonarrGetCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
get_all_indexer_settings_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_get_episode_details_command() {
|
||||||
|
let expected_episode_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetEpisodeDetails(expected_episode_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let get_episode_details_command = SonarrGetCommand::EpisodeDetails { episode_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrGetCommandHandler::with(&app_arc, get_episode_details_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_get_host_config_command() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::GetHostConfig.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let get_host_config_command = SonarrGetCommand::HostConfig;
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrGetCommandHandler::with(&app_arc, get_host_config_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_get_security_config_command() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::GetSecurityConfig.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let get_security_config_command = SonarrGetCommand::SecurityConfig;
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrGetCommandHandler::with(&app_arc, get_security_config_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_get_series_details_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetSeriesDetails(expected_series_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let get_series_details_command = SonarrGetCommand::SeriesDetails { series_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrGetCommandHandler::with(&app_arc, get_series_details_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_get_system_status_command() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::GetStatus.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let get_system_status_command = SonarrGetCommand::SystemStatus;
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrGetCommandHandler::with(&app_arc, get_system_status_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,292 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::Subcommand;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "list_command_handler_tests.rs"]
|
||||||
|
mod list_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrListCommand {
|
||||||
|
#[command(about = "List all items in the Sonarr blocklist")]
|
||||||
|
Blocklist,
|
||||||
|
#[command(about = "List all active downloads in Sonarr")]
|
||||||
|
Downloads,
|
||||||
|
#[command(about = "List disk space details for all provisioned root folders in Sonarr")]
|
||||||
|
DiskSpace,
|
||||||
|
#[command(about = "List the episodes for the series with the given ID")]
|
||||||
|
Episodes {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series whose episodes you wish to fetch",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "List the episode files for the series with the given ID")]
|
||||||
|
EpisodeFiles {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series whose episode files you wish to fetch",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Fetch all history events for the episode with the given ID")]
|
||||||
|
EpisodeHistory {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the episode whose history you wish to fetch",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
episode_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Fetch all Sonarr history events")]
|
||||||
|
History {
|
||||||
|
#[arg(long, help = "How many history events to fetch", default_value_t = 500)]
|
||||||
|
events: u64,
|
||||||
|
},
|
||||||
|
#[command(about = "List all Sonarr indexers")]
|
||||||
|
Indexers,
|
||||||
|
#[command(about = "List all Sonarr language profiles")]
|
||||||
|
LanguageProfiles,
|
||||||
|
#[command(about = "Fetch Sonarr logs")]
|
||||||
|
Logs {
|
||||||
|
#[arg(long, help = "How many log events to fetch", default_value_t = 500)]
|
||||||
|
events: u64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "Output the logs in the same format as they appear in the log files"
|
||||||
|
)]
|
||||||
|
output_in_log_format: bool,
|
||||||
|
},
|
||||||
|
#[command(about = "List all Sonarr quality profiles")]
|
||||||
|
QualityProfiles,
|
||||||
|
#[command(about = "List all queued events")]
|
||||||
|
QueuedEvents,
|
||||||
|
#[command(about = "List all root folders in Sonarr")]
|
||||||
|
RootFolders,
|
||||||
|
#[command(
|
||||||
|
about = "Fetch all history events for the given season corresponding to the series with the given ID."
|
||||||
|
)]
|
||||||
|
SeasonHistory {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series whose history you wish to fetch and list",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The season number to fetch history events for",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
season_number: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "List all series in your Sonarr library")]
|
||||||
|
Series,
|
||||||
|
#[command(about = "Fetch all history events for the series with the given ID")]
|
||||||
|
SeriesHistory {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series whose history you wish to fetch",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "List all Sonarr tags")]
|
||||||
|
Tags,
|
||||||
|
#[command(about = "List all Sonarr tasks")]
|
||||||
|
Tasks,
|
||||||
|
#[command(about = "List all Sonarr updates")]
|
||||||
|
Updates,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrListCommand> for Command {
|
||||||
|
fn from(value: SonarrListCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::List(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrListCommandHandler<'a, 'b> {
|
||||||
|
app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrListCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrListCommand> for SonarrListCommandHandler<'a, 'b> {
|
||||||
|
fn with(
|
||||||
|
app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrListCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrListCommandHandler {
|
||||||
|
app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrListCommand::Blocklist => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetBlocklist.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Downloads => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetDownloads.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::DiskSpace => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetDiskSpace.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Episodes { series_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetEpisodes(series_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::EpisodeFiles { series_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetEpisodeFiles(series_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::EpisodeHistory { episode_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetEpisodeHistory(episode_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::History { events: items } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetHistory(items).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Indexers => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetIndexers.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::LanguageProfiles => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetLanguageProfiles.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Logs {
|
||||||
|
events,
|
||||||
|
output_in_log_format,
|
||||||
|
} => {
|
||||||
|
let logs = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetLogs(events).into())
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if output_in_log_format {
|
||||||
|
let log_lines = &self.app.lock().await.data.sonarr_data.logs.items;
|
||||||
|
|
||||||
|
serde_json::to_string_pretty(log_lines)?
|
||||||
|
} else {
|
||||||
|
serde_json::to_string_pretty(&logs)?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
SonarrListCommand::QualityProfiles => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetQualityProfiles.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::QueuedEvents => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetQueuedEvents.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::RootFolders => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetRootFolders.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::SeasonHistory {
|
||||||
|
series_id,
|
||||||
|
season_number,
|
||||||
|
} => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetSeasonHistory((series_id, season_number)).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Series => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::ListSeries.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::SeriesHistory { series_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetSeriesHistory(series_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Tags => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetTags.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Tasks => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetTasks.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrListCommand::Updates => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetUpdates.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,513 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::cli::{
|
||||||
|
sonarr::{list_command_handler::SonarrListCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
};
|
||||||
|
use crate::Cli;
|
||||||
|
use clap::CommandFactory;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_list_command_from() {
|
||||||
|
let command = SonarrListCommand::Series;
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(SonarrCommand::List(command)));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use super::*;
|
||||||
|
use clap::{error::ErrorKind, Parser};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
use rstest::rstest;
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_list_commands_have_no_arg_requirements(
|
||||||
|
#[values(
|
||||||
|
"blocklist",
|
||||||
|
"series",
|
||||||
|
"downloads",
|
||||||
|
"disk-space",
|
||||||
|
"quality-profiles",
|
||||||
|
"indexers",
|
||||||
|
"queued-events",
|
||||||
|
"root-folders",
|
||||||
|
"tags",
|
||||||
|
"tasks",
|
||||||
|
"updates",
|
||||||
|
"language-profiles"
|
||||||
|
)]
|
||||||
|
subcommand: &str,
|
||||||
|
) {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "list", subcommand]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_episodes_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episodes"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_episode_files_requires_series_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episode-files"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_episode_history_requires_series_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episode-history"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_episode_history_success() {
|
||||||
|
let expected_args = SonarrListCommand::EpisodeHistory { episode_id: 1 };
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"list",
|
||||||
|
"episode-history",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::List(episode_history_command))) =
|
||||||
|
result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(episode_history_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_history_events_flag_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "history", "--events"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_history_default_values() {
|
||||||
|
let expected_args = SonarrListCommand::History { events: 500 };
|
||||||
|
let result = Cli::try_parse_from(["managarr", "sonarr", "list", "history"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::List(history_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(history_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_logs_events_flag_requires_arguments() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "logs", "--events"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_logs_default_values() {
|
||||||
|
let expected_args = SonarrListCommand::Logs {
|
||||||
|
events: 500,
|
||||||
|
output_in_log_format: false,
|
||||||
|
};
|
||||||
|
let result = Cli::try_parse_from(["managarr", "sonarr", "list", "logs"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::List(logs_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(logs_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_episodes_success() {
|
||||||
|
let expected_args = SonarrListCommand::Episodes { series_id: 1 };
|
||||||
|
let result =
|
||||||
|
Cli::try_parse_from(["managarr", "sonarr", "list", "episodes", "--series-id", "1"]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::List(episodes_command))) = result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(episodes_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_episode_files_success() {
|
||||||
|
let expected_args = SonarrListCommand::EpisodeFiles { series_id: 1 };
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"list",
|
||||||
|
"episode-files",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::List(episode_files_command))) =
|
||||||
|
result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(episode_files_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_season_history_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"list",
|
||||||
|
"season-history",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_season_history_requires_season_number() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"list",
|
||||||
|
"season-history",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_season_history_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"list",
|
||||||
|
"season-history",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_series_history_requires_series_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "series-history"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_list_series_history_success() {
|
||||||
|
let expected_args = SonarrListCommand::SeriesHistory { series_id: 1 };
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"list",
|
||||||
|
"series-history",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::List(series_command))) = result.unwrap().command {
|
||||||
|
assert_eq!(series_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use rstest::rstest;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::cli::sonarr::list_command_handler::{SonarrListCommand, SonarrListCommandHandler};
|
||||||
|
use crate::cli::CliCommandHandler;
|
||||||
|
use crate::models::sonarr_models::SonarrSerdeable;
|
||||||
|
use crate::models::Serdeable;
|
||||||
|
use crate::network::sonarr_network::SonarrEvent;
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
network::{MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
#[case(SonarrListCommand::Blocklist, SonarrEvent::GetBlocklist)]
|
||||||
|
#[case(SonarrListCommand::Downloads, SonarrEvent::GetDownloads)]
|
||||||
|
#[case(SonarrListCommand::DiskSpace, SonarrEvent::GetDiskSpace)]
|
||||||
|
#[case(SonarrListCommand::Indexers, SonarrEvent::GetIndexers)]
|
||||||
|
#[case(SonarrListCommand::QualityProfiles, SonarrEvent::GetQualityProfiles)]
|
||||||
|
#[case(SonarrListCommand::QueuedEvents, SonarrEvent::GetQueuedEvents)]
|
||||||
|
#[case(SonarrListCommand::RootFolders, SonarrEvent::GetRootFolders)]
|
||||||
|
#[case(SonarrListCommand::Series, SonarrEvent::ListSeries)]
|
||||||
|
#[case(SonarrListCommand::Tags, SonarrEvent::GetTags)]
|
||||||
|
#[case(SonarrListCommand::Tasks, SonarrEvent::GetTasks)]
|
||||||
|
#[case(SonarrListCommand::Updates, SonarrEvent::GetUpdates)]
|
||||||
|
#[case(SonarrListCommand::LanguageProfiles, SonarrEvent::GetLanguageProfiles)]
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_list_command(
|
||||||
|
#[case] list_command: SonarrListCommand,
|
||||||
|
#[case] expected_sonarr_event: SonarrEvent,
|
||||||
|
) {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(expected_sonarr_event.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
|
||||||
|
let result = SonarrListCommandHandler::with(&app_arc, list_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_list_episodes_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetEpisodes(expected_series_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_episodes_command = SonarrListCommand::Episodes { series_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrListCommandHandler::with(&app_arc, list_episodes_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_list_episode_files_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetEpisodeFiles(expected_series_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_episode_files_command = SonarrListCommand::EpisodeFiles { series_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrListCommandHandler::with(&app_arc, list_episode_files_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_list_history_command() {
|
||||||
|
let expected_events = 1000;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetHistory(expected_events).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_history_command = SonarrListCommand::History { events: 1000 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrListCommandHandler::with(&app_arc, list_history_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_list_logs_command() {
|
||||||
|
let expected_events = 1000;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetLogs(expected_events).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_logs_command = SonarrListCommand::Logs {
|
||||||
|
events: 1000,
|
||||||
|
output_in_log_format: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrListCommandHandler::with(&app_arc, list_logs_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_list_series_history_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetSeriesHistory(expected_series_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_series_history_command = SonarrListCommand::SeriesHistory { series_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrListCommandHandler::with(&app_arc, list_series_history_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_list_episode_history_command() {
|
||||||
|
let expected_episode_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetEpisodeHistory(expected_episode_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_episode_history_command = SonarrListCommand::EpisodeHistory { episode_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrListCommandHandler::with(&app_arc, list_episode_history_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_list_season_history_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let expected_season_number = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetSeasonHistory((expected_series_id, expected_season_number)).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_season_history_command = SonarrListCommand::SeasonHistory {
|
||||||
|
series_id: 1,
|
||||||
|
season_number: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrListCommandHandler::with(&app_arc, list_season_history_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,97 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::Subcommand;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "manual_search_command_handler_tests.rs"]
|
||||||
|
mod manual_search_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrManualSearchCommand {
|
||||||
|
#[command(about = "Trigger a manual search of releases for the episode with the given ID")]
|
||||||
|
Episode {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the episode whose releases you wish to fetch and list",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
episode_id: i64,
|
||||||
|
},
|
||||||
|
#[command(
|
||||||
|
about = "Trigger a manual search of releases for the given season corresponding to the series with the given ID.\nNote that when downloading a season release, ensure that the release includes 'fullSeason: true', otherwise you'll run into issues"
|
||||||
|
)]
|
||||||
|
Season {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series whose releases you wish to fetch and list",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
#[arg(long, help = "The season number to search for", required = true)]
|
||||||
|
season_number: i64,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrManualSearchCommand> for Command {
|
||||||
|
fn from(value: SonarrManualSearchCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::ManualSearch(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrManualSearchCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrManualSearchCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrManualSearchCommand>
|
||||||
|
for SonarrManualSearchCommandHandler<'a, 'b>
|
||||||
|
{
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrManualSearchCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrManualSearchCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrManualSearchCommand::Episode { episode_id } => {
|
||||||
|
println!("Searching for episode releases. This may take a minute...");
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetEpisodeReleases(episode_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrManualSearchCommand::Season {
|
||||||
|
series_id,
|
||||||
|
season_number,
|
||||||
|
} => {
|
||||||
|
println!("Searching for season releases. This may take a minute...");
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetSeasonReleases((series_id, season_number)).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,188 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::cli::{
|
||||||
|
sonarr::{manual_search_command_handler::SonarrManualSearchCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
};
|
||||||
|
use crate::Cli;
|
||||||
|
use clap::CommandFactory;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_manual_search_command_from() {
|
||||||
|
let command = SonarrManualSearchCommand::Episode { episode_id: 1 };
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
result,
|
||||||
|
Command::Sonarr(SonarrCommand::ManualSearch(command))
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use super::*;
|
||||||
|
use clap::error::ErrorKind;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_manual_season_search_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"manual-search",
|
||||||
|
"season",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_manual_season_search_requires_season_number() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"manual-search",
|
||||||
|
"season",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_manual_season_search_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"manual-search",
|
||||||
|
"season",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_manual_episode_search_requires_episode_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "manual-search", "episode"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_manual_episode_search_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"manual-search",
|
||||||
|
"episode",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{
|
||||||
|
sonarr::manual_search_command_handler::{
|
||||||
|
SonarrManualSearchCommand, SonarrManualSearchCommandHandler,
|
||||||
|
},
|
||||||
|
CliCommandHandler,
|
||||||
|
},
|
||||||
|
models::{sonarr_models::SonarrSerdeable, Serdeable},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_manual_episode_search_command() {
|
||||||
|
let expected_episode_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetEpisodeReleases(expected_episode_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let manual_episode_search_command = SonarrManualSearchCommand::Episode { episode_id: 1 };
|
||||||
|
|
||||||
|
let result = SonarrManualSearchCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
manual_episode_search_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_manual_season_search_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let expected_season_number = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetSeasonReleases((expected_series_id, expected_season_number)).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let manual_season_search_command = SonarrManualSearchCommand::Season {
|
||||||
|
series_id: 1,
|
||||||
|
season_number: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrManualSearchCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
manual_season_search_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,297 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use add_command_handler::{SonarrAddCommand, SonarrAddCommandHandler};
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::Subcommand;
|
||||||
|
use delete_command_handler::{SonarrDeleteCommand, SonarrDeleteCommandHandler};
|
||||||
|
use download_command_handler::{SonarrDownloadCommand, SonarrDownloadCommandHandler};
|
||||||
|
use edit_command_handler::{SonarrEditCommand, SonarrEditCommandHandler};
|
||||||
|
use get_command_handler::{SonarrGetCommand, SonarrGetCommandHandler};
|
||||||
|
use list_command_handler::{SonarrListCommand, SonarrListCommandHandler};
|
||||||
|
use manual_search_command_handler::{SonarrManualSearchCommand, SonarrManualSearchCommandHandler};
|
||||||
|
use refresh_command_handler::{SonarrRefreshCommand, SonarrRefreshCommandHandler};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
use trigger_automatic_search_command_handler::{
|
||||||
|
SonarrTriggerAutomaticSearchCommand, SonarrTriggerAutomaticSearchCommandHandler,
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
models::sonarr_models::SonarrTaskName,
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::{CliCommandHandler, Command};
|
||||||
|
|
||||||
|
mod add_command_handler;
|
||||||
|
mod delete_command_handler;
|
||||||
|
mod download_command_handler;
|
||||||
|
mod edit_command_handler;
|
||||||
|
mod get_command_handler;
|
||||||
|
mod list_command_handler;
|
||||||
|
mod manual_search_command_handler;
|
||||||
|
mod refresh_command_handler;
|
||||||
|
mod trigger_automatic_search_command_handler;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "sonarr_command_tests.rs"]
|
||||||
|
mod sonarr_command_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrCommand {
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to add or create new resources within your Sonarr instance"
|
||||||
|
)]
|
||||||
|
Add(SonarrAddCommand),
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to delete resources from your Sonarr instance"
|
||||||
|
)]
|
||||||
|
Delete(SonarrDeleteCommand),
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to edit resources in your Sonarr instance"
|
||||||
|
)]
|
||||||
|
Edit(SonarrEditCommand),
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to fetch details of the resources in your Sonarr instance"
|
||||||
|
)]
|
||||||
|
Get(SonarrGetCommand),
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to download releases in your Sonarr instance"
|
||||||
|
)]
|
||||||
|
Download(SonarrDownloadCommand),
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to list attributes from your Sonarr instance"
|
||||||
|
)]
|
||||||
|
List(SonarrListCommand),
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to refresh the data in your Sonarr instance"
|
||||||
|
)]
|
||||||
|
Refresh(SonarrRefreshCommand),
|
||||||
|
#[command(subcommand, about = "Commands to manually search for releases")]
|
||||||
|
ManualSearch(SonarrManualSearchCommand),
|
||||||
|
#[command(
|
||||||
|
subcommand,
|
||||||
|
about = "Commands to trigger automatic searches for releases of different resources in your Sonarr instance"
|
||||||
|
)]
|
||||||
|
TriggerAutomaticSearch(SonarrTriggerAutomaticSearchCommand),
|
||||||
|
#[command(about = "Clear the blocklist")]
|
||||||
|
ClearBlocklist,
|
||||||
|
#[command(about = "Mark the Sonarr history item with the given ID as 'failed'")]
|
||||||
|
MarkHistoryItemAsFailed {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the history item you wish to mark as 'failed'",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
history_item_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Search for a new series to add to Sonarr")]
|
||||||
|
SearchNewSeries {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The title of the series you want to search for",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
query: String,
|
||||||
|
},
|
||||||
|
#[command(about = "Start the specified Sonarr task")]
|
||||||
|
StartTask {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The name of the task to trigger",
|
||||||
|
value_enum,
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
task_name: SonarrTaskName,
|
||||||
|
},
|
||||||
|
#[command(
|
||||||
|
about = "Test the indexer with the given ID. Note that a successful test returns an empty JSON body; i.e. '{}'"
|
||||||
|
)]
|
||||||
|
TestIndexer {
|
||||||
|
#[arg(long, help = "The ID of the indexer to test", required = true)]
|
||||||
|
indexer_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Test all Sonarr indexers")]
|
||||||
|
TestAllIndexers,
|
||||||
|
#[command(about = "Toggle monitoring for the specified episode")]
|
||||||
|
ToggleEpisodeMonitoring {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the episode to toggle monitoring on",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
episode_id: i64,
|
||||||
|
},
|
||||||
|
#[command(
|
||||||
|
about = "Toggle monitoring for the specified season that corresponds to the specified series ID"
|
||||||
|
)]
|
||||||
|
ToggleSeasonMonitoring {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series that the season belongs to",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The season number to toggle monitoring for",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
season_number: i64,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrCommand> for Command {
|
||||||
|
fn from(sonarr_command: SonarrCommand) -> Command {
|
||||||
|
Command::Sonarr(sonarr_command)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrCliHandler<'a, 'b> {
|
||||||
|
app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrCommand> for SonarrCliHandler<'a, 'b> {
|
||||||
|
fn with(
|
||||||
|
app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrCliHandler {
|
||||||
|
app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrCommand::Add(add_command) => {
|
||||||
|
SonarrAddCommandHandler::with(self.app, add_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::Delete(delete_command) => {
|
||||||
|
SonarrDeleteCommandHandler::with(self.app, delete_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::Edit(edit_command) => {
|
||||||
|
SonarrEditCommandHandler::with(self.app, edit_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::Download(download_command) => {
|
||||||
|
SonarrDownloadCommandHandler::with(self.app, download_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::Get(get_command) => {
|
||||||
|
SonarrGetCommandHandler::with(self.app, get_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::List(list_command) => {
|
||||||
|
SonarrListCommandHandler::with(self.app, list_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::Refresh(refresh_command) => {
|
||||||
|
SonarrRefreshCommandHandler::with(self.app, refresh_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::ManualSearch(manual_search_command) => {
|
||||||
|
SonarrManualSearchCommandHandler::with(self.app, manual_search_command, self.network)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::TriggerAutomaticSearch(trigger_automatic_search_command) => {
|
||||||
|
SonarrTriggerAutomaticSearchCommandHandler::with(
|
||||||
|
self.app,
|
||||||
|
trigger_automatic_search_command,
|
||||||
|
self.network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await?
|
||||||
|
}
|
||||||
|
SonarrCommand::ClearBlocklist => {
|
||||||
|
self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::GetBlocklist.into())
|
||||||
|
.await?;
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::ClearBlocklist.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrCommand::MarkHistoryItemAsFailed { history_item_id } => {
|
||||||
|
let _ = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::MarkHistoryItemAsFailed(history_item_id).into())
|
||||||
|
.await?;
|
||||||
|
"Sonarr history item marked as 'failed'".to_owned()
|
||||||
|
}
|
||||||
|
SonarrCommand::SearchNewSeries { query } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::SearchNewSeries(query).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrCommand::StartTask { task_name } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::StartTask(task_name).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrCommand::TestIndexer { indexer_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::TestIndexer(indexer_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrCommand::TestAllIndexers => {
|
||||||
|
println!("Testing all Sonarr indexers. This may take a minute...");
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::TestAllIndexers.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrCommand::ToggleEpisodeMonitoring { episode_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::ToggleEpisodeMonitoring(episode_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrCommand::ToggleSeasonMonitoring {
|
||||||
|
series_id,
|
||||||
|
season_number,
|
||||||
|
} => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(
|
||||||
|
SonarrEvent::ToggleSeasonMonitoring((series_id, season_number)).into(),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,89 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use clap::Subcommand;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "refresh_command_handler_tests.rs"]
|
||||||
|
mod refresh_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrRefreshCommand {
|
||||||
|
#[command(about = "Refresh all series data for all series in your Sonarr library")]
|
||||||
|
AllSeries,
|
||||||
|
#[command(about = "Refresh series data and scan disk for the series with the given ID")]
|
||||||
|
Series {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the series to refresh information on and to scan the disk for",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Refresh all downloads in Sonarr")]
|
||||||
|
Downloads,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrRefreshCommand> for Command {
|
||||||
|
fn from(value: SonarrRefreshCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::Refresh(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrRefreshCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrRefreshCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrRefreshCommand>
|
||||||
|
for SonarrRefreshCommandHandler<'a, 'b>
|
||||||
|
{
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrRefreshCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrRefreshCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> anyhow::Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrRefreshCommand::AllSeries => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::UpdateAllSeries.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrRefreshCommand::Series { series_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::UpdateAndScanSeries(series_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrRefreshCommand::Downloads => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::UpdateDownloads.into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,141 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
use crate::cli::{
|
||||||
|
sonarr::{refresh_command_handler::SonarrRefreshCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
};
|
||||||
|
use crate::Cli;
|
||||||
|
use clap::CommandFactory;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_refresh_command_from() {
|
||||||
|
let command = SonarrRefreshCommand::AllSeries;
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(SonarrCommand::Refresh(command)));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use super::*;
|
||||||
|
use clap::{error::ErrorKind, Parser};
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
use rstest::rstest;
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_refresh_commands_have_no_arg_requirements(
|
||||||
|
#[values("all-series", "downloads")] subcommand: &str,
|
||||||
|
) {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "refresh", subcommand]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_refresh_series_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "refresh", "series"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_refresh_series_success() {
|
||||||
|
let expected_args = SonarrRefreshCommand::Series { series_id: 1 };
|
||||||
|
let result = Cli::try_parse_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"refresh",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
|
||||||
|
if let Some(Command::Sonarr(SonarrCommand::Refresh(refresh_command))) =
|
||||||
|
result.unwrap().command
|
||||||
|
{
|
||||||
|
assert_eq!(refresh_command, expected_args);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use rstest::rstest;
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{app::App, cli::sonarr::refresh_command_handler::SonarrRefreshCommandHandler};
|
||||||
|
use crate::{
|
||||||
|
cli::{sonarr::refresh_command_handler::SonarrRefreshCommand, CliCommandHandler},
|
||||||
|
network::sonarr_network::SonarrEvent,
|
||||||
|
};
|
||||||
|
use crate::{
|
||||||
|
models::{sonarr_models::SonarrSerdeable, Serdeable},
|
||||||
|
network::{MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
#[case(SonarrRefreshCommand::AllSeries, SonarrEvent::UpdateAllSeries)]
|
||||||
|
#[case(SonarrRefreshCommand::Downloads, SonarrEvent::UpdateDownloads)]
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_refresh_command(
|
||||||
|
#[case] refresh_command: SonarrRefreshCommand,
|
||||||
|
#[case] expected_sonarr_event: SonarrEvent,
|
||||||
|
) {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(expected_sonarr_event.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
|
||||||
|
let result = SonarrRefreshCommandHandler::with(&app_arc, refresh_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_refresh_series_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::UpdateAndScanSeries(expected_series_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let refresh_series_command = SonarrRefreshCommand::Series { series_id: 1 };
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrRefreshCommandHandler::with(&app_arc, refresh_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,757 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::cli::{
|
||||||
|
sonarr::{list_command_handler::SonarrListCommand, SonarrCommand},
|
||||||
|
Command,
|
||||||
|
};
|
||||||
|
use crate::Cli;
|
||||||
|
use clap::CommandFactory;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_command_from() {
|
||||||
|
let command = SonarrCommand::List(SonarrListCommand::Series);
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(result, Command::Sonarr(command));
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use super::*;
|
||||||
|
use clap::error::ErrorKind;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
use rstest::rstest;
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_commands_that_have_no_arg_requirements(
|
||||||
|
#[values("clear-blocklist", "test-all-indexers")] subcommand: &str,
|
||||||
|
) {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", subcommand]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_mark_history_item_as_failed_requires_history_item_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "mark-history-item-as-failed"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_mark_history_item_as_failed_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"mark-history-item-as-failed",
|
||||||
|
"--history-item-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_search_new_series_requires_query() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "search-new-series"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_search_new_series_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"search-new-series",
|
||||||
|
"--query",
|
||||||
|
"halo",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_start_task_requires_task_name() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "start-task"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_start_task_task_name_validation() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"start-task",
|
||||||
|
"--task-name",
|
||||||
|
"test",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_start_task_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"start-task",
|
||||||
|
"--task-name",
|
||||||
|
"application-update-check",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_test_indexer_requires_indexer_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "test-indexer"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_test_indexer_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"test-indexer",
|
||||||
|
"--indexer-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_toggle_episode_monitoring_requires_episode_id() {
|
||||||
|
let result =
|
||||||
|
Cli::command().try_get_matches_from(["managarr", "sonarr", "toggle-episode-monitoring"]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_toggle_episode_monitoring_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"toggle-episode-monitoring",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_toggle_season_monitoring_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"toggle-season-monitoring",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_toggle_season_monitoring_requires_season_number() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"toggle-season-monitoring",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_toggle_season_monitoring_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"toggle-season-monitoring",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{
|
||||||
|
sonarr::{
|
||||||
|
add_command_handler::SonarrAddCommand, delete_command_handler::SonarrDeleteCommand,
|
||||||
|
download_command_handler::SonarrDownloadCommand, edit_command_handler::SonarrEditCommand,
|
||||||
|
get_command_handler::SonarrGetCommand, list_command_handler::SonarrListCommand,
|
||||||
|
manual_search_command_handler::SonarrManualSearchCommand,
|
||||||
|
refresh_command_handler::SonarrRefreshCommand,
|
||||||
|
trigger_automatic_search_command_handler::SonarrTriggerAutomaticSearchCommand,
|
||||||
|
SonarrCliHandler, SonarrCommand,
|
||||||
|
},
|
||||||
|
CliCommandHandler,
|
||||||
|
},
|
||||||
|
models::{
|
||||||
|
sonarr_models::{
|
||||||
|
BlocklistItem, BlocklistResponse, IndexerSettings, Series, SonarrReleaseDownloadBody,
|
||||||
|
SonarrSerdeable, SonarrTaskName,
|
||||||
|
},
|
||||||
|
Serdeable,
|
||||||
|
},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_clear_blocklist_command() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::GetBlocklist.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::BlocklistResponse(
|
||||||
|
BlocklistResponse {
|
||||||
|
records: vec![BlocklistItem::default()],
|
||||||
|
},
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::ClearBlocklist.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let claer_blocklist_command = SonarrCommand::ClearBlocklist;
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, claer_blocklist_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_mark_history_item_as_failed_command() {
|
||||||
|
let expected_history_item_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::MarkHistoryItemAsFailed(expected_history_item_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let mark_history_item_as_failed_command =
|
||||||
|
SonarrCommand::MarkHistoryItemAsFailed { history_item_id: 1 };
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
mark_history_item_as_failed_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_add_commands_to_the_add_command_handler() {
|
||||||
|
let expected_tag_name = "test".to_owned();
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::AddTag(expected_tag_name.clone()).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let add_tag_command = SonarrCommand::Add(SonarrAddCommand::Tag {
|
||||||
|
name: expected_tag_name,
|
||||||
|
});
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, add_tag_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_delete_commands_to_the_delete_command_handler() {
|
||||||
|
let expected_blocklist_item_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DeleteBlocklistItem(expected_blocklist_item_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let delete_blocklist_item_command =
|
||||||
|
SonarrCommand::Delete(SonarrDeleteCommand::BlocklistItem {
|
||||||
|
blocklist_item_id: 1,
|
||||||
|
});
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrCliHandler::with(&app_arc, delete_blocklist_item_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_download_commands_to_the_download_command_handler() {
|
||||||
|
let expected_params = SonarrReleaseDownloadBody {
|
||||||
|
guid: "1234".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
series_id: Some(1),
|
||||||
|
..SonarrReleaseDownloadBody::default()
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::DownloadRelease(expected_params).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let download_series_release_command =
|
||||||
|
SonarrCommand::Download(SonarrDownloadCommand::Series {
|
||||||
|
guid: "1234".to_owned(),
|
||||||
|
indexer_id: 1,
|
||||||
|
series_id: 1,
|
||||||
|
});
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrCliHandler::with(&app_arc, download_series_release_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_edit_commands_to_the_edit_command_handler() {
|
||||||
|
let expected_edit_all_indexer_settings = IndexerSettings {
|
||||||
|
id: 1,
|
||||||
|
maximum_size: 1,
|
||||||
|
minimum_age: 1,
|
||||||
|
retention: 1,
|
||||||
|
rss_sync_interval: 1,
|
||||||
|
};
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetAllIndexerSettings.into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::IndexerSettings(
|
||||||
|
IndexerSettings {
|
||||||
|
id: 1,
|
||||||
|
maximum_size: 2,
|
||||||
|
minimum_age: 2,
|
||||||
|
retention: 2,
|
||||||
|
rss_sync_interval: 2,
|
||||||
|
},
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let edit_all_indexer_settings_command =
|
||||||
|
SonarrCommand::Edit(SonarrEditCommand::AllIndexerSettings {
|
||||||
|
maximum_size: Some(1),
|
||||||
|
minimum_age: Some(1),
|
||||||
|
retention: Some(1),
|
||||||
|
rss_sync_interval: Some(1),
|
||||||
|
});
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
edit_all_indexer_settings_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_manual_search_commands_to_the_manual_search_command_handler(
|
||||||
|
) {
|
||||||
|
let expected_episode_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::GetEpisodeReleases(expected_episode_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let manual_episode_search_command =
|
||||||
|
SonarrCommand::ManualSearch(SonarrManualSearchCommand::Episode { episode_id: 1 });
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrCliHandler::with(&app_arc, manual_episode_search_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_trigger_automatic_search_commands_to_the_trigger_automatic_search_command_handler(
|
||||||
|
) {
|
||||||
|
let expected_episode_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::TriggerAutomaticEpisodeSearch(expected_episode_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let manual_episode_search_command =
|
||||||
|
SonarrCommand::TriggerAutomaticSearch(SonarrTriggerAutomaticSearchCommand::Episode {
|
||||||
|
episode_id: 1,
|
||||||
|
});
|
||||||
|
|
||||||
|
let result =
|
||||||
|
SonarrCliHandler::with(&app_arc, manual_episode_search_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_get_commands_to_the_get_command_handler() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::GetStatus.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let get_system_status_command = SonarrCommand::Get(SonarrGetCommand::SystemStatus);
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, get_system_status_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_list_commands_to_the_list_command_handler() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::ListSeries.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::SeriesVec(vec![
|
||||||
|
Series::default(),
|
||||||
|
])))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let list_series_command = SonarrCommand::List(SonarrListCommand::Series);
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, list_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_sonarr_cli_handler_delegates_refresh_commands_to_the_refresh_command_handler() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::UpdateAndScanSeries(expected_series_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let refresh_series_command =
|
||||||
|
SonarrCommand::Refresh(SonarrRefreshCommand::Series { series_id: 1 });
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, refresh_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_search_new_series_command() {
|
||||||
|
let expected_search_query = "halo".to_owned();
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::SearchNewSeries(expected_search_query).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let search_new_series_command = SonarrCommand::SearchNewSeries {
|
||||||
|
query: "halo".to_owned(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, search_new_series_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_start_task_command() {
|
||||||
|
let expected_task_name = SonarrTaskName::ApplicationUpdateCheck;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::StartTask(expected_task_name).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let start_task_command = SonarrCommand::StartTask {
|
||||||
|
task_name: SonarrTaskName::ApplicationUpdateCheck,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, start_task_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_test_indexer_command() {
|
||||||
|
let expected_indexer_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::TestIndexer(expected_indexer_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let test_indexer_command = SonarrCommand::TestIndexer { indexer_id: 1 };
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, test_indexer_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_test_all_indexers_command() {
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(SonarrEvent::TestAllIndexers.into()))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let test_all_indexers_command = SonarrCommand::TestAllIndexers;
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(&app_arc, test_all_indexers_command, &mut mock_network)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_list_toggle_episode_monitoring_command() {
|
||||||
|
let expected_episode_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::ToggleEpisodeMonitoring(expected_episode_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let toggle_episode_monitoring_command =
|
||||||
|
SonarrCommand::ToggleEpisodeMonitoring { episode_id: 1 };
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
toggle_episode_monitoring_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_list_toggle_season_monitoring_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let expected_season_number = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::ToggleSeasonMonitoring((expected_series_id, expected_season_number)).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let toggle_season_monitoring_command = SonarrCommand::ToggleSeasonMonitoring {
|
||||||
|
series_id: 1,
|
||||||
|
season_number: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrCliHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
toggle_season_monitoring_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,113 @@
|
|||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use anyhow::Result;
|
||||||
|
use clap::Subcommand;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{CliCommandHandler, Command},
|
||||||
|
network::{sonarr_network::SonarrEvent, NetworkTrait},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::SonarrCommand;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "trigger_automatic_search_command_handler_tests.rs"]
|
||||||
|
mod trigger_automatic_search_command_handler_tests;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq, Subcommand)]
|
||||||
|
pub enum SonarrTriggerAutomaticSearchCommand {
|
||||||
|
#[command(about = "Trigger an automatic search for the series with the specified ID")]
|
||||||
|
Series {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the series you want to trigger an automatic search for",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
},
|
||||||
|
#[command(
|
||||||
|
about = "Trigger an automatic search for the given season corresponding to the series with the given ID"
|
||||||
|
)]
|
||||||
|
Season {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The Sonarr ID of the series whose season you wish to trigger an automatic search for",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
series_id: i64,
|
||||||
|
#[arg(long, help = "The season number to search for", required = true)]
|
||||||
|
season_number: i64,
|
||||||
|
},
|
||||||
|
#[command(about = "Trigger an automatic search for the episode with the specified ID")]
|
||||||
|
Episode {
|
||||||
|
#[arg(
|
||||||
|
long,
|
||||||
|
help = "The ID of the episode you want to trigger an automatic search for",
|
||||||
|
required = true
|
||||||
|
)]
|
||||||
|
episode_id: i64,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<SonarrTriggerAutomaticSearchCommand> for Command {
|
||||||
|
fn from(value: SonarrTriggerAutomaticSearchCommand) -> Self {
|
||||||
|
Command::Sonarr(SonarrCommand::TriggerAutomaticSearch(value))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) struct SonarrTriggerAutomaticSearchCommandHandler<'a, 'b> {
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrTriggerAutomaticSearchCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrTriggerAutomaticSearchCommand>
|
||||||
|
for SonarrTriggerAutomaticSearchCommandHandler<'a, 'b>
|
||||||
|
{
|
||||||
|
fn with(
|
||||||
|
_app: &'a Arc<Mutex<App<'b>>>,
|
||||||
|
command: SonarrTriggerAutomaticSearchCommand,
|
||||||
|
network: &'a mut dyn NetworkTrait,
|
||||||
|
) -> Self {
|
||||||
|
SonarrTriggerAutomaticSearchCommandHandler {
|
||||||
|
_app,
|
||||||
|
command,
|
||||||
|
network,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle(self) -> Result<String> {
|
||||||
|
let result = match self.command {
|
||||||
|
SonarrTriggerAutomaticSearchCommand::Series { series_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::TriggerAutomaticSeriesSearch(series_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrTriggerAutomaticSearchCommand::Season {
|
||||||
|
series_id,
|
||||||
|
season_number,
|
||||||
|
} => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(
|
||||||
|
SonarrEvent::TriggerAutomaticSeasonSearch((series_id, season_number)).into(),
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
SonarrTriggerAutomaticSearchCommand::Episode { episode_id } => {
|
||||||
|
let resp = self
|
||||||
|
.network
|
||||||
|
.handle_network_event(SonarrEvent::TriggerAutomaticEpisodeSearch(episode_id).into())
|
||||||
|
.await?;
|
||||||
|
serde_json::to_string_pretty(&resp)?
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,256 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::cli::{
|
||||||
|
sonarr::{
|
||||||
|
trigger_automatic_search_command_handler::SonarrTriggerAutomaticSearchCommand, SonarrCommand,
|
||||||
|
},
|
||||||
|
Command,
|
||||||
|
};
|
||||||
|
use crate::Cli;
|
||||||
|
use clap::CommandFactory;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_sonarr_trigger_automatic_search_command_from() {
|
||||||
|
let command = SonarrTriggerAutomaticSearchCommand::Episode { episode_id: 1 };
|
||||||
|
|
||||||
|
let result = Command::from(command.clone());
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
result,
|
||||||
|
Command::Sonarr(SonarrCommand::TriggerAutomaticSearch(command))
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
mod cli {
|
||||||
|
use super::*;
|
||||||
|
use clap::error::ErrorKind;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_trigger_automatic_series_search_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"trigger-automatic-search",
|
||||||
|
"series",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_trigger_automatic_series_search_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"trigger-automatic-search",
|
||||||
|
"series",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_trigger_automatic_season_search_requires_series_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"trigger-automatic-search",
|
||||||
|
"season",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_trigger_automatic_season_search_requires_season_number() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"trigger-automatic-search",
|
||||||
|
"season",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_trigger_automatic_season_search_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"trigger-automatic-search",
|
||||||
|
"season",
|
||||||
|
"--series-id",
|
||||||
|
"1",
|
||||||
|
"--season-number",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_trigger_automatic_episode_search_requires_episode_id() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"trigger-automatic-search",
|
||||||
|
"episode",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_err());
|
||||||
|
assert_eq!(
|
||||||
|
result.unwrap_err().kind(),
|
||||||
|
ErrorKind::MissingRequiredArgument
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_trigger_automatic_episode_search_requirements_satisfied() {
|
||||||
|
let result = Cli::command().try_get_matches_from([
|
||||||
|
"managarr",
|
||||||
|
"sonarr",
|
||||||
|
"trigger-automatic-search",
|
||||||
|
"episode",
|
||||||
|
"--episode-id",
|
||||||
|
"1",
|
||||||
|
]);
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mod handler {
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use mockall::predicate::eq;
|
||||||
|
use serde_json::json;
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
app::App,
|
||||||
|
cli::{
|
||||||
|
sonarr::trigger_automatic_search_command_handler::{
|
||||||
|
SonarrTriggerAutomaticSearchCommand, SonarrTriggerAutomaticSearchCommandHandler,
|
||||||
|
},
|
||||||
|
CliCommandHandler,
|
||||||
|
},
|
||||||
|
models::{sonarr_models::SonarrSerdeable, Serdeable},
|
||||||
|
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_trigger_automatic_series_search_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::TriggerAutomaticSeriesSearch(expected_series_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let trigger_automatic_series_search_command =
|
||||||
|
SonarrTriggerAutomaticSearchCommand::Series { series_id: 1 };
|
||||||
|
|
||||||
|
let result = SonarrTriggerAutomaticSearchCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
trigger_automatic_series_search_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_trigger_automatic_season_search_command() {
|
||||||
|
let expected_series_id = 1;
|
||||||
|
let expected_season_number = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::TriggerAutomaticSeasonSearch((expected_series_id, expected_season_number))
|
||||||
|
.into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let trigger_automatic_season_search_command = SonarrTriggerAutomaticSearchCommand::Season {
|
||||||
|
series_id: 1,
|
||||||
|
season_number: 1,
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = SonarrTriggerAutomaticSearchCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
trigger_automatic_season_search_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_trigger_automatic_episode_search_command() {
|
||||||
|
let expected_episode_id = 1;
|
||||||
|
let mut mock_network = MockNetworkTrait::new();
|
||||||
|
mock_network
|
||||||
|
.expect_handle_network_event()
|
||||||
|
.with(eq::<NetworkEvent>(
|
||||||
|
SonarrEvent::TriggerAutomaticEpisodeSearch(expected_episode_id).into(),
|
||||||
|
))
|
||||||
|
.times(1)
|
||||||
|
.returning(|_| {
|
||||||
|
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
|
||||||
|
json!({"testResponse": "response"}),
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
||||||
|
let trigger_automatic_episode_search_command =
|
||||||
|
SonarrTriggerAutomaticSearchCommand::Episode { episode_id: 1 };
|
||||||
|
|
||||||
|
let result = SonarrTriggerAutomaticSearchCommandHandler::with(
|
||||||
|
&app_arc,
|
||||||
|
trigger_automatic_episode_search_command,
|
||||||
|
&mut mock_network,
|
||||||
|
)
|
||||||
|
.handle()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -19,6 +19,7 @@ pub enum Key {
|
|||||||
Home,
|
Home,
|
||||||
End,
|
End,
|
||||||
Tab,
|
Tab,
|
||||||
|
BackTab,
|
||||||
Delete,
|
Delete,
|
||||||
Ctrl(char),
|
Ctrl(char),
|
||||||
Char(char),
|
Char(char),
|
||||||
@@ -40,6 +41,7 @@ impl Display for Key {
|
|||||||
Key::Home => write!(f, "<home>"),
|
Key::Home => write!(f, "<home>"),
|
||||||
Key::End => write!(f, "<end>"),
|
Key::End => write!(f, "<end>"),
|
||||||
Key::Tab => write!(f, "<tab>"),
|
Key::Tab => write!(f, "<tab>"),
|
||||||
|
Key::BackTab => write!(f, "<shift-tab>"),
|
||||||
Key::Delete => write!(f, "<del>"),
|
Key::Delete => write!(f, "<del>"),
|
||||||
_ => write!(f, "<{self:?}>"),
|
_ => write!(f, "<{self:?}>"),
|
||||||
}
|
}
|
||||||
@@ -75,6 +77,11 @@ impl From<KeyEvent> for Key {
|
|||||||
KeyEvent {
|
KeyEvent {
|
||||||
code: KeyCode::End, ..
|
code: KeyCode::End, ..
|
||||||
} => Key::End,
|
} => Key::End,
|
||||||
|
KeyEvent {
|
||||||
|
code: KeyCode::BackTab,
|
||||||
|
modifiers: KeyModifiers::SHIFT,
|
||||||
|
..
|
||||||
|
} => Key::BackTab,
|
||||||
KeyEvent {
|
KeyEvent {
|
||||||
code: KeyCode::Tab, ..
|
code: KeyCode::Tab, ..
|
||||||
} => Key::Tab,
|
} => Key::Tab,
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ mod tests {
|
|||||||
#[case(Key::Home, "home")]
|
#[case(Key::Home, "home")]
|
||||||
#[case(Key::End, "end")]
|
#[case(Key::End, "end")]
|
||||||
#[case(Key::Tab, "tab")]
|
#[case(Key::Tab, "tab")]
|
||||||
|
#[case(Key::BackTab, "shift-tab")]
|
||||||
#[case(Key::Delete, "del")]
|
#[case(Key::Delete, "del")]
|
||||||
#[case(Key::Char('q'), "q")]
|
#[case(Key::Char('q'), "q")]
|
||||||
#[case(Key::Ctrl('q'), "ctrl-q")]
|
#[case(Key::Ctrl('q'), "ctrl-q")]
|
||||||
@@ -67,6 +68,19 @@ mod tests {
|
|||||||
assert_eq!(Key::from(KeyEvent::from(KeyCode::Tab)), Key::Tab);
|
assert_eq!(Key::from(KeyEvent::from(KeyCode::Tab)), Key::Tab);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_key_from_back_tab() {
|
||||||
|
assert_eq!(
|
||||||
|
Key::from(KeyEvent {
|
||||||
|
code: KeyCode::BackTab,
|
||||||
|
modifiers: KeyModifiers::SHIFT,
|
||||||
|
kind: KeyEventKind::Press,
|
||||||
|
state: KeyEventState::NONE
|
||||||
|
}),
|
||||||
|
Key::BackTab
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_key_from_delete() {
|
fn test_key_from_delete() {
|
||||||
assert_eq!(Key::from(KeyEvent::from(KeyCode::Delete)), Key::Delete);
|
assert_eq!(Key::from(KeyEvent::from(KeyCode::Delete)), Key::Delete);
|
||||||
|
|||||||
@@ -99,86 +99,96 @@ mod test_utils {
|
|||||||
|
|
||||||
#[macro_export]
|
#[macro_export]
|
||||||
macro_rules! test_iterable_scroll {
|
macro_rules! test_iterable_scroll {
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $block:expr, $context:expr) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $block:expr, $context:expr) => {
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
|
app.push_navigation_stack($block.into());
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.$servarr_data
|
||||||
.$data_ref
|
.$data_ref
|
||||||
.set_items(vec!["Test 1".to_owned(), "Test 2".to_owned()]);
|
.set_items(vec!["Test 1".to_owned(), "Test 2".to_owned()]);
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(&key, &mut app, &$block, &$context).handle();
|
||||||
|
|
||||||
assert_str_eq!(app.data.radarr_data.$data_ref.current_selection(), "Test 2");
|
pretty_assertions::assert_str_eq!(
|
||||||
|
app.data.$servarr_data.$data_ref.current_selection(),
|
||||||
|
"Test 2"
|
||||||
|
);
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(&key, &mut app, &$block, &$context).handle();
|
||||||
|
|
||||||
assert_str_eq!(app.data.radarr_data.$data_ref.current_selection(), "Test 1");
|
pretty_assertions::assert_str_eq!(
|
||||||
|
app.data.$servarr_data.$data_ref.current_selection(),
|
||||||
|
"Test 1"
|
||||||
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $items:ident, $block:expr, $context:expr, $field:ident) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $items:ident, $block:expr, $context:expr, $field:ident) => {
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
|
app.push_navigation_stack($block.into());
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.$servarr_data
|
||||||
.$data_ref
|
.$data_ref
|
||||||
.set_items(simple_stateful_iterable_vec!($items));
|
.set_items(simple_stateful_iterable_vec!($items));
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 2"
|
"Test 2"
|
||||||
);
|
);
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 1"
|
"Test 1"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident) => {
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.$data_ref.set_items($items);
|
app.push_navigation_stack($block.into());
|
||||||
|
app.data.$servarr_data.$data_ref.set_items($items);
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 2"
|
"Test 2"
|
||||||
);
|
);
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 1"
|
"Test 1"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident, $conversion_fn:ident) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident, $conversion_fn:ident) => {
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
fn $func(#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.$data_ref.set_items($items);
|
app.push_navigation_stack($block.into());
|
||||||
|
app.data.$servarr_data.$data_ref.set_items($items);
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.$servarr_data
|
||||||
.$data_ref
|
.$data_ref
|
||||||
.current_selection()
|
.current_selection()
|
||||||
.$field
|
.$field
|
||||||
@@ -186,12 +196,12 @@ mod test_utils {
|
|||||||
"Test 2"
|
"Test 2"
|
||||||
);
|
);
|
||||||
|
|
||||||
$handler::with(&key, &mut app, &$block, &$context).handle();
|
$handler::new(key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.$servarr_data
|
||||||
.$data_ref
|
.$data_ref
|
||||||
.current_selection()
|
.current_selection()
|
||||||
.$field
|
.$field
|
||||||
@@ -204,86 +214,96 @@ mod test_utils {
|
|||||||
|
|
||||||
#[macro_export]
|
#[macro_export]
|
||||||
macro_rules! test_iterable_home_and_end {
|
macro_rules! test_iterable_home_and_end {
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $block:expr, $context:expr) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $block:expr, $context:expr) => {
|
||||||
#[test]
|
#[test]
|
||||||
fn $func() {
|
fn $func() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.$data_ref.set_items(vec![
|
app.push_navigation_stack($block.into());
|
||||||
|
app.data.$servarr_data.$data_ref.set_items(vec![
|
||||||
"Test 1".to_owned(),
|
"Test 1".to_owned(),
|
||||||
"Test 2".to_owned(),
|
"Test 2".to_owned(),
|
||||||
"Test 3".to_owned(),
|
"Test 3".to_owned(),
|
||||||
]);
|
]);
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.end.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.end.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(app.data.radarr_data.$data_ref.current_selection(), "Test 3");
|
pretty_assertions::assert_str_eq!(
|
||||||
|
app.data.$servarr_data.$data_ref.current_selection(),
|
||||||
|
"Test 3"
|
||||||
|
);
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.home.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.home.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(app.data.radarr_data.$data_ref.current_selection(), "Test 1");
|
pretty_assertions::assert_str_eq!(
|
||||||
|
app.data.$servarr_data.$data_ref.current_selection(),
|
||||||
|
"Test 1"
|
||||||
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $items:ident, $block:expr, $context:expr, $field:ident) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $items:ident, $block:expr, $context:expr, $field:ident) => {
|
||||||
#[test]
|
#[test]
|
||||||
fn $func() {
|
fn $func() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
|
app.push_navigation_stack($block.into());
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.$servarr_data
|
||||||
.$data_ref
|
.$data_ref
|
||||||
.set_items(extended_stateful_iterable_vec!($items));
|
.set_items(extended_stateful_iterable_vec!($items));
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.end.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.end.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 3"
|
"Test 3"
|
||||||
);
|
);
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.home.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.home.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 1"
|
"Test 1"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident) => {
|
||||||
#[test]
|
#[test]
|
||||||
fn $func() {
|
fn $func() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.$data_ref.set_items($items);
|
app.push_navigation_stack($block.into());
|
||||||
|
app.data.$servarr_data.$data_ref.set_items($items);
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.end.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.end.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 3"
|
"Test 3"
|
||||||
);
|
);
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.home.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.home.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app.data.radarr_data.$data_ref.current_selection().$field,
|
app.data.$servarr_data.$data_ref.current_selection().$field,
|
||||||
"Test 1"
|
"Test 1"
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
($func:ident, $handler:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident, $conversion_fn:ident) => {
|
($func:ident, $handler:ident, $servarr_data:ident, $data_ref:ident, $items:expr, $block:expr, $context:expr, $field:ident, $conversion_fn:ident) => {
|
||||||
#[test]
|
#[test]
|
||||||
fn $func() {
|
fn $func() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.$data_ref.set_items($items);
|
app.push_navigation_stack($block.into());
|
||||||
|
app.data.$servarr_data.$data_ref.set_items($items);
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.end.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.end.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.$servarr_data
|
||||||
.$data_ref
|
.$data_ref
|
||||||
.current_selection()
|
.current_selection()
|
||||||
.$field
|
.$field
|
||||||
@@ -291,12 +311,12 @@ mod test_utils {
|
|||||||
"Test 3"
|
"Test 3"
|
||||||
);
|
);
|
||||||
|
|
||||||
$handler::with(&DEFAULT_KEYBINDINGS.home.key, &mut app, &$block, &$context).handle();
|
$handler::new(DEFAULT_KEYBINDINGS.home.key, &mut app, $block, $context).handle();
|
||||||
|
|
||||||
assert_str_eq!(
|
pretty_assertions::assert_str_eq!(
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.$servarr_data
|
||||||
.$data_ref
|
.$data_ref
|
||||||
.current_selection()
|
.current_selection()
|
||||||
.$field
|
.$field
|
||||||
@@ -310,19 +330,126 @@ mod test_utils {
|
|||||||
#[macro_export]
|
#[macro_export]
|
||||||
macro_rules! test_handler_delegation {
|
macro_rules! test_handler_delegation {
|
||||||
($handler:ident, $base:expr, $active_block:expr) => {
|
($handler:ident, $base:expr, $active_block:expr) => {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.push_navigation_stack($base.clone().into());
|
app.data.sonarr_data.history.set_items(vec![
|
||||||
app.push_navigation_stack($active_block.clone().into());
|
$crate::models::sonarr_models::SonarrHistoryItem::default(),
|
||||||
|
]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.root_folders
|
||||||
|
.set_items(vec![$crate::models::servarr_models::RootFolder::default()]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.indexers
|
||||||
|
.set_items(vec![$crate::models::servarr_models::Indexer::default()]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.blocklist
|
||||||
|
.set_items(vec![$crate::models::sonarr_models::BlocklistItem::default()]);
|
||||||
|
app.data.sonarr_data.add_searched_series =
|
||||||
|
Some($crate::models::stateful_table::StatefulTable::default());
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.radarr_data
|
||||||
|
.movies
|
||||||
|
.set_items(vec![$crate::models::radarr_models::Movie::default()]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.radarr_data
|
||||||
|
.collections
|
||||||
|
.set_items(vec![$crate::models::radarr_models::Collection::default()]);
|
||||||
|
app.data.radarr_data.collection_movies.set_items(vec![
|
||||||
|
$crate::models::radarr_models::CollectionMovie::default(),
|
||||||
|
]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.radarr_data
|
||||||
|
.indexers
|
||||||
|
.set_items(vec![$crate::models::servarr_models::Indexer::default()]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.radarr_data
|
||||||
|
.root_folders
|
||||||
|
.set_items(vec![$crate::models::servarr_models::RootFolder::default()]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.radarr_data
|
||||||
|
.blocklist
|
||||||
|
.set_items(vec![$crate::models::radarr_models::BlocklistItem::default()]);
|
||||||
|
app.data.radarr_data.add_searched_movies =
|
||||||
|
Some($crate::models::stateful_table::StatefulTable::default());
|
||||||
|
let mut movie_details_modal =
|
||||||
|
$crate::models::servarr_data::radarr::modals::MovieDetailsModal::default();
|
||||||
|
movie_details_modal.movie_history.set_items(vec![
|
||||||
|
$crate::models::radarr_models::MovieHistoryItem::default(),
|
||||||
|
]);
|
||||||
|
movie_details_modal
|
||||||
|
.movie_cast
|
||||||
|
.set_items(vec![$crate::models::radarr_models::Credit::default()]);
|
||||||
|
movie_details_modal
|
||||||
|
.movie_crew
|
||||||
|
.set_items(vec![$crate::models::radarr_models::Credit::default()]);
|
||||||
|
movie_details_modal
|
||||||
|
.movie_releases
|
||||||
|
.set_items(vec![$crate::models::radarr_models::RadarrRelease::default()]);
|
||||||
|
app.data.radarr_data.movie_details_modal = Some(movie_details_modal);
|
||||||
|
let mut season_details_modal =
|
||||||
|
$crate::models::servarr_data::sonarr::modals::SeasonDetailsModal::default();
|
||||||
|
season_details_modal.season_history.set_items(vec![
|
||||||
|
$crate::models::sonarr_models::SonarrHistoryItem::default(),
|
||||||
|
]);
|
||||||
|
season_details_modal.episode_details_modal =
|
||||||
|
Some($crate::models::servarr_data::sonarr::modals::EpisodeDetailsModal::default());
|
||||||
|
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
|
||||||
|
let mut series_history = $crate::models::stateful_table::StatefulTable::default();
|
||||||
|
series_history.set_items(vec![
|
||||||
|
$crate::models::sonarr_models::SonarrHistoryItem::default(),
|
||||||
|
]);
|
||||||
|
app.data.sonarr_data.series_history = Some(series_history);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.series
|
||||||
|
.set_items(vec![$crate::models::sonarr_models::Series::default()]);
|
||||||
|
app.push_navigation_stack($base.into());
|
||||||
|
app.push_navigation_stack($active_block.into());
|
||||||
|
|
||||||
$handler::with(
|
$handler::new(DEFAULT_KEYBINDINGS.esc.key, &mut app, $active_block, None).handle();
|
||||||
&DEFAULT_KEYBINDINGS.esc.key,
|
|
||||||
&mut app,
|
|
||||||
&$active_block,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(app.get_current_route(), &$base.into());
|
pretty_assertions::assert_eq!(app.get_current_route(), $base.into());
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! assert_delete_prompt {
|
||||||
|
($handler:ident, $block:expr, $expected_block:expr) => {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
|
||||||
|
$handler::new(DELETE_KEY, &mut app, $block, None).handle();
|
||||||
|
|
||||||
|
pretty_assertions::assert_eq!(app.get_current_route(), $expected_block.into());
|
||||||
|
};
|
||||||
|
|
||||||
|
($handler:ident, $app:expr, $block:expr, $expected_block:expr) => {
|
||||||
|
$handler::new(DELETE_KEY, &mut $app, $block, None).handle();
|
||||||
|
|
||||||
|
pretty_assertions::assert_eq!($app.get_current_route(), $expected_block.into());
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! assert_refresh_key {
|
||||||
|
($handler:ident, $block:expr) => {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.push_navigation_stack($block.into());
|
||||||
|
|
||||||
|
$handler::new(DEFAULT_KEYBINDINGS.refresh.key, &mut app, $block, None).handle();
|
||||||
|
|
||||||
|
pretty_assertions::assert_eq!(app.get_current_route(), $block.into());
|
||||||
|
assert!(app.should_refresh);
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,14 +1,24 @@
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
|
use crate::models::radarr_models::Movie;
|
||||||
|
use crate::models::sonarr_models::Series;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
|
use tokio_util::sync::CancellationToken;
|
||||||
|
|
||||||
|
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
||||||
use crate::app::App;
|
use crate::app::App;
|
||||||
use crate::event::Key;
|
use crate::event::Key;
|
||||||
|
use crate::handlers::handle_events;
|
||||||
use crate::handlers::{handle_clear_errors, handle_prompt_toggle};
|
use crate::handlers::{handle_clear_errors, handle_prompt_toggle};
|
||||||
|
use crate::models::servarr_data::radarr::radarr_data::ActiveRadarrBlock;
|
||||||
|
use crate::models::servarr_data::sonarr::sonarr_data::ActiveSonarrBlock;
|
||||||
|
use crate::models::HorizontallyScrollableText;
|
||||||
|
use crate::models::Route;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_handle_clear_errors() {
|
fn test_handle_clear_errors() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.error = "test error".to_owned().into();
|
app.error = "test error".to_owned().into();
|
||||||
|
|
||||||
handle_clear_errors(&mut app);
|
handle_clear_errors(&mut app);
|
||||||
@@ -17,17 +27,90 @@ mod tests {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn test_handle_prompt_toggle_left_right(#[values(Key::Left, Key::Right)] key: Key) {
|
#[case(ActiveRadarrBlock::Movies.into(), ActiveRadarrBlock::SearchMovie.into())]
|
||||||
let mut app = App::default();
|
#[case(ActiveSonarrBlock::Series.into(), ActiveSonarrBlock::SearchSeries.into())]
|
||||||
|
fn test_handle_events(#[case] base_block: Route, #[case] top_block: Route) {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.push_navigation_stack(base_block);
|
||||||
|
app.push_navigation_stack(top_block);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.series
|
||||||
|
.set_items(vec![Series::default()]);
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.radarr_data
|
||||||
|
.movies
|
||||||
|
.set_items(vec![Movie::default()]);
|
||||||
|
|
||||||
|
handle_events(DEFAULT_KEYBINDINGS.esc.key, &mut app);
|
||||||
|
|
||||||
|
assert_eq!(app.get_current_route(), base_block);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
#[case(0, ActiveSonarrBlock::Series, ActiveSonarrBlock::Series)]
|
||||||
|
#[case(1, ActiveRadarrBlock::Movies, ActiveRadarrBlock::Movies)]
|
||||||
|
fn test_handle_change_tabs<T>(#[case] index: usize, #[case] left_block: T, #[case] right_block: T)
|
||||||
|
where
|
||||||
|
T: Into<Route> + Copy,
|
||||||
|
{
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.error = "Test".into();
|
||||||
|
app.server_tabs.set_index(index);
|
||||||
|
|
||||||
|
handle_events(DEFAULT_KEYBINDINGS.previous_servarr.key, &mut app);
|
||||||
|
|
||||||
|
assert_eq!(app.server_tabs.get_active_route(), left_block.into());
|
||||||
|
assert_eq!(app.get_current_route(), left_block.into());
|
||||||
|
assert!(app.is_first_render);
|
||||||
|
assert_eq!(app.error, HorizontallyScrollableText::default());
|
||||||
|
assert!(app.cancellation_token.is_cancelled());
|
||||||
|
|
||||||
|
app.server_tabs.set_index(index);
|
||||||
|
app.is_first_render = false;
|
||||||
|
app.error = "Test".into();
|
||||||
|
app.cancellation_token = CancellationToken::new();
|
||||||
|
|
||||||
|
handle_events(DEFAULT_KEYBINDINGS.next_servarr.key, &mut app);
|
||||||
|
|
||||||
|
assert_eq!(app.server_tabs.get_active_route(), right_block.into());
|
||||||
|
assert_eq!(app.get_current_route(), right_block.into());
|
||||||
|
assert!(app.is_first_render);
|
||||||
|
assert_eq!(app.error, HorizontallyScrollableText::default());
|
||||||
|
assert!(app.cancellation_token.is_cancelled());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_handle_prompt_toggle_left_right_radarr(#[values(Key::Left, Key::Right)] key: Key) {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.push_navigation_stack(ActiveRadarrBlock::Movies.into());
|
||||||
|
|
||||||
assert!(!app.data.radarr_data.prompt_confirm);
|
assert!(!app.data.radarr_data.prompt_confirm);
|
||||||
|
|
||||||
handle_prompt_toggle(&mut app, &key);
|
handle_prompt_toggle(&mut app, key);
|
||||||
|
|
||||||
assert!(app.data.radarr_data.prompt_confirm);
|
assert!(app.data.radarr_data.prompt_confirm);
|
||||||
|
|
||||||
handle_prompt_toggle(&mut app, &key);
|
handle_prompt_toggle(&mut app, key);
|
||||||
|
|
||||||
assert!(!app.data.radarr_data.prompt_confirm);
|
assert!(!app.data.radarr_data.prompt_confirm);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_handle_prompt_toggle_left_right_sonarr(#[values(Key::Left, Key::Right)] key: Key) {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.push_navigation_stack(ActiveSonarrBlock::Series.into());
|
||||||
|
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
|
||||||
|
handle_prompt_toggle(&mut app, key);
|
||||||
|
|
||||||
|
assert!(app.data.sonarr_data.prompt_confirm);
|
||||||
|
|
||||||
|
handle_prompt_toggle(&mut app, key);
|
||||||
|
|
||||||
|
assert!(!app.data.sonarr_data.prompt_confirm);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,11 +1,13 @@
|
|||||||
use radarr_handlers::RadarrHandler;
|
use radarr_handlers::RadarrHandler;
|
||||||
|
use sonarr_handlers::SonarrHandler;
|
||||||
|
|
||||||
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
|
||||||
use crate::app::App;
|
use crate::app::App;
|
||||||
use crate::event::Key;
|
use crate::event::Key;
|
||||||
|
use crate::matches_key;
|
||||||
use crate::models::{HorizontallyScrollableText, Route};
|
use crate::models::{HorizontallyScrollableText, Route};
|
||||||
|
|
||||||
mod radarr_handlers;
|
mod radarr_handlers;
|
||||||
|
mod sonarr_handlers;
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[path = "handlers_tests.rs"]
|
#[path = "handlers_tests.rs"]
|
||||||
@@ -14,45 +16,48 @@ mod handlers_tests;
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[path = "handler_test_utils.rs"]
|
#[path = "handler_test_utils.rs"]
|
||||||
pub mod handler_test_utils;
|
pub mod handler_test_utils;
|
||||||
|
mod table_handler;
|
||||||
|
|
||||||
pub trait KeyEventHandler<'a, 'b, T: Into<Route>> {
|
pub trait KeyEventHandler<'a, 'b, T: Into<Route> + Copy> {
|
||||||
fn handle_key_event(&mut self) {
|
fn handle_key_event(&mut self) {
|
||||||
let key = self.get_key();
|
let key = self.get_key();
|
||||||
match key {
|
match key {
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.up.key => {
|
_ if matches_key!(up, key, self.ignore_alt_navigation()) => {
|
||||||
if self.is_ready() {
|
if self.is_ready() {
|
||||||
self.handle_scroll_up();
|
self.handle_scroll_up();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.down.key => {
|
_ if matches_key!(down, key, self.ignore_alt_navigation()) => {
|
||||||
if self.is_ready() {
|
if self.is_ready() {
|
||||||
self.handle_scroll_down();
|
self.handle_scroll_down();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.home.key => {
|
_ if matches_key!(home, key) => {
|
||||||
if self.is_ready() {
|
if self.is_ready() {
|
||||||
self.handle_home();
|
self.handle_home();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.end.key => {
|
_ if matches_key!(end, key) => {
|
||||||
if self.is_ready() {
|
if self.is_ready() {
|
||||||
self.handle_end();
|
self.handle_end();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.delete.key => {
|
_ if matches_key!(delete, key) => {
|
||||||
if self.is_ready() {
|
if self.is_ready() {
|
||||||
self.handle_delete();
|
self.handle_delete();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.left.key || *key == DEFAULT_KEYBINDINGS.right.key => {
|
_ if matches_key!(left, key, self.ignore_alt_navigation())
|
||||||
|
|| matches_key!(right, key, self.ignore_alt_navigation()) =>
|
||||||
|
{
|
||||||
self.handle_left_right_action()
|
self.handle_left_right_action()
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.submit.key => {
|
_ if matches_key!(submit, key) => {
|
||||||
if self.is_ready() {
|
if self.is_ready() {
|
||||||
self.handle_submit();
|
self.handle_submit();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.esc.key => self.handle_esc(),
|
_ if matches_key!(esc, key) => self.handle_esc(),
|
||||||
_ => {
|
_ => {
|
||||||
if self.is_ready() {
|
if self.is_ready() {
|
||||||
self.handle_char_key_event();
|
self.handle_char_key_event();
|
||||||
@@ -65,9 +70,10 @@ pub trait KeyEventHandler<'a, 'b, T: Into<Route>> {
|
|||||||
self.handle_key_event();
|
self.handle_key_event();
|
||||||
}
|
}
|
||||||
|
|
||||||
fn accepts(active_block: &'a T) -> bool;
|
fn accepts(active_block: T) -> bool;
|
||||||
fn with(key: &'a Key, app: &'a mut App<'b>, active_block: &'a T, context: &'a Option<T>) -> Self;
|
fn new(key: Key, app: &'a mut App<'b>, active_block: T, context: Option<T>) -> Self;
|
||||||
fn get_key(&self) -> &Key;
|
fn get_key(&self) -> Key;
|
||||||
|
fn ignore_alt_navigation(&self) -> bool;
|
||||||
fn is_ready(&self) -> bool;
|
fn is_ready(&self) -> bool;
|
||||||
fn handle_scroll_up(&mut self);
|
fn handle_scroll_up(&mut self);
|
||||||
fn handle_scroll_down(&mut self);
|
fn handle_scroll_down(&mut self);
|
||||||
@@ -81,8 +87,26 @@ pub trait KeyEventHandler<'a, 'b, T: Into<Route>> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn handle_events(key: Key, app: &mut App<'_>) {
|
pub fn handle_events(key: Key, app: &mut App<'_>) {
|
||||||
if let Route::Radarr(active_radarr_block, context) = *app.get_current_route() {
|
if matches_key!(next_servarr, key) {
|
||||||
RadarrHandler::with(&key, app, &active_radarr_block, &context).handle()
|
app.reset();
|
||||||
|
app.server_tabs.next();
|
||||||
|
app.pop_and_push_navigation_stack(app.server_tabs.get_active_route());
|
||||||
|
app.cancellation_token.cancel();
|
||||||
|
} else if matches_key!(previous_servarr, key) {
|
||||||
|
app.reset();
|
||||||
|
app.server_tabs.previous();
|
||||||
|
app.pop_and_push_navigation_stack(app.server_tabs.get_active_route());
|
||||||
|
app.cancellation_token.cancel();
|
||||||
|
} else {
|
||||||
|
match app.get_current_route() {
|
||||||
|
Route::Radarr(active_radarr_block, context) => {
|
||||||
|
RadarrHandler::new(key, app, active_radarr_block, context).handle()
|
||||||
|
}
|
||||||
|
Route::Sonarr(active_sonarr_block, context) => {
|
||||||
|
SonarrHandler::new(key, app, active_sonarr_block, context).handle()
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -92,13 +116,17 @@ fn handle_clear_errors(app: &mut App<'_>) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn handle_prompt_toggle(app: &mut App<'_>, key: &Key) {
|
fn handle_prompt_toggle(app: &mut App<'_>, key: Key) {
|
||||||
match key {
|
match key {
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.left.key || *key == DEFAULT_KEYBINDINGS.right.key => {
|
_ if matches_key!(left, key) || matches_key!(right, key) => match app.get_current_route() {
|
||||||
if let Route::Radarr(_, _) = *app.get_current_route() {
|
Route::Radarr(_, _) => {
|
||||||
app.data.radarr_data.prompt_confirm = !app.data.radarr_data.prompt_confirm;
|
app.data.radarr_data.prompt_confirm = !app.data.radarr_data.prompt_confirm
|
||||||
}
|
}
|
||||||
}
|
Route::Sonarr(_, _) => {
|
||||||
|
app.data.sonarr_data.prompt_confirm = !app.data.sonarr_data.prompt_confirm
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
},
|
||||||
_ => (),
|
_ => (),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -107,10 +135,10 @@ fn handle_prompt_toggle(app: &mut App<'_>, key: &Key) {
|
|||||||
macro_rules! handle_text_box_left_right_keys {
|
macro_rules! handle_text_box_left_right_keys {
|
||||||
($self:expr, $key:expr, $input:expr) => {
|
($self:expr, $key:expr, $input:expr) => {
|
||||||
match $self.key {
|
match $self.key {
|
||||||
_ if *$key == DEFAULT_KEYBINDINGS.left.key => {
|
_ if $key == $crate::app::key_binding::DEFAULT_KEYBINDINGS.left.key => {
|
||||||
$input.scroll_left();
|
$input.scroll_left();
|
||||||
}
|
}
|
||||||
_ if *$key == DEFAULT_KEYBINDINGS.right.key => {
|
_ if $key == $crate::app::key_binding::DEFAULT_KEYBINDINGS.right.key => {
|
||||||
$input.scroll_right();
|
$input.scroll_right();
|
||||||
}
|
}
|
||||||
_ => (),
|
_ => (),
|
||||||
@@ -122,13 +150,26 @@ macro_rules! handle_text_box_left_right_keys {
|
|||||||
macro_rules! handle_text_box_keys {
|
macro_rules! handle_text_box_keys {
|
||||||
($self:expr, $key:expr, $input:expr) => {
|
($self:expr, $key:expr, $input:expr) => {
|
||||||
match $self.key {
|
match $self.key {
|
||||||
_ if *$key == DEFAULT_KEYBINDINGS.backspace.key => {
|
_ if $crate::matches_key!(backspace, $key) => {
|
||||||
$input.pop();
|
$input.pop();
|
||||||
}
|
}
|
||||||
Key::Char(character) => {
|
Key::Char(character) => {
|
||||||
$input.push(*character);
|
$input.push(character);
|
||||||
}
|
}
|
||||||
_ => (),
|
_ => (),
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! handle_prompt_left_right_keys {
|
||||||
|
($self:expr, $confirm_prompt:expr, $data:ident) => {
|
||||||
|
if $self.app.data.$data.selected_block.get_active_block() == $confirm_prompt {
|
||||||
|
handle_prompt_toggle($self.app, $self.key);
|
||||||
|
} else if $crate::matches_key!(left, $self.key) {
|
||||||
|
$self.app.data.$data.selected_block.left();
|
||||||
|
} else {
|
||||||
|
$self.app.data.$data.selected_block.right();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ mod tests {
|
|||||||
|
|
||||||
use chrono::DateTime;
|
use chrono::DateTime;
|
||||||
use pretty_assertions::{assert_eq, assert_str_eq};
|
use pretty_assertions::{assert_eq, assert_str_eq};
|
||||||
|
use rstest::rstest;
|
||||||
use strum::IntoEnumIterator;
|
use strum::IntoEnumIterator;
|
||||||
|
|
||||||
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
||||||
@@ -11,250 +12,9 @@ mod tests {
|
|||||||
use crate::event::Key;
|
use crate::event::Key;
|
||||||
use crate::handlers::radarr_handlers::blocklist::{blocklist_sorting_options, BlocklistHandler};
|
use crate::handlers::radarr_handlers::blocklist::{blocklist_sorting_options, BlocklistHandler};
|
||||||
use crate::handlers::KeyEventHandler;
|
use crate::handlers::KeyEventHandler;
|
||||||
use crate::models::radarr_models::{
|
use crate::models::radarr_models::{BlocklistItem, BlocklistItemMovie};
|
||||||
BlocklistItem, BlocklistItemMovie, Language, Quality, QualityWrapper,
|
|
||||||
};
|
|
||||||
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, BLOCKLIST_BLOCKS};
|
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, BLOCKLIST_BLOCKS};
|
||||||
use crate::models::stateful_table::SortOption;
|
use crate::models::servarr_models::{Language, Quality, QualityWrapper};
|
||||||
|
|
||||||
mod test_handle_scroll_up_and_down {
|
|
||||||
use pretty_assertions::{assert_eq, assert_str_eq};
|
|
||||||
use rstest::rstest;
|
|
||||||
|
|
||||||
use crate::models::radarr_models::BlocklistItem;
|
|
||||||
use crate::{simple_stateful_iterable_vec, test_iterable_scroll};
|
|
||||||
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
test_iterable_scroll!(
|
|
||||||
test_blocklist_scroll,
|
|
||||||
BlocklistHandler,
|
|
||||||
blocklist,
|
|
||||||
simple_stateful_iterable_vec!(BlocklistItem, String, source_title),
|
|
||||||
ActiveRadarrBlock::Blocklist,
|
|
||||||
None,
|
|
||||||
source_title,
|
|
||||||
to_string
|
|
||||||
);
|
|
||||||
|
|
||||||
#[rstest]
|
|
||||||
fn test_blocklist_scroll_no_op_when_not_ready(
|
|
||||||
#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key,
|
|
||||||
) {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.is_loading = true;
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.set_items(simple_stateful_iterable_vec!(
|
|
||||||
BlocklistItem,
|
|
||||||
String,
|
|
||||||
source_title
|
|
||||||
));
|
|
||||||
|
|
||||||
BlocklistHandler::with(&key, &mut app, &ActiveRadarrBlock::Blocklist, &None).handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.current_selection()
|
|
||||||
.source_title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
|
|
||||||
BlocklistHandler::with(&key, &mut app, &ActiveRadarrBlock::Blocklist, &None).handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.current_selection()
|
|
||||||
.source_title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[rstest]
|
|
||||||
fn test_blocklist_sort_scroll(
|
|
||||||
#[values(DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key)] key: Key,
|
|
||||||
) {
|
|
||||||
let blocklist_field_vec = sort_options();
|
|
||||||
let mut app = App::default();
|
|
||||||
app.data.radarr_data.blocklist.sorting(sort_options());
|
|
||||||
|
|
||||||
if key == Key::Up {
|
|
||||||
for i in (0..blocklist_field_vec.len()).rev() {
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::BlocklistSortPrompt,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_ref()
|
|
||||||
.unwrap()
|
|
||||||
.current_selection(),
|
|
||||||
&blocklist_field_vec[i]
|
|
||||||
);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
for i in 0..blocklist_field_vec.len() {
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::BlocklistSortPrompt,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_ref()
|
|
||||||
.unwrap()
|
|
||||||
.current_selection(),
|
|
||||||
&blocklist_field_vec[(i + 1) % blocklist_field_vec.len()]
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
mod test_handle_home_end {
|
|
||||||
use pretty_assertions::{assert_eq, assert_str_eq};
|
|
||||||
|
|
||||||
use crate::models::radarr_models::BlocklistItem;
|
|
||||||
use crate::{extended_stateful_iterable_vec, test_iterable_home_and_end};
|
|
||||||
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
test_iterable_home_and_end!(
|
|
||||||
test_blocklist_home_and_end,
|
|
||||||
BlocklistHandler,
|
|
||||||
blocklist,
|
|
||||||
extended_stateful_iterable_vec!(BlocklistItem, String, source_title),
|
|
||||||
ActiveRadarrBlock::Blocklist,
|
|
||||||
None,
|
|
||||||
source_title,
|
|
||||||
to_string
|
|
||||||
);
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_blocklist_home_and_end_no_op_when_not_ready() {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.is_loading = true;
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.set_items(extended_stateful_iterable_vec!(
|
|
||||||
BlocklistItem,
|
|
||||||
String,
|
|
||||||
source_title
|
|
||||||
));
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.end.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::Blocklist,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.current_selection()
|
|
||||||
.source_title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.home.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::Blocklist,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.current_selection()
|
|
||||||
.source_title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_blocklist_sort_home_end() {
|
|
||||||
let blocklist_field_vec = sort_options();
|
|
||||||
let mut app = App::default();
|
|
||||||
app.data.radarr_data.blocklist.sorting(sort_options());
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.end.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::BlocklistSortPrompt,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_ref()
|
|
||||||
.unwrap()
|
|
||||||
.current_selection(),
|
|
||||||
&blocklist_field_vec[blocklist_field_vec.len() - 1]
|
|
||||||
);
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.home.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::BlocklistSortPrompt,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_ref()
|
|
||||||
.unwrap()
|
|
||||||
.current_selection(),
|
|
||||||
&blocklist_field_vec[0]
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
mod test_handle_delete {
|
mod test_handle_delete {
|
||||||
use pretty_assertions::assert_eq;
|
use pretty_assertions::assert_eq;
|
||||||
@@ -265,30 +25,27 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_delete_blocklist_item_prompt() {
|
fn test_delete_blocklist_item_prompt() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
|
|
||||||
BlocklistHandler::with(&DELETE_KEY, &mut app, &ActiveRadarrBlock::Blocklist, &None).handle();
|
BlocklistHandler::new(DELETE_KEY, &mut app, ActiveRadarrBlock::Blocklist, None).handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::DeleteBlocklistItemPrompt.into()
|
ActiveRadarrBlock::DeleteBlocklistItemPrompt.into()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_delete_blocklist_item_no_op_when_not_ready() {
|
fn test_delete_blocklist_item_no_op_when_not_ready() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
|
|
||||||
BlocklistHandler::with(&DELETE_KEY, &mut app, &ActiveRadarrBlock::Blocklist, &None).handle();
|
BlocklistHandler::new(DELETE_KEY, &mut app, ActiveRadarrBlock::Blocklist, None).handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -300,49 +57,46 @@ mod tests {
|
|||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn test_blocklist_tab_left(#[values(true, false)] is_ready: bool) {
|
fn test_blocklist_tab_left(#[values(true, false)] is_ready: bool) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = is_ready;
|
app.is_loading = is_ready;
|
||||||
app.data.radarr_data.main_tabs.set_index(3);
|
app.data.radarr_data.main_tabs.set_index(3);
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.left.key,
|
DEFAULT_KEYBINDINGS.left.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.data.radarr_data.main_tabs.get_active_route(),
|
app.data.radarr_data.main_tabs.get_active_route(),
|
||||||
&ActiveRadarrBlock::Downloads.into()
|
ActiveRadarrBlock::Downloads.into()
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Downloads.into()
|
|
||||||
);
|
);
|
||||||
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Downloads.into());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn test_blocklist_tab_right(#[values(true, false)] is_ready: bool) {
|
fn test_blocklist_tab_right(#[values(true, false)] is_ready: bool) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = is_ready;
|
app.is_loading = is_ready;
|
||||||
app.data.radarr_data.main_tabs.set_index(3);
|
app.data.radarr_data.main_tabs.set_index(3);
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.right.key,
|
DEFAULT_KEYBINDINGS.right.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.data.radarr_data.main_tabs.get_active_route(),
|
app.data.radarr_data.main_tabs.get_active_route(),
|
||||||
&ActiveRadarrBlock::RootFolders.into()
|
ActiveRadarrBlock::RootFolders.into()
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::RootFolders.into()
|
ActiveRadarrBlock::RootFolders.into()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -355,13 +109,13 @@ mod tests {
|
|||||||
active_radarr_block: ActiveRadarrBlock,
|
active_radarr_block: ActiveRadarrBlock,
|
||||||
#[values(DEFAULT_KEYBINDINGS.left.key, DEFAULT_KEYBINDINGS.right.key)] key: Key,
|
#[values(DEFAULT_KEYBINDINGS.left.key, DEFAULT_KEYBINDINGS.right.key)] key: Key,
|
||||||
) {
|
) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
|
|
||||||
BlocklistHandler::with(&key, &mut app, &active_radarr_block, &None).handle();
|
BlocklistHandler::new(key, &mut app, active_radarr_block, None).handle();
|
||||||
|
|
||||||
assert!(app.data.radarr_data.prompt_confirm);
|
assert!(app.data.radarr_data.prompt_confirm);
|
||||||
|
|
||||||
BlocklistHandler::with(&key, &mut app, &active_radarr_block, &None).handle();
|
BlocklistHandler::new(key, &mut app, active_radarr_block, None).handle();
|
||||||
|
|
||||||
assert!(!app.data.radarr_data.prompt_confirm);
|
assert!(!app.data.radarr_data.prompt_confirm);
|
||||||
}
|
}
|
||||||
@@ -379,38 +133,35 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_blocklist_submit() {
|
fn test_blocklist_submit() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
|
|
||||||
BlocklistHandler::with(&SUBMIT_KEY, &mut app, &ActiveRadarrBlock::Blocklist, &None).handle();
|
BlocklistHandler::new(SUBMIT_KEY, &mut app, ActiveRadarrBlock::Blocklist, None).handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::BlocklistItemDetails.into()
|
ActiveRadarrBlock::BlocklistItemDetails.into()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_blocklist_submit_no_op_when_not_ready() {
|
fn test_blocklist_submit_no_op_when_not_ready() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
|
|
||||||
BlocklistHandler::with(&SUBMIT_KEY, &mut app, &ActiveRadarrBlock::Blocklist, &None).handle();
|
BlocklistHandler::new(SUBMIT_KEY, &mut app, ActiveRadarrBlock::Blocklist, None).handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
#[case(
|
#[case(
|
||||||
ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
ActiveRadarrBlock::DeleteBlocklistItemPrompt,
|
ActiveRadarrBlock::DeleteBlocklistItemPrompt,
|
||||||
RadarrEvent::DeleteBlocklistItem(None)
|
RadarrEvent::DeleteBlocklistItem(3)
|
||||||
)]
|
)]
|
||||||
#[case(
|
#[case(
|
||||||
ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
@@ -422,20 +173,20 @@ mod tests {
|
|||||||
#[case] prompt_block: ActiveRadarrBlock,
|
#[case] prompt_block: ActiveRadarrBlock,
|
||||||
#[case] expected_action: RadarrEvent,
|
#[case] expected_action: RadarrEvent,
|
||||||
) {
|
) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
app.data.radarr_data.prompt_confirm = true;
|
app.data.radarr_data.prompt_confirm = true;
|
||||||
app.push_navigation_stack(base_route.into());
|
app.push_navigation_stack(base_route.into());
|
||||||
app.push_navigation_stack(prompt_block.into());
|
app.push_navigation_stack(prompt_block.into());
|
||||||
|
|
||||||
BlocklistHandler::with(&SUBMIT_KEY, &mut app, &prompt_block, &None).handle();
|
BlocklistHandler::new(SUBMIT_KEY, &mut app, prompt_block, None).handle();
|
||||||
|
|
||||||
assert!(app.data.radarr_data.prompt_confirm);
|
assert!(app.data.radarr_data.prompt_confirm);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.data.radarr_data.prompt_confirm_action,
|
app.data.radarr_data.prompt_confirm_action,
|
||||||
Some(expected_action)
|
Some(expected_action)
|
||||||
);
|
);
|
||||||
assert_eq!(app.get_current_route(), &base_route.into());
|
assert_eq!(app.get_current_route(), base_route.into());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
@@ -446,47 +197,16 @@ mod tests {
|
|||||||
)]
|
)]
|
||||||
prompt_block: ActiveRadarrBlock,
|
prompt_block: ActiveRadarrBlock,
|
||||||
) {
|
) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
app.push_navigation_stack(prompt_block.into());
|
app.push_navigation_stack(prompt_block.into());
|
||||||
|
|
||||||
BlocklistHandler::with(&SUBMIT_KEY, &mut app, &prompt_block, &None).handle();
|
BlocklistHandler::new(SUBMIT_KEY, &mut app, prompt_block, None).handle();
|
||||||
|
|
||||||
assert!(!app.data.radarr_data.prompt_confirm);
|
assert!(!app.data.radarr_data.prompt_confirm);
|
||||||
assert_eq!(app.data.radarr_data.prompt_confirm_action, None);
|
assert_eq!(app.data.radarr_data.prompt_confirm_action, None);
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_blocklist_sort_prompt_submit() {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.data.radarr_data.blocklist.sort_asc = true;
|
|
||||||
app.data.radarr_data.blocklist.sorting(sort_options());
|
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::BlocklistSortPrompt.into());
|
|
||||||
|
|
||||||
let mut expected_vec = blocklist_vec();
|
|
||||||
expected_vec.sort_by(|a, b| a.id.cmp(&b.id));
|
|
||||||
expected_vec.reverse();
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&SUBMIT_KEY,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::BlocklistSortPrompt,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
assert_eq!(app.data.radarr_data.blocklist.items, expected_vec);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -513,71 +233,45 @@ mod tests {
|
|||||||
#[case] base_block: ActiveRadarrBlock,
|
#[case] base_block: ActiveRadarrBlock,
|
||||||
#[case] prompt_block: ActiveRadarrBlock,
|
#[case] prompt_block: ActiveRadarrBlock,
|
||||||
) {
|
) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.push_navigation_stack(base_block.into());
|
app.push_navigation_stack(base_block.into());
|
||||||
app.push_navigation_stack(prompt_block.into());
|
app.push_navigation_stack(prompt_block.into());
|
||||||
app.data.radarr_data.prompt_confirm = true;
|
app.data.radarr_data.prompt_confirm = true;
|
||||||
|
|
||||||
BlocklistHandler::with(&ESC_KEY, &mut app, &prompt_block, &None).handle();
|
BlocklistHandler::new(ESC_KEY, &mut app, prompt_block, None).handle();
|
||||||
|
|
||||||
assert_eq!(app.get_current_route(), &base_block.into());
|
assert_eq!(app.get_current_route(), base_block.into());
|
||||||
assert!(!app.data.radarr_data.prompt_confirm);
|
assert!(!app.data.radarr_data.prompt_confirm);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_esc_blocklist_item_details() {
|
fn test_esc_blocklist_item_details() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::BlocklistItemDetails.into());
|
app.push_navigation_stack(ActiveRadarrBlock::BlocklistItemDetails.into());
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&ESC_KEY,
|
ESC_KEY,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::BlocklistItemDetails,
|
ActiveRadarrBlock::BlocklistItemDetails,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_blocklist_sort_prompt_block_esc() {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::BlocklistSortPrompt.into());
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&ESC_KEY,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::BlocklistSortPrompt,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn test_default_esc(#[values(true, false)] is_ready: bool) {
|
fn test_default_esc(#[values(true, false)] is_ready: bool) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = is_ready;
|
app.is_loading = is_ready;
|
||||||
app.error = "test error".to_owned().into();
|
app.error = "test error".to_owned().into();
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
|
|
||||||
DownloadsHandler::with(&ESC_KEY, &mut app, &ActiveRadarrBlock::Blocklist, &None).handle();
|
DownloadsHandler::new(ESC_KEY, &mut app, ActiveRadarrBlock::Blocklist, None).handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
assert!(app.error.text.is_empty());
|
assert!(app.error.text.is_empty());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -592,139 +286,83 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_refresh_blocklist_key() {
|
fn test_refresh_blocklist_key() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.refresh.key,
|
DEFAULT_KEYBINDINGS.refresh.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
assert!(app.should_refresh);
|
assert!(app.should_refresh);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_refresh_blocklist_key_no_op_when_not_ready() {
|
fn test_refresh_blocklist_key_no_op_when_not_ready() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.refresh.key,
|
DEFAULT_KEYBINDINGS.refresh.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
assert!(!app.should_refresh);
|
assert!(!app.should_refresh);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_clear_blocklist_key() {
|
fn test_clear_blocklist_key() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.clear.key,
|
DEFAULT_KEYBINDINGS.clear.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::BlocklistClearAllItemsPrompt.into()
|
ActiveRadarrBlock::BlocklistClearAllItemsPrompt.into()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_clear_blocklist_key_no_op_when_not_ready() {
|
fn test_clear_blocklist_key_no_op_when_not_ready() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.clear.key,
|
DEFAULT_KEYBINDINGS.clear.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Blocklist.into());
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_sort_key() {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.sort.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::Blocklist,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::BlocklistSortPrompt.into()
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
app.data.radarr_data.blocklist.sort.as_ref().unwrap().items,
|
|
||||||
blocklist_sorting_options()
|
|
||||||
);
|
|
||||||
assert!(!app.data.radarr_data.blocklist.sort_asc);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_sort_key_no_op_when_not_ready() {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.is_loading = true;
|
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Blocklist.into());
|
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
|
||||||
|
|
||||||
BlocklistHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.sort.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::Blocklist,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
app.get_current_route(),
|
|
||||||
&ActiveRadarrBlock::Blocklist.into()
|
|
||||||
);
|
|
||||||
assert!(app.data.radarr_data.blocklist.sort.is_none());
|
|
||||||
assert!(!app.data.radarr_data.blocklist.sort_asc);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
#[case(
|
#[case(
|
||||||
ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
ActiveRadarrBlock::DeleteBlocklistItemPrompt,
|
ActiveRadarrBlock::DeleteBlocklistItemPrompt,
|
||||||
RadarrEvent::DeleteBlocklistItem(None)
|
RadarrEvent::DeleteBlocklistItem(3)
|
||||||
)]
|
)]
|
||||||
#[case(
|
#[case(
|
||||||
ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
@@ -736,16 +374,16 @@ mod tests {
|
|||||||
#[case] prompt_block: ActiveRadarrBlock,
|
#[case] prompt_block: ActiveRadarrBlock,
|
||||||
#[case] expected_action: RadarrEvent,
|
#[case] expected_action: RadarrEvent,
|
||||||
) {
|
) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
app.push_navigation_stack(base_route.into());
|
app.push_navigation_stack(base_route.into());
|
||||||
app.push_navigation_stack(prompt_block.into());
|
app.push_navigation_stack(prompt_block.into());
|
||||||
|
|
||||||
BlocklistHandler::with(
|
BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.confirm.key,
|
DEFAULT_KEYBINDINGS.confirm.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&prompt_block,
|
prompt_block,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
@@ -754,7 +392,7 @@ mod tests {
|
|||||||
app.data.radarr_data.prompt_confirm_action,
|
app.data.radarr_data.prompt_confirm_action,
|
||||||
Some(expected_action)
|
Some(expected_action)
|
||||||
);
|
);
|
||||||
assert_eq!(app.get_current_route(), &base_route.into());
|
assert_eq!(app.get_current_route(), base_route.into());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -897,23 +535,55 @@ mod tests {
|
|||||||
fn test_blocklist_handler_accepts() {
|
fn test_blocklist_handler_accepts() {
|
||||||
ActiveRadarrBlock::iter().for_each(|active_radarr_block| {
|
ActiveRadarrBlock::iter().for_each(|active_radarr_block| {
|
||||||
if BLOCKLIST_BLOCKS.contains(&active_radarr_block) {
|
if BLOCKLIST_BLOCKS.contains(&active_radarr_block) {
|
||||||
assert!(BlocklistHandler::accepts(&active_radarr_block));
|
assert!(BlocklistHandler::accepts(active_radarr_block));
|
||||||
} else {
|
} else {
|
||||||
assert!(!BlocklistHandler::accepts(&active_radarr_block));
|
assert!(!BlocklistHandler::accepts(active_radarr_block));
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_blocklist_handler_ignore_alt_navigation(
|
||||||
|
#[values(true, false)] should_ignore_quit_key: bool,
|
||||||
|
) {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.should_ignore_quit_key = should_ignore_quit_key;
|
||||||
|
let handler = BlocklistHandler::new(
|
||||||
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
|
&mut app,
|
||||||
|
ActiveRadarrBlock::default(),
|
||||||
|
None,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(handler.ignore_alt_navigation(), should_ignore_quit_key);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_extract_blocklist_item_id() {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.data.radarr_data.blocklist.set_items(blocklist_vec());
|
||||||
|
|
||||||
|
let blocklist_item_id = BlocklistHandler::new(
|
||||||
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
|
&mut app,
|
||||||
|
ActiveRadarrBlock::Blocklist,
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.extract_blocklist_item_id();
|
||||||
|
|
||||||
|
assert_eq!(blocklist_item_id, 3);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_blocklist_handler_not_ready_when_loading() {
|
fn test_blocklist_handler_not_ready_when_loading() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
|
|
||||||
let handler = BlocklistHandler::with(
|
let handler = BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.esc.key,
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
);
|
);
|
||||||
|
|
||||||
assert!(!handler.is_ready());
|
assert!(!handler.is_ready());
|
||||||
@@ -921,14 +591,14 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_blocklist_handler_not_ready_when_blocklist_is_empty() {
|
fn test_blocklist_handler_not_ready_when_blocklist_is_empty() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = false;
|
app.is_loading = false;
|
||||||
|
|
||||||
let handler = BlocklistHandler::with(
|
let handler = BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.esc.key,
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
);
|
);
|
||||||
|
|
||||||
assert!(!handler.is_ready());
|
assert!(!handler.is_ready());
|
||||||
@@ -936,7 +606,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_blocklist_handler_ready_when_not_loading_and_blocklist_is_not_empty() {
|
fn test_blocklist_handler_ready_when_not_loading_and_blocklist_is_not_empty() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = false;
|
app.is_loading = false;
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
@@ -944,11 +614,11 @@ mod tests {
|
|||||||
.blocklist
|
.blocklist
|
||||||
.set_items(vec![BlocklistItem::default()]);
|
.set_items(vec![BlocklistItem::default()]);
|
||||||
|
|
||||||
let handler = BlocklistHandler::with(
|
let handler = BlocklistHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.esc.key,
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::Blocklist,
|
ActiveRadarrBlock::Blocklist,
|
||||||
&None,
|
None,
|
||||||
);
|
);
|
||||||
|
|
||||||
assert!(handler.is_ready());
|
assert!(handler.is_ready());
|
||||||
@@ -960,6 +630,7 @@ mod tests {
|
|||||||
id: 3,
|
id: 3,
|
||||||
source_title: "test 1".to_owned(),
|
source_title: "test 1".to_owned(),
|
||||||
languages: vec![Language {
|
languages: vec![Language {
|
||||||
|
id: 1,
|
||||||
name: "telgu".to_owned(),
|
name: "telgu".to_owned(),
|
||||||
}],
|
}],
|
||||||
quality: QualityWrapper {
|
quality: QualityWrapper {
|
||||||
@@ -968,6 +639,7 @@ mod tests {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
custom_formats: Some(vec![Language {
|
custom_formats: Some(vec![Language {
|
||||||
|
id: 2,
|
||||||
name: "nikki".to_owned(),
|
name: "nikki".to_owned(),
|
||||||
}]),
|
}]),
|
||||||
date: DateTime::from(DateTime::parse_from_rfc3339("2024-01-10T07:28:45Z").unwrap()),
|
date: DateTime::from(DateTime::parse_from_rfc3339("2024-01-10T07:28:45Z").unwrap()),
|
||||||
@@ -980,6 +652,7 @@ mod tests {
|
|||||||
id: 2,
|
id: 2,
|
||||||
source_title: "test 2".to_owned(),
|
source_title: "test 2".to_owned(),
|
||||||
languages: vec![Language {
|
languages: vec![Language {
|
||||||
|
id: 3,
|
||||||
name: "chinese".to_owned(),
|
name: "chinese".to_owned(),
|
||||||
}],
|
}],
|
||||||
quality: QualityWrapper {
|
quality: QualityWrapper {
|
||||||
@@ -989,9 +662,11 @@ mod tests {
|
|||||||
},
|
},
|
||||||
custom_formats: Some(vec![
|
custom_formats: Some(vec![
|
||||||
Language {
|
Language {
|
||||||
|
id: 4,
|
||||||
name: "alex".to_owned(),
|
name: "alex".to_owned(),
|
||||||
},
|
},
|
||||||
Language {
|
Language {
|
||||||
|
id: 5,
|
||||||
name: "English".to_owned(),
|
name: "English".to_owned(),
|
||||||
},
|
},
|
||||||
]),
|
]),
|
||||||
@@ -1005,6 +680,7 @@ mod tests {
|
|||||||
id: 1,
|
id: 1,
|
||||||
source_title: "test 3".to_owned(),
|
source_title: "test 3".to_owned(),
|
||||||
languages: vec![Language {
|
languages: vec![Language {
|
||||||
|
id: 1,
|
||||||
name: "english".to_owned(),
|
name: "english".to_owned(),
|
||||||
}],
|
}],
|
||||||
quality: QualityWrapper {
|
quality: QualityWrapper {
|
||||||
@@ -1013,6 +689,7 @@ mod tests {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
custom_formats: Some(vec![Language {
|
custom_formats: Some(vec![Language {
|
||||||
|
id: 2,
|
||||||
name: "English".to_owned(),
|
name: "English".to_owned(),
|
||||||
}]),
|
}]),
|
||||||
date: DateTime::from(DateTime::parse_from_rfc3339("2024-03-10T07:28:45Z").unwrap()),
|
date: DateTime::from(DateTime::parse_from_rfc3339("2024-03-10T07:28:45Z").unwrap()),
|
||||||
@@ -1023,15 +700,4 @@ mod tests {
|
|||||||
},
|
},
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|
||||||
fn sort_options() -> Vec<SortOption<BlocklistItem>> {
|
|
||||||
vec![SortOption {
|
|
||||||
name: "Test 1",
|
|
||||||
cmp_fn: Some(|a, b| {
|
|
||||||
b.source_title
|
|
||||||
.to_lowercase()
|
|
||||||
.cmp(&a.source_title.to_lowercase())
|
|
||||||
}),
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,35 +1,64 @@
|
|||||||
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
|
||||||
use crate::app::App;
|
use crate::app::App;
|
||||||
use crate::event::Key;
|
use crate::event::Key;
|
||||||
use crate::handlers::radarr_handlers::handle_change_tab_left_right_keys;
|
use crate::handlers::radarr_handlers::handle_change_tab_left_right_keys;
|
||||||
|
use crate::handlers::table_handler::TableHandlingConfig;
|
||||||
use crate::handlers::{handle_clear_errors, handle_prompt_toggle, KeyEventHandler};
|
use crate::handlers::{handle_clear_errors, handle_prompt_toggle, KeyEventHandler};
|
||||||
use crate::models::radarr_models::BlocklistItem;
|
use crate::models::radarr_models::BlocklistItem;
|
||||||
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, BLOCKLIST_BLOCKS};
|
use crate::models::servarr_data::radarr::radarr_data::{ActiveRadarrBlock, BLOCKLIST_BLOCKS};
|
||||||
use crate::models::stateful_table::SortOption;
|
use crate::models::stateful_table::SortOption;
|
||||||
use crate::models::Scrollable;
|
|
||||||
use crate::network::radarr_network::RadarrEvent;
|
use crate::network::radarr_network::RadarrEvent;
|
||||||
|
use crate::{handle_table_events, matches_key};
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[path = "blocklist_handler_tests.rs"]
|
#[path = "blocklist_handler_tests.rs"]
|
||||||
mod blocklist_handler_tests;
|
mod blocklist_handler_tests;
|
||||||
|
|
||||||
pub(super) struct BlocklistHandler<'a, 'b> {
|
pub(super) struct BlocklistHandler<'a, 'b> {
|
||||||
key: &'a Key,
|
key: Key,
|
||||||
app: &'a mut App<'b>,
|
app: &'a mut App<'b>,
|
||||||
active_radarr_block: &'a ActiveRadarrBlock,
|
active_radarr_block: ActiveRadarrBlock,
|
||||||
_context: &'a Option<ActiveRadarrBlock>,
|
_context: Option<ActiveRadarrBlock>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl BlocklistHandler<'_, '_> {
|
||||||
|
handle_table_events!(
|
||||||
|
self,
|
||||||
|
blocklist,
|
||||||
|
self.app.data.radarr_data.blocklist,
|
||||||
|
BlocklistItem
|
||||||
|
);
|
||||||
|
|
||||||
|
fn extract_blocklist_item_id(&self) -> i64 {
|
||||||
|
self.app.data.radarr_data.blocklist.current_selection().id
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a, 'b> {
|
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a, 'b> {
|
||||||
fn accepts(active_block: &'a ActiveRadarrBlock) -> bool {
|
fn handle(&mut self) {
|
||||||
BLOCKLIST_BLOCKS.contains(active_block)
|
let blocklist_table_handling_config =
|
||||||
|
TableHandlingConfig::new(ActiveRadarrBlock::Blocklist.into())
|
||||||
|
.sorting_block(ActiveRadarrBlock::BlocklistSortPrompt.into())
|
||||||
|
.sort_by_fn(|a: &BlocklistItem, b: &BlocklistItem| a.id.cmp(&b.id))
|
||||||
|
.sort_options(blocklist_sorting_options());
|
||||||
|
|
||||||
|
if !self.handle_blocklist_table_events(blocklist_table_handling_config) {
|
||||||
|
self.handle_key_event();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn with(
|
fn accepts(active_block: ActiveRadarrBlock) -> bool {
|
||||||
key: &'a Key,
|
BLOCKLIST_BLOCKS.contains(&active_block)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn ignore_alt_navigation(&self) -> bool {
|
||||||
|
self.app.should_ignore_quit_key
|
||||||
|
}
|
||||||
|
|
||||||
|
fn new(
|
||||||
|
key: Key,
|
||||||
app: &'a mut App<'b>,
|
app: &'a mut App<'b>,
|
||||||
active_block: &'a ActiveRadarrBlock,
|
active_block: ActiveRadarrBlock,
|
||||||
context: &'a Option<ActiveRadarrBlock>,
|
context: Option<ActiveRadarrBlock>,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
BlocklistHandler {
|
BlocklistHandler {
|
||||||
key,
|
key,
|
||||||
@@ -39,7 +68,7 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_key(&self) -> &Key {
|
fn get_key(&self) -> Key {
|
||||||
self.key
|
self.key
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -47,72 +76,16 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a,
|
|||||||
!self.app.is_loading && !self.app.data.radarr_data.blocklist.is_empty()
|
!self.app.is_loading && !self.app.data.radarr_data.blocklist.is_empty()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn handle_scroll_up(&mut self) {
|
fn handle_scroll_up(&mut self) {}
|
||||||
match self.active_radarr_block {
|
|
||||||
ActiveRadarrBlock::Blocklist => self.app.data.radarr_data.blocklist.scroll_up(),
|
|
||||||
ActiveRadarrBlock::BlocklistSortPrompt => self
|
|
||||||
.app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_mut()
|
|
||||||
.unwrap()
|
|
||||||
.scroll_up(),
|
|
||||||
_ => (),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_scroll_down(&mut self) {
|
fn handle_scroll_down(&mut self) {}
|
||||||
match self.active_radarr_block {
|
|
||||||
ActiveRadarrBlock::Blocklist => self.app.data.radarr_data.blocklist.scroll_down(),
|
|
||||||
ActiveRadarrBlock::BlocklistSortPrompt => self
|
|
||||||
.app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_mut()
|
|
||||||
.unwrap()
|
|
||||||
.scroll_down(),
|
|
||||||
_ => (),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_home(&mut self) {
|
fn handle_home(&mut self) {}
|
||||||
match self.active_radarr_block {
|
|
||||||
ActiveRadarrBlock::Blocklist => self.app.data.radarr_data.blocklist.scroll_to_top(),
|
|
||||||
ActiveRadarrBlock::BlocklistSortPrompt => self
|
|
||||||
.app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_mut()
|
|
||||||
.unwrap()
|
|
||||||
.scroll_to_top(),
|
|
||||||
_ => (),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_end(&mut self) {
|
fn handle_end(&mut self) {}
|
||||||
match self.active_radarr_block {
|
|
||||||
ActiveRadarrBlock::Blocklist => self.app.data.radarr_data.blocklist.scroll_to_bottom(),
|
|
||||||
ActiveRadarrBlock::BlocklistSortPrompt => self
|
|
||||||
.app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sort
|
|
||||||
.as_mut()
|
|
||||||
.unwrap()
|
|
||||||
.scroll_to_bottom(),
|
|
||||||
_ => (),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_delete(&mut self) {
|
fn handle_delete(&mut self) {
|
||||||
if self.active_radarr_block == &ActiveRadarrBlock::Blocklist {
|
if self.active_radarr_block == ActiveRadarrBlock::Blocklist {
|
||||||
self
|
self
|
||||||
.app
|
.app
|
||||||
.push_navigation_stack(ActiveRadarrBlock::DeleteBlocklistItemPrompt.into());
|
.push_navigation_stack(ActiveRadarrBlock::DeleteBlocklistItemPrompt.into());
|
||||||
@@ -132,8 +105,9 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a,
|
|||||||
match self.active_radarr_block {
|
match self.active_radarr_block {
|
||||||
ActiveRadarrBlock::DeleteBlocklistItemPrompt => {
|
ActiveRadarrBlock::DeleteBlocklistItemPrompt => {
|
||||||
if self.app.data.radarr_data.prompt_confirm {
|
if self.app.data.radarr_data.prompt_confirm {
|
||||||
self.app.data.radarr_data.prompt_confirm_action =
|
self.app.data.radarr_data.prompt_confirm_action = Some(RadarrEvent::DeleteBlocklistItem(
|
||||||
Some(RadarrEvent::DeleteBlocklistItem(None));
|
self.extract_blocklist_item_id(),
|
||||||
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
self.app.pop_navigation_stack();
|
self.app.pop_navigation_stack();
|
||||||
@@ -145,18 +119,6 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a,
|
|||||||
|
|
||||||
self.app.pop_navigation_stack();
|
self.app.pop_navigation_stack();
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::BlocklistSortPrompt => {
|
|
||||||
self
|
|
||||||
.app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.items
|
|
||||||
.sort_by(|a, b| a.id.cmp(&b.id));
|
|
||||||
self.app.data.radarr_data.blocklist.apply_sorting();
|
|
||||||
|
|
||||||
self.app.pop_navigation_stack();
|
|
||||||
}
|
|
||||||
ActiveRadarrBlock::Blocklist => {
|
ActiveRadarrBlock::Blocklist => {
|
||||||
self
|
self
|
||||||
.app
|
.app
|
||||||
@@ -173,7 +135,7 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a,
|
|||||||
self.app.pop_navigation_stack();
|
self.app.pop_navigation_stack();
|
||||||
self.app.data.radarr_data.prompt_confirm = false;
|
self.app.data.radarr_data.prompt_confirm = false;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::BlocklistItemDetails | ActiveRadarrBlock::BlocklistSortPrompt => {
|
ActiveRadarrBlock::BlocklistItemDetails => {
|
||||||
self.app.pop_navigation_stack();
|
self.app.pop_navigation_stack();
|
||||||
}
|
}
|
||||||
_ => handle_clear_errors(self.app),
|
_ => handle_clear_errors(self.app),
|
||||||
@@ -184,38 +146,28 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for BlocklistHandler<'a,
|
|||||||
let key = self.key;
|
let key = self.key;
|
||||||
match self.active_radarr_block {
|
match self.active_radarr_block {
|
||||||
ActiveRadarrBlock::Blocklist => match self.key {
|
ActiveRadarrBlock::Blocklist => match self.key {
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.refresh.key => {
|
_ if matches_key!(refresh, key) => {
|
||||||
self.app.should_refresh = true;
|
self.app.should_refresh = true;
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.clear.key => {
|
_ if matches_key!(clear, key) => {
|
||||||
self
|
self
|
||||||
.app
|
.app
|
||||||
.push_navigation_stack(ActiveRadarrBlock::BlocklistClearAllItemsPrompt.into());
|
.push_navigation_stack(ActiveRadarrBlock::BlocklistClearAllItemsPrompt.into());
|
||||||
}
|
}
|
||||||
_ if *key == DEFAULT_KEYBINDINGS.sort.key => {
|
|
||||||
self
|
|
||||||
.app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.blocklist
|
|
||||||
.sorting(blocklist_sorting_options());
|
|
||||||
self
|
|
||||||
.app
|
|
||||||
.push_navigation_stack(ActiveRadarrBlock::BlocklistSortPrompt.into());
|
|
||||||
}
|
|
||||||
_ => (),
|
_ => (),
|
||||||
},
|
},
|
||||||
ActiveRadarrBlock::DeleteBlocklistItemPrompt => {
|
ActiveRadarrBlock::DeleteBlocklistItemPrompt => {
|
||||||
if *key == DEFAULT_KEYBINDINGS.confirm.key {
|
if matches_key!(confirm, key) {
|
||||||
self.app.data.radarr_data.prompt_confirm = true;
|
self.app.data.radarr_data.prompt_confirm = true;
|
||||||
self.app.data.radarr_data.prompt_confirm_action =
|
self.app.data.radarr_data.prompt_confirm_action = Some(RadarrEvent::DeleteBlocklistItem(
|
||||||
Some(RadarrEvent::DeleteBlocklistItem(None));
|
self.extract_blocklist_item_id(),
|
||||||
|
));
|
||||||
|
|
||||||
self.app.pop_navigation_stack();
|
self.app.pop_navigation_stack();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::BlocklistClearAllItemsPrompt => {
|
ActiveRadarrBlock::BlocklistClearAllItemsPrompt => {
|
||||||
if *key == DEFAULT_KEYBINDINGS.confirm.key {
|
if matches_key!(confirm, key) {
|
||||||
self.app.data.radarr_data.prompt_confirm = true;
|
self.app.data.radarr_data.prompt_confirm = true;
|
||||||
self.app.data.radarr_data.prompt_confirm_action = Some(RadarrEvent::ClearBlocklist);
|
self.app.data.radarr_data.prompt_confirm_action = Some(RadarrEvent::ClearBlocklist);
|
||||||
|
|
||||||
|
|||||||
@@ -1,35 +1,59 @@
|
|||||||
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
|
||||||
use crate::app::App;
|
use crate::app::App;
|
||||||
use crate::event::Key;
|
use crate::event::Key;
|
||||||
|
use crate::handlers::table_handler::TableHandlingConfig;
|
||||||
use crate::handlers::KeyEventHandler;
|
use crate::handlers::KeyEventHandler;
|
||||||
|
use crate::models::radarr_models::CollectionMovie;
|
||||||
use crate::models::servarr_data::radarr::radarr_data::{
|
use crate::models::servarr_data::radarr::radarr_data::{
|
||||||
ActiveRadarrBlock, ADD_MOVIE_SELECTION_BLOCKS, COLLECTION_DETAILS_BLOCKS,
|
ActiveRadarrBlock, ADD_MOVIE_SELECTION_BLOCKS, COLLECTION_DETAILS_BLOCKS,
|
||||||
EDIT_COLLECTION_SELECTION_BLOCKS,
|
EDIT_COLLECTION_SELECTION_BLOCKS,
|
||||||
};
|
};
|
||||||
use crate::models::stateful_table::StatefulTable;
|
use crate::models::stateful_table::StatefulTable;
|
||||||
use crate::models::{BlockSelectionState, Scrollable};
|
use crate::models::BlockSelectionState;
|
||||||
|
use crate::{handle_table_events, matches_key};
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[path = "collection_details_handler_tests.rs"]
|
#[path = "collection_details_handler_tests.rs"]
|
||||||
mod collection_details_handler_tests;
|
mod collection_details_handler_tests;
|
||||||
|
|
||||||
pub(super) struct CollectionDetailsHandler<'a, 'b> {
|
pub(super) struct CollectionDetailsHandler<'a, 'b> {
|
||||||
key: &'a Key,
|
key: Key,
|
||||||
app: &'a mut App<'b>,
|
app: &'a mut App<'b>,
|
||||||
active_radarr_block: &'a ActiveRadarrBlock,
|
active_radarr_block: ActiveRadarrBlock,
|
||||||
_context: &'a Option<ActiveRadarrBlock>,
|
_context: Option<ActiveRadarrBlock>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl CollectionDetailsHandler<'_, '_> {
|
||||||
|
handle_table_events!(
|
||||||
|
self,
|
||||||
|
collection_movies,
|
||||||
|
self.app.data.radarr_data.collection_movies,
|
||||||
|
CollectionMovie
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for CollectionDetailsHandler<'a, 'b> {
|
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for CollectionDetailsHandler<'a, 'b> {
|
||||||
fn accepts(active_block: &'a ActiveRadarrBlock) -> bool {
|
fn handle(&mut self) {
|
||||||
COLLECTION_DETAILS_BLOCKS.contains(active_block)
|
let collection_movies_table_handling_config =
|
||||||
|
TableHandlingConfig::new(ActiveRadarrBlock::CollectionDetails.into());
|
||||||
|
|
||||||
|
if !self.handle_collection_movies_table_events(collection_movies_table_handling_config) {
|
||||||
|
self.handle_key_event();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn with(
|
fn accepts(active_block: ActiveRadarrBlock) -> bool {
|
||||||
key: &'a Key,
|
COLLECTION_DETAILS_BLOCKS.contains(&active_block)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn ignore_alt_navigation(&self) -> bool {
|
||||||
|
self.app.should_ignore_quit_key
|
||||||
|
}
|
||||||
|
|
||||||
|
fn new(
|
||||||
|
key: Key,
|
||||||
app: &'a mut App<'b>,
|
app: &'a mut App<'b>,
|
||||||
active_block: &'a ActiveRadarrBlock,
|
active_block: ActiveRadarrBlock,
|
||||||
_context: &'a Option<ActiveRadarrBlock>,
|
_context: Option<ActiveRadarrBlock>,
|
||||||
) -> CollectionDetailsHandler<'a, 'b> {
|
) -> CollectionDetailsHandler<'a, 'b> {
|
||||||
CollectionDetailsHandler {
|
CollectionDetailsHandler {
|
||||||
key,
|
key,
|
||||||
@@ -39,7 +63,7 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for CollectionDetailsHan
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_key(&self) -> &Key {
|
fn get_key(&self) -> Key {
|
||||||
self.key
|
self.key
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -47,41 +71,20 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for CollectionDetailsHan
|
|||||||
!self.app.is_loading && !self.app.data.radarr_data.collection_movies.is_empty()
|
!self.app.is_loading && !self.app.data.radarr_data.collection_movies.is_empty()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn handle_scroll_up(&mut self) {
|
fn handle_scroll_up(&mut self) {}
|
||||||
if ActiveRadarrBlock::CollectionDetails == *self.active_radarr_block {
|
|
||||||
self.app.data.radarr_data.collection_movies.scroll_up()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_scroll_down(&mut self) {
|
fn handle_scroll_down(&mut self) {}
|
||||||
if ActiveRadarrBlock::CollectionDetails == *self.active_radarr_block {
|
|
||||||
self.app.data.radarr_data.collection_movies.scroll_down()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_home(&mut self) {
|
fn handle_home(&mut self) {}
|
||||||
if ActiveRadarrBlock::CollectionDetails == *self.active_radarr_block {
|
|
||||||
self.app.data.radarr_data.collection_movies.scroll_to_top();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_end(&mut self) {
|
fn handle_end(&mut self) {}
|
||||||
if ActiveRadarrBlock::CollectionDetails == *self.active_radarr_block {
|
|
||||||
self
|
|
||||||
.app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.collection_movies
|
|
||||||
.scroll_to_bottom();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn handle_delete(&mut self) {}
|
fn handle_delete(&mut self) {}
|
||||||
|
|
||||||
fn handle_left_right_action(&mut self) {}
|
fn handle_left_right_action(&mut self) {}
|
||||||
|
|
||||||
fn handle_submit(&mut self) {
|
fn handle_submit(&mut self) {
|
||||||
if ActiveRadarrBlock::CollectionDetails == *self.active_radarr_block {
|
if ActiveRadarrBlock::CollectionDetails == self.active_radarr_block {
|
||||||
let tmdb_id = self
|
let tmdb_id = self
|
||||||
.app
|
.app
|
||||||
.data
|
.data
|
||||||
@@ -111,7 +114,7 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for CollectionDetailsHan
|
|||||||
.into(),
|
.into(),
|
||||||
);
|
);
|
||||||
self.app.data.radarr_data.selected_block =
|
self.app.data.radarr_data.selected_block =
|
||||||
BlockSelectionState::new(&ADD_MOVIE_SELECTION_BLOCKS);
|
BlockSelectionState::new(ADD_MOVIE_SELECTION_BLOCKS);
|
||||||
self.app.data.radarr_data.add_movie_modal = Some((&self.app.data.radarr_data).into());
|
self.app.data.radarr_data.add_movie_modal = Some((&self.app.data.radarr_data).into());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -129,19 +132,19 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for CollectionDetailsHan
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn handle_char_key_event(&mut self) {
|
fn handle_char_key_event(&mut self) {
|
||||||
if *self.active_radarr_block == ActiveRadarrBlock::CollectionDetails
|
if self.active_radarr_block == ActiveRadarrBlock::CollectionDetails
|
||||||
&& *self.key == DEFAULT_KEYBINDINGS.edit.key
|
&& matches_key!(edit, self.key)
|
||||||
{
|
{
|
||||||
self.app.push_navigation_stack(
|
self.app.push_navigation_stack(
|
||||||
(
|
(
|
||||||
ActiveRadarrBlock::EditCollectionPrompt,
|
ActiveRadarrBlock::EditCollectionPrompt,
|
||||||
Some(*self.active_radarr_block),
|
Some(self.active_radarr_block),
|
||||||
)
|
)
|
||||||
.into(),
|
.into(),
|
||||||
);
|
);
|
||||||
self.app.data.radarr_data.edit_collection_modal = Some((&self.app.data.radarr_data).into());
|
self.app.data.radarr_data.edit_collection_modal = Some((&self.app.data.radarr_data).into());
|
||||||
self.app.data.radarr_data.selected_block =
|
self.app.data.radarr_data.selected_block =
|
||||||
BlockSelectionState::new(&EDIT_COLLECTION_SELECTION_BLOCKS);
|
BlockSelectionState::new(EDIT_COLLECTION_SELECTION_BLOCKS);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use pretty_assertions::assert_str_eq;
|
use pretty_assertions::assert_str_eq;
|
||||||
|
use rstest::rstest;
|
||||||
use strum::IntoEnumIterator;
|
use strum::IntoEnumIterator;
|
||||||
|
|
||||||
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
||||||
@@ -12,142 +13,6 @@ mod tests {
|
|||||||
use crate::models::servarr_data::radarr::radarr_data::{
|
use crate::models::servarr_data::radarr::radarr_data::{
|
||||||
ActiveRadarrBlock, COLLECTION_DETAILS_BLOCKS,
|
ActiveRadarrBlock, COLLECTION_DETAILS_BLOCKS,
|
||||||
};
|
};
|
||||||
use crate::models::HorizontallyScrollableText;
|
|
||||||
|
|
||||||
mod test_handle_scroll_up_and_down {
|
|
||||||
use rstest::rstest;
|
|
||||||
|
|
||||||
use crate::{simple_stateful_iterable_vec, test_iterable_scroll};
|
|
||||||
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
test_iterable_scroll!(
|
|
||||||
test_collection_details_scroll,
|
|
||||||
CollectionDetailsHandler,
|
|
||||||
collection_movies,
|
|
||||||
simple_stateful_iterable_vec!(CollectionMovie, HorizontallyScrollableText),
|
|
||||||
ActiveRadarrBlock::CollectionDetails,
|
|
||||||
None,
|
|
||||||
title,
|
|
||||||
to_string
|
|
||||||
);
|
|
||||||
|
|
||||||
#[rstest]
|
|
||||||
fn test_collection_details_scroll_no_op_when_not_ready(
|
|
||||||
#[values(
|
|
||||||
DEFAULT_KEYBINDINGS.up.key, DEFAULT_KEYBINDINGS.down.key
|
|
||||||
)]
|
|
||||||
key: Key,
|
|
||||||
) {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.is_loading = true;
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.collection_movies
|
|
||||||
.set_items(simple_stateful_iterable_vec!(
|
|
||||||
CollectionMovie,
|
|
||||||
HorizontallyScrollableText
|
|
||||||
));
|
|
||||||
|
|
||||||
CollectionDetailsHandler::with(&key, &mut app, &ActiveRadarrBlock::CollectionDetails, &None)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.collection_movies
|
|
||||||
.current_selection()
|
|
||||||
.title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
|
|
||||||
CollectionDetailsHandler::with(&key, &mut app, &ActiveRadarrBlock::CollectionDetails, &None)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.collection_movies
|
|
||||||
.current_selection()
|
|
||||||
.title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
mod test_handle_home_end {
|
|
||||||
use crate::{extended_stateful_iterable_vec, test_iterable_home_and_end};
|
|
||||||
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
test_iterable_home_and_end!(
|
|
||||||
test_collection_details_home_end,
|
|
||||||
CollectionDetailsHandler,
|
|
||||||
collection_movies,
|
|
||||||
extended_stateful_iterable_vec!(CollectionMovie, HorizontallyScrollableText),
|
|
||||||
ActiveRadarrBlock::CollectionDetails,
|
|
||||||
None,
|
|
||||||
title,
|
|
||||||
to_string
|
|
||||||
);
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_collection_details_home_end_no_op_when_not_ready() {
|
|
||||||
let mut app = App::default();
|
|
||||||
app.is_loading = true;
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.collection_movies
|
|
||||||
.set_items(extended_stateful_iterable_vec!(
|
|
||||||
CollectionMovie,
|
|
||||||
HorizontallyScrollableText
|
|
||||||
));
|
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.end.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.collection_movies
|
|
||||||
.current_selection()
|
|
||||||
.title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
|
||||||
&DEFAULT_KEYBINDINGS.home.key,
|
|
||||||
&mut app,
|
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
|
||||||
&None,
|
|
||||||
)
|
|
||||||
.handle();
|
|
||||||
|
|
||||||
assert_str_eq!(
|
|
||||||
app
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.collection_movies
|
|
||||||
.current_selection()
|
|
||||||
.title
|
|
||||||
.to_string(),
|
|
||||||
"Test 1"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
mod test_handle_submit {
|
mod test_handle_submit {
|
||||||
use bimap::BiMap;
|
use bimap::BiMap;
|
||||||
@@ -163,7 +28,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_collection_details_submit() {
|
fn test_collection_details_submit() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.radarr_data
|
||||||
@@ -171,24 +36,24 @@ mod tests {
|
|||||||
.set_items(vec![CollectionMovie::default()]);
|
.set_items(vec![CollectionMovie::default()]);
|
||||||
app.data.radarr_data.quality_profile_map =
|
app.data.radarr_data.quality_profile_map =
|
||||||
BiMap::from_iter([(1, "B - Test 2".to_owned()), (0, "A - Test 1".to_owned())]);
|
BiMap::from_iter([(1, "B - Test 2".to_owned()), (0, "A - Test 1".to_owned())]);
|
||||||
app.data.radarr_data.selected_block = BlockSelectionState::new(&ADD_MOVIE_SELECTION_BLOCKS);
|
app.data.radarr_data.selected_block = BlockSelectionState::new(ADD_MOVIE_SELECTION_BLOCKS);
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.radarr_data
|
||||||
.selected_block
|
.selected_block
|
||||||
.set_index(ADD_MOVIE_SELECTION_BLOCKS.len() - 1);
|
.set_index(0, ADD_MOVIE_SELECTION_BLOCKS.len() - 1);
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
CollectionDetailsHandler::new(
|
||||||
&SUBMIT_KEY,
|
SUBMIT_KEY,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&(
|
(
|
||||||
ActiveRadarrBlock::AddMoviePrompt,
|
ActiveRadarrBlock::AddMoviePrompt,
|
||||||
Some(ActiveRadarrBlock::CollectionDetails)
|
Some(ActiveRadarrBlock::CollectionDetails)
|
||||||
)
|
)
|
||||||
@@ -205,7 +70,7 @@ mod tests {
|
|||||||
.is_empty());
|
.is_empty());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.data.radarr_data.selected_block.get_active_block(),
|
app.data.radarr_data.selected_block.get_active_block(),
|
||||||
&ActiveRadarrBlock::AddMovieSelectRootFolder
|
ActiveRadarrBlock::AddMovieSelectRootFolder
|
||||||
);
|
);
|
||||||
assert!(!app
|
assert!(!app
|
||||||
.data
|
.data
|
||||||
@@ -240,7 +105,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_collection_details_submit_no_op_when_not_ready() {
|
fn test_collection_details_submit_no_op_when_not_ready() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
||||||
app
|
app
|
||||||
@@ -249,24 +114,24 @@ mod tests {
|
|||||||
.collection_movies
|
.collection_movies
|
||||||
.set_items(vec![CollectionMovie::default()]);
|
.set_items(vec![CollectionMovie::default()]);
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
CollectionDetailsHandler::new(
|
||||||
&SUBMIT_KEY,
|
SUBMIT_KEY,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::CollectionDetails.into()
|
ActiveRadarrBlock::CollectionDetails.into()
|
||||||
);
|
);
|
||||||
assert!(app.data.radarr_data.add_movie_modal.is_none());
|
assert!(app.data.radarr_data.add_movie_modal.is_none());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_collection_details_submit_movie_already_in_library() {
|
fn test_collection_details_submit_movie_already_in_library() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
.radarr_data
|
.radarr_data
|
||||||
@@ -278,17 +143,17 @@ mod tests {
|
|||||||
.movies
|
.movies
|
||||||
.set_items(vec![Movie::default()]);
|
.set_items(vec![Movie::default()]);
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
CollectionDetailsHandler::new(
|
||||||
&SUBMIT_KEY,
|
SUBMIT_KEY,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::ViewMovieOverview.into()
|
ActiveRadarrBlock::ViewMovieOverview.into()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -302,7 +167,7 @@ mod tests {
|
|||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn test_esc_collection_details(#[values(true, false)] is_ready: bool) {
|
fn test_esc_collection_details(#[values(true, false)] is_ready: bool) {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = is_ready;
|
app.is_loading = is_ready;
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::Collections.into());
|
app.push_navigation_stack(ActiveRadarrBlock::Collections.into());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
||||||
@@ -312,38 +177,38 @@ mod tests {
|
|||||||
.collection_movies
|
.collection_movies
|
||||||
.set_items(vec![CollectionMovie::default()]);
|
.set_items(vec![CollectionMovie::default()]);
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
CollectionDetailsHandler::new(
|
||||||
&ESC_KEY,
|
ESC_KEY,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::Collections.into()
|
ActiveRadarrBlock::Collections.into()
|
||||||
);
|
);
|
||||||
assert!(app.data.radarr_data.collection_movies.items.is_empty());
|
assert!(app.data.radarr_data.collection_movies.items.is_empty());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_esc_view_movie_overview() {
|
fn test_esc_view_movie_overview() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::ViewMovieOverview.into());
|
app.push_navigation_stack(ActiveRadarrBlock::ViewMovieOverview.into());
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
CollectionDetailsHandler::new(
|
||||||
&ESC_KEY,
|
ESC_KEY,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::ViewMovieOverview,
|
ActiveRadarrBlock::ViewMovieOverview,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::CollectionDetails.into()
|
ActiveRadarrBlock::CollectionDetails.into()
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -367,13 +232,13 @@ mod tests {
|
|||||||
test_edit_collection_key!(
|
test_edit_collection_key!(
|
||||||
CollectionDetailsHandler,
|
CollectionDetailsHandler,
|
||||||
ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
ActiveRadarrBlock::CollectionDetails
|
Some(ActiveRadarrBlock::CollectionDetails)
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_edit_key_no_op_when_not_ready() {
|
fn test_edit_key_no_op_when_not_ready() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
app.push_navigation_stack(ActiveRadarrBlock::CollectionDetails.into());
|
||||||
let mut radarr_data = create_test_radarr_data();
|
let mut radarr_data = create_test_radarr_data();
|
||||||
@@ -387,17 +252,17 @@ mod tests {
|
|||||||
}]);
|
}]);
|
||||||
app.data.radarr_data = radarr_data;
|
app.data.radarr_data = radarr_data;
|
||||||
|
|
||||||
CollectionDetailsHandler::with(
|
CollectionDetailsHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.edit.key,
|
DEFAULT_KEYBINDINGS.edit.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
)
|
)
|
||||||
.handle();
|
.handle();
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
&ActiveRadarrBlock::CollectionDetails.into()
|
ActiveRadarrBlock::CollectionDetails.into()
|
||||||
);
|
);
|
||||||
assert!(app.data.radarr_data.edit_collection_modal.is_none());
|
assert!(app.data.radarr_data.edit_collection_modal.is_none());
|
||||||
}
|
}
|
||||||
@@ -407,23 +272,39 @@ mod tests {
|
|||||||
fn test_collection_details_handler_accepts() {
|
fn test_collection_details_handler_accepts() {
|
||||||
ActiveRadarrBlock::iter().for_each(|active_radarr_block| {
|
ActiveRadarrBlock::iter().for_each(|active_radarr_block| {
|
||||||
if COLLECTION_DETAILS_BLOCKS.contains(&active_radarr_block) {
|
if COLLECTION_DETAILS_BLOCKS.contains(&active_radarr_block) {
|
||||||
assert!(CollectionDetailsHandler::accepts(&active_radarr_block));
|
assert!(CollectionDetailsHandler::accepts(active_radarr_block));
|
||||||
} else {
|
} else {
|
||||||
assert!(!CollectionDetailsHandler::accepts(&active_radarr_block));
|
assert!(!CollectionDetailsHandler::accepts(active_radarr_block));
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn test_collection_details_handler_ignore_alt_navigation(
|
||||||
|
#[values(true, false)] should_ignore_quit_key: bool,
|
||||||
|
) {
|
||||||
|
let mut app = App::test_default();
|
||||||
|
app.should_ignore_quit_key = should_ignore_quit_key;
|
||||||
|
let handler = CollectionDetailsHandler::new(
|
||||||
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
|
&mut app,
|
||||||
|
ActiveRadarrBlock::default(),
|
||||||
|
None,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(handler.ignore_alt_navigation(), should_ignore_quit_key);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_collection_details_handler_not_ready_when_loading() {
|
fn test_collection_details_handler_not_ready_when_loading() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = true;
|
app.is_loading = true;
|
||||||
|
|
||||||
let handler = CollectionDetailsHandler::with(
|
let handler = CollectionDetailsHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.esc.key,
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
);
|
);
|
||||||
|
|
||||||
assert!(!handler.is_ready());
|
assert!(!handler.is_ready());
|
||||||
@@ -431,14 +312,14 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_collection_details_handler_not_ready_when_collection_movies_is_empty() {
|
fn test_collection_details_handler_not_ready_when_collection_movies_is_empty() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = false;
|
app.is_loading = false;
|
||||||
|
|
||||||
let handler = CollectionDetailsHandler::with(
|
let handler = CollectionDetailsHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.esc.key,
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
);
|
);
|
||||||
|
|
||||||
assert!(!handler.is_ready());
|
assert!(!handler.is_ready());
|
||||||
@@ -446,7 +327,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_collection_details_handler_ready_when_not_loading_and_collection_movies_is_not_empty() {
|
fn test_collection_details_handler_ready_when_not_loading_and_collection_movies_is_not_empty() {
|
||||||
let mut app = App::default();
|
let mut app = App::test_default();
|
||||||
app.is_loading = false;
|
app.is_loading = false;
|
||||||
app
|
app
|
||||||
.data
|
.data
|
||||||
@@ -454,11 +335,11 @@ mod tests {
|
|||||||
.collection_movies
|
.collection_movies
|
||||||
.set_items(vec![CollectionMovie::default()]);
|
.set_items(vec![CollectionMovie::default()]);
|
||||||
|
|
||||||
let handler = CollectionDetailsHandler::with(
|
let handler = CollectionDetailsHandler::new(
|
||||||
&DEFAULT_KEYBINDINGS.esc.key,
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
&mut app,
|
&mut app,
|
||||||
&ActiveRadarrBlock::CollectionDetails,
|
ActiveRadarrBlock::CollectionDetails,
|
||||||
&None,
|
None,
|
||||||
);
|
);
|
||||||
|
|
||||||
assert!(handler.is_ready());
|
assert!(handler.is_ready());
|
||||||
|
|||||||