From 22f6e89400e4f0f2797bd378cb19e57ed6d580a6 Mon Sep 17 00:00:00 2001
From: Benex254 <benextempest@gmail.com>
Date: Fri, 16 Aug 2024 14:24:42 +0300
Subject: [PATCH] fix:preferred server not reflecting in command

---
 fastanime/cli/commands/download.py | 15 +++++++++------
 fastanime/cli/commands/search.py   | 19 +++++++++++--------
 2 files changed, 20 insertions(+), 14 deletions(-)

diff --git a/fastanime/cli/commands/download.py b/fastanime/cli/commands/download.py
index 5988c15..65b49dd 100644
--- a/fastanime/cli/commands/download.py
+++ b/fastanime/cli/commands/download.py
@@ -138,13 +138,16 @@ def download(
                     # prompt for server selection
                     servers = {server["server"]: server for server in streams}
                 servers_names = list(servers.keys())
-                if config.use_fzf:
-                    server = fzf.run(servers_names, "Select an link: ")
+                if config.server in servers_names:
+                    server = config.server
                 else:
-                    server = fuzzy_inquirer(
-                        servers_names,
-                        "Select link",
-                    )
+                    if config.use_fzf:
+                        server = fzf.run(servers_names, "Select an link: ")
+                    else:
+                        server = fuzzy_inquirer(
+                            servers_names,
+                            "Select link",
+                        )
                 stream_link = filter_by_quality(
                     config.quality, servers[server]["links"]
                 )
diff --git a/fastanime/cli/commands/search.py b/fastanime/cli/commands/search.py
index 2ed3632..47f7899 100644
--- a/fastanime/cli/commands/search.py
+++ b/fastanime/cli/commands/search.py
@@ -153,15 +153,18 @@ def search(config: Config, anime_title: str, episode_range: str):
                     # prompt for server selection
                     servers = {server["server"]: server for server in streams}
                 servers_names = list(servers.keys())
-                if config.use_fzf:
-                    server = fzf.run(servers_names, "Select an link: ")
-                elif config.use_rofi:
-                    server = Rofi.run(servers_names, "Select an link")
+                if config.server in servers_names:
+                    server = config.server
                 else:
-                    server = fuzzy_inquirer(
-                        servers_names,
-                        "Select link",
-                    )
+                    if config.use_fzf:
+                        server = fzf.run(servers_names, "Select an link: ")
+                    elif config.use_rofi:
+                        server = Rofi.run(servers_names, "Select an link")
+                    else:
+                        server = fuzzy_inquirer(
+                            servers_names,
+                            "Select link",
+                        )
                 stream_link = filter_by_quality(
                     config.quality, servers[server]["links"]
                 )