booker 2.1.0 → 2.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 13dc462a7bedf194daa250fdd6fc0b7bdcc61374873c23fb3a03e48e73c5e287
-  data.tar.gz: bffeeb80575559e1404dd81543ab5b36a87fe74cb8e3c908495c5afb0196f0cd
+  metadata.gz: a0f32b2a81f8c03acdc8c56ce6f9554a1229aced8a731d103fbfd53f1d03b5d3
+  data.tar.gz: 2d0386dcc31468311faaac5e5ec749d19623ad40f9f6b083a86b77cd9632be01
 SHA512:
-  metadata.gz: 11fc60dbe348abd38052010e3073f7ad524bc81b0aeb1017741f25a1a8b3108654443c91444c40d07d065907f9b59117768ab5f2e9923b3fbd529f335c5082c7
-  data.tar.gz: 130643b4c4920f03d273cdfde85955295ca6fd701031fffe6228aad52e26e8b5e7f11c784d6ad5f914371785ad321074e155fff2a5eab657030bb45a60671d9c
+  metadata.gz: 963773985475ebf1a6e9c2fa340ceb79a607e4dd052e7c0dab1507881c0cd736f881388e7bd0445e8381dd69d561ead3bc14a9d940aed1cd08c07eb3770dc593
+  data.tar.gz: 279afd708175be5626eea44dfe3733ad435f4bb527dc4c9995bb97027376df4551b84c464f9d9ac69150cecddeb440b15b3d03f7eb8b1d706ef26589e8d03e4b

data/lib/booker/cli.rb

@@ -13,10 +13,10 @@ module Booker
 
     # a bookmark id is "<source index>_<browser id>". safari uses a uuid where
     # chrome and firefox number theirs, so the second half is anything id-shaped
-    # rather than digits; bare digits stay valid for single-source configs
-    # written before the prefix existed. completion inserts urls now, but ids
-    # are still first class - --bookmark takes them, the picker returns them
-    BOOKMARK_ID = /\A(?:\d+_[a-z0-9-]+|[0-9_]+)\z/i
+    # rather than digits. a bare number is a search: `booker 2024` is looking
+    # for a year. completion inserts urls now, but ids are still first class -
+    # --bookmark takes them, the picker returns them
+    BOOKMARK_ID = /\A\d+_[a-z0-9-]+\z/i
 
     def initialize(args)
       parse args
@@ -83,7 +83,9 @@ module Booker
       case @mode
       when :bookmark
         pexit "Error: ".red + "booker --bookmark expects bookmark id" if args.empty?
-        open_bookmark(args)
+        # a bare number is an id from before ids carried their source, when
+        # there was only ever source 0
+        open_bookmark(args.map { |id| /\A\d+\z/.match?(id) ? "0_#{id}" : id })
       when :install
         installer.install(args.empty? ? %w[completion config bookmarks] : args)
       when :search
@@ -98,7 +100,9 @@ module Booker
       when :complete_raw
         Bookmarks.new(args.join(" ")).autocomplete_raw
       end
-    rescue OptionParser::InvalidOption => e
+    # ParseError rather than InvalidOption: `booker --list=x` is a
+    # NeedlessArgument, and every other subclass is just as much a usage error
+    rescue OptionParser::ParseError => e
       pexit "Error: ".red + e.message
     end
 
@@ -146,8 +150,10 @@ module Booker
     def open_search(term)
       puts "searching ".grn + term
       search = Config.default.searcher
-      term = term.tr(" ", "+")
-      openweb(search + term)  # No shell escape needed - it's a URL
+      # encoded, not just spaces to "+": a literal + reads back as a space, &
+      # ends the query and # starts a fragment, so `c++` searched for "c"
+      require "uri"
+      openweb(search + URI.encode_www_form_component(term))
     end
 
     # a term matching bookmarks is offered as a choice; one matching nothing is

data/lib/booker/config.rb

@@ -57,6 +57,10 @@ module Booker
 
   # configuration
   class Config
+    # raised by #write rather than replacing a file that exists but would not
+    # parse: the defaults in memory are a fallback, not the user's settings
+    class Unreadable < StandardError; end
+
     VALID = [:searcher, :bookmarks, :browser, :picker].freeze
     HOME = ENV.fetch("HOME", "/usr/local/").freeze
     YAMLCONF = (HOME + "/.booker.yml").freeze
@@ -87,7 +91,7 @@ module Booker
 
       # prune bad config keys
       @config.each_key do |k|
-        if !VALID.include? k.to_sym
+        if !VALID.include? k
           warn "Failure:".red + " Bad key found in config file: #{k}"
           exit 1
         end
@@ -111,7 +115,15 @@ module Booker
 
     def read(file)
       # #write emits symbol keys, so they have to be permitted on the way back
-      YAML.safe_load_file(file, permitted_classes: [Symbol])
+      conf = YAML.safe_load_file(file, permitted_classes: [Symbol])
+      # an empty file is no config at all, which is fine to write over
+      return false if conf.nil?
+      raise Psych::Exception, "not a mapping" unless conf.is_a?(Hash)
+
+      # symbols throughout, whatever the file used: `searcher:` is how yaml is
+      # usually written, but the readers look up :searcher, so a config spelled
+      # the usual way passed validation and then read back as nil
+      conf.transform_keys { |k| k.to_s.to_sym }
     rescue Errno::ENOENT
       unless @quiet_missing
         warn "Warning: ".yel +
