Dumb af systemctl thingy
This commit is contained in:
Maix0
commit
7807eb0c42
7 changed files with 691 additions and 0 deletions
1
.envrc
Normal file
1
.envrc
Normal file
|
|
@ -0,0 +1 @@
|
|||
use flake . --impure
|
||||
12
Cargo.toml
Normal file
12
Cargo.toml
Normal file
|
|
@ -0,0 +1,12 @@
|
|||
[package]
|
||||
name = "botloc"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
[dependencies]
|
||||
axum = { version = "0.7.5", features = ["multipart", "macros"] }
|
||||
serde = { version = "1.0.203", features = ["derive"] }
|
||||
serde_json = "1.0.118"
|
||||
tokio = { version = "1.38.0", features = ["full"] }
|
||||
tracing = "0.1.40"
|
||||
tracing-subscriber = "0.3.18"
|
||||
314
flake.lock
generated
Normal file
314
flake.lock
generated
Normal file
|
|
@ -0,0 +1,314 @@
|
|||
{
|
||||
"nodes": {
|
||||
"cargo-semver-checks": {
|
||||
"inputs": {
|
||||
"flake-utils": "flake-utils",
|
||||
"naersk": "naersk",
|
||||
"nixpkgs": "nixpkgs_2"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1718446388,
|
||||
"narHash": "sha256-/4a5CS5EdI+g1Y/nzOeAmI04olLDCk8SDhQbPqL82s4=",
|
||||
"owner": "Maix0",
|
||||
"repo": "cargo-semver-checks-flake",
|
||||
"rev": "10641cbb61ecb2ec40c4847080a08c8d4ee8bec0",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "Maix0",
|
||||
"repo": "cargo-semver-checks-flake",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"cargo-workspace": {
|
||||
"inputs": {
|
||||
"cargo-workspaces-git": "cargo-workspaces-git",
|
||||
"flake-utils": "flake-utils_2",
|
||||
"naersk": "naersk_2",
|
||||
"nixpkgs": "nixpkgs_4"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1718446390,
|
||||
"narHash": "sha256-YAgg27LQwK6BfOIut/U8zqDN87L0SxU9SRC98Bsp16w=",
|
||||
"owner": "Maix0",
|
||||
"repo": "cargo-ws-flake",
|
||||
"rev": "bbb8a52d1a294549f81a52c226fa3c085e44e5af",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "Maix0",
|
||||
"repo": "cargo-ws-flake",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"cargo-workspaces-git": {
|
||||
"flake": false,
|
||||
"locked": {
|
||||
"lastModified": 1683825807,
|
||||
"narHash": "sha256-WlJu8FLaTMzR5JCU5eASHfmJ3C4IdwlPZ6//m5xOb1k=",
|
||||
"owner": "pksunkara",
|
||||
"repo": "cargo-workspaces",
|
||||
"rev": "8bf9c4ab519eeed45e83cbc64aa646431bddd899",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "pksunkara",
|
||||
"ref": "v0.2.41",
|
||||
"repo": "cargo-workspaces",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"flake-utils": {
|
||||
"inputs": {
|
||||
"systems": "systems"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1710146030,
|
||||
"narHash": "sha256-SZ5L6eA7HJ/nmkzGG7/ISclqe6oZdOZTNoesiInkXPQ=",
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"rev": "b1d9ab70662946ef0850d488da1c9019f3a9752a",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"flake-utils_2": {
|
||||
"inputs": {
|
||||
"systems": "systems_2"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1710146030,
|
||||
"narHash": "sha256-SZ5L6eA7HJ/nmkzGG7/ISclqe6oZdOZTNoesiInkXPQ=",
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"rev": "b1d9ab70662946ef0850d488da1c9019f3a9752a",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"flake-utils_3": {
|
||||
"inputs": {
|
||||
"systems": "systems_3"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1710146030,
|
||||
"narHash": "sha256-SZ5L6eA7HJ/nmkzGG7/ISclqe6oZdOZTNoesiInkXPQ=",
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"rev": "b1d9ab70662946ef0850d488da1c9019f3a9752a",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"naersk": {
|
||||
"inputs": {
|
||||
"nixpkgs": "nixpkgs"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1717067539,
|
||||
"narHash": "sha256-oIs5EF+6VpHJRvvpVWuqCYJMMVW/6h59aYUv9lABLtY=",
|
||||
"owner": "nix-community",
|
||||
"repo": "naersk",
|
||||
"rev": "fa19d8c135e776dc97f4dcca08656a0eeb28d5c0",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "nix-community",
|
||||
"repo": "naersk",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"naersk_2": {
|
||||
"inputs": {
|
||||
"nixpkgs": "nixpkgs_3"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1717067539,
|
||||
"narHash": "sha256-oIs5EF+6VpHJRvvpVWuqCYJMMVW/6h59aYUv9lABLtY=",
|
||||
"owner": "nix-community",
|
||||
"repo": "naersk",
|
||||
"rev": "fa19d8c135e776dc97f4dcca08656a0eeb28d5c0",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "nix-community",
|
||||
"repo": "naersk",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"nixpkgs": {
|
||||
"locked": {
|
||||
"lastModified": 0,
|
||||
"narHash": "sha256-yZKhxVIKd2lsbOqYd5iDoUIwsRZFqE87smE2Vzf6Ck0=",
|
||||
"path": "/nix/store/5jgh89kgmrb687c254wxdac4cj5hqjw8-source",
|
||||
"type": "path"
|
||||
},
|
||||
"original": {
|
||||
"id": "nixpkgs",
|
||||
"type": "indirect"
|
||||
}
|
||||
},
|
||||
"nixpkgs_2": {
|
||||
"locked": {
|
||||
"lastModified": 1718276985,
|
||||
"narHash": "sha256-u1fA0DYQYdeG+5kDm1bOoGcHtX0rtC7qs2YA2N1X++I=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "3f84a279f1a6290ce154c5531378acc827836fbb",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "NixOS",
|
||||
"ref": "nixpkgs-unstable",
|
||||
"repo": "nixpkgs",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"nixpkgs_3": {
|
||||
"locked": {
|
||||
"lastModified": 0,
|
||||
"narHash": "sha256-yZKhxVIKd2lsbOqYd5iDoUIwsRZFqE87smE2Vzf6Ck0=",
|
||||
"path": "/nix/store/5jgh89kgmrb687c254wxdac4cj5hqjw8-source",
|
||||
"type": "path"
|
||||
},
|
||||
"original": {
|
||||
"id": "nixpkgs",
|
||||
"type": "indirect"
|
||||
}
|
||||
},
|
||||
"nixpkgs_4": {
|
||||
"locked": {
|
||||
"lastModified": 1718276985,
|
||||
"narHash": "sha256-u1fA0DYQYdeG+5kDm1bOoGcHtX0rtC7qs2YA2N1X++I=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "3f84a279f1a6290ce154c5531378acc827836fbb",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "NixOS",
|
||||
"ref": "nixpkgs-unstable",
|
||||
"repo": "nixpkgs",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"nixpkgs_5": {
|
||||
"locked": {
|
||||
"lastModified": 1719223410,
|
||||
"narHash": "sha256-jtIo8xR0Zp4SalIwmD+OdCwHF4l7OU6PD63UUK4ckt4=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "efb39c6052f3ce51587cf19733f5f4e5d515aa13",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "NixOS",
|
||||
"ref": "nixpkgs-unstable",
|
||||
"repo": "nixpkgs",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"nixpkgs_6": {
|
||||
"locked": {
|
||||
"lastModified": 1718428119,
|
||||
"narHash": "sha256-WdWDpNaq6u1IPtxtYHHWpl5BmabtpmLnMAx0RdJ/vo8=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "e6cea36f83499eb4e9cd184c8a8e823296b50ad5",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "NixOS",
|
||||
"ref": "nixpkgs-unstable",
|
||||
"repo": "nixpkgs",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"root": {
|
||||
"inputs": {
|
||||
"cargo-semver-checks": "cargo-semver-checks",
|
||||
"cargo-workspace": "cargo-workspace",
|
||||
"flake-utils": "flake-utils_3",
|
||||
"nixpkgs": "nixpkgs_5",
|
||||
"rust-overlay": "rust-overlay"
|
||||
}
|
||||
},
|
||||
"rust-overlay": {
|
||||
"inputs": {
|
||||
"nixpkgs": "nixpkgs_6"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1719281921,
|
||||
"narHash": "sha256-LIBMfhM9pMOlEvBI757GOK5l0R58SRi6YpwfYMbf4yc=",
|
||||
"owner": "oxalica",
|
||||
"repo": "rust-overlay",
|
||||
"rev": "b6032d3a404d8a52ecfc8571ff0c26dfbe221d07",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "oxalica",
|
||||
"repo": "rust-overlay",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"systems": {
|
||||
"locked": {
|
||||
"lastModified": 1681028828,
|
||||
"narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"systems_2": {
|
||||
"locked": {
|
||||
"lastModified": 1681028828,
|
||||
"narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"type": "github"
|
||||
}
|
||||
},
|
||||
"systems_3": {
|
||||
"locked": {
|
||||
"lastModified": 1681028828,
|
||||
"narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
"owner": "nix-systems",
|
||||
"repo": "default",
|
||||
"type": "github"
|
||||
}
|
||||
}
|
||||
},
|
||||
"root": "root",
|
||||
"version": 7
|
||||
}
|
||||
99
flake.nix
Normal file
99
flake.nix
Normal file
|
|
@ -0,0 +1,99 @@
|
|||
{
|
||||
description = "A basic flake with a shell";
|
||||
inputs = {
|
||||
nixpkgs.url = "github:NixOS/nixpkgs/nixpkgs-unstable";
|
||||
flake-utils.url = "github:numtide/flake-utils";
|
||||
rust-overlay.url = "github:oxalica/rust-overlay";
|
||||
cargo-workspace.url = "github:Maix0/cargo-ws-flake";
|
||||
cargo-semver-checks.url = "github:Maix0/cargo-semver-checks-flake";
|
||||
};
|
||||
outputs = {
|
||||
self,
|
||||
nixpkgs,
|
||||
flake-utils,
|
||||
rust-overlay,
|
||||
...
|
||||
} @ inputs:
|
||||
flake-utils.lib.eachDefaultSystem (system: let
|
||||
pkgs = import nixpkgs {
|
||||
inherit system;
|
||||
overlays = [(import rust-overlay)];
|
||||
};
|
||||
packageIf = name: packageDef:
|
||||
if builtins.hasAttr name inputs
|
||||
then [(packageDef inputs.${name})]
|
||||
else [];
|
||||
buildRustToolchain = toolchainDef:
|
||||
if builtins.isString toolchainDef
|
||||
then let
|
||||
split = pkgs.lib.strings.splitString "/" toolchainDef;
|
||||
toolchainType =
|
||||
if (builtins.length split) > 3
|
||||
then throw "You can only specify a single version using `{type}/{version}/{profile}`"
|
||||
else if (builtins.length split) >= 1
|
||||
then builtins.elemAt split 0
|
||||
else throw "You must specify at least a toolchain version with this format `{type}/{version}/{profile}`";
|
||||
toolchainVersion =
|
||||
if (builtins.length split) > 3
|
||||
then throw "You can only specify a single version using `{type}/{version}/{profile}`"
|
||||
else if (builtins.length split) >= 2
|
||||
then builtins.elemAt split 1
|
||||
else "latest";
|
||||
toolchainProfile =
|
||||
if (builtins.length split) > 3
|
||||
then throw "You can only specify a single version using `{type}/{version}/{profile}`"
|
||||
else if (builtins.length split) >= 3
|
||||
then builtins.elemAt split 2
|
||||
else "default";
|
||||
in
|
||||
buildRustToolchain {
|
||||
toolchain = toolchainType;
|
||||
version = toolchainVersion;
|
||||
profile = toolchainProfile;
|
||||
}
|
||||
else if builtins.isAttrs toolchainDef
|
||||
then
|
||||
if toolchainDef.toolchain == "stable"
|
||||
then pkgs.rust-bin.stable.${toolchainDef.version}.${toolchainDef.profile}.override (builtins.removeAttrs toolchainDef ["toolchain" "version" "profile"])
|
||||
else if toolchainDef.toolchain == "beta"
|
||||
then pkgs.rust-bin.beta.${toolchainDef.version}.${toolchainDef.profile}.override (builtins.removeAttrs toolchainDef ["toolchain" "version" "profile"])
|
||||
else if toolchainDef.toolchain == "nightly"
|
||||
then
|
||||
if toolchainDef.version == "latest"
|
||||
then pkgs.rust-bin.selectLatestNightlyWith (toolchain: toolchain.${toolchainDef.profile}.override (builtins.removeAttrs toolchainDef ["toolchain" "version" "profile"]))
|
||||
else pkgs.rust-bin.nightly.${toolchainDef.version}.${toolchainDef.profile}.override (builtins.removeAttrs toolchainDef ["toolchain" "version" "profile"])
|
||||
else throw "toolchain version isn't valid (not 'stable' or 'beta' or 'nightly')"
|
||||
else throw "toolchainDef isn't a string or an attr describing the toolchain";
|
||||
in {
|
||||
devShell = let
|
||||
rust_dev = buildRustToolchain "stable/latest/default";
|
||||
# {
|
||||
# extensions = [];
|
||||
# targets = ["x86_64-unknown-linux-gnu"];
|
||||
# };
|
||||
in
|
||||
pkgs.mkShell {
|
||||
packages = with pkgs;
|
||||
[
|
||||
rust_dev
|
||||
mold
|
||||
openssl
|
||||
pkg-config
|
||||
]
|
||||
++ (packageIf "cargo-semver-checks" (p: p.packages.${system}.default))
|
||||
++ (packageIf "cargo-workspace" (p: p.packages.${system}.default));
|
||||
|
||||
shellHook = ''
|
||||
export RUST_STD="${rust_dev}/share/doc/rust/html/std/index.html"
|
||||
|
||||
if [ -z $CARGO_TARGET_DIR ] then
|
||||
: "''${XDG_CACHE_HOME:="''$HOME/.cache"}"
|
||||
local hash path
|
||||
hash="$(sha1sum - <<< "$PWD" | head -c40)"
|
||||
path="''${PWD//[^a-zA-Z0-9]/}"
|
||||
export CARGO_TARGET_DIR="''${XDG_CACHE_HOME}/rust-targets/$hash-$path"
|
||||
fi
|
||||
'';
|
||||
};
|
||||
});
|
||||
}
|
||||
206
src/main.rs
Normal file
206
src/main.rs
Normal file
|
|
@ -0,0 +1,206 @@
|
|||
//! Run with
|
||||
//!
|
||||
//! ```not_rust
|
||||
//! cargo run -p example-readme
|
||||
//! ```
|
||||
|
||||
use axum::{
|
||||
http::StatusCode,
|
||||
response::{Html, IntoResponse, Redirect},
|
||||
routing::{get, post},
|
||||
Json, Router,
|
||||
};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||
use tracing::{error, info};
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
// initialize tracing
|
||||
tracing_subscriber::fmt::init();
|
||||
|
||||
// build our application with a route
|
||||
let app = Router::new()
|
||||
// `GET /` goes to `root`
|
||||
.route("/", get(root))
|
||||
.route("/status", get(status))
|
||||
.route("/stop", get(stop))
|
||||
.route("/start", get(start))
|
||||
.route("/restart", get(restart))
|
||||
.route("/config", get(get_config).post(post_config));
|
||||
|
||||
// run our app with hyper
|
||||
let listener = tokio::net::TcpListener::bind("127.0.0.1:3000")
|
||||
.await
|
||||
.unwrap();
|
||||
tracing::debug!("listening on {}", listener.local_addr().unwrap());
|
||||
axum::serve(listener, app).await.unwrap();
|
||||
}
|
||||
|
||||
// basic handler that responds with a static string
|
||||
async fn root() -> Html<&'static str> {
|
||||
Html(
|
||||
r#"
|
||||
<a href="/restart">restart</a><br>
|
||||
<a href="/stop">stop</a><br>
|
||||
<a href="/start">start</a><br>
|
||||
<a href="/status">status</a><br>
|
||||
<a href="/config">config</a><br>
|
||||
"#,
|
||||
)
|
||||
}
|
||||
|
||||
async fn restart() -> Redirect {
|
||||
info!("Requested to restart the bot");
|
||||
tokio::spawn(async {
|
||||
tokio::process::Command::new("systemctl")
|
||||
.args(["--user", "restart", "botloc.service"])
|
||||
.spawn()
|
||||
.unwrap()
|
||||
});
|
||||
Redirect::to("/")
|
||||
}
|
||||
|
||||
async fn start() -> Redirect {
|
||||
info!("Requested to start the bot");
|
||||
tokio::spawn(async {
|
||||
tokio::process::Command::new("systemctl")
|
||||
.args(["--user", "start", "botloc.service"])
|
||||
.spawn()
|
||||
.unwrap()
|
||||
});
|
||||
Redirect::to("/")
|
||||
}
|
||||
|
||||
async fn stop() -> Redirect {
|
||||
info!("Requested to stop the bot");
|
||||
tokio::spawn(async {
|
||||
tokio::process::Command::new("systemctl")
|
||||
.args(["--user", "stop", "botloc.service"])
|
||||
.spawn()
|
||||
.unwrap()
|
||||
});
|
||||
Redirect::to("/")
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
struct BotConfig {
|
||||
piscine: Vec<String>,
|
||||
}
|
||||
|
||||
async fn get_config() -> Result<Json<BotConfig>, (StatusCode, String)> {
|
||||
info!("Requested config");
|
||||
let Ok(mut file) = tokio::fs::File::open(std::env::var("CONFIG_PATH").map_err(|e| {
|
||||
error!("Failed to open config file: {e}");
|
||||
(
|
||||
StatusCode::INTERNAL_SERVER_ERROR,
|
||||
"please set env CONFIG_PATH".to_string(),
|
||||
)
|
||||
})?)
|
||||
.await
|
||||
.map_err(|e| {
|
||||
error!("Failed to open config file: {e}");
|
||||
e
|
||||
}) else {
|
||||
return Err((
|
||||
StatusCode::INTERNAL_SERVER_ERROR,
|
||||
format!(
|
||||
"failed to open config file: {}",
|
||||
std::env::var("CONFIG_PATH").unwrap()
|
||||
),
|
||||
));
|
||||
};
|
||||
|
||||
let mut s = String::new();
|
||||
if file
|
||||
.read_to_string(&mut s)
|
||||
.await
|
||||
.map_err(|e| {
|
||||
error!("Failed to open config file: {e}");
|
||||
e
|
||||
})
|
||||
.is_err()
|
||||
{
|
||||
return Err((
|
||||
StatusCode::INTERNAL_SERVER_ERROR,
|
||||
format!(
|
||||
"Failed to read config file at {}",
|
||||
std::env::var("CONFIG_PATH").unwrap()
|
||||
),
|
||||
));
|
||||
};
|
||||
let Ok(val) = serde_json::from_str::<BotConfig>(&s).map_err(|e| {
|
||||
error!("Failed to open config file: {e}");
|
||||
e
|
||||
}) else {
|
||||
return Err((
|
||||
StatusCode::INTERNAL_SERVER_ERROR,
|
||||
format!(
|
||||
"Failed to read config file as json at {}",
|
||||
std::env::var("CONFIG_PATH").unwrap()
|
||||
),
|
||||
));
|
||||
};
|
||||
Ok(Json(val))
|
||||
}
|
||||
|
||||
async fn post_config(Json(body): Json<BotConfig>) -> (StatusCode, String) {
|
||||
info!("Posted config");
|
||||
let Ok(mut file) = tokio::fs::File::open(
|
||||
match std::env::var("CONFIG_PATH").map_err(|e| {
|
||||
error!("Unset env {e}");
|
||||
(
|
||||
StatusCode::INTERNAL_SERVER_ERROR,
|
||||
"please set env CONFIG_PATH".to_string(),
|
||||
)
|
||||
}) {
|
||||
Ok(v) => v,
|
||||
Err(e) => return e,
|
||||
},
|
||||
)
|
||||
.await
|
||||
.map_err(|e| {
|
||||
error!("Failed to open config file: {e}");
|
||||
e
|
||||
}) else {
|
||||
return (
|
||||
StatusCode::INTERNAL_SERVER_ERROR,
|
||||
format!(
|
||||
"failed to open config file: {}",
|
||||
std::env::var("CONFIG_PATH").unwrap()
|
||||
),
|
||||
);
|
||||
};
|
||||
|
||||
match serde_json::to_string_pretty(&body) {
|
||||
Err(e) => {
|
||||
error!("Failed to convert to json {e}");
|
||||
return (StatusCode::OK, "Done".to_string());
|
||||
}
|
||||
Ok(s) => match file.write(s.as_bytes()).await {
|
||||
Err(e) => error!("Got an error when writing file: {e}"),
|
||||
Ok(_) => (),
|
||||
},
|
||||
}
|
||||
|
||||
(StatusCode::OK, "Done".to_string())
|
||||
}
|
||||
|
||||
async fn status() -> Result<String, StatusCode> {
|
||||
info!("Requested status");
|
||||
let mut output = tokio::process::Command::new("systemctl")
|
||||
.args(["--user", "status", "botloc.service"])
|
||||
.output()
|
||||
.await
|
||||
// let mut output = child.wait_with_output().await
|
||||
.map_err(|e| {
|
||||
error!("Error with systemctl status {e}");
|
||||
StatusCode::INTERNAL_SERVER_ERROR
|
||||
})?;
|
||||
output.stdout.push(b'\n');
|
||||
output.stdout.append(&mut output.stderr);
|
||||
String::from_utf8(output.stdout).map_err(|e| {
|
||||
error!("Error with systemctl status output {e}");
|
||||
StatusCode::INTERNAL_SERVER_ERROR
|
||||
})
|
||||
}
|
||||
58
src/utils/memory_db_util.rs
Normal file
58
src/utils/memory_db_util.rs
Normal file
|
|
@ -0,0 +1,58 @@
|
|||
use std::{
|
||||
collections::HashMap,
|
||||
sync::{Arc, Mutex},
|
||||
thread::spawn,
|
||||
time::SystemTime,
|
||||
};
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct AxumState {
|
||||
db: Arc<Mutex<HashMap<String, ItemOauthAxum>>>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct ItemOauthAxum {
|
||||
pub verifier: String,
|
||||
pub created_at: SystemTime,
|
||||
}
|
||||
|
||||
impl AxumState {
|
||||
pub fn new() -> Self {
|
||||
let db: Arc<Mutex<HashMap<String, ItemOauthAxum>>> = Arc::new(Mutex::new(HashMap::new()));
|
||||
let db_binding = Arc::clone(&db);
|
||||
spawn(move || loop {
|
||||
std::thread::sleep(std::time::Duration::from_secs(10));
|
||||
let mut db = db_binding.lock().unwrap();
|
||||
let now = SystemTime::now();
|
||||
db.retain(|_, item| now.duration_since(item.created_at).unwrap().as_secs() < 900);
|
||||
});
|
||||
AxumState {
|
||||
db: Arc::clone(&db),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn get(&self, key: String) -> Option<String> {
|
||||
let db = self.db.lock().unwrap();
|
||||
if let Some(item) = db.get(&key) {
|
||||
Some(item.verifier.clone())
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
pub fn set(&self, key: String, value: String) {
|
||||
let mut db = self.db.lock().unwrap();
|
||||
db.insert(
|
||||
key,
|
||||
ItemOauthAxum {
|
||||
verifier: value,
|
||||
created_at: SystemTime::now(),
|
||||
},
|
||||
);
|
||||
}
|
||||
|
||||
pub fn get_all_items(&self) -> Vec<ItemOauthAxum> {
|
||||
let db = self.db.lock().unwrap();
|
||||
db.values().cloned().collect::<Vec<ItemOauthAxum>>()
|
||||
}
|
||||
}
|
||||
1
src/utils/mod.rs
Normal file
1
src/utils/mod.rs
Normal file
|
|
@ -0,0 +1 @@
|
|||
pub mod memory_db_utils;
|
||||
Loading…
Add table
Add a link
Reference in a new issue