Compare commits

...

29 Commits

Author SHA1 Message Date
Oystein Kristoffer Tveit 80e0447bcb
default.nix: filter `flake.lock` from source 2024-10-30 01:53:59 +01:00
Oystein Kristoffer Tveit 14703dc733
flake.nix: opinionated app 2024-10-30 01:53:36 +01:00
Oystein Kristoffer Tveit 012cdb4658 Add openapi docs 2024-10-30 01:53:08 +01:00
Oystein Kristoffer Tveit 355d2ad13d
module.nix: relax hardening 2024-10-22 19:49:52 +02:00
Oystein Kristoffer Tveit 9934b11766
nix: use unwrapped program by default in module 2024-10-20 23:06:32 +02:00
Oystein Kristoffer Tveit cfb6bbd7a7
default.nix: better source filtering 2024-10-20 23:06:32 +02:00
Oystein Kristoffer Tveit b1f8cf9ba2
systemd integration
- Add watchdog timeout support
- Add native journald logging support
- Add application state notifications
- Add verbosity flag
2024-10-20 23:06:32 +02:00
Oystein Kristoffer Tveit 64bcef4307
default.nix: filter source 2024-10-20 19:48:23 +02:00
Oystein Kristoffer Tveit 0def9bc340
flake.lock: bump 2024-10-20 19:48:04 +02:00
Oystein Kristoffer Tveit ae975f946d
flake.nix: fix rust-analyzer 2024-10-20 19:47:55 +02:00
Oystein Kristoffer Tveit 1413af8111
Cargo.lock: bump 2024-10-20 19:47:36 +02:00
Oystein Kristoffer Tveit c15feac958
Don't fail on not being able to show image 2024-10-20 19:45:53 +02:00
Oystein Kristoffer Tveit 91ab1ad771
module.nix: move greg user under sway condition 2024-10-20 19:45:53 +02:00
Oystein Kristoffer Tveit a476f877d2
systemd hardening 2024-10-20 19:45:53 +02:00
Oystein Kristoffer Tveit c5f606f910
module.nix: add `enableDebug` option 2024-10-20 02:08:39 +02:00
Oystein Kristoffer Tveit 6bb8d28eff
Add picture of grzeg 2024-10-20 01:24:09 +02:00
Oystein Kristoffer Tveit 5e3df86a2a
Split mpv setup to separate module 2024-10-20 00:04:59 +02:00
Oystein Kristoffer Tveit a52f2a2e49
module.nix: use sway as graphical backend 2024-10-19 23:26:16 +02:00
Oystein Kristoffer Tveit c1438f2480
Add nixos module and overlay 2024-08-04 20:33:22 +02:00
Oystein Kristoffer Tveit 0ff6440457
Add --mpv-config-file option, with default content
This fixes some issues with seeking in output from yt-dlp
2024-08-04 19:00:05 +02:00
Oystein Kristoffer Tveit a0202215aa
README.md: add startup commands 2024-08-04 04:19:00 +02:00
Oystein Kristoffer Tveit e9190f7879
default.nix: init, flake.nix: add app 2024-08-04 04:10:59 +02:00
Oystein Kristoffer Tveit a89cb24c86
Cargo.toml: add some metadata 2024-08-04 03:19:55 +02:00
Oystein Kristoffer Tveit a5bedf4d87
Cargo.toml: optimize release profile 2024-08-04 03:19:42 +02:00
Oystein Kristoffer Tveit 894cc4e146
Use mpvipc-async version 0.1.0 2024-08-04 03:13:11 +02:00
Oystein Kristoffer Tveit 24b0307f0f
.envrc: init 2024-08-04 03:13:10 +02:00
Oystein Kristoffer Tveit f0b0e9f248
flake.nix: replace fenix with rust-overlay 2024-08-04 03:13:10 +02:00
Oystein Kristoffer Tveit 0d4841a9ec
README: move TODO list to gitea issue tracker 2024-07-28 16:29:17 +02:00
Oystein Kristoffer Tveit 18d186b997
nix support 2024-04-19 00:20:16 +02:00
16 changed files with 1935 additions and 646 deletions

1
.envrc Normal file
View File

@ -0,0 +1 @@
use flake

1
.gitignore vendored
View File

@ -1 +1,2 @@
/target /target
result

1294
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@ -2,18 +2,33 @@
name = "greg-ng" name = "greg-ng"
version = "0.1.0" version = "0.1.0"
edition = "2021" edition = "2021"
license = "MIT"
authors = ["oysteikt@pvv.ntnu.no"]
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]
anyhow = "1.0.82" anyhow = "1.0.82"
axum = { version = "0.6.20", features = ["macros"] } axum = { version = "0.7.7", features = ["macros"] }
clap = { version = "4.4.1", features = ["derive"] } clap = { version = "4.4.1", features = ["derive"] }
clap-verbosity-flag = "2.2.2"
env_logger = "0.10.0" env_logger = "0.10.0"
log = "0.4.20" log = "0.4.20"
mpvipc = "1.3.0" mpvipc-async = { git = "https://git.pvv.ntnu.no/oysteikt/mpvipc-async.git", rev = "v0.1.0" }
sd-notify = "0.4.3"
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"
tempfile = "3.11.0"
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]
strip = true
lto = true
codegen-units = 1

View File

@ -2,23 +2,20 @@
New implementation of https://github.com/Programvareverkstedet/grzegorz New implementation of https://github.com/Programvareverkstedet/grzegorz
## Feature wishlist ## Test it out
- [ ] Feature parity with old grzegorz ```sh
- [X] Rest API # NixOS
- [ ] Rest API docs nix run "git+https://git.pvv.ntnu.no/Projects/greg-ng#" -- --mpv-socket-path /tmp/mpv.sock
- [ ] Metadata fetcher
- [ ] Init mpv with image of grzegorz # Other (after git clone and rust toolchain has been set up)
- [ ] Save playlists to machine cargo run -- --mpv-socket-path /tmp/mpv.sock
- [ ] Cache playlist contents to disk ```
- [ ] Expose service through mpd protocol
- [ ] Users with playlists and songs (and auth?) See also https://git.pvv.ntnu.no/Projects/grzegorz-clients for frontend alternatives
- [ ] Some kind of fair scheduling for each user
- [ ] Max time to avoid playlist songs ## Debugging
- [ ] Expose video/media stream so others can listen at home
- [ ] Syncope support >:) ```sh
- [ ] Jitsi support >:))) RUST_LOG=greg_ng=trace,mpvipc=trace cargo run -- --mpv-socket-path /tmp/mpv.sock
- [ ] Show other media while playing music, like grafana or bustimes ```
- [ ] Soft shuffle
- [ ] Libre.fm integration
- [ ] Karaoke mode lmao

