Compare commits
No commits in common. "main" and "v0.1.5" have entirely different histories.
9 changed files with 47 additions and 161 deletions
|
@ -16,9 +16,17 @@ jobs:
|
||||||
source "$HOME/.cargo/env"
|
source "$HOME/.cargo/env"
|
||||||
cargo build --release
|
cargo build --release
|
||||||
shell: bash
|
shell: bash
|
||||||
|
- uses: https://code.forgejo.org/actions/upload-artifact@v3
|
||||||
|
with:
|
||||||
|
name: teamspeak-query-lib
|
||||||
|
path: ${{ github.workspace }}/target/release/teamspeak-query-lib
|
||||||
|
- uses: https://code.forgejo.org/actions/download-artifact@v3
|
||||||
|
with:
|
||||||
|
name: teamspeak-query-lib
|
||||||
|
path: /tmp/artifacts
|
||||||
|
shell: bash
|
||||||
- id: copy-ts-control-artificat
|
- id: copy-ts-control-artificat
|
||||||
run: |
|
run: |
|
||||||
cp ${{ github.workspace }}/target/release/teamspeak-query-lib /tmp/artifacts
|
|
||||||
cp ${{ github.workspace }}/ts-control /tmp/artifacts
|
cp ${{ github.workspace }}/ts-control /tmp/artifacts
|
||||||
shell: bash
|
shell: bash
|
||||||
- uses: https://code.forgejo.org/actions/forgejo-release@v1
|
- uses: https://code.forgejo.org/actions/forgejo-release@v1
|
||||||
|
|
2
Cargo.lock
generated
2
Cargo.lock
generated
|
@ -182,7 +182,7 @@ dependencies = [
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "teamspeak-query-lib"
|
name = "teamspeak-query-lib"
|
||||||
version = "0.1.6"
|
version = "0.1.5"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"clap",
|
"clap",
|
||||||
"serde",
|
"serde",
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
[package]
|
[package]
|
||||||
name = "teamspeak-query-lib"
|
name = "teamspeak-query-lib"
|
||||||
version = "0.1.6"
|
version = "0.1.5"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
|
|
4
Makefile
4
Makefile
|
@ -2,13 +2,11 @@
|
||||||
|
|
||||||
INSTALL_DIR = $(HOME)/.local/bin
|
INSTALL_DIR = $(HOME)/.local/bin
|
||||||
|
|
||||||
export PATH := target/debug:$(PATH)
|
|
||||||
|
|
||||||
build:
|
build:
|
||||||
@cargo build
|
@cargo build
|
||||||
|
|
||||||
run: build
|
run: build
|
||||||
./ts-control
|
@PATH=$(PWD)/target/debug:$(PATH) ./ts-control
|
||||||
|
|
||||||
install:
|
install:
|
||||||
mkdir -p "$(INSTALL_DIR)"
|
mkdir -p "$(INSTALL_DIR)"
|
||||||
|
|
12
src/cli.rs
12
src/cli.rs
|
@ -1,4 +1,4 @@
|
||||||
use clap::{Args, Parser, Subcommand, ValueEnum};
|
use clap::{Args, Parser, Subcommand};
|
||||||
use telnet::Telnet;
|
use telnet::Telnet;
|
||||||
|
|
||||||
use crate::parameter::ParameterList;
|
use crate::parameter::ParameterList;
|
||||||
|
@ -17,7 +17,6 @@ struct Cli {
|
||||||
|
|
||||||
#[derive(Subcommand)]
|
#[derive(Subcommand)]
|
||||||
pub enum Commands {
|
pub enum Commands {
|
||||||
Info,
|
|
||||||
Channels(ChannelsArgs),
|
Channels(ChannelsArgs),
|
||||||
Clients,
|
Clients,
|
||||||
Fetch(FetchArgs),
|
Fetch(FetchArgs),
|
||||||
|
@ -80,18 +79,9 @@ pub struct UpdateArgs {
|
||||||
speakers: Option<bool>,
|
speakers: Option<bool>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone, ValueEnum)]
|
|
||||||
pub enum EventArgsFilter {
|
|
||||||
All,
|
|
||||||
Mine,
|
|
||||||
Others,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Args)]
|
#[derive(Args)]
|
||||||
pub struct EventArgs {
|
pub struct EventArgs {
|
||||||
pub event: Vec<EventType>,
|
pub event: Vec<EventType>,
|
||||||
#[arg(long, short)]
|
|
||||||
pub filter: Option<EventArgsFilter>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl FetchArgs {
|
impl FetchArgs {
|
||||||
|
|
|
@ -1,7 +1,6 @@
|
||||||
use telnet::Telnet;
|
use telnet::Telnet;
|
||||||
|
|
||||||
use crate::{commands, wrappers};
|
use crate::{commands, wrappers};
|
||||||
use crate::cli::EventArgsFilter;
|
|
||||||
use crate::models::{Client, Event, EventType};
|
use crate::models::{Client, Event, EventType};
|
||||||
use crate::response::Response;
|
use crate::response::Response;
|
||||||
|
|
||||||
|
@ -28,30 +27,19 @@ pub fn handle_event_response(connection: &mut Telnet, event: Event, known_client
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn try_get_self_clid(connection: &mut Telnet) -> Option<i32> {
|
pub fn loop_response_reader(connection: &mut Telnet, self_clid: i32) {
|
||||||
wrappers::get_self_clid(connection)
|
|
||||||
.unwrap_or_default()
|
|
||||||
.parse()
|
|
||||||
.map(Some)
|
|
||||||
.unwrap_or(None)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn loop_response_reader(connection: &mut Telnet, filter: &EventArgsFilter) {
|
|
||||||
let mut known_clients = wrappers::get_clients(connection).unwrap_or_else(|_| Vec::new());
|
let mut known_clients = wrappers::get_clients(connection).unwrap_or_else(|_| Vec::new());
|
||||||
|
|
||||||
let mut self_clid: Option<i32> = try_get_self_clid(connection);
|
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
match commands::read_response(connection, true, String::new()) {
|
match commands::read_response(connection, true, String::new()) {
|
||||||
Ok(response) => {
|
Ok(response) => {
|
||||||
if let Response::Event(event_type, params) = response {
|
if let Response::Event(event_type, params) = response {
|
||||||
let event = Event::new(connection, event_type, params, &known_clients);
|
let event = Event::new(connection, event_type, params, &known_clients);
|
||||||
|
|
||||||
if self_clid.is_none() {
|
if let Some(client) = &event.client {
|
||||||
self_clid = try_get_self_clid(connection);
|
if client.clid == self_clid {
|
||||||
}
|
continue;
|
||||||
if !event.should_handle(filter, self_clid) {
|
}
|
||||||
continue;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
handle_event_response(connection, event, &mut known_clients);
|
handle_event_response(connection, event, &mut known_clients);
|
||||||
|
|
53
src/main.rs
53
src/main.rs
|
@ -1,8 +1,11 @@
|
||||||
use std::io::{Error, ErrorKind};
|
use std::io::{Error, ErrorKind};
|
||||||
|
use std::num::ParseIntError;
|
||||||
|
|
||||||
use telnet::Telnet;
|
use telnet::Telnet;
|
||||||
|
|
||||||
use crate::cli::{Commands, EventArgsFilter};
|
use crate::cli::Commands;
|
||||||
|
use crate::models::Channel;
|
||||||
|
use crate::models::Client;
|
||||||
|
|
||||||
mod wrappers;
|
mod wrappers;
|
||||||
mod commands;
|
mod commands;
|
||||||
|
@ -13,10 +16,14 @@ mod models;
|
||||||
mod response;
|
mod response;
|
||||||
mod command_utils;
|
mod command_utils;
|
||||||
|
|
||||||
fn result_or_error<T>(res: Result<Option<T>, String>, result_type: &str) -> Result<T, Error> {
|
fn channel_or_error(channel_res: Result<Option<Channel>, String>) -> Result<Channel, Error> {
|
||||||
let error_action = format!("find {}", result_type);
|
channel_res.map_err(|err| make_action_error("find channel", err))?
|
||||||
res.map_err(|err| make_action_error(&error_action, err))?
|
.ok_or_else(|| make_action_error("find channel", String::from("Not Found.")))
|
||||||
.ok_or_else(|| make_action_error(&error_action, String::from("Not Found.")))
|
}
|
||||||
|
|
||||||
|
fn client_or_error(client_res: Result<Option<Client>, String>) -> Result<Client, Error> {
|
||||||
|
client_res.map_err(|err| make_action_error("find client", err))?
|
||||||
|
.ok_or_else(|| make_action_error("find client", String::from("Not Found.")))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn connect() -> Result<Telnet, Error> {
|
fn connect() -> Result<Telnet, Error> {
|
||||||
|
@ -47,26 +54,6 @@ fn main() -> Result<(), Error> {
|
||||||
// You can check for the existence of subcommands, and if found use their
|
// You can check for the existence of subcommands, and if found use their
|
||||||
// matches just as you would the top level cmd
|
// matches just as you would the top level cmd
|
||||||
match cli {
|
match cli {
|
||||||
Commands::Info => {
|
|
||||||
let client = match wrappers::find_self(&mut connection) {
|
|
||||||
Ok(client) => client,
|
|
||||||
Err(msg) => return Err(make_action_error("find self", msg))
|
|
||||||
};
|
|
||||||
|
|
||||||
let channel_name: String = match wrappers::find_channel(&mut connection, "cid", &client.cid.to_string(), true) {
|
|
||||||
Ok(channel) => match channel {
|
|
||||||
Some(channel) => channel.channel_name,
|
|
||||||
None => return Err(to_other_error("Channel not found.".to_string()))
|
|
||||||
},
|
|
||||||
Err(msg) => return Err(make_action_error("find self", msg))
|
|
||||||
};
|
|
||||||
|
|
||||||
println!("Client: {}", client.client_nickname);
|
|
||||||
println!("Channel: {}", channel_name);
|
|
||||||
println!("Channel ID: {}", client.cid);
|
|
||||||
println!("Client ID: {}", client.clid);
|
|
||||||
}
|
|
||||||
|
|
||||||
Commands::Channels(args) => {
|
Commands::Channels(args) => {
|
||||||
match wrappers::get_channels(&mut connection, args.spacers) {
|
match wrappers::get_channels(&mut connection, args.spacers) {
|
||||||
Ok(channels) => {
|
Ok(channels) => {
|
||||||
|
@ -102,7 +89,7 @@ fn main() -> Result<(), Error> {
|
||||||
}
|
}
|
||||||
|
|
||||||
if args.want_client() {
|
if args.want_client() {
|
||||||
let client = result_or_error(args.client(&mut connection), "client")?;
|
let client = client_or_error(args.client(&mut connection))?;
|
||||||
|
|
||||||
match wrappers::fetch_client(&mut connection, &[client]) {
|
match wrappers::fetch_client(&mut connection, &[client]) {
|
||||||
Ok(_) => println!("Successfully fetched client."),
|
Ok(_) => println!("Successfully fetched client."),
|
||||||
|
@ -113,7 +100,7 @@ fn main() -> Result<(), Error> {
|
||||||
}
|
}
|
||||||
|
|
||||||
if args.want_channel() {
|
if args.want_channel() {
|
||||||
let channel = result_or_error(args.channel(&mut connection), "channel")?;
|
let channel = channel_or_error(args.channel(&mut connection))?;
|
||||||
|
|
||||||
match wrappers::fetch_channel(&mut connection, channel) {
|
match wrappers::fetch_channel(&mut connection, channel) {
|
||||||
Ok(_) => println!("Successfully fetched channel."),
|
Ok(_) => println!("Successfully fetched channel."),
|
||||||
|
@ -137,8 +124,8 @@ fn main() -> Result<(), Error> {
|
||||||
}
|
}
|
||||||
|
|
||||||
Commands::Move(args) => {
|
Commands::Move(args) => {
|
||||||
let channel = result_or_error(args.channel(&mut connection), "channel")?;
|
let channel = channel_or_error(args.channel(&mut connection))?;
|
||||||
let client = result_or_error(args.client(&mut connection), "client")?;
|
let client = client_or_error(args.client(&mut connection))?;
|
||||||
|
|
||||||
match wrappers::move_client(&mut connection, &channel, &[client]) {
|
match wrappers::move_client(&mut connection, &channel, &[client]) {
|
||||||
Ok(resp) => println!("Successfully moved client: {}", resp),
|
Ok(resp) => println!("Successfully moved client: {}", resp),
|
||||||
|
@ -158,12 +145,16 @@ fn main() -> Result<(), Error> {
|
||||||
}
|
}
|
||||||
|
|
||||||
Commands::Events(args) => {
|
Commands::Events(args) => {
|
||||||
let filter = args.filter.unwrap_or(EventArgsFilter::All);
|
|
||||||
loop {
|
loop {
|
||||||
command_utils::events::register_events(&mut connection, args.event.clone())
|
command_utils::events::register_events(&mut connection, args.event.clone())
|
||||||
.map_err(to_other_error)?;
|
.map_err(to_other_error)?;
|
||||||
|
|
||||||
command_utils::events::loop_response_reader(&mut connection, &filter);
|
let self_clid: i32 = wrappers::get_self_clid(&mut connection)
|
||||||
|
.map_err(|msg| make_action_error("get self clid", msg))?
|
||||||
|
.parse()
|
||||||
|
.map_err(|err: ParseIntError| make_action_error("parse clid", err.to_string()))?;
|
||||||
|
|
||||||
|
command_utils::events::loop_response_reader(&mut connection, self_clid);
|
||||||
|
|
||||||
// loop_response_reader failed. Let's try to reconnect after 1 second.
|
// loop_response_reader failed. Let's try to reconnect after 1 second.
|
||||||
std::thread::sleep(std::time::Duration::from_secs(1));
|
std::thread::sleep(std::time::Duration::from_secs(1));
|
||||||
|
|
|
@ -4,7 +4,6 @@ use std::str::FromStr;
|
||||||
use serde::{Deserialize, Serialize, Serializer};
|
use serde::{Deserialize, Serialize, Serializer};
|
||||||
use serde::ser::SerializeStruct;
|
use serde::ser::SerializeStruct;
|
||||||
use telnet::Telnet;
|
use telnet::Telnet;
|
||||||
use crate::cli::EventArgsFilter;
|
|
||||||
|
|
||||||
use crate::models::{Channel, Client};
|
use crate::models::{Channel, Client};
|
||||||
use crate::parameter::{parameter_find, ParameterList};
|
use crate::parameter::{parameter_find, ParameterList};
|
||||||
|
@ -266,23 +265,6 @@ impl Event {
|
||||||
_ => String::from(""),
|
_ => String::from(""),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn is_mine(&self, self_clid: Option<i32>) -> bool {
|
|
||||||
if let Some(self_clid) = self_clid {
|
|
||||||
if let Some(client) = &self.client {
|
|
||||||
return client.clid == self_clid;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
false
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn should_handle(&self, filter: &EventArgsFilter, self_clid: Option<i32>) -> bool {
|
|
||||||
match filter {
|
|
||||||
EventArgsFilter::All => true,
|
|
||||||
EventArgsFilter::Mine => self.is_mine(self_clid),
|
|
||||||
EventArgsFilter::Others => !self.is_mine(self_clid),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Serialize for Event {
|
impl Serialize for Event {
|
||||||
|
@ -294,7 +276,6 @@ impl Serialize for Event {
|
||||||
x.serialize_field("type", &self.event_type)?;
|
x.serialize_field("type", &self.event_type)?;
|
||||||
x.serialize_field("channel", &self.channel)?;
|
x.serialize_field("channel", &self.channel)?;
|
||||||
x.serialize_field("client", &self.client)?;
|
x.serialize_field("client", &self.client)?;
|
||||||
x.serialize_field("params", &self.params)?;
|
|
||||||
x.serialize_field("message", &self.get_message())?;
|
x.serialize_field("message", &self.get_message())?;
|
||||||
x.end()
|
x.end()
|
||||||
}
|
}
|
||||||
|
|
82
ts-control
82
ts-control
|
@ -4,14 +4,12 @@ actions="quick
|
||||||
move
|
move
|
||||||
fetch-client
|
fetch-client
|
||||||
fetch-channel
|
fetch-channel
|
||||||
migrate
|
|
||||||
away
|
away
|
||||||
not away
|
not away
|
||||||
back
|
back
|
||||||
message
|
message
|
||||||
message-user
|
message-user
|
||||||
events
|
events"
|
||||||
ntfy-events"
|
|
||||||
|
|
||||||
_ts_control_get_entity() {
|
_ts_control_get_entity() {
|
||||||
entity=$(_ts_control_single_or_dmenu "$(teamspeak-query-lib "$1s")" "$2")
|
entity=$(_ts_control_single_or_dmenu "$(teamspeak-query-lib "$1s")" "$2")
|
||||||
|
@ -46,78 +44,18 @@ _ts_control_fetch() {
|
||||||
teamspeak-query-lib fetch "--strict-$1" "--$1" "$target"
|
teamspeak-query-lib fetch "--strict-$1" "--$1" "$target"
|
||||||
}
|
}
|
||||||
|
|
||||||
_ts_control_migrate() {
|
|
||||||
target=$(_ts_control_get_entity "channel" "$2")
|
|
||||||
current_channel=$(teamspeak-query-lib info | grep "Channel: " | sed 's/Channel: //')
|
|
||||||
_ts_control_move_self "$target"
|
|
||||||
teamspeak-query-lib fetch "--strict-channel" "--channel" "$current_channel"
|
|
||||||
}
|
|
||||||
|
|
||||||
_ts_control_single_or_dmenu() {
|
_ts_control_single_or_dmenu() {
|
||||||
filter="$2"
|
options=$(echo "$1" | grep "$2")
|
||||||
if [ "$filter" == "^$" ]; then
|
|
||||||
filter=""
|
|
||||||
fi
|
|
||||||
options=$(echo "$1" | grep "$filter")
|
|
||||||
count=$(echo "$options" | wc -l)
|
count=$(echo "$options" | wc -l)
|
||||||
if [ "$count" -eq 1 ]; then
|
if [ "$count" -eq 1 ]; then
|
||||||
echo "$options"
|
echo "$options"
|
||||||
else
|
else
|
||||||
echo "$1" | $DMENU
|
echo "$options" | $DMENU
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
handle_ntfy_events() {
|
# Add '$' to $1 to add 'End of line' to the regex for grep
|
||||||
while read -r data; do
|
action=$(_ts_control_single_or_dmenu "$actions" "$1$")
|
||||||
msg=$(echo "$data" | jq -r --unbuffered '.message')
|
|
||||||
type=$(echo "$data" | jq -r --unbuffered '.type')
|
|
||||||
mode=$(echo "$data" | jq -r --unbuffered '.params.targetmode')
|
|
||||||
client_nickname=$(echo "$data" | jq -r --unbuffered '.client.client_nickname')
|
|
||||||
|
|
||||||
echo "$data"
|
|
||||||
|
|
||||||
if [ "$type" = "NotifyTextMessage" ] && [ "$mode" != "1" ]; then
|
|
||||||
continue # Skip all messages that are not direct messages
|
|
||||||
fi
|
|
||||||
|
|
||||||
title="TS3 Event"
|
|
||||||
case $type in
|
|
||||||
"NotifyClientPoke")
|
|
||||||
title="TS3 Poke"
|
|
||||||
;;
|
|
||||||
"NotifyTextMessage")
|
|
||||||
title="TS3 Message"
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
|
|
||||||
echo "($title) $target: $msg"
|
|
||||||
|
|
||||||
payload=$(jq -n \
|
|
||||||
--arg topic "$TS3_NTFY_TOPIC" \
|
|
||||||
--arg webhook "$TS3_NTFY_WEBHOOK&client=$client_nickname" \
|
|
||||||
--arg msg "$msg" \
|
|
||||||
--arg type "$title" \
|
|
||||||
'{
|
|
||||||
topic: $topic,
|
|
||||||
message: $msg,
|
|
||||||
title: $type,
|
|
||||||
actions: [{
|
|
||||||
action: "http",
|
|
||||||
label: "TS response",
|
|
||||||
method: "POST",
|
|
||||||
url: $webhook,
|
|
||||||
clear: true
|
|
||||||
}]
|
|
||||||
}')
|
|
||||||
|
|
||||||
curl -sSL \
|
|
||||||
-H "Authorization: Bearer $TS3_NTFY_TOKEN" \
|
|
||||||
-d "$payload" \
|
|
||||||
"$TS3_NTFY_HOST"
|
|
||||||
done
|
|
||||||
}
|
|
||||||
|
|
||||||
action=$(_ts_control_single_or_dmenu "$actions" "^$1$")
|
|
||||||
|
|
||||||
case $action in
|
case $action in
|
||||||
"quick")
|
"quick")
|
||||||
|
@ -136,9 +74,6 @@ case $action in
|
||||||
"fetch-channel")
|
"fetch-channel")
|
||||||
_ts_control_fetch channel "$2"
|
_ts_control_fetch channel "$2"
|
||||||
;;
|
;;
|
||||||
"migrate")
|
|
||||||
_ts_control_migrate "$2"
|
|
||||||
;;
|
|
||||||
"not away")
|
"not away")
|
||||||
teamspeak-query-lib move "Not Away From Keyboard"
|
teamspeak-query-lib move "Not Away From Keyboard"
|
||||||
teamspeak-query-lib update --microphone=false --speakers=false
|
teamspeak-query-lib update --microphone=false --speakers=false
|
||||||
|
@ -164,12 +99,7 @@ case $action in
|
||||||
teamspeak-query-lib message --strict-client --client "$user" "$message"
|
teamspeak-query-lib message --strict-client --client "$user" "$message"
|
||||||
;;
|
;;
|
||||||
"events")
|
"events")
|
||||||
teamspeak-query-lib events --filter=others NotifyClientMoved NotifyClientEnterView NotifyClientLeftView NotifyTextMessage NotifyClientPoke \
|
teamspeak-query-lib events NotifyClientMoved NotifyClientEnterView NotifyClientLeftView NotifyTextMessage NotifyClientPoke \
|
||||||
| jq -r --unbuffered '.message' \
|
| jq -r --unbuffered '.message' \
|
||||||
| tee >(xargs -I{} notify-send "TS3 Event" "{}")
|
| tee >(xargs -I{} notify-send "TS3 Event" "{}")
|
||||||
;;
|
|
||||||
"ntfy-events")
|
|
||||||
teamspeak-query-lib events --filter=others NotifyClientPoke NotifyTextMessage \
|
|
||||||
| handle_ntfy_events
|
|
||||||
;;
|
|
||||||
esac
|
esac
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue