--- /dev/null
+use clap::StructOpt;
+use jsonrpc_client_transports::{RpcError, TypedSubscriptionStream};
+use jsonrpc_core::{futures_util::StreamExt, Value};
+use std::{path::PathBuf, time::Duration};
+use tokio::{select, time::sleep};
+
+use crate::cli::{GroupPermission, LinkState};
+
+mod cli;
+#[allow(clippy::too_many_arguments)]
+mod jsonrpc;
+mod tcp;
+
+const DEFAULT_TCP: &str = "127.0.0.1:7583";
+const DEFAULT_SOCKET_SUFFIX: &str = "signal-cli/socket";
+
+#[tokio::main]
+async fn main() -> Result<(), anyhow::Error> {
+ let cli = cli::Cli::parse();
+
+ let client = connect(&cli)
+ .await
+ .map_err(|e| anyhow::anyhow!("Failed to connect to socket: {e}"))?;
+
+ let result = match cli.command {
+ cli::CliCommands::Receive { timeout } => {
+ let mut stream = client
+ .subscribe_receive(cli.account)
+ .map_err(|e| anyhow::anyhow!("JSON-RPC command failed: {:?}", e))?;
+
+ {
+ while let Some(v) = stream_next(timeout, &mut stream).await {
+ let v = v.map_err(|e| anyhow::anyhow!("JSON-RPC command failed: {:?}", e))?;
+ println!("{v}");
+ }
+ }
+ return Ok(());
+ }
+ cli::CliCommands::AddDevice { uri } => client.add_device(cli.account, uri).await,
+ cli::CliCommands::Block {
+ recipient,
+ group_id,
+ } => client.block(cli.account, recipient, group_id).await,
+ cli::CliCommands::GetUserStatus { recipient } => {
+ client.get_user_status(cli.account, recipient).await
+ }
+ cli::CliCommands::JoinGroup { uri } => client.join_group(cli.account, uri).await,
+ cli::CliCommands::Link { name } => {
+ let url = client
+ .start_link(cli.account)
+ .await
+ .map_err(|e| anyhow::anyhow!("JSON-RPC command startLink failed: {e:?}",))?
+ .device_link_uri;
+ println!("{}", url);
+ client.finish_link(url, name).await
+ }
+ cli::CliCommands::ListAccounts => client.list_accounts().await,
+ cli::CliCommands::ListContacts => client.list_contacts(cli.account).await,
+ cli::CliCommands::ListDevices => client.list_devices(cli.account).await,
+ cli::CliCommands::ListGroups { detailed: _ } => client.list_groups(cli.account).await,
+ cli::CliCommands::ListIdentities { number } => {
+ client.list_identities(cli.account, number).await
+ }
+ cli::CliCommands::ListStickerPacks => client.list_sticker_packs(cli.account).await,
+ cli::CliCommands::QuitGroup {
+ group_id,
+ delete,
+ admin,
+ } => {
+ client
+ .quit_group(cli.account, group_id, delete, admin)
+ .await
+ }
+ cli::CliCommands::Register { voice, captcha } => {
+ client.register(cli.account, voice, captcha).await
+ }
+ cli::CliCommands::RemoveContact { recipient, forget } => {
+ client.remove_contact(cli.account, recipient, forget).await
+ }
+ cli::CliCommands::RemoveDevice { device_id } => {
+ client.remove_device(cli.account, device_id).await
+ }
+ cli::CliCommands::RemovePin => client.remove_pin(cli.account).await,
+ cli::CliCommands::RemoteDelete {
+ target_timestamp,
+ recipient,
+ group_id,
+ note_to_self,
+ } => {
+ client
+ .remote_delete(
+ cli.account,
+ target_timestamp,
+ recipient,
+ group_id,
+ note_to_self,
+ )
+ .await
+ }
+ cli::CliCommands::Send {
+ recipient,
+ group_id,
+ note_to_self,
+ end_session,
+ message,
+ attachment,
+ mention,
+ quote_timestamp,
+ quote_author,
+ quote_message,
+ quote_mention,
+ sticker,
+ } => {
+ client
+ .send(
+ cli.account,
+ recipient,
+ group_id,
+ note_to_self,
+ end_session,
+ message.unwrap_or_default(),
+ attachment,
+ mention,
+ quote_timestamp,
+ quote_author,
+ quote_message,
+ quote_mention,
+ sticker,
+ )
+ .await
+ }
+ cli::CliCommands::SendContacts => client.send_contacts(cli.account).await,
+ cli::CliCommands::SendReaction {
+ recipient,
+ group_id,
+ note_to_self,
+ emoji,
+ target_author,
+ target_timestamp,
+ remove,
+ } => {
+ client
+ .send_reaction(
+ cli.account,
+ recipient,
+ group_id,
+ note_to_self,
+ emoji,
+ target_author,
+ target_timestamp,
+ remove,
+ )
+ .await
+ }
+ cli::CliCommands::SendReceipt {
+ recipient,
+ target_timestamp,
+ r#type,
+ } => {
+ client
+ .send_receipt(
+ cli.account,
+ recipient,
+ target_timestamp,
+ match r#type {
+ cli::ReceiptType::Read => "read".to_owned(),
+ cli::ReceiptType::Viewed => "viewed".to_owned(),
+ },
+ )
+ .await
+ }
+ cli::CliCommands::SendSyncRequest => client.send_sync_request(cli.account).await,
+ cli::CliCommands::SendTyping {
+ recipient,
+ group_id,
+ stop,
+ } => {
+ client
+ .send_typing(cli.account, recipient, group_id, stop)
+ .await
+ }
+ cli::CliCommands::SetPin { pin } => client.set_pin(cli.account, pin).await,
+ cli::CliCommands::SubmitRateLimitChallenge { challenge, captcha } => {
+ client
+ .submit_rate_limit_challenge(cli.account, challenge, captcha)
+ .await
+ }
+ cli::CliCommands::Trust {
+ recipient,
+ trust_all_known_keys,
+ verified_safety_number,
+ } => {
+ client
+ .trust(
+ cli.account,
+ recipient,
+ trust_all_known_keys,
+ verified_safety_number,
+ )
+ .await
+ }
+ cli::CliCommands::Unblock {
+ recipient,
+ group_id,
+ } => client.unblock(cli.account, recipient, group_id).await,
+ cli::CliCommands::Unregister { delete_account } => {
+ client.unregister(cli.account, delete_account).await
+ }
+ cli::CliCommands::UpdateAccount { device_name } => {
+ client.update_account(cli.account, device_name).await
+ }
+ cli::CliCommands::UpdateConfiguration {
+ read_receipts,
+ unidentified_delivery_indicators,
+ typing_indicators,
+ link_previews,
+ } => {
+ client
+ .update_configuration(
+ cli.account,
+ read_receipts,
+ unidentified_delivery_indicators,
+ typing_indicators,
+ link_previews,
+ )
+ .await
+ }
+ cli::CliCommands::UpdateContact {
+ recipient,
+ expiration,
+ name,
+ } => {
+ client
+ .update_contact(cli.account, recipient, name, expiration)
+ .await
+ }
+ cli::CliCommands::UpdateGroup {
+ group_id,
+ name,
+ description,
+ avatar,
+ member,
+ remove_member,
+ admin,
+ remove_admin,
+ reset_link,
+ link,
+ set_permission_add_member,
+ set_permission_edit_details,
+ set_permission_send_messages,
+ expiration,
+ } => {
+ client
+ .update_group(
+ cli.account,
+ group_id,
+ name,
+ description,
+ avatar,
+ member,
+ remove_member,
+ admin,
+ remove_admin,
+ reset_link,
+ link.map(|link| match link {
+ LinkState::Enabled => "enabled".to_owned(),
+ LinkState::EnabledWithApproval => "enabledWithApproval".to_owned(),
+ LinkState::Disabled => "disabled".to_owned(),
+ }),
+ set_permission_add_member.map(|p| match p {
+ GroupPermission::EveryMember => "everyMember".to_owned(),
+ GroupPermission::OnlyAdmins => "onlyAdmins".to_owned(),
+ }),
+ set_permission_edit_details.map(|p| match p {
+ GroupPermission::EveryMember => "everyMember".to_owned(),
+ GroupPermission::OnlyAdmins => "onlyAdmins".to_owned(),
+ }),
+ set_permission_send_messages.map(|p| match p {
+ GroupPermission::EveryMember => "everyMember".to_owned(),
+ GroupPermission::OnlyAdmins => "onlyAdmins".to_owned(),
+ }),
+ expiration,
+ )
+ .await
+ }
+ cli::CliCommands::UpdateProfile {
+ given_name,
+ family_name,
+ about,
+ about_emoji,
+ avatar,
+ remove_avatar,
+ } => {
+ client
+ .update_profile(
+ cli.account,
+ given_name,
+ family_name,
+ about,
+ about_emoji,
+ avatar,
+ remove_avatar,
+ )
+ .await
+ }
+ cli::CliCommands::UploadStickerPack { path } => {
+ client.upload_sticker_pack(cli.account, path).await
+ }
+ cli::CliCommands::Verify {
+ verification_code,
+ pin,
+ } => client.verify(cli.account, verification_code, pin).await,
+ cli::CliCommands::Version => client.version().await,
+ };
+
+ result
+ .map(|v| println!("{v}"))
+ .map_err(|e| anyhow::anyhow!("JSON-RPC command failed: {e:?}",))?;
+ Ok(())
+}
+
+async fn connect(cli: &cli::Cli) -> Result<jsonrpc::SignalCliClient, RpcError> {
+ if let Some(tcp) = cli.json_rpc_tcp {
+ let socket_addr = tcp.unwrap_or_else(|| DEFAULT_TCP.parse().unwrap());
+ jsonrpc::connect_tcp(socket_addr).await
+ } else {
+ let socket_path = cli
+ .json_rpc_socket
+ .clone()
+ .unwrap_or(None)
+ .or_else(|| {
+ std::env::var_os("XDG_RUNTIME_DIR").map(|runtime_dir| {
+ PathBuf::from(runtime_dir)
+ .join(DEFAULT_SOCKET_SUFFIX)
+ .into()
+ })
+ })
+ .unwrap_or_else(|| ("/run".to_owned() + DEFAULT_SOCKET_SUFFIX).into());
+ jsonrpc::connect_unix(socket_path).await
+ }
+}
+
+async fn stream_next(
+ timeout: f64,
+ stream: &mut TypedSubscriptionStream<Value>,
+) -> Option<Result<Value, RpcError>> {
+ if timeout < 0.0 {
+ stream.next().await
+ } else {
+ select! {
+ v = stream.next() => v,
+ _= sleep(Duration::from_millis((timeout * 1000.0) as u64)) => None,
+ }
+ }
+}