3
assets/default-mpv.conf Normal file
View File

@ -0,0 +1,3 @@
[youtube]
profile-cond=path:find('youtu%.?be')
--ytdl-format="bestvideo[height<=?720][vcodec!~='vp0?9']+bestaudio/best"

BIN
assets/the_man.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 157 KiB

52
default.nix Normal file
View File

@ -0,0 +1,52 @@
{
lib
, fetchFromGitHub
, rustPlatform
, makeWrapper
, mpv
, wrapped ? false
}:
rustPlatform.buildRustPackage rec {
pname = "greg-ng";
version = "0.1.0";
src = builtins.filterSource (path: type: let
baseName = baseNameOf (toString path);
in !(lib.any (b: b) [
(!(lib.cleanSourceFilter path type))
(type == "directory" && lib.elem baseName [
".direnv"
".git"
"target"
"result"
])
(type == "regular" && lib.elem baseName [
"flake.nix"
"flake.lock"
"default.nix"
"module.nix"
".envrc"
])
])) ./.;
nativeBuildInputs = [ makeWrapper ];
cargoLock = {
lockFile = ./Cargo.lock;
outputHashes = {
"mpvipc-async-0.1.0" = "sha256-2TQ2d4q9/DTxTZe9kOAoKBhsmegRZw32x3G2hbluS6U=";
};
};
postInstall = lib.optionalString wrapped ''
wrapProgram $out/bin/greg-ng \
--prefix PATH : '${lib.makeBinPath [ mpv ]}'
'';
meta = with lib; {
license = licenses.mit;
maintainers = with maintainers; [ h7x4 ];
platforms = platforms.linux ++ platforms.darwin;
mainProgram = "greg-ng";
};
}

48
flake.lock Normal file
View File

@ -0,0 +1,48 @@
{
"nodes": {
"nixpkgs": {
"locked": {
"lastModified": 1729256560,
"narHash": "sha256-/uilDXvCIEs3C9l73JTACm4quuHUsIHcns1c+cHUJwA=",
"owner": "NixOS",
"repo": "nixpkgs",
"rev": "4c2fcb090b1f3e5b47eaa7bd33913b574a11e0a0",
"type": "github"
},
"original": {
"owner": "NixOS",
"ref": "nixos-unstable",
"repo": "nixpkgs",
"type": "github"
}
},
"root": {
"inputs": {
"nixpkgs": "nixpkgs",
"rust-overlay": "rust-overlay"
}
},
"rust-overlay": {
"inputs": {
"nixpkgs": [
"nixpkgs"
]
},
"locked": {
"lastModified": 1729391507,
"narHash": "sha256-as0I9xieJUHf7kiK2a9znDsVZQTFWhM1pLivII43Gi0=",
"owner": "oxalica",
"repo": "rust-overlay",
"rev": "784981a9feeba406de38c1c9a3decf966d853cca",
"type": "github"
},
"original": {
"owner": "oxalica",
"repo": "rust-overlay",
"type": "github"
}
}
},
"root": "root",
"version": 7
}

75
flake.nix Normal file
View File

@ -0,0 +1,75 @@
{
inputs = {
nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
rust-overlay.url = "github:oxalica/rust-overlay";
rust-overlay.inputs.nixpkgs.follows = "nixpkgs";
};
outputs = { self, nixpkgs, rust-overlay }:
let
inherit (nixpkgs) lib;
systems = [
"x86_64-linux"
"aarch64-linux"
"x86_64-darwin"
"aarch64-darwin"
"armv7l-linux"
];
forAllSystems = f: lib.genAttrs systems (system: let
pkgs = import nixpkgs {
inherit system;
overlays = [
(import rust-overlay)
];
};
rust-bin = rust-overlay.lib.mkRustBin { } pkgs.buildPackages;
toolchain = rust-bin.stable.latest.default.override {
extensions = [ "rust-src" "rust-analyzer" "rust-std" ];
};
in f system pkgs toolchain);
in {
apps = forAllSystems (system: pkgs: _: {
default = self.apps.${system}.greg-ng;
greg-ng = let
package = self.packages.${system}.greg-ng-wrapped;
in {
type = "app";
program = toString (pkgs.writeShellScript "greg-ng" ''
${lib.getExe package} --mpv-socket-path /tmp/greg-ng-mpv.sock -vvvv
'');
};
});
devShells = forAllSystems (system: pkgs: toolchain: {
default = pkgs.mkShell {
nativeBuildInputs = [
toolchain
pkgs.mpv
];
RUST_SRC_PATH = "${toolchain}/lib/rustlib/src/rust/library";
};
});
overlays = {
default = self.overlays.greg-ng;
greg-ng = final: prev: {
inherit (self.packages.${prev.system}) greg-ng;
};
};
packages = forAllSystems (system: pkgs: _: {
default = self.packages.${system}.greg-ng;
greg-ng = pkgs.callPackage ./default.nix { };
greg-ng-wrapped = pkgs.callPackage ./default.nix {
wrapped = true;
};
});
} // {
nixosModules.default = ./module.nix;
};
}

197
module.nix Normal file
View File

