From 211a9c329f4658fd498b721637c2634043bec047 Mon Sep 17 00:00:00 2001
From: Maurice Parker <mo@vincode.io>
Date: Thu, 23 Apr 2020 11:42:00 -0500
Subject: [PATCH] Make screen name entry tolerant of @ usage.

---
 .../AddTwitterFeedWindowController.swift      | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/Mac/MainWindow/AddFeed/AddTwitterFeedWindowController.swift b/Mac/MainWindow/AddFeed/AddTwitterFeedWindowController.swift
index 72189485f..024a7e8ca 100644
--- a/Mac/MainWindow/AddFeed/AddTwitterFeedWindowController.swift
+++ b/Mac/MainWindow/AddFeed/AddTwitterFeedWindowController.swift
@@ -106,8 +106,13 @@ class AddTwitterFeedWindowController : NSWindowController, AddFeedWindowControll
 			let atUsername = accountPopupButton.selectedItem?.title else { return }
 		
 		let username = String(atUsername[atUsername.index(atUsername.startIndex, offsetBy: 1)..<atUsername.endIndex])
-		
-		guard let url = TwitterFeedProvider.buildURL(type, username: username, screenName: userEnteredScreenSearch, searchField: userEnteredScreenSearch) else { return }
+
+		var screenSearch = userEnteredScreenSearch
+		if let screenName = screenSearch, type == .screenName && screenName.starts(with: "@") {
+			screenSearch = String(screenName[screenName.index(screenName.startIndex, offsetBy: 1)..<screenName.endIndex])
+		}
+
+		guard let url = TwitterFeedProvider.buildURL(type, username: username, screenName: screenSearch, searchField: screenSearch) else { return }
 		
 		let container = selectedContainer()!
 		AddWebFeedDefaultContainer.saveDefaultContainer(container)
@@ -150,10 +155,12 @@ private extension AddTwitterFeedWindowController {
 			accountLabel.isHidden = true
 			accountPopupButton.isHidden = true
 			
-			var screenName = screenSearchTextField.stringValue
-			if !screenName.isEmpty && !screenName.starts(with: "@") {
-				screenName = "@\(screenName)"
-				typeDescriptionLabel.stringValue = NSLocalizedString("Tweets from \(screenName)", comment: "Home Timeline")
+			var screenSearch = userEnteredScreenSearch
+			if screenSearch != nil {
+				if let screenName = screenSearch, screenName.starts(with: "@") {
+					screenSearch = String(screenName[screenName.index(screenName.startIndex, offsetBy: 1)..<screenName.endIndex])
+				}
+				typeDescriptionLabel.stringValue = NSLocalizedString("Tweets from @\(screenSearch!)", comment: "Home Timeline")
 			} else {
 				typeDescriptionLabel.stringValue = ""
 			}