From 39c9cc440c513d8d5599f0ba502f8642c4d4cdaf Mon Sep 17 00:00:00 2001
From: Maurice Parker <mo@vincode.io>
Date: Wed, 8 Sep 2021 19:58:40 -0500
Subject: [PATCH] Add optional article theme toolbar item

---
 Mac/AppAssets.swift                       |  5 +++
 Mac/MainWindow/MainWindowController.swift | 48 +++++++++++++++++++++++
 2 files changed, 53 insertions(+)

diff --git a/Mac/AppAssets.swift b/Mac/AppAssets.swift
index be4065a9f..74c69df40 100644
--- a/Mac/AppAssets.swift
+++ b/Mac/AppAssets.swift
@@ -69,6 +69,11 @@ struct AppAssets {
 		return RSImage(named: "articleExtractorOn")!
 	}()
 
+	@available(macOS 11.0, *)
+	static var articleTheme: RSImage = {
+		return NSImage(systemSymbolName: "doc.richtext", accessibilityDescription: nil)!
+	}()
+
 	@available(macOS 11.0, *)
 	static var cleanUpImage: RSImage = {
 		return NSImage(systemSymbolName: "wind", accessibilityDescription: nil)!
diff --git a/Mac/MainWindow/MainWindowController.swift b/Mac/MainWindow/MainWindowController.swift
index fe988889e..cba22d362 100644
--- a/Mac/MainWindow/MainWindowController.swift
+++ b/Mac/MainWindow/MainWindowController.swift
@@ -44,6 +44,7 @@ class MainWindowController : NSWindowController, NSUserInterfaceValidations {
 	private var timelineContainerViewController: TimelineContainerViewController?
 	private var detailViewController: DetailViewController?
 	private var currentSearchField: NSSearchField? = nil
+	private let articleThemeMenuToolbarItem = NSMenuToolbarItem(itemIdentifier: .articleThemeMenu)
 	private var searchString: String? = nil
 	private var lastSentSearchString: String? = nil
 	private var timelineSourceMode: TimelineSourceMode = .regular {
@@ -98,6 +99,9 @@ class MainWindowController : NSWindowController, NSUserInterfaceValidations {
 		NotificationCenter.default.addObserver(self, selector: #selector(unreadCountDidChange(_:)), name: .UnreadCountDidChange, object: nil)
 		NotificationCenter.default.addObserver(self, selector: #selector(displayNameDidChange(_:)), name: .DisplayNameDidChange, object: nil)
 		
+		NotificationCenter.default.addObserver(self, selector: #selector(articleThemeNamesDidChangeNotification(_:)), name: .ArticleThemeNamesDidChangeNotification, object: nil)
+		NotificationCenter.default.addObserver(self, selector: #selector(currentArticleThemeDidChangeNotification(_:)), name: .CurrentArticleThemeDidChangeNotification, object: nil)
+		
 		DispatchQueue.main.async {
 			self.updateWindowTitle()
 		}
@@ -150,6 +154,14 @@ class MainWindowController : NSWindowController, NSUserInterfaceValidations {
 		updateWindowTitleIfNecessary(note.object)
 	}
 
+	@objc func articleThemeNamesDidChangeNotification(_ note: Notification) {
+		buildArticleThemeMenu()
+	}
+
+	@objc func currentArticleThemeDidChangeNotification(_ note: Notification) {
+		buildArticleThemeMenu()
+	}
+
 	private func updateWindowTitleIfNecessary(_ noteObject: Any?) {
 		
 		if let folder = currentFeedOrFolder as? Folder, let noteObject = noteObject as? Folder {
@@ -506,6 +518,10 @@ class MainWindowController : NSWindowController, NSUserInterfaceValidations {
 		timelineContainerViewController?.toggleReadFilter()
 	}
 	
+	@objc func selectArticleTheme(_ menuItem: NSMenuItem) {
+		ArticleThemesManager.shared.currentThemeName = menuItem.title
+	}
+	
 }
 
 // MARK: NSWindowDelegate
@@ -726,6 +742,7 @@ extension NSToolbarItem.Identifier {
 	static let readerView = NSToolbarItem.Identifier("readerView")
 	static let openInBrowser = NSToolbarItem.Identifier("openInBrowser")
 	static let share = NSToolbarItem.Identifier("share")
+	static let articleThemeMenu = NSToolbarItem.Identifier("articleThemeMenu")
 	static let cleanUp = NSToolbarItem.Identifier("cleanUp")
 }
 
@@ -795,6 +812,14 @@ extension MainWindowController: NSToolbarDelegate {
 				let title = NSLocalizedString("Open in Browser", comment: "Open in Browser")
 				return buildToolbarButton(.openInBrowser, title, AppAssets.openInBrowserImage, "openArticleInBrowser:")
 			
+			case .articleThemeMenu:
+				articleThemeMenuToolbarItem.image = AppAssets.articleTheme
+				let description = NSLocalizedString("Article Theme", comment: "Article Theme")
+				articleThemeMenuToolbarItem.toolTip = description
+				articleThemeMenuToolbarItem.label = description
+				buildArticleThemeMenu()
+				return articleThemeMenuToolbarItem
+
 			case .search:
 				let toolbarItem = NSSearchToolbarItem(itemIdentifier: .search)
 				let description = NSLocalizedString("Search", comment: "Search")
@@ -832,6 +857,7 @@ extension MainWindowController: NSToolbarDelegate {
 				.readerView,
 				.openInBrowser,
 				.share,
+				.articleThemeMenu,
 				.search,
 				.cleanUp
 			]
@@ -1387,5 +1413,27 @@ private extension MainWindowController {
 		return menu
 	}
 
+	func buildArticleThemeMenu() {
+		let articleThemeMenu = NSMenu()
+		
+		let defaultThemeItem = NSMenuItem()
+		defaultThemeItem.title = ArticleTheme.defaultTheme.name
+		defaultThemeItem.action = #selector(selectArticleTheme(_:))
+		defaultThemeItem.state = defaultThemeItem.title == ArticleThemesManager.shared.currentThemeName ? .on : .off
+		articleThemeMenu.addItem(defaultThemeItem)
+
+		articleThemeMenu.addItem(NSMenuItem.separator())
+
+		for themeName in ArticleThemesManager.shared.themeNames {
+			let themeItem = NSMenuItem()
+			themeItem.title = themeName
+			themeItem.action = #selector(selectArticleTheme(_:))
+			themeItem.state = themeItem.title == ArticleThemesManager.shared.currentThemeName ? .on : .off
+			articleThemeMenu.addItem(themeItem)
+		}
+
+		articleThemeMenuToolbarItem.menu = articleThemeMenu
+	}
+
 }