gopher: download files that can't be shown

plugins/gopher/bebop_gopher/plugin.py

@@ -6,8 +6,10 @@ from typing import Optional
 
 from bebop.browser.browser import Browser
 from bebop.command_line import CommandLine
+from bebop.downloads import get_download_path
 from bebop.links import Links
 from bebop.metalines import LineType
+from bebop.mime import MimeType
 from bebop.navigation import parse_url, parse_host_and_port, unparse_url
 from bebop.page import Page
 from bebop.plugins import PluginCommand, SchemePlugin
@@ -36,6 +38,8 @@ class ItemType(Enum):
     _missing_ = lambda s: ItemType.FILE
 
 
+# Types that can be parsed as a page (see `parse_source`).
+PARSABLE_TYPES = (ItemType.FILE, ItemType.DIR)
 # Types that are not rendered by this plugin; should be handled by a separate
 # program, but for now we simply do nothing with them.
 UNHANDLED_TYPES = (
@@ -129,14 +133,28 @@ class GopherPlugin(SchemePlugin):
         timeout = browser.config["connect_timeout"]
         try:
             response = request(host, port, path, timeout)
-            page = parse_response(response, item_type)
         except GopherPluginException as exc:
             browser.set_status_error("Error: " + exc.message)
             return None
 
-        browser.load_page(page)
         url = f"gopher://{host}:{port}/{item_type.value}{encoded_path}"
+        if item_type in PARSABLE_TYPES:
+            page = parse_response(response, item_type)
+            browser.load_page(page)
             browser.current_url = url
+        else:
+            download_dir = browser.config["download_path"]
+            filepath = get_download_path(url, download_dir=download_dir)
+            try:
+                with open(filepath, "wb") as download_file:
+                    download_file.write(response)
+            except OSError as exc:
+                browser.set_status_error(f"Failed to save {url} ({exc})")
+                return None
+            else:
+                browser.set_status(f"Downloaded {url}.")
+                browser.last_download = None, filepath
+
         return url
 
     def use_command(self, browser: Browser, name: str, text: str):
@@ -212,7 +230,11 @@ def get_page_from_source(source: str, item_type: ItemType):
 
 
 def parse_source(source: str, item_type: ItemType):
-    """Generate metalines and a Links instance for this source text."""
+    """Generate metalines and a Links instance for this source text.
+
+    The item_type must be a type that can be parsed: FILE or DIR. Any other
+    item type will silently result in no metalines.
+    """
     metalines = []
     links = Links()