@ -0,0 +1,197 @@
{ config, pkgs, lib, ... }:
let
cfg = config.services.greg-ng;
in
{
options.services.greg-ng = {
enable = lib.mkEnableOption "greg-ng, an mpv based media player";
package = lib.mkPackageOption pkgs "greg-ng" { };
mpvPackage = lib.mkPackageOption pkgs "mpv" { };
enableSway = lib.mkEnableOption "sway as the main window manager";
enablePipewire = lib.mkEnableOption "pipewire" // { default = true; };
logLevel = lib.mkOption {
type = lib.types.enum [ "quiet" "error" "warn" "info" "debug" "trace" ];
default = "debug";
description = "Log level.";
apply = level: {
"quiet" = "-q";
"error" = "";
"warn" = "-v";
"info" = "-vv";
"debug" = "-vvv";
"trace" = "-vvvv";
}.${level};
};
# TODO: create some better descriptions
settings = {
host = lib.mkOption {
type = lib.types.str;
default = "localhost";
example = "0.0.0.0";
description = ''
Which host to bind to.
'';
};
port = lib.mkOption {
type = lib.types.port;
default = 8008;
example = 10008;
description = ''
Which port to bind to.
'';
};
mpv-socket-path = lib.mkOption {
type = lib.types.str;
default = "%t/greg-ng-mpv.sock";
description = ''
Path to the mpv socket.
'';
};
mpv-executable-path = lib.mkOption {
type = lib.types.str;
default = lib.getExe cfg.mpvPackage;
defaultText = lib.literalExpression ''
lib.getExe config.services.greg-ng.mpvPackage
'';
description = ''
Path to the mpv executable.
'';
};
mpv-config-file = lib.mkOption {
type = with lib.types; nullOr str;
default = null;
description = ''
Path to the mpv config file.
'';
};
auto-start-mpv = lib.mkOption {
type = lib.types.bool;
default = true;
description = ''
Whether to automatically start mpv.
'';
};
force-auto-start = lib.mkOption {
type = lib.types.bool;
default = true;
description = ''
Whether to force auto starting mpv.
'';
};
};
};
config = lib.mkMerge [
(lib.mkIf cfg.enable {
systemd.user.services.greg-ng = {
description = "greg-ng, an mpv based media player";
wantedBy = [ "graphical-session.target" ];
partOf = [ "graphical-session.target" ];
serviceConfig = {
Type = "notify";
ExecStart = let
args = lib.cli.toGNUCommandLineShell { } (cfg.settings // {
systemd = true;
});
in "${lib.getExe cfg.package} ${cfg.logLevel} ${args}";
Restart = "always";
RestartSec = 3;
WatchdogSec = lib.mkDefault 15;
TimeoutStartSec = lib.mkDefault 30;
RestrictAddressFamilies = [ "AF_UNIX" "AF_INET" "AF_INET6" ];
AmbientCapabilities = [ "" ];
CapabilityBoundingSet = [ "" ];
DeviceAllow = [ "" ];
LockPersonality = true;
# Might work, but wouldn't bet on it with embedded lua in mpv
MemoryDenyWriteExecute = false;
NoNewPrivileges = true;
# MPV and mesa tries to talk directly to the GPU.
PrivateDevices = false;
PrivateMounts = true;
PrivateTmp = true;
PrivateUsers = true;
ProcSubset = "pid";
ProtectClock = true;
ProtectControlGroups = true;
# MPV wants ~/.cache
ProtectHome = false;
ProtectHostname = true;
ProtectKernelLogs = true;
ProtectKernelModules = true;
ProtectKernelTunables = true;
ProtectProc = "invisible";
# I'll figure it out sometime
# ProtectSystem = "full";
RemoveIPC = true;
UMask = "0077";
RestrictNamespaces = true;
RestrictRealtime = true;
RestrictSUIDSGID = true;
SystemCallArchitectures = "native";
# Something brokey
# SystemCallFilter = [
# "@system-service"
# "~@privileged"
# "~@resources"
# ];
};
};
})
(lib.mkIf (cfg.enable && cfg.enablePipewire) {
services.pipewire = {
enable = true;
alsa.enable = true;
alsa.support32Bit = true;
pulse.enable = true;
};
})
(lib.mkIf (cfg.enable && cfg.enableSway) {
programs.sway = {
enable = true;
wrapperFeatures.gtk = true;
};
xdg.portal = {
enable = true;
wlr.enable = true;
extraPortals = [ pkgs.xdg-desktop-portal-gtk ];
};
users = {
users.greg = {
isNormalUser = true;
group = "greg";
uid = 2000;
description = "loud gym bro";
};
groups.greg.gid = 2000;
};
services.greetd = {
enable = true;
settings = rec {
initial_session = {
command = "${pkgs.sway}/bin/sway";
user = "greg";
};
default_session = initial_session;
};
};
})
];
}

View File

@ -1,4 +1,4 @@
mod base; mod base;
mod rest_wrapper_v1; mod rest_wrapper_v1;
pub use rest_wrapper_v1::rest_api_routes; pub use rest_wrapper_v1::{rest_api_routes, rest_api_docs};

View File

