Compare commits
1 Commits
syncope
...
openapi-do
Author | SHA1 | Date | |
---|---|---|---|
361c683b5f
|
1444
Cargo.lock
generated
1444
Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
25
Cargo.toml
25
Cargo.toml
@ -9,29 +9,34 @@ readme = "README.md"
|
|||||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
regex = "1.5.4"
|
aide = { version = "0.13.4", features = [
|
||||||
lazy_static = "1.4.0"
|
"axum",
|
||||||
url = "2.2.2"
|
"axum-extra",
|
||||||
swayipc = "3.0"
|
"axum-extra-query",
|
||||||
|
"axum-extra-form",
|
||||||
|
"axum-ws",
|
||||||
|
"macros",
|
||||||
|
"scalar",
|
||||||
|
] }
|
||||||
anyhow = "1.0.82"
|
anyhow = "1.0.82"
|
||||||
axum = { version = "0.7.7", features = ["macros", "ws"] }
|
axum = { version = "0.6.20", features = ["macros", "ws"] }
|
||||||
|
axum-jsonschema = { version = "0.8.0", features = ["aide"] }
|
||||||
|
axum-macros = "0.4.1"
|
||||||
clap = { version = "4.4.1", features = ["derive"] }
|
clap = { version = "4.4.1", features = ["derive"] }
|
||||||
clap-verbosity-flag = "2.2.2"
|
clap-verbosity-flag = "2.2.2"
|
||||||
env_logger = "0.10.0"
|
env_logger = "0.10.0"
|
||||||
futures = "0.3.31"
|
|
||||||
log = "0.4.20"
|
log = "0.4.20"
|
||||||
mpvipc-async = { git = "https://git.pvv.ntnu.no/Grzegorz/mpvipc-async.git", branch = "main" }
|
mpvipc-async = { git = "https://git.pvv.ntnu.no/oysteikt/mpvipc-async.git", rev = "v0.1.0" }
|
||||||
sd-notify = "0.4.3"
|
sd-notify = "0.4.3"
|
||||||
|
schemars = "0.8.16"
|
||||||
serde = { version = "1.0.188", features = ["derive"] }
|
serde = { version = "1.0.188", features = ["derive"] }
|
||||||
serde_json = "1.0.105"
|
serde_json = "1.0.105"
|
||||||
systemd-journal-logger = "2.2.0"
|
systemd-journal-logger = "2.2.0"
|
||||||
tempfile = "3.11.0"
|
tempfile = "3.11.0"
|
||||||
|
serde_urlencoded = "0.7.1"
|
||||||
tokio = { version = "1.32.0", features = ["full"] }
|
tokio = { version = "1.32.0", features = ["full"] }
|
||||||
tower = { version = "0.4.13", features = ["full"] }
|
tower = { version = "0.4.13", features = ["full"] }
|
||||||
tower-http = { version = "0.4.3", features = ["full"] }
|
tower-http = { version = "0.4.3", features = ["full"] }
|
||||||
utoipa = { version = "5.1.3", features = ["axum_extras"] }
|
|
||||||
utoipa-axum = "0.1.2"
|
|
||||||
utoipa-swagger-ui = { version = "8.0.3", features = ["axum", "vendored"] }
|
|
||||||
|
|
||||||
[profile.release]
|
[profile.release]
|
||||||
strip = true
|
strip = true
|
||||||
|
17
README.md
17
README.md
@ -6,29 +6,16 @@ New implementation of https://github.com/Programvareverkstedet/grzegorz
|
|||||||
|
|
||||||
```sh
|
```sh
|
||||||
# NixOS
|
# NixOS
|
||||||
nix run "git+https://git.pvv.ntnu.no/Grzegorz/greg-ng#" -- --mpv-socket-path /tmp/mpv.sock
|
nix run "git+https://git.pvv.ntnu.no/Projects/greg-ng#" -- --mpv-socket-path /tmp/mpv.sock
|
||||||
|
|
||||||
# Other (after git clone and rust toolchain has been set up)
|
# Other (after git clone and rust toolchain has been set up)
|
||||||
cargo run -- --mpv-socket-path /tmp/mpv.sock
|
cargo run -- --mpv-socket-path /tmp/mpv.sock
|
||||||
```
|
```
|
||||||
|
|
||||||
See also https://git.pvv.ntnu.no/Grzegorz/grzegorz-clients for frontend alternatives
|
See also https://git.pvv.ntnu.no/Projects/grzegorz-clients for frontend alternatives
|
||||||
|
|
||||||
## Debugging
|
## Debugging
|
||||||
|
|
||||||
```sh
|
```sh
|
||||||
RUST_LOG=greg_ng=trace,mpvipc=trace cargo run -- --mpv-socket-path /tmp/mpv.sock
|
RUST_LOG=greg_ng=trace,mpvipc=trace cargo run -- --mpv-socket-path /tmp/mpv.sock
|
||||||
```
|
```
|
||||||
|
|
||||||
Custom api call in curl examples
|
|
||||||
|
|
||||||
LOL with input command. (utilizing ydotools)
|
|
||||||
```sh
|
|
||||||
curl -X POST -H "Content-Type: application/json" -d '{"keys": "42:1 38:1 38:0 24:1 24:0 38:1 38:0 42:0"}' http://localhost:8008/api/v2/sway/input/keys
|
|
||||||
```
|
|
||||||
|
|
||||||
Launching DEFAULT_BROWSER with url, in --kiosk mode
|
|
||||||
```sh
|
|
||||||
curl -X POST -H "Content-Type: application/json" -d '{"url": "https://www.youtube.com/watch?v=dQw4w9WgXcQ"}' http://localhost:8008/api/v2/sway/browser/launch
|
|
||||||
```
|
|
||||||
|
|
||||||
|
@ -22,7 +22,6 @@ rustPlatform.buildRustPackage rec {
|
|||||||
])
|
])
|
||||||
(type == "regular" && lib.elem baseName [
|
(type == "regular" && lib.elem baseName [
|
||||||
"flake.nix"
|
"flake.nix"
|
||||||
"flake.lock"
|
|
||||||
"default.nix"
|
"default.nix"
|
||||||
"module.nix"
|
"module.nix"
|
||||||
".envrc"
|
".envrc"
|
||||||
@ -34,7 +33,7 @@ rustPlatform.buildRustPackage rec {
|
|||||||
cargoLock = {
|
cargoLock = {
|
||||||
lockFile = ./Cargo.lock;
|
lockFile = ./Cargo.lock;
|
||||||
outputHashes = {
|
outputHashes = {
|
||||||
"mpvipc-async-0.1.0" = "sha256-V22wdnVVCBzayqkwb2d0msG7YypVss0cGBihtXrHtuM=";
|
"mpvipc-async-0.1.0" = "sha256-2TQ2d4q9/DTxTZe9kOAoKBhsmegRZw32x3G2hbluS6U=";
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -38,9 +38,7 @@
|
|||||||
package = self.packages.${system}.greg-ng-wrapped;
|
package = self.packages.${system}.greg-ng-wrapped;
|
||||||
in {
|
in {
|
||||||
type = "app";
|
type = "app";
|
||||||
program = toString (pkgs.writeShellScript "greg-ng" ''
|
program = lib.getExe package;
|
||||||
${lib.getExe package} --mpv-socket-path /tmp/greg-ng-mpv.sock -vvvv
|
|
||||||
'');
|
|
||||||
};
|
};
|
||||||
});
|
});
|
||||||
|
|
||||||
|
34
module.nix
34
module.nix
@ -11,10 +11,6 @@ in
|
|||||||
mpvPackage = lib.mkPackageOption pkgs "mpv" { };
|
mpvPackage = lib.mkPackageOption pkgs "mpv" { };
|
||||||
|
|
||||||
enableSway = lib.mkEnableOption "sway as the main window manager";
|
enableSway = lib.mkEnableOption "sway as the main window manager";
|
||||||
|
|
||||||
enableFirefox = lib.mkEnableOption "include fiorefox browser for external urls";
|
|
||||||
|
|
||||||
enableInput = lib.mkEnableOption "enable input maipulation with ydotool,";
|
|
||||||
|
|
||||||
enablePipewire = lib.mkEnableOption "pipewire" // { default = true; };
|
enablePipewire = lib.mkEnableOption "pipewire" // { default = true; };
|
||||||
|
|
||||||
@ -139,20 +135,18 @@ in
|
|||||||
ProtectKernelModules = true;
|
ProtectKernelModules = true;
|
||||||
ProtectKernelTunables = true;
|
ProtectKernelTunables = true;
|
||||||
ProtectProc = "invisible";
|
ProtectProc = "invisible";
|
||||||
# I'll figure it out sometime
|
ProtectSystem = "full";
|
||||||
# ProtectSystem = "full";
|
|
||||||
RemoveIPC = true;
|
RemoveIPC = true;
|
||||||
UMask = "0077";
|
UMask = "0077";
|
||||||
RestrictNamespaces = true;
|
RestrictNamespaces = true;
|
||||||
RestrictRealtime = true;
|
RestrictRealtime = true;
|
||||||
RestrictSUIDSGID = true;
|
RestrictSUIDSGID = true;
|
||||||
SystemCallArchitectures = "native";
|
SystemCallArchitectures = "native";
|
||||||
# Something brokey
|
SystemCallFilter = [
|
||||||
# SystemCallFilter = [
|
"@system-service"
|
||||||
# "@system-service"
|
"~@privileged"
|
||||||
# "~@privileged"
|
"~@resources"
|
||||||
# "~@resources"
|
];
|
||||||
# ];
|
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
})
|
})
|
||||||
@ -164,27 +158,12 @@ in
|
|||||||
pulse.enable = true;
|
pulse.enable = true;
|
||||||
};
|
};
|
||||||
})
|
})
|
||||||
(lib.mkIf (cfg.enable && cfg.enableInput) {
|
|
||||||
programs.ydotool = {
|
|
||||||
enable = true;
|
|
||||||
};
|
|
||||||
})
|
|
||||||
(lib.mkIf (cfg.enable && cfg.enableSway) {
|
(lib.mkIf (cfg.enable && cfg.enableSway) {
|
||||||
programs.sway = {
|
programs.sway = {
|
||||||
enable = true;
|
enable = true;
|
||||||
wrapperFeatures.gtk = true;
|
wrapperFeatures.gtk = true;
|
||||||
};
|
};
|
||||||
|
|
||||||
(lib.mkIf (cfg.enableFirefox) {
|
|
||||||
programs.firefox = {
|
|
||||||
enable = true;
|
|
||||||
preferences = {
|
|
||||||
media.autoplay.default = "0";
|
|
||||||
};
|
|
||||||
};
|
|
||||||
environment.sessionVariables.DEFAULT_BROWSER = "${pkgs.firefox}/bin/firefox";
|
|
||||||
})
|
|
||||||
|
|
||||||
xdg.portal = {
|
xdg.portal = {
|
||||||
enable = true;
|
enable = true;
|
||||||
wlr.enable = true;
|
wlr.enable = true;
|
||||||
@ -195,7 +174,6 @@ in
|
|||||||
users.greg = {
|
users.greg = {
|
||||||
isNormalUser = true;
|
isNormalUser = true;
|
||||||
group = "greg";
|
group = "greg";
|
||||||
extraGroups [ "audio" "video" "input" "ydotool" ];
|
|
||||||
uid = 2000;
|
uid = 2000;
|
||||||
description = "loud gym bro";
|
description = "loud gym bro";
|
||||||
};
|
};
|
||||||
|
@ -1,8 +1,4 @@
|
|||||||
mod base;
|
mod base;
|
||||||
// mod rest_wrapper_v1;
|
mod rest_wrapper_v1;
|
||||||
mod rest_wrapper_v2;
|
|
||||||
mod websocket_v1;
|
|
||||||
|
|
||||||
// pub use rest_wrapper_v1::{rest_api_docs as rest_api_docs_v1, rest_api_routes as rest_api_routes_v1};
|
pub use rest_wrapper_v1::rest_api_routes;
|
||||||
pub use rest_wrapper_v2::{rest_api_docs as rest_api_docs_v2, rest_api_routes as rest_api_routes_v2};
|
|
||||||
pub use websocket_v1::websocket_api;
|
|
||||||
|
244
src/api/base.rs
244
src/api/base.rs
@ -170,247 +170,3 @@ pub async fn playlist_set_looping(mpv: Mpv, r#loop: bool) -> anyhow::Result<()>
|
|||||||
.await
|
.await
|
||||||
.map_err(|e| e.into())
|
.map_err(|e| e.into())
|
||||||
}
|
}
|
||||||
|
|
||||||
// pub async fn sway_run_command(command: String) -> anyhow::Result<()> {
|
|
||||||
// tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
// let mut connection = swayipc::Connection::new()?;
|
|
||||||
// connection.run_command(&command)?;
|
|
||||||
// Ok(())
|
|
||||||
// })
|
|
||||||
// .await
|
|
||||||
// .map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
// }
|
|
||||||
|
|
||||||
//only to check if workspace exists.
|
|
||||||
fn get_workspace_names(connection: &mut swayipc::Connection) -> anyhow::Result<Vec<String>> {
|
|
||||||
let workspaces = connection.get_workspaces()?;
|
|
||||||
Ok(workspaces.iter().map(|w| w.name.clone()).collect())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn sway_get_workspace_names() -> anyhow::Result<Vec<String>> {
|
|
||||||
tokio::task::spawn_blocking(|| -> anyhow::Result<Vec<String>> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
get_workspace_names(&mut connection)
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
||||||
|
|
||||||
fn is_valid_workspace(workspace: &str, connection: &mut swayipc::Connection) -> anyhow::Result<bool> {
|
|
||||||
let workspace_names = get_workspace_names(connection)?;
|
|
||||||
Ok(workspace_names.contains(&workspace.to_string()) ||
|
|
||||||
workspace.parse::<u32>()
|
|
||||||
.map(|num| num >= 1 && num <= 10)
|
|
||||||
.unwrap_or(false))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn sway_change_workspace(workspace: String) -> anyhow::Result<()> {
|
|
||||||
tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
|
|
||||||
if !is_valid_workspace(&workspace, &mut connection)? {
|
|
||||||
anyhow::bail!("Invalid workspace name");
|
|
||||||
}
|
|
||||||
|
|
||||||
connection.run_command(&format!("workspace {}", workspace))?;
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
||||||
|
|
||||||
use url::Url;
|
|
||||||
|
|
||||||
pub async fn sway_launch_browser(url: &str) -> anyhow::Result<()> {
|
|
||||||
// Validate URL
|
|
||||||
let url = Url::parse(url)
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(format!("Invalid URL: {}", e)))?;
|
|
||||||
|
|
||||||
// Ensure URL scheme is http or https
|
|
||||||
if url.scheme() != "http" && url.scheme() != "https" {
|
|
||||||
anyhow::bail!("URL must use http or https protocol");
|
|
||||||
}
|
|
||||||
|
|
||||||
tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
// connection.run_command(&format!("exec xdg-open {}", url))?;
|
|
||||||
// connection.run_command(&format!("exec firefox --kiosk {}", url))?; //moved to firefox to pin in kiosk mode. potentially add --new-window
|
|
||||||
|
|
||||||
//get the DEFAULT_BROWSER env var
|
|
||||||
let default_browser = std::env::var("DEFAULT_BROWSER").unwrap_or("firefox".to_string());
|
|
||||||
connection.run_command(&format!("exec {} --kiosk {}", default_browser, url))?; // set default browser in kiosk mode
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn sway_close_workspace(workspace: String) -> anyhow::Result<()> {
|
|
||||||
tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
|
|
||||||
// Validate workspace exists
|
|
||||||
if !is_valid_workspace(&workspace, &mut connection)? {
|
|
||||||
anyhow::bail!("Invalid workspace name");
|
|
||||||
}
|
|
||||||
|
|
||||||
// // Get workspace tree and find all nodes in target workspace
|
|
||||||
// let tree = connection.get_tree()?;
|
|
||||||
// let workspace_nodes = tree
|
|
||||||
// .nodes
|
|
||||||
// .iter()
|
|
||||||
// .flat_map(|output| &output.nodes) // Get workspaces
|
|
||||||
// .find(|ws| ws.name.as_ref().map_or(false, |name| name == &workspace));
|
|
||||||
|
|
||||||
// // Kill all nodes in workspace if found
|
|
||||||
// if let Some(ws) = workspace_nodes {
|
|
||||||
// for container in ws.nodes.iter() {
|
|
||||||
// // Close each container in the workspace
|
|
||||||
// connection.run_command(&format!("[con_id={}] kill", container.id))?;
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
// Get workspace tree and find all nodes in target workspace
|
|
||||||
let tree = connection.get_tree()?;
|
|
||||||
let workspace_nodes = tree
|
|
||||||
.nodes
|
|
||||||
.iter()
|
|
||||||
.flat_map(|output| &output.nodes) // Get workspaces
|
|
||||||
.find(|ws| ws.name.as_ref().map_or(false, |name| name == &workspace));
|
|
||||||
|
|
||||||
// Kill all non-MPV nodes in workspace if found
|
|
||||||
if let Some(ws) = workspace_nodes {
|
|
||||||
for container in ws.nodes.iter() {
|
|
||||||
// Check if window is MPV
|
|
||||||
let is_mpv = container.window_properties.as_ref()
|
|
||||||
.map(|props| props.class.as_ref()
|
|
||||||
.map_or(false, |class| class == "mpv"))
|
|
||||||
.unwrap_or(false);
|
|
||||||
|
|
||||||
if !is_mpv {
|
|
||||||
// Close container only if not MPV
|
|
||||||
connection.run_command(&format!("[con_id={}] kill", container.id))?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
use regex::Regex;
|
|
||||||
use lazy_static::lazy_static;
|
|
||||||
|
|
||||||
lazy_static! {
|
|
||||||
static ref KEYPRESS_PATTERN: Regex = Regex::new(r"^(\d+:[01]\s*)+$").unwrap();
|
|
||||||
static ref CLEANUP_PATTERN: Regex = Regex::new(r"[^0-9: \t]").unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
fn validate_keypress_string(input: &str) -> anyhow::Result<String> {
|
|
||||||
let cleaned = CLEANUP_PATTERN.replace_all(input, "").to_string();
|
|
||||||
let cleaned = cleaned.trim();
|
|
||||||
|
|
||||||
if !KEYPRESS_PATTERN.is_match(cleaned) {
|
|
||||||
anyhow::bail!("Invalid keypress string");
|
|
||||||
}
|
|
||||||
Ok(cleaned.to_string())
|
|
||||||
}
|
|
||||||
|
|
||||||
//to simulate keypresses 42:1 38:1 38:0 24:1 24:0 38:1 38:0 42:0 -> LOL
|
|
||||||
pub async fn input(keys: String) -> anyhow::Result<()> {
|
|
||||||
let validated_input = validate_keypress_string(&keys)?;
|
|
||||||
|
|
||||||
tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
connection.run_command(&format!("exec ydotool key {}", validated_input))?;
|
|
||||||
// instead of running through swaycmf
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
||||||
|
|
||||||
// simulate mouse movement
|
|
||||||
pub async fn mouse_move(x: i32, y: i32) -> anyhow::Result<()> {
|
|
||||||
tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
connection.run_command(&format!("exec ydotool mousemove -x {} -y {}", x, y))?;
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
//simulate scroll
|
|
||||||
pub async fn mouse_scroll(x: i32, y: i32) -> anyhow::Result<()> {
|
|
||||||
tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
connection.run_command(&format!("exec ydotool mousemove -w -x {} -y {}", x, y))?;
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub enum MouseButton {
|
|
||||||
Left,
|
|
||||||
Right,
|
|
||||||
Middle,
|
|
||||||
Side,
|
|
||||||
Extra,
|
|
||||||
Forward,
|
|
||||||
Back,
|
|
||||||
Task,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl MouseButton {
|
|
||||||
fn to_base_value(&self) -> u8 {
|
|
||||||
match self {
|
|
||||||
MouseButton::Left => 0x00,
|
|
||||||
MouseButton::Right => 0x01,
|
|
||||||
MouseButton::Middle => 0x02,
|
|
||||||
MouseButton::Side => 0x03,
|
|
||||||
MouseButton::Extra => 0x04,
|
|
||||||
MouseButton::Forward => 0x05,
|
|
||||||
MouseButton::Back => 0x06,
|
|
||||||
MouseButton::Task => 0x07,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn from_str(s: &str) -> anyhow::Result<Self> {
|
|
||||||
match s.to_uppercase().as_str() {
|
|
||||||
"LEFT" => Ok(MouseButton::Left),
|
|
||||||
"RIGHT" => Ok(MouseButton::Right),
|
|
||||||
"MIDDLE" => Ok(MouseButton::Middle),
|
|
||||||
"SIDE" => Ok(MouseButton::Side),
|
|
||||||
"EXTRA" => Ok(MouseButton::Extra),
|
|
||||||
"FORWARD" => Ok(MouseButton::Forward),
|
|
||||||
"BACK" => Ok(MouseButton::Back),
|
|
||||||
"TASK" => Ok(MouseButton::Task),
|
|
||||||
_ => anyhow::bail!("Invalid mouse button"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn click_value(&self) -> u8 {
|
|
||||||
self.to_base_value() | 0xC0 // Combines DOWN (0x40) and UP (0x80)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn mouse_click(button: String) -> anyhow::Result<()> {
|
|
||||||
let mouse_button = MouseButton::from_str(&button)?;
|
|
||||||
let click_value = mouse_button.click_value();
|
|
||||||
|
|
||||||
tokio::task::spawn_blocking(move || -> anyhow::Result<()> {
|
|
||||||
let mut connection = swayipc::Connection::new()?;
|
|
||||||
connection.run_command(&format!("exec ydotool click {:#04x}", click_value))?;
|
|
||||||
Ok(())
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
.map_err(|e| swayipc::Error::CommandFailed(e.to_string()))?
|
|
||||||
}
|
|
@ -1,17 +1,16 @@
|
|||||||
|
use std::ops::Deref;
|
||||||
|
|
||||||
|
use aide::{axum::IntoApiResponse, operation::OperationIo, OperationOutput};
|
||||||
|
use axum_jsonschema::JsonSchemaRejection;
|
||||||
|
|
||||||
use axum::{
|
use axum::{
|
||||||
extract::{Query, State},
|
async_trait, extract::{rejection::{FailedToDeserializeQueryString, QueryRejection}, FromRequest, FromRequestParts, State}, http::{request::Parts, StatusCode}, response::{IntoResponse, Response}, routing::{delete, get, post}, Json, Router
|
||||||
http::StatusCode,
|
|
||||||
response::{IntoResponse, Response},
|
|
||||||
routing::{delete, get, post},
|
|
||||||
Json, Router,
|
|
||||||
};
|
};
|
||||||
use mpvipc_async::Mpv;
|
use mpvipc_async::Mpv;
|
||||||
|
use schemars::JsonSchema;
|
||||||
|
use serde::{de::DeserializeOwned, Serialize};
|
||||||
use serde_json::{json, Value};
|
use serde_json::{json, Value};
|
||||||
|
|
||||||
use utoipa::OpenApi;
|
|
||||||
use utoipa_axum::{router::OpenApiRouter, routes};
|
|
||||||
use utoipa_swagger_ui::SwaggerUi;
|
|
||||||
|
|
||||||
use super::base;
|
use super::base;
|
||||||
|
|
||||||
pub fn rest_api_routes(mpv: Mpv) -> Router {
|
pub fn rest_api_routes(mpv: Mpv) -> Router {
|
||||||
@ -35,64 +34,13 @@ pub fn rest_api_routes(mpv: Mpv) -> Router {
|
|||||||
.with_state(mpv)
|
.with_state(mpv)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn rest_api_docs(mpv: Mpv) -> Router {
|
// #[derive(FromRequest, OperationIo)]
|
||||||
let (router, api) = OpenApiRouter::with_openapi(ApiDoc::openapi())
|
// #[from_request(via(axum_jsonschema::Json), rejection(RestResponse))]
|
||||||
.routes(routes!(loadfile))
|
// #[aide(
|
||||||
.routes(routes!(play_get, play_set))
|
// input_with = "axum_jsonschema::Json<T>",
|
||||||
.routes(routes!(volume_get, volume_set))
|
// output_with = "axum_jsonschema::Json<T>",
|
||||||
.routes(routes!(time_get, time_set))
|
// json_schema
|
||||||
.routes(routes!(playlist_get, playlist_remove_or_clear))
|
// )]
|
||||||
.routes(routes!(playlist_next))
|
|
||||||
.routes(routes!(playlist_previous))
|
|
||||||
.routes(routes!(playlist_goto))
|
|
||||||
.routes(routes!(playlist_move))
|
|
||||||
.routes(routes!(playlist_get_looping, playlist_set_looping))
|
|
||||||
.routes(routes!(shuffle))
|
|
||||||
.with_state(mpv)
|
|
||||||
.split_for_parts();
|
|
||||||
|
|
||||||
router.merge(SwaggerUi::new("/docs/v1").url("/docs/v1/openapi.json", api))
|
|
||||||
}
|
|
||||||
|
|
||||||
// NOTE: the openapi stuff is very heavily duplicated and introduces
|
|
||||||
// a lot of maintenance overhead and boilerplate. It should theoretically
|
|
||||||
// be possible to infer a lot of this from axum, but I haven't found a
|
|
||||||
// good library that does this and works properly yet (I have tried some
|
|
||||||
// but they all had issues). Feel free to replace this with a better solution.
|
|
||||||
|
|
||||||
#[derive(OpenApi)]
|
|
||||||
#[openapi(info(
|
|
||||||
description = "The legacy Grzegorz Brzeczyszczykiewicz API, used to control a running mpv instance",
|
|
||||||
version = "1.0.0",
|
|
||||||
))]
|
|
||||||
struct ApiDoc;
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, utoipa::ToSchema)]
|
|
||||||
struct EmptySuccessResponse {
|
|
||||||
success: bool,
|
|
||||||
error: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, utoipa::ToSchema)]
|
|
||||||
struct SuccessResponse {
|
|
||||||
#[schema(example = true)]
|
|
||||||
success: bool,
|
|
||||||
#[schema(example = false)]
|
|
||||||
error: bool,
|
|
||||||
#[schema(example = json!({ some: "arbitrary json value" }))]
|
|
||||||
value: Value,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, utoipa::ToSchema)]
|
|
||||||
struct ErrorResponse {
|
|
||||||
#[schema(example = "error....")]
|
|
||||||
error: String,
|
|
||||||
#[schema(example = "error....")]
|
|
||||||
errortext: String,
|
|
||||||
#[schema(example = false)]
|
|
||||||
success: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct RestResponse(anyhow::Result<Value>);
|
pub struct RestResponse(anyhow::Result<Value>);
|
||||||
|
|
||||||
impl From<anyhow::Result<Value>> for RestResponse {
|
impl From<anyhow::Result<Value>> for RestResponse {
|
||||||
@ -120,183 +68,169 @@ impl IntoResponse for RestResponse {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl aide::OperationOutput for RestResponse {
|
||||||
|
type Inner = anyhow::Result<Value>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// -------
|
||||||
|
|
||||||
|
// impl<T> aide::OperationInput for Query<T> {}
|
||||||
|
|
||||||
|
// #[derive(FromRequest, OperationIo)]
|
||||||
|
// #[from_request(via(axum_jsonschema::Json), rejection(RestResponse))]
|
||||||
|
// #[aide(
|
||||||
|
// input_with = "axum_jsonschema::Json<T>",
|
||||||
|
// output_with = "axum_jsonschema::Json<T>",
|
||||||
|
// json_schema
|
||||||
|
// )]
|
||||||
|
// pub struct Json<T>(pub T);
|
||||||
|
|
||||||
|
// impl<T> IntoResponse for Json<T>
|
||||||
|
// where
|
||||||
|
// T: Serialize,
|
||||||
|
// {
|
||||||
|
// fn into_response(self) -> axum::response::Response {
|
||||||
|
// axum::Json(self.0).into_response()
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
|
#[derive(OperationIo)]
|
||||||
|
#[aide(json_schema)]
|
||||||
|
pub struct Query<T>(pub T);
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl <T, S> FromRequestParts<S> for Query<T>
|
||||||
|
where
|
||||||
|
T: JsonSchema + DeserializeOwned,
|
||||||
|
S: Send + Sync,
|
||||||
|
{
|
||||||
|
type Rejection = QueryRejection;
|
||||||
|
|
||||||
|
async fn from_request_parts(parts: &mut Parts, _state: &S) -> Result<Self, Self::Rejection> {
|
||||||
|
let axum::extract::Query(query) = axum::extract::Query::try_from_uri(&parts.uri)?;
|
||||||
|
Ok(Query(query))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T> Deref for Query<T> {
|
||||||
|
type Target = T;
|
||||||
|
|
||||||
|
fn deref(&self) -> &Self::Target {
|
||||||
|
&self.0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn rest_api_route_docs(mpv: Mpv) -> Router {
|
||||||
|
use aide::axum::ApiRouter;
|
||||||
|
use aide::axum::routing::{delete, get, post};
|
||||||
|
|
||||||
|
let mut api = aide::openapi::OpenApi::default();
|
||||||
|
|
||||||
|
let x = ApiRouter::new()
|
||||||
|
// .api_route("/load", get(loadfile))
|
||||||
|
.api_route("/play", get(play_get))
|
||||||
|
.finish_api(&mut api);
|
||||||
|
// .with_state(mpv);
|
||||||
|
|
||||||
|
todo!()
|
||||||
|
}
|
||||||
|
|
||||||
|
// ----------
|
||||||
|
|
||||||
|
pub fn rest_api_routes(mpv: Mpv) -> Router {
|
||||||
|
Router::new()
|
||||||
|
.route("/load", post(loadfile))
|
||||||
|
.route("/play", get(play_get))
|
||||||
|
.route("/play", post(play_set))
|
||||||
|
.route("/volume", get(volume_get))
|
||||||
|
.route("/volume", post(volume_set))
|
||||||
|
.route("/time", get(time_get))
|
||||||
|
.route("/time", post(time_set))
|
||||||
|
.route("/playlist", get(playlist_get))
|
||||||
|
.route("/playlist/next", post(playlist_next))
|
||||||
|
.route("/playlist/previous", post(playlist_previous))
|
||||||
|
.route("/playlist/goto", post(playlist_goto))
|
||||||
|
.route("/playlist", delete(playlist_remove_or_clear))
|
||||||
|
.route("/playlist/move", post(playlist_move))
|
||||||
|
.route("/playlist/shuffle", post(shuffle))
|
||||||
|
.route("/playlist/loop", get(playlist_get_looping))
|
||||||
|
.route("/playlist/loop", post(playlist_set_looping))
|
||||||
|
.with_state(mpv)
|
||||||
|
}
|
||||||
|
|
||||||
// -------------------//
|
// -------------------//
|
||||||
// Boilerplate galore //
|
// Boilerplate galore //
|
||||||
// -------------------//
|
// -------------------//
|
||||||
|
|
||||||
// TODO: These could possibly be generated with a proc macro
|
// TODO: These could possibly be generated with a proc macro
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct LoadFileArgs {
|
struct LoadFileArgs {
|
||||||
path: String,
|
path: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Add item to playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/load",
|
|
||||||
params(LoadFileArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn loadfile(State(mpv): State<Mpv>, Query(query): Query<LoadFileArgs>) -> RestResponse {
|
async fn loadfile(State(mpv): State<Mpv>, Query(query): Query<LoadFileArgs>) -> RestResponse {
|
||||||
base::loadfile(mpv, &query.path).await.into()
|
base::loadfile(mpv, &query.path).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Check whether the player is paused or playing
|
async fn play_get(State(mpv): State<Mpv>) -> impl IntoApiResponse {
|
||||||
#[utoipa::path(
|
RestResponse::from(base::play_get(mpv).await)
|
||||||
get,
|
|
||||||
path = "/play",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn play_get(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::play_get(mpv).await.into()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct PlaySetArgs {
|
struct PlaySetArgs {
|
||||||
play: String,
|
play: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set whether the player is paused or playing
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/play",
|
|
||||||
params(PlaySetArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn play_set(State(mpv): State<Mpv>, Query(query): Query<PlaySetArgs>) -> RestResponse {
|
async fn play_set(State(mpv): State<Mpv>, Query(query): Query<PlaySetArgs>) -> RestResponse {
|
||||||
let play = query.play.to_lowercase() == "true";
|
let play = query.play.to_lowercase() == "true";
|
||||||
base::play_set(mpv, play).await.into()
|
base::play_set(mpv, play).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the current player volume
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/volume",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn volume_get(State(mpv): State<Mpv>) -> RestResponse {
|
async fn volume_get(State(mpv): State<Mpv>) -> RestResponse {
|
||||||
base::volume_get(mpv).await.into()
|
base::volume_get(mpv).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct VolumeSetArgs {
|
struct VolumeSetArgs {
|
||||||
volume: f64,
|
volume: f64,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set the player volume
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/volume",
|
|
||||||
params(VolumeSetArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn volume_set(State(mpv): State<Mpv>, Query(query): Query<VolumeSetArgs>) -> RestResponse {
|
async fn volume_set(State(mpv): State<Mpv>, Query(query): Query<VolumeSetArgs>) -> RestResponse {
|
||||||
base::volume_set(mpv, query.volume).await.into()
|
base::volume_set(mpv, query.volume).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get current playback position
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/time",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn time_get(State(mpv): State<Mpv>) -> RestResponse {
|
async fn time_get(State(mpv): State<Mpv>) -> RestResponse {
|
||||||
base::time_get(mpv).await.into()
|
base::time_get(mpv).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct TimeSetArgs {
|
struct TimeSetArgs {
|
||||||
pos: Option<f64>,
|
pos: Option<f64>,
|
||||||
percent: Option<f64>,
|
percent: Option<f64>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set playback position
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/time",
|
|
||||||
params(TimeSetArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn time_set(State(mpv): State<Mpv>, Query(query): Query<TimeSetArgs>) -> RestResponse {
|
async fn time_set(State(mpv): State<Mpv>, Query(query): Query<TimeSetArgs>) -> RestResponse {
|
||||||
base::time_set(mpv, query.pos, query.percent).await.into()
|
base::time_set(mpv, query.pos, query.percent).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Get the current playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/playlist",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_get(State(mpv): State<Mpv>) -> RestResponse {
|
async fn playlist_get(State(mpv): State<Mpv>) -> RestResponse {
|
||||||
base::playlist_get(mpv).await.into()
|
base::playlist_get(mpv).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Go to the next item in the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/next",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_next(State(mpv): State<Mpv>) -> RestResponse {
|
async fn playlist_next(State(mpv): State<Mpv>) -> RestResponse {
|
||||||
base::playlist_next(mpv).await.into()
|
base::playlist_next(mpv).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Go back to the previous item in the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/previous",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_previous(State(mpv): State<Mpv>) -> RestResponse {
|
async fn playlist_previous(State(mpv): State<Mpv>) -> RestResponse {
|
||||||
base::playlist_previous(mpv).await.into()
|
base::playlist_previous(mpv).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct PlaylistGotoArgs {
|
struct PlaylistGotoArgs {
|
||||||
index: usize,
|
index: usize,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Go to a specific item in the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/goto",
|
|
||||||
params(PlaylistGotoArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_goto(
|
async fn playlist_goto(
|
||||||
State(mpv): State<Mpv>,
|
State(mpv): State<Mpv>,
|
||||||
Query(query): Query<PlaylistGotoArgs>,
|
Query(query): Query<PlaylistGotoArgs>,
|
||||||
@ -304,21 +238,11 @@ async fn playlist_goto(
|
|||||||
base::playlist_goto(mpv, query.index).await.into()
|
base::playlist_goto(mpv, query.index).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct PlaylistRemoveOrClearArgs {
|
struct PlaylistRemoveOrClearArgs {
|
||||||
index: Option<usize>,
|
index: Option<usize>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Clears a single item or the entire playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
delete,
|
|
||||||
path = "/playlist",
|
|
||||||
params(PlaylistRemoveOrClearArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_remove_or_clear(
|
async fn playlist_remove_or_clear(
|
||||||
State(mpv): State<Mpv>,
|
State(mpv): State<Mpv>,
|
||||||
Query(query): Query<PlaylistRemoveOrClearArgs>,
|
Query(query): Query<PlaylistRemoveOrClearArgs>,
|
||||||
@ -329,22 +253,12 @@ async fn playlist_remove_or_clear(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct PlaylistMoveArgs {
|
struct PlaylistMoveArgs {
|
||||||
index1: usize,
|
index1: usize,
|
||||||
index2: usize,
|
index2: usize,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Move a playlist item to a different position
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/move",
|
|
||||||
params(PlaylistMoveArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_move(
|
async fn playlist_move(
|
||||||
State(mpv): State<Mpv>,
|
State(mpv): State<Mpv>,
|
||||||
Query(query): Query<PlaylistMoveArgs>,
|
Query(query): Query<PlaylistMoveArgs>,
|
||||||
@ -354,47 +268,19 @@ async fn playlist_move(
|
|||||||
.into()
|
.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Shuffle the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/shuffle",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn shuffle(State(mpv): State<Mpv>) -> RestResponse {
|
async fn shuffle(State(mpv): State<Mpv>) -> RestResponse {
|
||||||
base::shuffle(mpv).await.into()
|
base::shuffle(mpv).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Check whether the playlist is looping
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/playlist/loop",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_get_looping(State(mpv): State<Mpv>) -> RestResponse {
|
async fn playlist_get_looping(State(mpv): State<Mpv>) -> RestResponse {
|
||||||
base::playlist_get_looping(mpv).await.into()
|
base::playlist_get_looping(mpv).await.into()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
#[derive(serde::Deserialize, JsonSchema)]
|
||||||
struct PlaylistSetLoopingArgs {
|
struct PlaylistSetLoopingArgs {
|
||||||
r#loop: bool,
|
r#loop: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Set whether the playlist should loop
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/loop",
|
|
||||||
params(PlaylistSetLoopingArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_set_looping(
|
async fn playlist_set_looping(
|
||||||
State(mpv): State<Mpv>,
|
State(mpv): State<Mpv>,
|
||||||
Query(query): Query<PlaylistSetLoopingArgs>,
|
Query(query): Query<PlaylistSetLoopingArgs>,
|
||||||
|
@ -1,590 +0,0 @@
|
|||||||
use axum::{
|
|
||||||
extract::{Query, State},
|
|
||||||
http::StatusCode,
|
|
||||||
response::{IntoResponse, Response},
|
|
||||||
routing::{delete, get, post},
|
|
||||||
Json, Router,
|
|
||||||
};
|
|
||||||
use mpvipc_async::Mpv;
|
|
||||||
use serde_json::{json, Value};
|
|
||||||
|
|
||||||
use utoipa::OpenApi;
|
|
||||||
use utoipa_axum::{router::OpenApiRouter, routes};
|
|
||||||
use utoipa_swagger_ui::SwaggerUi;
|
|
||||||
///use futures::FutureExt;
|
|
||||||
|
|
||||||
use super::base;
|
|
||||||
|
|
||||||
pub fn rest_api_routes(mpv: Mpv) -> Router {
|
|
||||||
Router::new()
|
|
||||||
.route("/load", post(loadfile))
|
|
||||||
.route("/play", get(play_get))
|
|
||||||
.route("/play", post(play_set))
|
|
||||||
.route("/volume", get(volume_get))
|
|
||||||
.route("/volume", post(volume_set))
|
|
||||||
.route("/time", get(time_get))
|
|
||||||
.route("/time", post(time_set))
|
|
||||||
.route("/playlist", get(playlist_get))
|
|
||||||
.route("/playlist/next", post(playlist_next))
|
|
||||||
.route("/playlist/previous", post(playlist_previous))
|
|
||||||
.route("/playlist/goto", post(playlist_goto))
|
|
||||||
.route("/playlist", delete(playlist_remove_or_clear))
|
|
||||||
.route("/playlist/move", post(playlist_move))
|
|
||||||
.route("/playlist/shuffle", post(shuffle))
|
|
||||||
.route("/playlist/loop", get(playlist_get_looping))
|
|
||||||
.route("/playlist/loop", post(playlist_set_looping))
|
|
||||||
// .route("/sway/command", post(sway_run_command))
|
|
||||||
.route("/sway/workspace/close", post(sway_close_workspace_handler))
|
|
||||||
.route("/sway/workspace/change", post(sway_change_workspace_handler))
|
|
||||||
.route("/sway/workspace/list", get(sway_get_workspace_names_handler))
|
|
||||||
.route("/sway/browser/launch", post(sway_launch_browser_handler))
|
|
||||||
.route("/input/keys", post(input_handler))
|
|
||||||
.route("/input/mouse", post(mouse_move_handler))
|
|
||||||
.route("/input/scroll", post(mouse_scroll_handler))
|
|
||||||
.route("/input/click", post(mouse_click_handler))
|
|
||||||
.with_state(mpv)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn rest_api_docs(mpv: Mpv) -> Router {
|
|
||||||
let (router, api) = OpenApiRouter::with_openapi(ApiDoc::openapi())
|
|
||||||
.routes(routes!(loadfile))
|
|
||||||
.routes(routes!(play_get, play_set))
|
|
||||||
.routes(routes!(volume_get, volume_set))
|
|
||||||
.routes(routes!(time_get, time_set))
|
|
||||||
.routes(routes!(playlist_get, playlist_remove_or_clear))
|
|
||||||
.routes(routes!(playlist_next))
|
|
||||||
.routes(routes!(playlist_previous))
|
|
||||||
.routes(routes!(playlist_goto))
|
|
||||||
.routes(routes!(playlist_move))
|
|
||||||
.routes(routes!(playlist_get_looping, playlist_set_looping))
|
|
||||||
.routes(routes!(shuffle))
|
|
||||||
.with_state(mpv)
|
|
||||||
.split_for_parts();
|
|
||||||
|
|
||||||
router.merge(SwaggerUi::new("/docs/v2").url("/docs/v2/openapi.json", api))
|
|
||||||
}
|
|
||||||
|
|
||||||
// NOTE: the openapi stuff is very heavily duplicated and introduces
|
|
||||||
// a lot of maintenance overhead and boilerplate. It should theoretically
|
|
||||||
// be possible to infer a lot of this from axum, but I haven't found a
|
|
||||||
// good library that does this and works properly yet (I have tried some
|
|
||||||
// but they all had issues). Feel free to replace this with a better solution.
|
|
||||||
|
|
||||||
#[derive(OpenApi)]
|
|
||||||
#[openapi(info(
|
|
||||||
description = "The legacy Grzegorz Brzeczyszczykiewicz API, used to control a running mpv instance",
|
|
||||||
version = "1.0.0",
|
|
||||||
))]
|
|
||||||
struct ApiDoc;
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, utoipa::ToSchema)]
|
|
||||||
struct EmptySuccessResponse {
|
|
||||||
success: bool,
|
|
||||||
error: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, utoipa::ToSchema)]
|
|
||||||
struct SuccessResponse {
|
|
||||||
#[schema(example = true)]
|
|
||||||
success: bool,
|
|
||||||
#[schema(example = false)]
|
|
||||||
error: bool,
|
|
||||||
#[schema(example = json!({ some: "arbitrary json value" }))]
|
|
||||||
value: Value,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Serialize, utoipa::ToSchema)]
|
|
||||||
struct ErrorResponse {
|
|
||||||
#[schema(example = "error....")]
|
|
||||||
error: String,
|
|
||||||
#[schema(example = "error....")]
|
|
||||||
errortext: String,
|
|
||||||
#[schema(example = false)]
|
|
||||||
success: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct RestResponse(anyhow::Result<Value>);
|
|
||||||
|
|
||||||
impl From<anyhow::Result<Value>> for RestResponse {
|
|
||||||
fn from(result: anyhow::Result<Value>) -> Self {
|
|
||||||
Self(result.map(|value| json!({ "success": true, "error": false, "value": value })))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<anyhow::Result<()>> for RestResponse {
|
|
||||||
fn from(result: anyhow::Result<()>) -> Self {
|
|
||||||
Self(result.map(|_| json!({ "success": true, "error": false })))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl IntoResponse for RestResponse {
|
|
||||||
fn into_response(self) -> Response {
|
|
||||||
match self.0 {
|
|
||||||
Ok(value) => (StatusCode::OK, Json(value)).into_response(),
|
|
||||||
Err(err) => (
|
|
||||||
StatusCode::INTERNAL_SERVER_ERROR,
|
|
||||||
Json(json!({ "error": err.to_string(), "errortext": err.to_string(), "success": false })),
|
|
||||||
)
|
|
||||||
.into_response(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// -------------------//
|
|
||||||
// Boilerplate galore //
|
|
||||||
// -------------------//
|
|
||||||
|
|
||||||
// TODO: These could possibly be generated with a proc macro
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct LoadFileArgs {
|
|
||||||
path: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Add item to playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/load",
|
|
||||||
params(LoadFileArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn loadfile(State(mpv): State<Mpv>, Query(query): Query<LoadFileArgs>) -> RestResponse {
|
|
||||||
base::loadfile(mpv, &query.path).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Check whether the player is paused or playing
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/play",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn play_get(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::play_get(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct PlaySetArgs {
|
|
||||||
play: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set whether the player is paused or playing
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/play",
|
|
||||||
params(PlaySetArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn play_set(State(mpv): State<Mpv>, Query(query): Query<PlaySetArgs>) -> RestResponse {
|
|
||||||
let play = query.play.to_lowercase() == "true";
|
|
||||||
base::play_set(mpv, play).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Get the current player volume
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/volume",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn volume_get(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::volume_get(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct VolumeSetArgs {
|
|
||||||
volume: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set the player volume
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/volume",
|
|
||||||
params(VolumeSetArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn volume_set(State(mpv): State<Mpv>, Query(query): Query<VolumeSetArgs>) -> RestResponse {
|
|
||||||
base::volume_set(mpv, query.volume).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Get current playback position
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/time",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn time_get(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::time_get(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct TimeSetArgs {
|
|
||||||
pos: Option<f64>,
|
|
||||||
percent: Option<f64>,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set playback position
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/time",
|
|
||||||
params(TimeSetArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn time_set(State(mpv): State<Mpv>, Query(query): Query<TimeSetArgs>) -> RestResponse {
|
|
||||||
base::time_set(mpv, query.pos, query.percent).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Get the current playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/playlist",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_get(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::playlist_get(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Go to the next item in the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/next",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_next(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::playlist_next(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Go back to the previous item in the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/previous",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_previous(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::playlist_previous(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct PlaylistGotoArgs {
|
|
||||||
index: usize,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Go to a specific item in the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/goto",
|
|
||||||
params(PlaylistGotoArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_goto(
|
|
||||||
State(mpv): State<Mpv>,
|
|
||||||
Query(query): Query<PlaylistGotoArgs>,
|
|
||||||
) -> RestResponse {
|
|
||||||
base::playlist_goto(mpv, query.index).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct PlaylistRemoveOrClearArgs {
|
|
||||||
index: Option<usize>,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Clears a single item or the entire playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
delete,
|
|
||||||
path = "/playlist",
|
|
||||||
params(PlaylistRemoveOrClearArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_remove_or_clear(
|
|
||||||
State(mpv): State<Mpv>,
|
|
||||||
Query(query): Query<PlaylistRemoveOrClearArgs>,
|
|
||||||
) -> RestResponse {
|
|
||||||
match query.index {
|
|
||||||
Some(index) => base::playlist_remove(mpv, index).await.into(),
|
|
||||||
None => base::playlist_clear(mpv).await.into(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct PlaylistMoveArgs {
|
|
||||||
index1: usize,
|
|
||||||
index2: usize,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Move a playlist item to a different position
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/move",
|
|
||||||
params(PlaylistMoveArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_move(
|
|
||||||
State(mpv): State<Mpv>,
|
|
||||||
Query(query): Query<PlaylistMoveArgs>,
|
|
||||||
) -> RestResponse {
|
|
||||||
base::playlist_move(mpv, query.index1, query.index2)
|
|
||||||
.await
|
|
||||||
.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Shuffle the playlist
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/shuffle",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn shuffle(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::shuffle(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Check whether the playlist is looping
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/playlist/loop",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = SuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_get_looping(State(mpv): State<Mpv>) -> RestResponse {
|
|
||||||
base::playlist_get_looping(mpv).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
||||||
struct PlaylistSetLoopingArgs {
|
|
||||||
r#loop: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Set whether the playlist should loop
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/playlist/loop",
|
|
||||||
params(PlaylistSetLoopingArgs),
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn playlist_set_looping(
|
|
||||||
State(mpv): State<Mpv>,
|
|
||||||
Query(query): Query<PlaylistSetLoopingArgs>,
|
|
||||||
) -> RestResponse {
|
|
||||||
base::playlist_set_looping(mpv, query.r#loop).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// #[derive(serde::Deserialize, utoipa::ToSchema)]
|
|
||||||
// struct SwayCommandBody {
|
|
||||||
// command: String,
|
|
||||||
// }
|
|
||||||
|
|
||||||
// #[utoipa::path(
|
|
||||||
// post,
|
|
||||||
// path = "/sway/command",
|
|
||||||
// request_body = SwayCommandBody,
|
|
||||||
// responses(
|
|
||||||
// (status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
// (status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
// )
|
|
||||||
// )]
|
|
||||||
// async fn sway_run_command(Json(body): Json<SwayCommandBody>) -> RestResponse {
|
|
||||||
// base::sway_run_command(body.command).await.map_err(anyhow::Error::new).into()
|
|
||||||
// }
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::ToSchema)]
|
|
||||||
struct SwayBrowserBody {
|
|
||||||
url: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/sway/browser/launch",
|
|
||||||
request_body = SwayBrowserBody,
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn sway_launch_browser_handler(Json(body): Json<SwayBrowserBody>) -> RestResponse {
|
|
||||||
base::sway_launch_browser(&body.url).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::ToSchema)]
|
|
||||||
struct SwayWorkspaceBody {
|
|
||||||
workspace: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/sway/workspace/close",
|
|
||||||
request_body = SwayWorkspaceBody,
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn sway_close_workspace_handler(Json(body): Json<SwayWorkspaceBody>) -> RestResponse {
|
|
||||||
base::sway_close_workspace(body.workspace).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/sway/workspace/change",
|
|
||||||
request_body = SwayWorkspaceBody,
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = EmptySuccessResponse),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn sway_change_workspace_handler(Json(body): Json<SwayWorkspaceBody>) -> RestResponse {
|
|
||||||
base::sway_change_workspace(body.workspace).await.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
get,
|
|
||||||
path = "/sway/workspace/list",
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = Vec<String>),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn sway_get_workspace_names_handler() -> RestResponse {
|
|
||||||
base::sway_get_workspace_names().await
|
|
||||||
.map(|workspaces| json!(workspaces))
|
|
||||||
.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::ToSchema)]
|
|
||||||
struct KeyboardInput {
|
|
||||||
keys: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/input/keys",
|
|
||||||
request_body = KeyboardInput,
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = Vec<String>),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn input_handler(
|
|
||||||
Json(payload): Json<KeyboardInput>
|
|
||||||
) -> RestResponse {
|
|
||||||
base::input(payload.keys)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}))
|
|
||||||
.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#[derive(serde::Deserialize, utoipa::ToSchema)]
|
|
||||||
struct MouseMove {
|
|
||||||
x: i32,
|
|
||||||
y: i32,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/input/mouse",
|
|
||||||
request_body = MouseMove,
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = Vec<String>),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn mouse_move_handler(
|
|
||||||
Json(payload): Json<MouseMove>
|
|
||||||
) -> RestResponse {
|
|
||||||
base::mouse_move(payload.x, payload.y)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}))
|
|
||||||
.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/input/scroll",
|
|
||||||
request_body = MouseMove,
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = Vec<String>),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn mouse_scroll_handler(
|
|
||||||
Json(payload): Json<MouseMove>
|
|
||||||
) -> RestResponse {
|
|
||||||
base::mouse_scroll(payload.x, payload.y)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}))
|
|
||||||
.into()
|
|
||||||
}
|
|
||||||
|
|
||||||
//click
|
|
||||||
#[derive(serde::Deserialize, utoipa::ToSchema)]
|
|
||||||
struct MouseClick {
|
|
||||||
button: String
|
|
||||||
}
|
|
||||||
|
|
||||||
#[utoipa::path(
|
|
||||||
post,
|
|
||||||
path = "/input/click",
|
|
||||||
request_body = MouseClick,
|
|
||||||
responses(
|
|
||||||
(status = 200, description = "Success", body = Vec<String>),
|
|
||||||
(status = 500, description = "Internal server error", body = ErrorResponse),
|
|
||||||
)
|
|
||||||
)]
|
|
||||||
async fn mouse_click_handler(
|
|
||||||
Json(payload): Json<MouseClick>
|
|
||||||
) -> RestResponse {
|
|
||||||
base::mouse_click(payload.button)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}))
|
|
||||||
.into()
|
|
||||||
}
|
|
@ -1,507 +0,0 @@
|
|||||||
use std::{
|
|
||||||
net::SocketAddr,
|
|
||||||
sync::{Arc, Mutex},
|
|
||||||
};
|
|
||||||
|
|
||||||
use anyhow::Context;
|
|
||||||
use futures::{stream::FuturesUnordered, StreamExt};
|
|
||||||
use serde::{Deserialize, Serialize};
|
|
||||||
|
|
||||||
use axum::{
|
|
||||||
extract::{
|
|
||||||
ws::{Message, WebSocket},
|
|
||||||
ConnectInfo, State, WebSocketUpgrade,
|
|
||||||
},
|
|
||||||
response::IntoResponse,
|
|
||||||
routing::any,
|
|
||||||
Router,
|
|
||||||
};
|
|
||||||
use mpvipc_async::{
|
|
||||||
LoopProperty, Mpv, MpvExt, NumberChangeOptions, Playlist, PlaylistAddTypeOptions, SeekOptions,
|
|
||||||
Switch,
|
|
||||||
};
|
|
||||||
use serde_json::{json, Value};
|
|
||||||
use tokio::{select, sync::watch};
|
|
||||||
|
|
||||||
use crate::util::IdPool;
|
|
||||||
|
|
||||||
use super::base;
|
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
struct WebsocketState {
|
|
||||||
mpv: Mpv,
|
|
||||||
id_pool: Arc<Mutex<IdPool>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn websocket_api(mpv: Mpv, id_pool: Arc<Mutex<IdPool>>) -> Router {
|
|
||||||
let state = WebsocketState { mpv, id_pool };
|
|
||||||
Router::new()
|
|
||||||
.route("/", any(websocket_handler))
|
|
||||||
.with_state(state)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn websocket_handler(
|
|
||||||
ws: WebSocketUpgrade,
|
|
||||||
ConnectInfo(addr): ConnectInfo<SocketAddr>,
|
|
||||||
State(WebsocketState { mpv, id_pool }): State<WebsocketState>,
|
|
||||||
) -> impl IntoResponse {
|
|
||||||
let mpv = mpv.clone();
|
|
||||||
let id = match id_pool.lock().unwrap().request_id() {
|
|
||||||
Ok(id) => id,
|
|
||||||
Err(e) => {
|
|
||||||
log::error!("Failed to get id from id pool: {:?}", e);
|
|
||||||
return axum::http::StatusCode::INTERNAL_SERVER_ERROR.into_response();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
ws.on_upgrade(move |socket| handle_connection(socket, addr, mpv, id, id_pool))
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
|
|
||||||
pub struct InitialState {
|
|
||||||
pub cached_timestamp: Option<f64>,
|
|
||||||
pub chapters: Vec<Value>,
|
|
||||||
pub connections: u64,
|
|
||||||
pub current_percent_pos: Option<f64>,
|
|
||||||
pub current_track: String,
|
|
||||||
pub duration: f64,
|
|
||||||
pub is_looping: bool,
|
|
||||||
pub is_muted: bool,
|
|
||||||
pub is_playing: bool,
|
|
||||||
pub is_paused_for_cache: bool,
|
|
||||||
pub playlist: Playlist,
|
|
||||||
pub tracks: Vec<Value>,
|
|
||||||
pub volume: f64,
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_initial_state(mpv: &Mpv, id_pool: Arc<Mutex<IdPool>>) -> InitialState {
|
|
||||||
let cached_timestamp = mpv
|
|
||||||
.get_property_value("demuxer-cache-state")
|
|
||||||
.await
|
|
||||||
.unwrap_or(None)
|
|
||||||
.and_then(|v| {
|
|
||||||
v.as_object()
|
|
||||||
.and_then(|o| o.get("data"))
|
|
||||||
.and_then(|v| v.as_object())
|
|
||||||
.and_then(|o| o.get("cache-end"))
|
|
||||||
.and_then(|v| v.as_f64())
|
|
||||||
});
|
|
||||||
let chapters = match mpv.get_property_value("chapter-list").await {
|
|
||||||
Ok(Some(Value::Array(chapters))) => chapters,
|
|
||||||
_ => vec![],
|
|
||||||
};
|
|
||||||
let connections = id_pool.lock().unwrap().id_count();
|
|
||||||
let current_percent_pos = mpv.get_property("percent-pos").await.unwrap_or(None);
|
|
||||||
let current_track = mpv.get_file_path().await.unwrap_or("".to_string());
|
|
||||||
let duration = mpv.get_duration().await.unwrap_or(0.0);
|
|
||||||
let is_looping =
|
|
||||||
mpv.playlist_is_looping().await.unwrap_or(LoopProperty::No) != LoopProperty::No;
|
|
||||||
let is_muted = mpv
|
|
||||||
.get_property("mute")
|
|
||||||
.await
|
|
||||||
.unwrap_or(Some(false))
|
|
||||||
.unwrap_or(false);
|
|
||||||
let is_playing = mpv.is_playing().await.unwrap_or(false);
|
|
||||||
let is_paused_for_cache = mpv
|
|
||||||
.get_property("paused-for-cache")
|
|
||||||
.await
|
|
||||||
.unwrap_or(Some(false))
|
|
||||||
.unwrap_or(false);
|
|
||||||
let playlist = mpv.get_playlist().await.unwrap_or(Playlist(vec![]));
|
|
||||||
let tracks = match mpv.get_property_value("track-list").await {
|
|
||||||
Ok(Some(Value::Array(tracks))) => tracks
|
|
||||||
.into_iter()
|
|
||||||
.filter(|t| {
|
|
||||||
t.as_object()
|
|
||||||
.and_then(|o| o.get("type"))
|
|
||||||
.and_then(|t| t.as_str())
|
|
||||||
.unwrap_or("")
|
|
||||||
== "sub"
|
|
||||||
})
|
|
||||||
.collect(),
|
|
||||||
_ => vec![],
|
|
||||||
};
|
|
||||||
let volume = mpv.get_volume().await.unwrap_or(0.0);
|
|
||||||
// TODO: use default when new version is released
|
|
||||||
InitialState {
|
|
||||||
cached_timestamp,
|
|
||||||
chapters,
|
|
||||||
connections,
|
|
||||||
current_percent_pos,
|
|
||||||
current_track,
|
|
||||||
duration,
|
|
||||||
is_looping,
|
|
||||||
is_muted,
|
|
||||||
is_playing,
|
|
||||||
is_paused_for_cache,
|
|
||||||
playlist,
|
|
||||||
tracks,
|
|
||||||
volume,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const DEFAULT_PROPERTY_SUBSCRIPTIONS: [&str; 11] = [
|
|
||||||
"chapter-list",
|
|
||||||
"demuxer-cache-state",
|
|
||||||
"duration",
|
|
||||||
"loop-playlist",
|
|
||||||
"mute",
|
|
||||||
"pause",
|
|
||||||
"paused-for-cache",
|
|
||||||
"percent-pos",
|
|
||||||
"playlist",
|
|
||||||
"track-list",
|
|
||||||
"volume",
|
|
||||||
];
|
|
||||||
|
|
||||||
async fn setup_default_subscribes(mpv: &Mpv) -> anyhow::Result<()> {
|
|
||||||
let mut futures = FuturesUnordered::new();
|
|
||||||
|
|
||||||
futures.extend(
|
|
||||||
DEFAULT_PROPERTY_SUBSCRIPTIONS
|
|
||||||
.iter()
|
|
||||||
.map(|property| mpv.observe_property(0, property)),
|
|
||||||
);
|
|
||||||
|
|
||||||
while let Some(result) = futures.next().await {
|
|
||||||
result?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn handle_connection(
|
|
||||||
mut socket: WebSocket,
|
|
||||||
addr: SocketAddr,
|
|
||||||
mpv: Mpv,
|
|
||||||
channel_id: u64,
|
|
||||||
id_pool: Arc<Mutex<IdPool>>,
|
|
||||||
) {
|
|
||||||
// TODO: There is an asynchronous gap between gathering the initial state and subscribing to the properties
|
|
||||||
// This could lead to missing events if they happen in that gap. Send initial state, but also ensure
|
|
||||||
// that there is an additional "initial state" sent upon subscription to all properties to ensure that
|
|
||||||
// the state is correct.
|
|
||||||
let initial_state = get_initial_state(&mpv, id_pool.clone()).await;
|
|
||||||
|
|
||||||
let message = Message::Text(
|
|
||||||
json!({
|
|
||||||
"type": "initial_state",
|
|
||||||
"value": initial_state,
|
|
||||||
})
|
|
||||||
.to_string(),
|
|
||||||
);
|
|
||||||
|
|
||||||
socket.send(message).await.unwrap();
|
|
||||||
|
|
||||||
setup_default_subscribes(&mpv).await.unwrap();
|
|
||||||
|
|
||||||
let id_count_watch_receiver = id_pool.lock().unwrap().get_id_count_watch_receiver();
|
|
||||||
|
|
||||||
let connection_loop_result = tokio::spawn(connection_loop(
|
|
||||||
socket,
|
|
||||||
addr,
|
|
||||||
mpv.clone(),
|
|
||||||
channel_id,
|
|
||||||
id_count_watch_receiver,
|
|
||||||
));
|
|
||||||
|
|
||||||
match connection_loop_result.await {
|
|
||||||
Ok(Ok(())) => {
|
|
||||||
log::trace!("Connection loop ended for {:?}", addr);
|
|
||||||
}
|
|
||||||
Ok(Err(e)) => {
|
|
||||||
log::error!("Error in connection loop for {:?}: {:?}", addr, e);
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
log::error!("Error in connection loop for {:?}: {:?}", addr, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
match mpv.unobserve_property(channel_id).await {
|
|
||||||
Ok(()) => {
|
|
||||||
log::trace!("Unsubscribed from properties for {:?}", addr);
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
log::error!(
|
|
||||||
"Error unsubscribing from properties for {:?}: {:?}",
|
|
||||||
addr,
|
|
||||||
e
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
match id_pool.lock().unwrap().release_id(channel_id) {
|
|
||||||
Ok(()) => {
|
|
||||||
log::trace!("Released id {} for {:?}", channel_id, addr);
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
log::error!("Error releasing id {} for {:?}: {:?}", channel_id, addr, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn connection_loop(
|
|
||||||
mut socket: WebSocket,
|
|
||||||
addr: SocketAddr,
|
|
||||||
mpv: Mpv,
|
|
||||||
channel_id: u64,
|
|
||||||
mut id_count_watch_receiver: watch::Receiver<u64>,
|
|
||||||
) -> Result<(), anyhow::Error> {
|
|
||||||
let mut event_stream = mpv.get_event_stream().await;
|
|
||||||
loop {
|
|
||||||
select! {
|
|
||||||
id_count = id_count_watch_receiver.changed() => {
|
|
||||||
if let Err(e) = id_count {
|
|
||||||
anyhow::bail!("Error reading id count watch receiver for {:?}: {:?}", addr, e);
|
|
||||||
}
|
|
||||||
|
|
||||||
let message = Message::Text(json!({
|
|
||||||
"type": "connection_count",
|
|
||||||
"value": id_count_watch_receiver.borrow().clone(),
|
|
||||||
}).to_string());
|
|
||||||
|
|
||||||
socket.send(message).await?;
|
|
||||||
}
|
|
||||||
message = socket.recv() => {
|
|
||||||
log::trace!("Received command from {:?}: {:?}", addr, message);
|
|
||||||
|
|
||||||
let ws_message_content = message
|
|
||||||
.ok_or(anyhow::anyhow!("Event stream ended for {:?}", addr))
|
|
||||||
.and_then(|message| {
|
|
||||||
match message {
|
|
||||||
Ok(message) => Ok(message),
|
|
||||||
err => Err(anyhow::anyhow!("Error reading message for {:?}: {:?}", addr, err)),
|
|
||||||
}
|
|
||||||
})?;
|
|
||||||
|
|
||||||
if let Message::Close(_) = ws_message_content {
|
|
||||||
log::trace!("Closing connection for {:?}", addr);
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Message::Ping(xs) = ws_message_content {
|
|
||||||
log::trace!("Ponging {:?} with {:?}", addr, xs);
|
|
||||||
socket.send(Message::Pong(xs)).await?;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let message_content = match ws_message_content {
|
|
||||||
Message::Text(text) => text,
|
|
||||||
m => anyhow::bail!("Unexpected message type: {:?}", m),
|
|
||||||
};
|
|
||||||
|
|
||||||
let message_json = match serde_json::from_str::<Value>(&message_content) {
|
|
||||||
Ok(json) => json,
|
|
||||||
Err(e) => anyhow::bail!("Error parsing message from {:?}: {:?}", addr, e),
|
|
||||||
};
|
|
||||||
|
|
||||||
log::trace!("Handling command from {:?}: {:?}", addr, message_json);
|
|
||||||
|
|
||||||
// TODO: handle errors
|
|
||||||
match handle_message(message_json, mpv.clone(), channel_id).await {
|
|
||||||
Ok(Some(response)) => {
|
|
||||||
log::trace!("Handled command from {:?} successfully, sending response", addr);
|
|
||||||
let message = Message::Text(json!({
|
|
||||||
"type": "response",
|
|
||||||
"value": response,
|
|
||||||
}).to_string());
|
|
||||||
socket.send(message).await?;
|
|
||||||
}
|
|
||||||
Ok(None) => {
|
|
||||||
log::trace!("Handled command from {:?} successfully", addr);
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
log::error!("Error handling message from {:?}: {:?}", addr, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
event = event_stream.next() => {
|
|
||||||
match event {
|
|
||||||
Some(Ok(event)) => {
|
|
||||||
log::trace!("Sending event to {:?}: {:?}", addr, event);
|
|
||||||
let message = Message::Text(json!({
|
|
||||||
"type": "event",
|
|
||||||
"value": event,
|
|
||||||
}).to_string());
|
|
||||||
socket.send(message).await?;
|
|
||||||
}
|
|
||||||
Some(Err(e)) => {
|
|
||||||
log::error!("Error reading event stream for {:?}: {:?}", addr, e);
|
|
||||||
anyhow::bail!("Error reading event stream for {:?}: {:?}", addr, e);
|
|
||||||
}
|
|
||||||
None => {
|
|
||||||
log::trace!("Event stream ended for {:?}", addr);
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
|
|
||||||
#[serde(tag = "type", rename_all = "snake_case")]
|
|
||||||
pub enum WSCommand {
|
|
||||||
// Subscribe { property: String },
|
|
||||||
// UnsubscribeAll,
|
|
||||||
Load { urls: Vec<String> },
|
|
||||||
TogglePlayback,
|
|
||||||
Volume { volume: f64 },
|
|
||||||
Time { time: f64 },
|
|
||||||
PlaylistNext,
|
|
||||||
PlaylistPrevious,
|
|
||||||
PlaylistGoto { position: usize },
|
|
||||||
PlaylistClear,
|
|
||||||
PlaylistRemove { positions: Vec<usize> },
|
|
||||||
PlaylistMove { from: usize, to: usize },
|
|
||||||
Shuffle,
|
|
||||||
SetSubtitleTrack { track: Option<usize> },
|
|
||||||
SetLooping { value: bool },
|
|
||||||
// SwayCommand { command: String },
|
|
||||||
SwayLaunchBrowser { url: String },
|
|
||||||
SwayCloseWorkspace { workspace: String },
|
|
||||||
SwayChangeWorkspace { workspace: String },
|
|
||||||
SwayGetWorkspaces,
|
|
||||||
// New input commands
|
|
||||||
InputKeys { keys: String },
|
|
||||||
MouseMove { x: i32, y: i32 },
|
|
||||||
MouseScroll { x: i32, y: i32 },
|
|
||||||
MouseClick { button: String },
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn handle_message(
|
|
||||||
message: Value,
|
|
||||||
mpv: Mpv,
|
|
||||||
_channel_id: u64,
|
|
||||||
) -> anyhow::Result<Option<Value>> {
|
|
||||||
let command =
|
|
||||||
serde_json::from_value::<WSCommand>(message).context("Failed to parse message")?;
|
|
||||||
|
|
||||||
log::trace!("Successfully parsed message: {:?}", command);
|
|
||||||
|
|
||||||
match command {
|
|
||||||
// WSCommand::Subscribe { property } => {
|
|
||||||
// mpv.observe_property(channel_id, &property).await?;
|
|
||||||
// Ok(None)
|
|
||||||
// }
|
|
||||||
// WSCommand::UnsubscribeAll => {
|
|
||||||
// mpv.unobserve_property(channel_id).await?;
|
|
||||||
// Ok(None)
|
|
||||||
// }
|
|
||||||
WSCommand::Load { urls } => {
|
|
||||||
for url in urls {
|
|
||||||
mpv.playlist_add(
|
|
||||||
&url,
|
|
||||||
PlaylistAddTypeOptions::File,
|
|
||||||
mpvipc_async::PlaylistAddOptions::Append,
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
}
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::TogglePlayback => {
|
|
||||||
mpv.set_playback(mpvipc_async::Switch::Toggle).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::Volume { volume } => {
|
|
||||||
mpv.set_volume(volume, NumberChangeOptions::Absolute)
|
|
||||||
.await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::Time { time } => {
|
|
||||||
mpv.seek(time, SeekOptions::AbsolutePercent).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::PlaylistNext => {
|
|
||||||
mpv.next().await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::PlaylistPrevious => {
|
|
||||||
mpv.prev().await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::PlaylistGoto { position } => {
|
|
||||||
mpv.playlist_play_id(position).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::PlaylistClear => {
|
|
||||||
mpv.playlist_clear().await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: this could lead to a race condition between `playlist_remove_id` commands
|
|
||||||
WSCommand::PlaylistRemove { mut positions } => {
|
|
||||||
positions.sort();
|
|
||||||
|
|
||||||
for position in positions.iter().rev() {
|
|
||||||
mpv.playlist_remove_id(*position).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
|
|
||||||
WSCommand::PlaylistMove { from, to } => {
|
|
||||||
mpv.playlist_move_id(from, to).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::Shuffle => {
|
|
||||||
mpv.playlist_shuffle().await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::SetSubtitleTrack { track } => {
|
|
||||||
mpv.set_property("sid", track).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::SetLooping { value } => {
|
|
||||||
mpv.set_loop_playlist(if value { Switch::On } else { Switch::Off })
|
|
||||||
.await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
// WSCommand::SwayCommand { command } => {
|
|
||||||
// base::sway_run_command(command).await?;
|
|
||||||
// Ok(None)
|
|
||||||
// }
|
|
||||||
WSCommand::SwayLaunchBrowser { url } => {
|
|
||||||
base::sway_launch_browser(&url).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::SwayCloseWorkspace { workspace } => {
|
|
||||||
base::sway_close_workspace(workspace).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::SwayChangeWorkspace { workspace } => {
|
|
||||||
base::sway_change_workspace(workspace).await?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::SwayGetWorkspaces => {
|
|
||||||
let workspaces = base::sway_get_workspace_names().await?;
|
|
||||||
Ok(Some(json!(workspaces)))
|
|
||||||
}
|
|
||||||
WSCommand::InputKeys { keys } => {
|
|
||||||
base::input(keys)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}))
|
|
||||||
.context("Failed to execute input command")?;
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::MouseMove { x, y } => {
|
|
||||||
let _ = base::mouse_move(x, y)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}));
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::MouseScroll { x, y } => {
|
|
||||||
let _ = base::mouse_scroll(x, y)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}));
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
WSCommand::MouseClick { button } => {
|
|
||||||
let _ = base::mouse_click(button)
|
|
||||||
.await
|
|
||||||
.map(|_| json!({}));
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
103
src/main.rs
103
src/main.rs
@ -1,22 +1,16 @@
|
|||||||
use anyhow::Context;
|
use anyhow::Context;
|
||||||
use axum::Router;
|
use axum::{Router, Server};
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use clap_verbosity_flag::Verbosity;
|
use clap_verbosity_flag::Verbosity;
|
||||||
use futures::StreamExt;
|
|
||||||
use mpv_setup::{connect_to_mpv, create_mpv_config_file, show_grzegorz_image};
|
use mpv_setup::{connect_to_mpv, create_mpv_config_file, show_grzegorz_image};
|
||||||
use mpvipc_async::{Event, Mpv, MpvDataType, MpvExt};
|
use mpvipc_async::Mpv;
|
||||||
use std::{
|
use std::net::{IpAddr, SocketAddr};
|
||||||
net::{IpAddr, SocketAddr},
|
|
||||||
sync::{Arc, Mutex},
|
|
||||||
};
|
|
||||||
use systemd_journal_logger::JournalLog;
|
use systemd_journal_logger::JournalLog;
|
||||||
use tempfile::NamedTempFile;
|
use tempfile::NamedTempFile;
|
||||||
use tokio::task::JoinHandle;
|
|
||||||
use util::IdPool;
|
|
||||||
|
|
||||||
mod api;
|
mod api;
|
||||||
mod mpv_setup;
|
mod mpv_setup;
|
||||||
mod util;
|
// mod mpv_broker;
|
||||||
|
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
struct Args {
|
struct Args {
|
||||||
@ -94,71 +88,10 @@ async fn setup_systemd_watchdog_thread() -> anyhow::Result<()> {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn systemd_update_play_status(playing: bool, current_song: &Option<String>) {
|
|
||||||
sd_notify::notify(
|
|
||||||
false,
|
|
||||||
&[sd_notify::NotifyState::Status(&format!(
|
|
||||||
"{} {:?}",
|
|
||||||
if playing { "[PLAY]" } else { "[STOP]" },
|
|
||||||
if let Some(song) = current_song {
|
|
||||||
song
|
|
||||||
} else {
|
|
||||||
""
|
|
||||||
}
|
|
||||||
))],
|
|
||||||
)
|
|
||||||
.unwrap_or_else(|e| log::warn!("Failed to update systemd status with current song: {}", e));
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn setup_systemd_notifier(mpv: Mpv) -> anyhow::Result<JoinHandle<()>> {
|
|
||||||
let handle = tokio::spawn(async move {
|
|
||||||
log::debug!("Starting systemd notifier thread");
|
|
||||||
let mut event_stream = mpv.get_event_stream().await;
|
|
||||||
|
|
||||||
mpv.observe_property(100, "media-title").await.unwrap();
|
|
||||||
mpv.observe_property(100, "pause").await.unwrap();
|
|
||||||
|
|
||||||
let mut current_song: Option<String> = mpv.get_property("media-title").await.unwrap();
|
|
||||||
let mut playing = !mpv.get_property("pause").await.unwrap().unwrap_or(false);
|
|
||||||
|
|
||||||
systemd_update_play_status(playing, ¤t_song);
|
|
||||||
|
|
||||||
loop {
|
|
||||||
if let Some(Ok(Event::PropertyChange { name, data, .. })) = event_stream.next().await {
|
|
||||||
match (name.as_str(), data) {
|
|
||||||
("media-title", Some(MpvDataType::String(s))) => {
|
|
||||||
current_song = Some(s);
|
|
||||||
}
|
|
||||||
("media-title", None) => {
|
|
||||||
current_song = None;
|
|
||||||
}
|
|
||||||
("pause", Some(MpvDataType::Bool(b))) => {
|
|
||||||
playing = !b;
|
|
||||||
}
|
|
||||||
(event_name, _) => {
|
|
||||||
log::trace!(
|
|
||||||
"Received unexpected property change on systemd notifier thread: {}",
|
|
||||||
event_name
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
systemd_update_play_status(playing, ¤t_song)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
Ok(handle)
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn shutdown(mpv: Mpv, proc: Option<tokio::process::Child>) {
|
async fn shutdown(mpv: Mpv, proc: Option<tokio::process::Child>) {
|
||||||
log::info!("Shutting down");
|
log::info!("Shutting down");
|
||||||
sd_notify::notify(false, &[sd_notify::NotifyState::Stopping]).unwrap_or_else(|e| {
|
sd_notify::notify(false, &[sd_notify::NotifyState::Stopping])
|
||||||
log::warn!(
|
.unwrap_or_else(|e| log::warn!("Failed to notify systemd that the service is stopping: {}", e));
|
||||||
"Failed to notify systemd that the service is stopping: {}",
|
|
||||||
e
|
|
||||||
)
|
|
||||||
});
|
|
||||||
|
|
||||||
mpv.disconnect()
|
mpv.disconnect()
|
||||||
.await
|
.await
|
||||||
@ -206,10 +139,6 @@ async fn main() -> anyhow::Result<()> {
|
|||||||
.await
|
.await
|
||||||
.context("Failed to connect to mpv")?;
|
.context("Failed to connect to mpv")?;
|
||||||
|
|
||||||
if systemd_mode {
|
|
||||||
setup_systemd_notifier(mpv.clone()).await?;
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Err(e) = show_grzegorz_image(mpv.clone()).await {
|
if let Err(e) = show_grzegorz_image(mpv.clone()).await {
|
||||||
log::warn!("Could not show Grzegorz image: {}", e);
|
log::warn!("Could not show Grzegorz image: {}", e);
|
||||||
}
|
}
|
||||||
@ -228,21 +157,11 @@ async fn main() -> anyhow::Result<()> {
|
|||||||
let socket_addr = SocketAddr::new(addr, args.port);
|
let socket_addr = SocketAddr::new(addr, args.port);
|
||||||
log::info!("Starting API on {}", socket_addr);
|
log::info!("Starting API on {}", socket_addr);
|
||||||
|
|
||||||
let id_pool = Arc::new(Mutex::new(IdPool::new_with_max_limit(1024)));
|
let app = Router::new().nest("/api", api::rest_api_routes(mpv.clone()));
|
||||||
|
let server = match Server::try_bind(&socket_addr.clone())
|
||||||
let app = Router::new()
|
|
||||||
// .nest("/api/v1", api::rest_api_routes_v1(mpv.clone()))
|
|
||||||
.nest("/api/v2", api::rest_api_routes_v2(mpv.clone()))
|
|
||||||
.nest("/ws", api::websocket_api(mpv.clone(), id_pool.clone()))
|
|
||||||
// .merge(api::rest_api_docs_v1(mpv.clone()))
|
|
||||||
.merge(api::rest_api_docs_v2(mpv.clone()))
|
|
||||||
.into_make_service_with_connect_info::<SocketAddr>();
|
|
||||||
|
|
||||||
let listener = match tokio::net::TcpListener::bind(&socket_addr)
|
|
||||||
.await
|
|
||||||
.context(format!("Failed to bind API server to '{}'", &socket_addr))
|
.context(format!("Failed to bind API server to '{}'", &socket_addr))
|
||||||
{
|
{
|
||||||
Ok(listener) => listener,
|
Ok(server) => server,
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
log::error!("{}", e);
|
log::error!("{}", e);
|
||||||
shutdown(mpv, proc).await;
|
shutdown(mpv, proc).await;
|
||||||
@ -273,7 +192,7 @@ async fn main() -> anyhow::Result<()> {
|
|||||||
log::info!("Received Ctrl-C, exiting");
|
log::info!("Received Ctrl-C, exiting");
|
||||||
shutdown(mpv, Some(proc)).await;
|
shutdown(mpv, Some(proc)).await;
|
||||||
}
|
}
|
||||||
result = axum::serve(listener, app) => {
|
result = server.serve(app.into_make_service()) => {
|
||||||
log::info!("API server exited");
|
log::info!("API server exited");
|
||||||
shutdown(mpv, Some(proc)).await;
|
shutdown(mpv, Some(proc)).await;
|
||||||
result?;
|
result?;
|
||||||
@ -285,7 +204,7 @@ async fn main() -> anyhow::Result<()> {
|
|||||||
log::info!("Received Ctrl-C, exiting");
|
log::info!("Received Ctrl-C, exiting");
|
||||||
shutdown(mpv.clone(), None).await;
|
shutdown(mpv.clone(), None).await;
|
||||||
}
|
}
|
||||||
result = axum::serve(listener, app) => {
|
result = server.serve(app.into_make_service()) => {
|
||||||
log::info!("API server exited");
|
log::info!("API server exited");
|
||||||
shutdown(mpv.clone(), None).await;
|
shutdown(mpv.clone(), None).await;
|
||||||
result?;
|
result?;
|
||||||
|
@ -11,9 +11,6 @@ const DEFAULT_MPV_CONFIG_CONTENT: &str = include_str!("../assets/default-mpv.con
|
|||||||
|
|
||||||
const THE_MAN_PNG: &[u8] = include_bytes!("../assets/the_man.png");
|
const THE_MAN_PNG: &[u8] = include_bytes!("../assets/the_man.png");
|
||||||
|
|
||||||
// https://mpv.io/manual/master/#options-ytdl
|
|
||||||
const YTDL_HOOK_ARGS: [&str; 2] = ["try_ytdl_first=yes", "thumbnails=none"];
|
|
||||||
|
|
||||||
pub fn create_mpv_config_file(args_config_file: Option<String>) -> anyhow::Result<NamedTempFile> {
|
pub fn create_mpv_config_file(args_config_file: Option<String>) -> anyhow::Result<NamedTempFile> {
|
||||||
let file_content = if let Some(path) = args_config_file {
|
let file_content = if let Some(path) = args_config_file {
|
||||||
if !Path::new(&path).exists() {
|
if !Path::new(&path).exists() {
|
||||||
@ -81,13 +78,6 @@ pub async fn connect_to_mpv<'a>(
|
|||||||
.arg("--force-window")
|
.arg("--force-window")
|
||||||
.arg("--fullscreen")
|
.arg("--fullscreen")
|
||||||
.arg("--no-config")
|
.arg("--no-config")
|
||||||
.arg("--ytdl=yes")
|
|
||||||
.args(
|
|
||||||
YTDL_HOOK_ARGS
|
|
||||||
.into_iter()
|
|
||||||
.map(|x| format!("--script-opts=ytdl_hook-{}", x))
|
|
||||||
.collect::<Vec<_>>(),
|
|
||||||
)
|
|
||||||
.arg(format!(
|
.arg(format!(
|
||||||
"--include={}",
|
"--include={}",
|
||||||
&args.config_file.path().to_string_lossy()
|
&args.config_file.path().to_string_lossy()
|
||||||
@ -142,3 +132,4 @@ pub async fn show_grzegorz_image(mpv: Mpv) -> anyhow::Result<()> {
|
|||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,3 +0,0 @@
|
|||||||
mod id_pool;
|
|
||||||
|
|
||||||
pub use id_pool::IdPool;
|
|
@ -1,145 +0,0 @@
|
|||||||
use std::{collections::BTreeSet, fmt::Debug};
|
|
||||||
|
|
||||||
use tokio::sync::watch;
|
|
||||||
|
|
||||||
/// A relatively naive ID pool implementation.
|
|
||||||
pub struct IdPool {
|
|
||||||
max_id: u64,
|
|
||||||
free_ids: BTreeSet<u64>,
|
|
||||||
id_count: u64,
|
|
||||||
id_count_watch_sender: watch::Sender<u64>,
|
|
||||||
id_count_watch_receiver: watch::Receiver<u64>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Debug for IdPool {
|
|
||||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
||||||
f.debug_struct("IdPool")
|
|
||||||
.field("max_id", &self.max_id)
|
|
||||||
.field("free_ids", &self.free_ids)
|
|
||||||
.field("id_count", &self.id_count)
|
|
||||||
.finish()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for IdPool {
|
|
||||||
fn default() -> Self {
|
|
||||||
let (id_count_watch_sender, id_count_watch_receiver) = watch::channel(0);
|
|
||||||
Self {
|
|
||||||
max_id: u64::MAX,
|
|
||||||
free_ids: BTreeSet::new(),
|
|
||||||
id_count: 0,
|
|
||||||
id_count_watch_sender,
|
|
||||||
id_count_watch_receiver,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//TODO: thiserror
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
||||||
pub enum IdPoolError {
|
|
||||||
NoFreeIds,
|
|
||||||
IdNotInUse(u64),
|
|
||||||
IdOutOfBound(u64),
|
|
||||||
}
|
|
||||||
|
|
||||||
impl IdPool {
|
|
||||||
pub fn new_with_max_limit(max_id: u64) -> Self {
|
|
||||||
let (id_count_watch_sender, id_count_watch_receiver) = watch::channel(0);
|
|
||||||
Self {
|
|
||||||
max_id,
|
|
||||||
free_ids: BTreeSet::new(),
|
|
||||||
id_count: 0,
|
|
||||||
id_count_watch_sender,
|
|
||||||
id_count_watch_receiver,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn id_count(&self) -> u64 {
|
|
||||||
self.id_count - self.free_ids.len() as u64
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn id_is_used(&self, id: u64) -> Result<bool, IdPoolError> {
|
|
||||||
if id > self.max_id {
|
|
||||||
Err(IdPoolError::IdOutOfBound(id))
|
|
||||||
} else if self.free_ids.contains(&id) {
|
|
||||||
return Ok(false);
|
|
||||||
} else {
|
|
||||||
return Ok(id <= self.id_count);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn request_id(&mut self) -> Result<u64, IdPoolError> {
|
|
||||||
if !self.free_ids.is_empty() {
|
|
||||||
let id = self.free_ids.pop_first().unwrap();
|
|
||||||
self.update_watch();
|
|
||||||
Ok(id)
|
|
||||||
} else if self.id_count < self.max_id {
|
|
||||||
self.id_count += 1;
|
|
||||||
self.update_watch();
|
|
||||||
Ok(self.id_count)
|
|
||||||
} else {
|
|
||||||
Err(IdPoolError::NoFreeIds)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn release_id(&mut self, id: u64) -> Result<(), IdPoolError> {
|
|
||||||
if !self.id_is_used(id)? {
|
|
||||||
Err(IdPoolError::IdNotInUse(id))
|
|
||||||
} else {
|
|
||||||
self.free_ids.insert(id);
|
|
||||||
self.update_watch();
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn update_watch(&self) {
|
|
||||||
self.id_count_watch_sender.send(self.id_count()).unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get_id_count_watch_receiver(&self) -> watch::Receiver<u64> {
|
|
||||||
self.id_count_watch_receiver.clone()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_id_pool() {
|
|
||||||
let mut pool = IdPool::new_with_max_limit(10);
|
|
||||||
assert_eq!(pool.request_id(), Ok(1));
|
|
||||||
assert_eq!(pool.request_id(), Ok(2));
|
|
||||||
assert_eq!(pool.request_id(), Ok(3));
|
|
||||||
assert_eq!(pool.request_id(), Ok(4));
|
|
||||||
assert_eq!(pool.id_count(), 4);
|
|
||||||
assert_eq!(pool.request_id(), Ok(5));
|
|
||||||
assert_eq!(pool.request_id(), Ok(6));
|
|
||||||
assert_eq!(pool.request_id(), Ok(7));
|
|
||||||
assert_eq!(pool.request_id(), Ok(8));
|
|
||||||
assert_eq!(pool.request_id(), Ok(9));
|
|
||||||
assert_eq!(pool.request_id(), Ok(10));
|
|
||||||
assert_eq!(pool.id_count(), 10);
|
|
||||||
assert_eq!(pool.request_id(), Err(IdPoolError::NoFreeIds));
|
|
||||||
assert_eq!(pool.release_id(5), Ok(()));
|
|
||||||
assert_eq!(pool.release_id(5), Err(IdPoolError::IdNotInUse(5)));
|
|
||||||
assert_eq!(pool.id_count(), 9);
|
|
||||||
assert_eq!(pool.request_id(), Ok(5));
|
|
||||||
assert_eq!(pool.release_id(11), Err(IdPoolError::IdOutOfBound(11)));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_id_pool_watch() {
|
|
||||||
let mut pool = IdPool::new_with_max_limit(10);
|
|
||||||
let receiver = pool.get_id_count_watch_receiver();
|
|
||||||
|
|
||||||
assert_eq!(receiver.borrow().clone(), 0);
|
|
||||||
pool.request_id().unwrap();
|
|
||||||
assert_eq!(receiver.borrow().clone(), 1);
|
|
||||||
pool.request_id().unwrap();
|
|
||||||
assert_eq!(receiver.borrow().clone(), 2);
|
|
||||||
pool.release_id(1).unwrap();
|
|
||||||
assert_eq!(receiver.borrow().clone(), 1);
|
|
||||||
}
|
|
||||||
}
|
|
Reference in New Issue
Block a user