@@ -123,6 +135,7 @@ module Booker
       # every psych failure means the same thing here - an unusable config to
       # fall back from rather than crash on. rescuing SyntaxError alone let
       # anchors through as an AliasesNotEnabled backtrace under psych 4
+      @unreadable = true
       warn "Warning: ".red +
         "YAML configuration file could not be read. Using defaults."
       false
@@ -130,6 +143,11 @@ module Booker
 
     # used for creating and updating the default configuration
     def write(k = nil, v = nil)
+      if @unreadable
+        raise Unreadable, "#{YAMLCONF} could not be read - fix or remove it, " \
+          "it was left as is"
+      end
+
       @config[k] = v unless k.nil? # update users yaml config file
       File.write(YAMLCONF, @config.to_yaml)
     end

data/lib/booker/installer.rb

@@ -290,7 +290,13 @@ module Booker
 
           input = gets
           raise "No input provided" if input.nil?
-          selection = input.chomp.to_i
+
+          # to_i alone read any typo as 0 - all sources - and a negative number
+          # as an index from the end of the list
+          selection = Integer(input.strip, 10, exception: false)
+          unless selection&.between?(0, bms.length)
+            raise "Invalid selection: #{input.strip.inspect}"
+          end
 
           if selection == 0
             # User selected "ALL" - save array of all paths
@@ -325,6 +331,8 @@ module Booker
     def install_config
       config.write
       puts "Success: ".grn + "example config file written to ~/.booker"
+    rescue Config::Unreadable => e
+      pexit "Failure: ".red + e.message
     rescue
       pexit "Failure: ".red + "could not write example config file to ~/.booker"
     end

data/lib/booker/parsers.rb

@@ -50,22 +50,30 @@ module Booker
     class Chrome < Base
       def parse
         begin
-          local_bookmarks = JSON.parse(File.read(@file_path))
-          @chrome_bookmarks = local_bookmarks["roots"]["bookmark_bar"]["children"]
+          roots = JSON.parse(File.read(@file_path))["roots"].select { |_, r| r.is_a?(Hash) }
         rescue
           warn "Warning: ".yel + "Bookmarks file not found or invalid."
           warn "Suggest: ".grn + "booker --install bookmarks"
           return
         end
 
-        parse_recursive
+        # every root, not just the bar: "other bookmarks" is where chrome files
+        # anything saved without picking a folder, and "mobile bookmarks" is
+        # what sync brings over from a phone. the bar keeps the bare "|" it
+        # always had; the others are folders like any other, named by chrome
+        top = Folder.new([], "|")
+        roots.each do |key, root|
+          if key == "bookmark_bar"
+            parse_recursive(Folder.new(root["children"], "|"))
+          else
+            parse_folder(top, root)
+          end
+        end
       end
 
       private
 
-      def parse_recursive(root = nil)
-        root = Folder.new(@chrome_bookmarks, "|") if root.nil?
-
+      def parse_recursive(root)
         root.json.each { |x| parse_link(root.title, x) }
         root.json.each { |x| parse_folder(root, x) }
       end
@@ -143,15 +151,19 @@ module Booker
           # Recursive CTE query to build folder paths
           query = <<-SQL
             WITH RECURSIVE bookmark_tree(id, parent, title, url, path) AS (
-              -- Start with bookmarks toolbar root
+              -- Start with the children of every user-facing root: the menu is
+              -- where firefox saves a bookmark by default, so the toolbar alone
+              -- missed most of them. toolbar paths stay unprefixed, as before
               SELECT b.id, b.parent, b.title, p.url,
-                     CAST(b.title as TEXT) as path
+                     CAST(CASE r.guid
+                       WHEN 'toolbar_____' THEN b.title
+                       ELSE COALESCE(r.title, '') || '/' || b.title
+                     END as TEXT) as path
               FROM moz_bookmarks b
+              INNER JOIN moz_bookmarks r ON b.parent = r.id
               LEFT JOIN moz_places p ON b.fk = p.id
-              WHERE b.parent = (
-                SELECT id FROM moz_bookmarks
-                WHERE guid = 'toolbar_____'
-              )
+              WHERE r.guid IN
+                ('toolbar_____', 'menu________', 'unfiled_____', 'mobile______')
 
               UNION ALL
 
@@ -217,7 +229,14 @@ module Booker
         # sniff the head rather than reading the whole file: on the binary path
         # the contents are thrown away and plutil reads it again from disk, and
         # a real Bookmarks.plist grows with the user's bookmarks
-        head = File.binread(path, 512).to_s
+        head = begin
+          File.binread(path, 512).to_s
+        rescue Errno::EPERM, Errno::EACCES
+          # macOS privacy protection, when this process lacks Full Disk Access.
+          # the file is binary on a mac anyway, so leave it to plutil - which
+          # may have been granted access on its own (--install safari, [B])
+          ""
+        end
 
         # Already an XML plist (fixtures, non-macOS). Binary plists need plutil
         # (macOS-only); the JSON converter rejects <data>/<date> fields, so xml1.

data/lib/booker/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Booker
-  VERSION = "2.1.0"
+  VERSION = "2.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: booker
 version: !ruby/object:Gem::Version
-  version: 2.1.0
+  version: 2.2.0
 platform: ruby
 authors:
 - Jeremy Warner