added an option to follow the poster or the mentioned users

 in a post, bound to alt+f closes #58.

Author: aryanchoudharypro

src/commands.rs

@@ -110,6 +110,7 @@ pub enum UiCommand {
 	ApplyPending(Option<usize>),
 	HomePressed,
 	ToggleContentWarning,
+	ToggleFollow,
 	ToggleWindowVisibility,
 	SetQuickActionKeysEnabled(bool),
 	SwitchTimelineByIndex(usize),
@@ -1437,6 +1438,76 @@ pub fn handle_ui_command(cmd: UiCommand, ctx: &mut UiCommandContext<'_>) {
 				let _ = launch_default_browser(&url, BrowserLaunchFlags::Default);
 			}
 		}
+		UiCommand::ToggleFollow => {
+			let Some(status) = get_selected_status(state) else {
+				live_region::announce(live_region, "No post selected");
+				return;
+			};
+			let target = status.reblog.as_ref().map_or(status, std::convert::AsRef::as_ref);
+
+			let mut all_users: Vec<crate::mastodon::Account> = Vec::new();
+
+			if status.reblog.is_some() {
+				all_users.push(status.account.clone());
+			}
+			all_users.push(target.account.clone());
+
+			let mut all_mentions: Vec<crate::mastodon::Mention> = target.mentions.clone();
+			for (url, text) in crate::html::extract_mention_links(&target.content) {
+				if all_mentions.iter().any(|m| m.url == url) {
+					continue;
+				}
+				let acct = acct_from_mention_link(&text, &url);
+				let username = acct.split('@').next().unwrap_or("").to_string();
+				all_mentions.push(crate::mastodon::Mention { id: String::new(), username, acct, url });
+			}
+			for mention in all_mentions {
+				if !all_users.iter().any(|u| u.acct == mention.acct) {
+					all_users.push(crate::mastodon::Account {
+						id: mention.id.clone(),
+						username: mention.username.clone(),
+						acct: mention.acct.clone(),
+						display_name: String::new(),
+						url: mention.url,
+						note: String::new(),
+						followers_count: 0,
+						following_count: 0,
+						statuses_count: 0,
+						fields: Vec::new(),
+						created_at: String::new(),
+						locked: false,
+						bot: false,
+						discoverable: None,
+						source: None,
+					});
+				}
+			}
+
+			let selected_user = if all_users.len() == 1 {
+				all_users[0].clone()
+			} else {
+				if let Some((acc, _)) = dialogs::prompt_for_account_list(
+					frame,
+					"Select User",
+					"Select user to follow/unfollow:",
+					&all_users,
+				) {
+					acc
+				} else {
+					return;
+				}
+			};
+
+			if let Some(net) = &state.network_handle {
+				net.send(NetworkCommand::ToggleFollow {
+					account_id: if selected_user.id.is_empty() { None } else { Some(selected_user.id) },
+					acct: selected_user.acct.clone(),
+					target_name: selected_user.username.clone(),
+				});
+			} else {
+				live_region::announce(live_region, "Network not available");
+			}
+		}
 		UiCommand::ViewInBrowser => {
 			let Some(status) = get_selected_status(state) else {
 				live_region::announce(live_region, "No post selected");

src/main.rs

@@ -38,9 +38,9 @@ pub(crate) use crate::ui::ids::{
 	ID_DELETE_POST, ID_DIRECT_TIMELINE, ID_EDIT_POST, ID_EDIT_PROFILE, ID_FAVORITE, ID_FAVORITES_TIMELINE,
 	ID_FEDERATED_TIMELINE, ID_LOAD_MORE, ID_LOCAL_TIMELINE, ID_MANAGE_ACCOUNTS, ID_NEW_POST, ID_OPEN_LINKS,
 	ID_OPEN_USER_TIMELINE_BY_INPUT, ID_OPTIONS, ID_PIN_POST, ID_QUOTE, ID_REFRESH, ID_REPLY, ID_REPLY_AUTHOR,
-	ID_SEARCH, ID_TRAY_EXIT, ID_TRAY_TOGGLE, ID_UI_WAKE, ID_VIEW_BOOSTS, ID_VIEW_FAVORITES, ID_VIEW_HASHTAGS,
-	ID_VIEW_HELP, ID_VIEW_IN_BROWSER, ID_VIEW_MENTIONS, ID_VIEW_POST, ID_VIEW_PROFILE, ID_VIEW_QUOTED_THREAD,
-	ID_VIEW_THREAD, ID_VIEW_USER_TIMELINE, ID_VOTE, KEY_DELETE,
+	ID_SEARCH, ID_TOGGLE_FOLLOW, ID_TRAY_EXIT, ID_TRAY_TOGGLE, ID_UI_WAKE, ID_VIEW_BOOSTS, ID_VIEW_FAVORITES,
+	ID_VIEW_HASHTAGS, ID_VIEW_HELP, ID_VIEW_IN_BROWSER, ID_VIEW_MENTIONS, ID_VIEW_POST, ID_VIEW_PROFILE,
+	ID_VIEW_QUOTED_THREAD, ID_VIEW_THREAD, ID_VIEW_USER_TIMELINE, ID_VOTE, KEY_DELETE,
 };
 use crate::{
 	accounts::{start_add_account_flow, switch_to_account},

src/network.rs

@@ -114,6 +114,11 @@ pub enum NetworkCommand {
 		reblogs: bool,
 		action: RelationshipAction,
 	},
+	ToggleFollow {
+		account_id: Option<String>,
+		acct: String,
+		target_name: String,
+	},
 	UnfollowAccount {
 		account_id: String,
 		target_name: String,
@@ -803,6 +808,37 @@ fn network_loop(
 					NetworkResponse::RelationshipUpdated { _account_id: account_id, target_name, action, result },
 				);
 			}
+			Ok(NetworkCommand::ToggleFollow { account_id, acct, target_name }) => {
+				let resolved_id = if let Some(id) = account_id {
+					Some(id)
+				} else if let Ok(account) = client.lookup_account(access_token, &acct) {
+					Some(account.id)
+				} else {
+					None
+				};
+
+				if let Some(id) = resolved_id {
+					if let Ok(mut rels) = client.get_relationships(access_token, slice::from_ref(&id)) {
+						if let Some(rel) = rels.pop() {
+							let (action, result) = if rel.following {
+								(RelationshipAction::Unfollow, client.unfollow_account(access_token, &id))
+							} else if rel.requested {
+								(RelationshipAction::CancelFollowRequest, client.unfollow_account(access_token, &id))
+							} else {
+								(
+									RelationshipAction::Follow,
+									client.follow_account_with_options(access_token, &id, true),
+								)
+							};
+							send_response(
+								responses,
+								ui_waker,
+								NetworkResponse::RelationshipUpdated { _account_id: id, target_name, action, result },
+							);
+						}
+					}
+				}
+			}
 			Ok(NetworkCommand::UnfollowAccount { account_id, target_name, action }) => {
 				let result = client.unfollow_account(access_token, &account_id);
 				send_response(

src/ui/ids.rs

@@ -38,6 +38,7 @@ define_ids! {
 	ID_COPY_POST,
 	ID_VIEW_POST,
 	// User actions
+	ID_TOGGLE_FOLLOW,
 	ID_VIEW_PROFILE,
 	ID_VIEW_USER_TIMELINE,
 	ID_OPEN_USER_TIMELINE_BY_INPUT,

src/ui/menu.rs

@@ -4,9 +4,9 @@ use crate::{
 	AppState, ContextMenuState, ID_BOOKMARK, ID_BOOST, ID_CHECK_FOR_UPDATES, ID_CLOSE_TIMELINE, ID_COPY_POST,
 	ID_DELETE_POST, ID_DIRECT_TIMELINE, ID_EDIT_POST, ID_EDIT_PROFILE, ID_FAVORITE, ID_FEDERATED_TIMELINE,
 	ID_LOAD_MORE, ID_LOCAL_TIMELINE, ID_MANAGE_ACCOUNTS, ID_NEW_POST, ID_OPEN_LINKS, ID_OPEN_USER_TIMELINE_BY_INPUT,
-	ID_OPTIONS, ID_PIN_POST, ID_QUOTE, ID_REFRESH, ID_REPLY, ID_REPLY_AUTHOR, ID_SEARCH, ID_VIEW_BOOSTS,
-	ID_VIEW_FAVORITES, ID_VIEW_HASHTAGS, ID_VIEW_HELP, ID_VIEW_IN_BROWSER, ID_VIEW_MENTIONS, ID_VIEW_PROFILE,
-	ID_VIEW_QUOTED_THREAD, ID_VIEW_THREAD, ID_VIEW_USER_TIMELINE, commands::get_selected_status,
+	ID_OPTIONS, ID_PIN_POST, ID_QUOTE, ID_REFRESH, ID_REPLY, ID_REPLY_AUTHOR, ID_SEARCH, ID_TOGGLE_FOLLOW,
+	ID_VIEW_BOOSTS, ID_VIEW_FAVORITES, ID_VIEW_HASHTAGS, ID_VIEW_HELP, ID_VIEW_IN_BROWSER, ID_VIEW_MENTIONS,
+	ID_VIEW_PROFILE, ID_VIEW_QUOTED_THREAD, ID_VIEW_THREAD, ID_VIEW_USER_TIMELINE, commands::get_selected_status,
 };
 
 pub fn build_menu_bar() -> MenuBar {
@@ -45,6 +45,9 @@ pub fn build_menu_bar() -> MenuBar {
 	post_menu
 		.append(ID_QUOTE, "&Quote...\tCtrl+Q", "Quote this post", ItemKind::Normal)
 		.expect("Failed to append quote menu item");
+	post_menu
+		.append(ID_TOGGLE_FOLLOW, "Toggle &Follow\tAlt+F", "Follow or unfollow the author", ItemKind::Normal)
+		.expect("Failed to append toggle follow menu item");
 	post_menu
 		.append(ID_VIEW_PROFILE, "View &Profile\tCtrl+P", "View profile of selected post's author", ItemKind::Normal)
 		.expect("Failed to append view profile menu item");

src/ui/window.rs

@@ -6,9 +6,9 @@ use crate::{
 	ContextMenuState, ID_BOOKMARK, ID_BOOST, ID_CHECK_FOR_UPDATES, ID_CLOSE_TIMELINE, ID_COPY_POST, ID_DELETE_POST,
 	ID_DIRECT_TIMELINE, ID_EDIT_POST, ID_EDIT_PROFILE, ID_FAVORITE, ID_FEDERATED_TIMELINE, ID_LOAD_MORE,
 	ID_LOCAL_TIMELINE, ID_MANAGE_ACCOUNTS, ID_NEW_POST, ID_OPEN_LINKS, ID_OPEN_USER_TIMELINE_BY_INPUT, ID_OPTIONS,
-	ID_PIN_POST, ID_QUOTE, ID_REFRESH, ID_REPLY, ID_REPLY_AUTHOR, ID_SEARCH, ID_VIEW_BOOSTS, ID_VIEW_FAVORITES,
-	ID_VIEW_HASHTAGS, ID_VIEW_HELP, ID_VIEW_IN_BROWSER, ID_VIEW_MENTIONS, ID_VIEW_POST, ID_VIEW_PROFILE,
-	ID_VIEW_QUOTED_THREAD, ID_VIEW_THREAD, ID_VIEW_USER_TIMELINE, KEY_DELETE, UiCommand,
+	ID_PIN_POST, ID_QUOTE, ID_REFRESH, ID_REPLY, ID_REPLY_AUTHOR, ID_SEARCH, ID_TOGGLE_FOLLOW, ID_VIEW_BOOSTS,
+	ID_VIEW_FAVORITES, ID_VIEW_HASHTAGS, ID_VIEW_HELP, ID_VIEW_IN_BROWSER, ID_VIEW_MENTIONS, ID_VIEW_POST,
+	ID_VIEW_PROFILE, ID_VIEW_QUOTED_THREAD, ID_VIEW_THREAD, ID_VIEW_USER_TIMELINE, KEY_DELETE, UiCommand,
 	config::{AutoloadMode, SortOrder},
 	ui::menu::build_menu_bar,
 	ui_wake::UiCommandSender,
@@ -207,6 +207,14 @@ pub fn bind_input_handlers(
 				}
 			}
 
+			if !ctrl && !shift && alt {
+				if k == 70 {
+					let _ = ui_tx_list_key.send(UiCommand::ToggleFollow);
+					event.skip(false);
+					return;
+				}
+			}
+
 			if ctrl && shift && !alt {
 				match k {
 					314 => {
@@ -659,6 +667,12 @@ pub fn bind_input_handlers(
 	let shutdown_menu = is_shutting_down;
 	let frame_menu = parts.frame;
 	frame_menu.on_menu_selected(move |event| match event.get_id() {
+		ID_TOGGLE_FOLLOW => {
+			if shutdown_menu.get() {
+				return;
+			}
+			let _ = ui_tx_menu.send(UiCommand::ToggleFollow);
+		}
 		ID_VIEW_PROFILE => {
 			if shutdown_menu.get() {
 				return;