@ -1,62 +1,61 @@
use std::sync::Arc; use mpvipc_async::{
LoopProperty, Mpv, MpvExt, NumberChangeOptions, PlaylistAddOptions, PlaylistAddTypeOptions,
use log::trace; SeekOptions, Switch,
use mpvipc::{
Mpv, NumberChangeOptions, PlaylistAddOptions, PlaylistAddTypeOptions, SeekOptions, Switch,
}; };
use serde_json::{json, Value}; use serde_json::{json, Value};
use tokio::sync::Mutex;
/// Add item to playlist /// Add item to playlist
pub async fn loadfile(mpv: Arc<Mutex<Mpv>>, path: &str) -> anyhow::Result<()> { pub async fn loadfile(mpv: Mpv, path: &str) -> anyhow::Result<()> {
trace!("api::loadfile({:?})", path); log::trace!("api::loadfile({:?})", path);
mpv.lock().await.playlist_add( mpv.playlist_add(
path, path,
PlaylistAddTypeOptions::File, PlaylistAddTypeOptions::File,
PlaylistAddOptions::Append, PlaylistAddOptions::Append,
)?; )
.await?;
Ok(()) Ok(())
} }
/// Check whether the player is paused or playing /// Check whether the player is paused or playing
pub async fn play_get(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> { pub async fn play_get(mpv: Mpv) -> anyhow::Result<Value> {
trace!("api::play_get()"); log::trace!("api::play_get()");
let paused: bool = mpv.lock().await.get_property("pause")?; let paused: bool = !mpv.is_playing().await?;
Ok(json!(!paused)) Ok(json!(!paused))
} }
/// Set whether the player is paused or playing /// Set whether the player is paused or playing
pub async fn play_set(mpv: Arc<Mutex<Mpv>>, should_play: bool) -> anyhow::Result<()> { pub async fn play_set(mpv: Mpv, should_play: bool) -> anyhow::Result<()> {
trace!("api::play_set({:?})", should_play); log::trace!("api::play_set({:?})", should_play);
mpv.lock() mpv.set_playback(if should_play { Switch::On } else { Switch::Off })
.await .await
.set_property("pause", !should_play)
.map_err(|e| e.into()) .map_err(|e| e.into())
} }
/// Get the current player volume /// Get the current player volume
pub async fn volume_get(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> { pub async fn volume_get(mpv: Mpv) -> anyhow::Result<Value> {
trace!("api::volume_get()"); log::trace!("api::volume_get()");
let volume: f64 = mpv.lock().await.get_property("volume")?; let volume: f64 = mpv.get_volume().await?;
Ok(json!(volume)) Ok(json!(volume))
} }
/// Set the player volume /// Set the player volume
pub async fn volume_set(mpv: Arc<Mutex<Mpv>>, value: f64) -> anyhow::Result<()> { pub async fn volume_set(mpv: Mpv, value: f64) -> anyhow::Result<()> {
trace!("api::volume_set({:?})", value); log::trace!("api::volume_set({:?})", value);
mpv.lock() mpv.set_volume(value, NumberChangeOptions::Absolute)
.await .await
.set_volume(value, NumberChangeOptions::Absolute)
.map_err(|e| e.into()) .map_err(|e| e.into())
} }
/// Get current playback position /// Get current playback position
pub async fn time_get(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> { pub async fn time_get(mpv: Mpv) -> anyhow::Result<Value> {
trace!("api::time_get()"); log::trace!("api::time_get()");
let current: f64 = mpv.lock().await.get_property("time-pos")?; let current: Option<f64> = mpv.get_time_pos().await?;
let remaining: f64 = mpv.lock().await.get_property("time-remaining")?; let remaining: Option<f64> = mpv.get_time_remaining().await?;
let total = current + remaining; let total = match (current, remaining) {
(Some(c), Some(r)) => Some(c + r),
(_, _) => None,
};
Ok(json!({ Ok(json!({
"current": current, "current": current,
@ -66,22 +65,16 @@ pub async fn time_get(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> {
} }
/// Set playback position /// Set playback position
pub async fn time_set( pub async fn time_set(mpv: Mpv, pos: Option<f64>, percent: Option<f64>) -> anyhow::Result<()> {
mpv: Arc<Mutex<Mpv>>, log::trace!("api::time_set({:?}, {:?})", pos, percent);
pos: Option<f64>,
percent: Option<f64>,
) -> anyhow::Result<()> {
trace!("api::time_set({:?}, {:?})", pos, percent);
if pos.is_some() && percent.is_some() { if pos.is_some() && percent.is_some() {
anyhow::bail!("pos and percent cannot be provided at the same time"); anyhow::bail!("pos and percent cannot be provided at the same time");
} }
if let Some(pos) = pos { if let Some(pos) = pos {
mpv.lock().await.seek(pos, SeekOptions::Absolute)?; mpv.seek(pos, SeekOptions::Absolute).await?;
} else if let Some(percent) = percent { } else if let Some(percent) = percent {
mpv.lock() mpv.seek(percent, SeekOptions::AbsolutePercent).await?;
.await
.seek(percent, SeekOptions::AbsolutePercent)?;
} else { } else {
anyhow::bail!("Either pos or percent must be provided"); anyhow::bail!("Either pos or percent must be provided");
}; };
@ -90,10 +83,10 @@ pub async fn time_set(
} }
/// Get the current playlist /// Get the current playlist
pub async fn playlist_get(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> { pub async fn playlist_get(mpv: Mpv) -> anyhow::Result<Value> {
trace!("api::playlist_get()"); log::trace!("api::playlist_get()");
let playlist: mpvipc::Playlist = mpv.lock().await.get_playlist()?; let playlist: mpvipc_async::Playlist = mpv.get_playlist().await?;
let is_playing: bool = mpv.lock().await.get_property("pause")?; let is_playing: bool = mpv.is_playing().await?;
let items: Vec<Value> = playlist let items: Vec<Value> = playlist
.0 .0
@ -104,7 +97,7 @@ pub async fn playlist_get(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> {
"index": i, "index": i,
"current": item.current, "current": item.current,
"playing": is_playing, "playing": is_playing,
"filename": item.filename, "filename": item.title.as_ref().unwrap_or(&item.filename),
"data": { "data": {
"fetching": true, "fetching": true,
} }
@ -116,74 +109,64 @@ pub async fn playlist_get(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> {
} }
/// Skip to the next item in the playlist /// Skip to the next item in the playlist
pub async fn playlist_next(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<()> { pub async fn playlist_next(mpv: Mpv) -> anyhow::Result<()> {
trace!("api::playlist_next()"); log::trace!("api::playlist_next()");
mpv.lock().await.next().map_err(|e| e.into()) mpv.next().await.map_err(|e| e.into())
} }
/// Go back to the previous item in the playlist /// Go back to the previous item in the playlist
pub async fn playlist_previous(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<()> { pub async fn playlist_previous(mpv: Mpv) -> anyhow::Result<()> {
trace!("api::playlist_previous()"); log::trace!("api::playlist_previous()");
mpv.lock().await.prev().map_err(|e| e.into()) mpv.prev().await.map_err(|e| e.into())
} }
/// Go chosen item in the playlist /// Go chosen item in the playlist
pub async fn playlist_goto(mpv: Arc<Mutex<Mpv>>, index: usize) -> anyhow::Result<()> { pub async fn playlist_goto(mpv: Mpv, index: usize) -> anyhow::Result<()> {
trace!("api::playlist_goto({:?})", index); log::trace!("api::playlist_goto({:?})", index);
mpv.lock() mpv.playlist_play_id(index).await.map_err(|e| e.into())
.await
.playlist_play_id(index)
.map_err(|e| e.into())
} }
/// Clears the playlist /// Clears the playlist
pub async fn playlist_clear(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<()> { pub async fn playlist_clear(mpv: Mpv) -> anyhow::Result<()> {
trace!("api::playlist_clear()"); log::trace!("api::playlist_clear()");
mpv.lock().await.playlist_clear().map_err(|e| e.into()) mpv.playlist_clear().await.map_err(|e| e.into())
} }
/// Remove an item from the playlist by index /// Remove an item from the playlist by index
pub async fn playlist_remove(mpv: Arc<Mutex<Mpv>>, index: usize) -> anyhow::Result<()> { pub async fn playlist_remove(mpv: Mpv, index: usize) -> anyhow::Result<()> {
trace!("api::playlist_remove({:?})", index); log::trace!("api::playlist_remove({:?})", index);
mpv.lock() mpv.playlist_remove_id(index).await.map_err(|e| e.into())
.await
.playlist_remove_id(index)
.map_err(|e| e.into())
} }
/// Move an item in the playlist from one index to another /// Move an item in the playlist from one index to another
pub async fn playlist_move(mpv: Arc<Mutex<Mpv>>, from: usize, to: usize) -> anyhow::Result<()> { pub async fn playlist_move(mpv: Mpv, from: usize, to: usize) -> anyhow::Result<()> {
trace!("api::playlist_move({:?}, {:?})", from, to); log::trace!("api::playlist_move({:?}, {:?})", from, to);
mpv.lock() mpv.playlist_move_id(from, to).await.map_err(|e| e.into())
.await
.playlist_move_id(from, to)
.map_err(|e| e.into())
} }
/// Shuffle the playlist /// Shuffle the playlist
pub async fn shuffle(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<()> { pub async fn shuffle(mpv: Mpv) -> anyhow::Result<()> {
trace!("api::shuffle()"); log::trace!("api::shuffle()");
mpv.lock().await.playlist_shuffle().map_err(|e| e.into()) mpv.playlist_shuffle().await.map_err(|e| e.into())
} }
/// See whether it loops the playlist or not /// See whether it loops the playlist or not
pub async fn playlist_get_looping(mpv: Arc<Mutex<Mpv>>) -> anyhow::Result<Value> { pub async fn playlist_get_looping(mpv: Mpv) -> anyhow::Result<Value> {
trace!("api::playlist_get_looping()"); log::trace!("api::playlist_get_looping()");
let loop_playlist = mpv.lock().await.get_property_string("loop-playlist")? == "inf";
Ok(json!(loop_playlist)) let loop_status = match mpv.playlist_is_looping().await? {
LoopProperty::No => false,
LoopProperty::Inf => true,
LoopProperty::N(_) => true,
};
Ok(json!(loop_status))
} }
pub async fn playlist_set_looping(mpv: Arc<Mutex<Mpv>>, r#loop: bool) -> anyhow::Result<()> { pub async fn playlist_set_looping(mpv: Mpv, r#loop: bool) -> anyhow::Result<()> {
trace!("api::playlist_set_looping({:?})", r#loop); log::trace!("api::playlist_set_looping({:?})", r#loop);
if r#loop {
mpv.lock() mpv.set_loop_playlist(if r#loop { Switch::On } else { Switch::Off })
.await .await
.set_loop_playlist(Switch::On)
.map_err(|e| e.into()) .map_err(|e| e.into())
} else {
mpv.lock()
.await
.set_loop_playlist(Switch::Off)
.map_err(|e| e.into())
}
} }

View File

@ -1,5 +1,3 @@
use std::sync::Arc;
use axum::{ use axum::{
extract::{Query, State}, extract::{Query, State},
http::StatusCode, http::StatusCode,
@ -7,13 +5,16 @@ use axum::{
routing::{delete, get, post}, routing::{delete, get, post},
Json, Router, Json, Router,
}; };
use mpvipc::Mpv; use mpvipc_async::Mpv;
use serde_json::{json, Value}; use serde_json::{json, Value};
use tokio::sync::Mutex;
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: Arc<Mutex<Mpv>>) -> Router { pub fn rest_api_routes(mpv: Mpv) -> Router {
Router::new() Router::new()
.route("/load", post(loadfile)) .route("/load", post(loadfile))
.route("/play", get(play_get)) .route("/play", get(play_get))
@ -34,6 +35,69 @@ pub fn rest_api_routes(mpv: Arc<Mutex<Mpv>>) -> Router {
.with_state(mpv) .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").url("/docs/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 {
@ -54,7 +118,7 @@ impl IntoResponse for RestResponse {
Ok(value) => (StatusCode::OK, Json(value)).into_response(), Ok(value) => (StatusCode::OK, Json(value)).into_response(),
Err(err) => ( Err(err) => (
StatusCode::INTERNAL_SERVER_ERROR, StatusCode::INTERNAL_SERVER_ERROR,
Json(json!({ "error": err.to_string(), "success": false })), Json(json!({ "error": err.to_string(), "errortext": err.to_string(), "success": false })),
) )
.into_response(), .into_response(),
} }
@ -67,99 +131,202 @@ impl IntoResponse for RestResponse {
// TODO: These could possibly be generated with a proc macro // TODO: These could possibly be generated with a proc macro
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
struct LoadFileArgs { struct LoadFileArgs {
path: String, path: String,
} }
async fn loadfile(
State(mpv): State<Arc<Mutex<Mpv>>>, /// Add item to playlist
Query(query): Query<LoadFileArgs>, #[utoipa::path(
) -> RestResponse { 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() base::loadfile(mpv, &query.path).await.into()
} }
async fn play_get(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::play_get(mpv).await.into()
} }
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
struct PlaySetArgs { struct PlaySetArgs {
play: String, play: String,
} }
async fn play_set( /// Set whether the player is paused or playing
State(mpv): State<Arc<Mutex<Mpv>>>, #[utoipa::path(
Query(query): Query<PlaySetArgs>, post,
) -> RestResponse { 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"; let play = query.play.to_lowercase() == "true";
base::play_set(mpv, play).await.into() base::play_set(mpv, play).await.into()
} }
async fn volume_get(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::volume_get(mpv).await.into()
} }
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
struct VolumeSetArgs { struct VolumeSetArgs {
volume: f64, volume: f64,
} }
async fn volume_set( /// Set the player volume
State(mpv): State<Arc<Mutex<Mpv>>>, #[utoipa::path(
Query(query): Query<VolumeSetArgs>, post,
) -> RestResponse { 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() base::volume_set(mpv, query.volume).await.into()
} }
async fn time_get(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::time_get(mpv).await.into()
} }
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
struct TimeSetArgs { struct TimeSetArgs {
pos: Option<f64>, pos: Option<f64>,
percent: Option<f64>, percent: Option<f64>,
} }
async fn time_set( /// Set playback position
State(mpv): State<Arc<Mutex<Mpv>>>, #[utoipa::path(
Query(query): Query<TimeSetArgs>, post,
) -> RestResponse { 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() base::time_set(mpv, query.pos, query.percent).await.into()
} }
async fn playlist_get(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::playlist_get(mpv).await.into()
} }
async fn playlist_next(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::playlist_next(mpv).await.into()
} }
async fn playlist_previous(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::playlist_previous(mpv).await.into()
} }
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
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<Arc<Mutex<Mpv>>>, State(mpv): State<Mpv>,
Query(query): Query<PlaylistGotoArgs>, Query(query): Query<PlaylistGotoArgs>,
) -> RestResponse { ) -> RestResponse {
base::playlist_goto(mpv, query.index).await.into() base::playlist_goto(mpv, query.index).await.into()
} }
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
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<Arc<Mutex<Mpv>>>, State(mpv): State<Mpv>,
Query(query): Query<PlaylistRemoveOrClearArgs>, Query(query): Query<PlaylistRemoveOrClearArgs>,
) -> RestResponse { ) -> RestResponse {
match query.index { match query.index {
@ -168,14 +335,24 @@ async fn playlist_remove_or_clear(
} }
} }
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
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<Arc<Mutex<Mpv>>>, State(mpv): State<Mpv>,
Query(query): Query<PlaylistMoveArgs>, Query(query): Query<PlaylistMoveArgs>,
) -> RestResponse { ) -> RestResponse {
base::playlist_move(mpv, query.index1, query.index2) base::playlist_move(mpv, query.index1, query.index2)
@ -183,21 +360,49 @@ async fn playlist_move(
.into() .into()
} }
async fn shuffle(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::shuffle(mpv).await.into()
} }
async fn playlist_get_looping(State(mpv): State<Arc<Mutex<Mpv>>>) -> RestResponse { /// 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() base::playlist_get_looping(mpv).await.into()
} }
#[derive(serde::Deserialize)] #[derive(serde::Deserialize, utoipa::IntoParams)]
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<Arc<Mutex<Mpv>>>, State(mpv): State<Mpv>,
Query(query): Query<PlaylistSetLoopingArgs>, Query(query): Query<PlaylistSetLoopingArgs>,
) -> RestResponse { ) -> RestResponse {
base::playlist_set_looping(mpv, query.r#loop).await.into() base::playlist_set_looping(mpv, query.r#loop).await.into()

View File

@ -1,19 +1,15 @@
use anyhow::Context; use anyhow::Context;
use axum::{Router, Server}; use axum::Router;
use clap::Parser; use clap::Parser;
use mpvipc::Mpv; use clap_verbosity_flag::Verbosity;
use std::{ use mpv_setup::{connect_to_mpv, create_mpv_config_file, show_grzegorz_image};
fs::create_dir_all, use mpvipc_async::Mpv;
net::{IpAddr, SocketAddr}, use std::net::{IpAddr, SocketAddr};
path::Path, use systemd_journal_logger::JournalLog;
sync::Arc, use tempfile::NamedTempFile;
};
use tokio::{
process::{Child, Command},
sync::Mutex,
};
mod api; mod api;
mod mpv_setup;
#[derive(Parser)] #[derive(Parser)]
struct Args { struct Args {
@ -23,12 +19,21 @@ struct Args {
#[clap(short, long, default_value = "8008")] #[clap(short, long, default_value = "8008")]
port: u16, port: u16,
#[command(flatten)]
verbose: Verbosity,
#[clap(long)]
systemd: bool,
#[clap(long, value_name = "PATH", default_value = "/run/mpv/mpv.sock")] #[clap(long, value_name = "PATH", default_value = "/run/mpv/mpv.sock")]
mpv_socket_path: String, mpv_socket_path: String,
#[clap(long, value_name = "PATH")] #[clap(long, value_name = "PATH")]
mpv_executable_path: Option<String>, mpv_executable_path: Option<String>,
#[clap(long, value_name = "PATH")]
mpv_config_file: Option<String>,
#[clap(long, default_value = "true")] #[clap(long, default_value = "true")]
auto_start_mpv: bool, auto_start_mpv: bool,
@ -36,90 +41,16 @@ struct Args {
force_auto_start: bool, force_auto_start: bool,
} }
struct MpvConnectionArgs { struct MpvConnectionArgs<'a> {
socket_path: String, socket_path: String,
executable_path: Option<String>, executable_path: Option<String>,
config_file: &'a NamedTempFile,
auto_start: bool, auto_start: bool,
force_auto_start: bool, force_auto_start: bool,
} }
async fn connect_to_mpv(args: &MpvConnectionArgs) -> anyhow::Result<(Mpv, Option<Child>)> { /// Helper function to resolve a hostname to an IP address.
log::debug!("Connecting to mpv"); /// Why is this not in the standard library? >:(
debug_assert!(
!args.force_auto_start || args.auto_start,
"force_auto_start requires auto_start"
);
let socket_path = Path::new(&args.socket_path);
if !socket_path.exists() {
log::debug!("Mpv socket not found at {}", &args.socket_path);
if !args.auto_start {
panic!("Mpv socket not found at {}", &args.socket_path);
}
log::debug!("Ensuring parent dir of mpv socket exists");
let parent_dir = Path::new(&args.socket_path)
.parent()
.context("Failed to get parent dir of mpv socket")?;
if !parent_dir.is_dir() {
create_dir_all(parent_dir).context("Failed to create parent dir of mpv socket")?;
}
} else {
log::debug!("Existing mpv socket found at {}", &args.socket_path);
if args.force_auto_start {
log::debug!("Removing mpv socket");
std::fs::remove_file(&args.socket_path)?;
}
}
let process_handle = if args.auto_start {
log::info!("Starting mpv with socket at {}", &args.socket_path);
// TODO: try to fetch mpv from PATH
Some(
Command::new(args.executable_path.as_deref().unwrap_or("mpv"))
.arg(format!("--input-ipc-server={}", &args.socket_path))
.arg("--idle")
.arg("--force-window")
// .arg("--fullscreen")
// .arg("--no-terminal")
// .arg("--load-unsafe-playlists")
.arg("--keep-open") // Keep last frame of video on end of video
.spawn()
.context("Failed to start mpv")?,
)
} else {
None
};
// Wait for mpv to create the socket
if tokio::time::timeout(tokio::time::Duration::from_millis(500), async {
while !&socket_path.exists() {
log::debug!("Waiting for mpv socket at {}", &args.socket_path);
tokio::time::sleep(tokio::time::Duration::from_millis(10)).await;
}
})
.await
.is_err()
{
return Err(anyhow::anyhow!(
"Failed to connect to mpv socket: {}",
&args.socket_path
));
}
Ok((
Mpv::connect(&args.socket_path).context(format!(
"Failed to connect to mpv socket: {}",
&args.socket_path
))?,
process_handle,
))
}
async fn resolve(host: &str) -> anyhow::Result<IpAddr> { async fn resolve(host: &str) -> anyhow::Result<IpAddr> {
let addr = format!("{}:0", host); let addr = format!("{}:0", host);
let addresses = tokio::net::lookup_host(addr).await?; let addresses = tokio::net::lookup_host(addr).await?;
@ -130,42 +61,147 @@ async fn resolve(host: &str) -> anyhow::Result<IpAddr> {
.ok_or_else(|| anyhow::anyhow!("Failed to resolve address")) .ok_or_else(|| anyhow::anyhow!("Failed to resolve address"))
} }
/// Helper function that spawns a tokio thread that
/// continuously sends a ping to systemd watchdog, if enabled.
async fn setup_systemd_watchdog_thread() -> anyhow::Result<()> {
let mut watchdog_microsecs: u64 = 0;
if sd_notify::watchdog_enabled(true, &mut watchdog_microsecs) {
watchdog_microsecs = watchdog_microsecs.div_ceil(2);
tokio::spawn(async move {
log::debug!(
"Starting systemd watchdog thread with {} millisecond interval",
watchdog_microsecs.div_ceil(1000)
);
loop {
tokio::time::sleep(tokio::time::Duration::from_micros(watchdog_microsecs)).await;
if let Err(err) = sd_notify::notify(false, &[sd_notify::NotifyState::Watchdog]) {
log::warn!("Failed to notify systemd watchdog: {}", err);
} else {
log::trace!("Ping sent to systemd watchdog");
}
}
});
} else {
log::info!("Watchdog not enabled, skipping");
}
Ok(())
}
async fn shutdown(mpv: Mpv, proc: Option<tokio::process::Child>) {
log::info!("Shutting down");
sd_notify::notify(false, &[sd_notify::NotifyState::Stopping]).unwrap_or_else(|e| {
log::warn!(
"Failed to notify systemd that the service is stopping: {}",
e
)
});
mpv.disconnect()
.await
.unwrap_or_else(|e| log::warn!("Failed to disconnect from mpv: {}", e));
if let Some(mut proc) = proc {
proc.kill()
.await
.unwrap_or_else(|e| log::warn!("Failed to kill mpv process: {}", e));
}
}
#[tokio::main] #[tokio::main]
async fn main() -> anyhow::Result<()> { async fn main() -> anyhow::Result<()> {
env_logger::init();
let args = Args::parse(); let args = Args::parse();
let systemd_mode = args.systemd && sd_notify::booted().unwrap_or(false);
if systemd_mode {
JournalLog::new()
.context("Failed to initialize journald logging")?
.install()
.context("Failed to install journald logger")?;
log::set_max_level(args.verbose.log_level_filter());
log::debug!("Running with systemd integration");
setup_systemd_watchdog_thread().await?;
} else {
env_logger::Builder::new()
.filter_level(args.verbose.log_level_filter())
.init();
log::info!("Running without systemd integration");
}
let mpv_config_file = create_mpv_config_file(args.mpv_config_file)?;
let (mpv, proc) = connect_to_mpv(&MpvConnectionArgs { let (mpv, proc) = connect_to_mpv(&MpvConnectionArgs {
socket_path: args.mpv_socket_path, socket_path: args.mpv_socket_path,
executable_path: args.mpv_executable_path, executable_path: args.mpv_executable_path,
config_file: &mpv_config_file,
auto_start: args.auto_start_mpv, auto_start: args.auto_start_mpv,
force_auto_start: args.force_auto_start, force_auto_start: args.force_auto_start,
}) })
.await?; .await
.context("Failed to connect to mpv")?;
let addr = SocketAddr::new(resolve(&args.host).await?, args.port); if let Err(e) = show_grzegorz_image(mpv.clone()).await {
log::info!("Starting API on {}", addr); log::warn!("Could not show Grzegorz image: {}", e);
}
let mpv = Arc::new(Mutex::new(mpv)); let addr = match resolve(&args.host)
let app = Router::new().nest("/api", api::rest_api_routes(mpv)); .await
.context(format!("Failed to resolve address: {}", &args.host))
{
Ok(addr) => addr,
Err(e) => {
log::error!("{}", e);
shutdown(mpv, proc).await;
return Err(e);
}
};
let socket_addr = SocketAddr::new(addr, args.port);
log::info!("Starting API on {}", socket_addr);
let app = Router::new()
.nest("/api", api::rest_api_routes(mpv.clone()))
.merge(api::rest_api_docs(mpv.clone()));
let listener = match tokio::net::TcpListener::bind(&socket_addr)
.await
.context(format!("Failed to bind API server to '{}'", &socket_addr))
{
Ok(listener) => listener,
Err(e) => {
log::error!("{}", e);
shutdown(mpv, proc).await;
return Err(e);
}
};
if systemd_mode {
match sd_notify::notify(false, &[sd_notify::NotifyState::Ready])
.context("Failed to notify systemd that the service is ready")
{
Ok(_) => log::trace!("Notified systemd that the service is ready"),
Err(e) => {
log::error!("{}", e);
shutdown(mpv, proc).await;
return Err(e);
}
}
}
if let Some(mut proc) = proc { if let Some(mut proc) = proc {
tokio::select! { tokio::select! {
exit_status = proc.wait() => { exit_status = proc.wait() => {
log::warn!("mpv process exited with status: {}", exit_status?); log::warn!("mpv process exited with status: {}", exit_status?);
shutdown(mpv, Some(proc)).await;
} }
_ = tokio::signal::ctrl_c() => { _ = tokio::signal::ctrl_c() => {
log::info!("Received Ctrl-C, exiting"); log::info!("Received Ctrl-C, exiting");
proc.kill().await?; shutdown(mpv, Some(proc)).await;
} }
result = async { result = axum::serve(listener, app.into_make_service()) => {
match Server::try_bind(&addr.clone()).context("Failed to bind server") {
Ok(server) => server.serve(app.into_make_service()).await.context("Failed to serve app"),
Err(err) => Err(err),
}
} => {
log::info!("API server exited"); log::info!("API server exited");
proc.kill().await?; shutdown(mpv, Some(proc)).await;
result?; result?;
} }
} }
@ -173,12 +209,17 @@ async fn main() -> anyhow::Result<()> {
tokio::select! { tokio::select! {
_ = tokio::signal::ctrl_c() => { _ = tokio::signal::ctrl_c() => {
log::info!("Received Ctrl-C, exiting"); log::info!("Received Ctrl-C, exiting");
shutdown(mpv.clone(), None).await;
} }
_ = Server::bind(&addr.clone()).serve(app.into_make_service()) => { result = axum::serve(listener, app.into_make_service()) => {
log::info!("API server exited"); log::info!("API server exited");
shutdown(mpv.clone(), None).await;
result?;
} }
} }
} }
std::mem::drop(mpv_config_file);
Ok(()) Ok(())
} }

135
src/mpv_setup.rs Normal file
View File

@ -0,0 +1,135 @@
use std::{fs::create_dir_all, io::Write, path::Path};
use anyhow::Context;
use mpvipc_async::{Mpv, MpvExt};
use tempfile::NamedTempFile;
use tokio::process::{Child, Command};
use crate::MpvConnectionArgs;
const DEFAULT_MPV_CONFIG_CONTENT: &str = include_str!("../assets/default-mpv.conf");
const THE_MAN_PNG: &[u8] = include_bytes!("../assets/the_man.png");
pub fn create_mpv_config_file(args_config_file: Option<String>) -> anyhow::Result<NamedTempFile> {
let file_content = if let Some(path) = args_config_file {
if !Path::new(&path).exists() {
anyhow::bail!("Mpv config file not found at {}", &path);
}
std::fs::read_to_string(&path).context("Failed to read mpv config file")?
} else {
DEFAULT_MPV_CONFIG_CONTENT.to_string()
};
let tmpfile = tempfile::Builder::new()
.prefix("mpv-")
.rand_bytes(8)
.suffix(".conf")
.tempfile()?;
tmpfile.reopen()?.write_all(file_content.as_bytes())?;
Ok(tmpfile)
}
pub async fn connect_to_mpv<'a>(
args: &MpvConnectionArgs<'a>,
) -> anyhow::Result<(Mpv, Option<Child>)> {
log::debug!("Connecting to mpv");
debug_assert!(
!args.force_auto_start || args.auto_start,
"force_auto_start requires auto_start"
);
let socket_path = Path::new(&args.socket_path);
if !socket_path.exists() {
log::debug!("Mpv socket not found at {}", &args.socket_path);
if !args.auto_start {
panic!("Mpv socket not found at {}", &args.socket_path);
}
log::debug!("Ensuring parent dir of mpv socket exists");
let parent_dir = Path::new(&args.socket_path)
.parent()
.context("Failed to get parent dir of mpv socket")?;
if !parent_dir.is_dir() {
create_dir_all(parent_dir).context("Failed to create parent dir of mpv socket")?;
}
} else {
log::debug!("Existing mpv socket found at {}", &args.socket_path);
if args.force_auto_start {
log::debug!("Removing mpv socket");
std::fs::remove_file(&args.socket_path)?;
}
}
let process_handle = if args.auto_start {
log::info!("Starting mpv with socket at {}", &args.socket_path);
// TODO: try to fetch mpv from PATH
Some(
Command::new(args.executable_path.as_deref().unwrap_or("mpv"))
.arg(format!("--input-ipc-server={}", &args.socket_path))
.arg("--idle")
.arg("--force-window")
.arg("--fullscreen")
.arg("--no-config")
.arg(format!(
"--include={}",
&args.config_file.path().to_string_lossy()
))
// .arg("--no-terminal")
.arg("--load-unsafe-playlists")
.arg("--keep-open") // Keep last frame of video on end of video
.spawn()
.context("Failed to start mpv")?,
)
} else {
None
};
// Wait for mpv to create the socket
if tokio::time::timeout(tokio::time::Duration::from_millis(500), async {
while !&socket_path.exists() {
log::debug!("Waiting for mpv socket at {}", &args.socket_path);
tokio::time::sleep(tokio::time::Duration::from_millis(10)).await;
}
})
.await
.is_err()
{
return Err(anyhow::anyhow!(
"Failed to connect to mpv socket: {}",
&args.socket_path
));
}
Ok((
Mpv::connect(&args.socket_path).await.context(format!(
"Failed to connect to mpv socket: {}",
&args.socket_path
))?,
process_handle,
))
}
pub async fn show_grzegorz_image(mpv: Mpv) -> anyhow::Result<()> {
let path = std::env::temp_dir().join("the_man.png");
std::fs::write(path.as_path(), THE_MAN_PNG)?;
mpv.playlist_clear().await?;
mpv.playlist_add(
path.to_string_lossy().as_ref(),
mpvipc_async::PlaylistAddTypeOptions::File,
mpvipc_async::PlaylistAddOptions::Append,
)
.await?;
mpv.next().await?;
Ok(())
}