snibbets 2.0.7

data/lib/snibbets/menu.rb

@@ -0,0 +1,136 @@
+# frozen_string_literal: true
+
+module Snibbets
+  # Menu functions
+  module Menu
+    class << self
+      def find_query_in_options(filename, res, query)
+        options = res.map { |m| "#{filename} #{m['title']}" }
+        words = query.split(/ +/)
+        words.delete_if { |word| options.none? { |o| o =~ /#{word}/i } }
+        words.map(&:downcase).join(' ')
+      end
+
+      def remove_items_without_query(filename, res, query)
+        q = find_query_in_options(filename, res, query).split(/ /)
+        res.delete_if do |opt|
+          q.none? do |word|
+            "#{filename} #{opt['title']}" =~ /#{word}/i
+          end
+        end
+        res
+      end
+
+      def gum_menu(executable, res, title, query, filename)
+        unless query.nil? || query.empty?
+          res = remove_items_without_query(filename, res, query)
+          return res[0] if res.count == 1
+        end
+
+        if res.count.zero?
+          warn 'No matches found'
+          Process.exit 1
+        end
+
+        options = res.map { |m| m['title'] }
+
+        puts title
+        args = [
+          "--height=#{options.count}"
+        ]
+        selection = `echo #{Shellwords.escape(options.join("\n"))} | #{executable} filter #{args.join(' ')}`.strip
+        Process.exit 1 if selection.empty?
+
+        res.select { |m| m['title'] =~ /#{Regexp.escape(selection)}/ }[0]
+      end
+
+      def fzf_menu(executable, res, title, query, filename)
+        orig_options = res.dup
+        unless query.nil? || query.empty?
+          res = remove_items_without_query(filename, res, query)
+          return res[0] if res.count == 1
+        end
+
+        res = orig_options if res.count.zero?
+
+        options = res.map { |m| "#{filename}: #{m['title']}" }
+        q = query.nil? ? '' : find_query_in_options(filename, res, query)
+        args = [
+          "--height=#{options.count + 2}",
+          %(--prompt="#{title} > "),
+          '-1',
+          %(--header="#{filename}"),
+          # '--header-first',
+          '--reverse',
+          '--no-info',
+          %(--query="#{q}"),
+          '--tac'
+        ]
+        selection = `echo #{Shellwords.escape(options.join("\n"))} | #{executable} #{args.join(' ')}`.strip
+        Process.exit 1 if selection.empty?
+
+        result = res.select { |m| m['title'] == selection.sub(/^.*?: /, '') }
+
+        result[0]
+      end
+
+      # Generate a numbered menu, items passed must have a title property
+      def console_menu(res, title, filename, query: nil)
+        unless query.nil? || query.empty?
+          res = remove_items_without_query(filename, res, query)
+          return res[0] if res.count == 1
+        end
+
+        if res.count.zero?
+          warn 'No matches found'
+          Process.exit 1
+        end
+
+        stty_save = `stty -g`.chomp
+
+        trap('INT') do
+          system('stty', stty_save)
+          Process.exit 1
+        end
+
+        # Generate a numbered menu, items passed must have a title property('INT') { system('stty', stty_save); exit }
+        counter = 1
+        $stderr.puts
+        res.each do |m|
+          $stderr.printf("%<counter>2d) %<title>s\n", counter: counter, title: m['title'])
+          counter += 1
+        end
+        $stderr.puts
+
+        begin
+          $stderr.printf(title.sub(/:?$/, ': '), res.length)
+          while (line = Readline.readline('', true))
+            unless line =~ /^[0-9]/
+              system('stty', stty_save) # Restore
+              exit
+            end
+            line = line.to_i
+            return res[line - 1] if line.positive? && line <= res.length
+
+            warn 'Out of range'
+            console_menu(res, title)
+          end
+        rescue Interrupt
+          system('stty', stty_save)
+          exit
+        end
+      end
+
+      def menu(res, filename: nil, title: 'Select one', query: nil)
+        query&.remove_spotlight_tags!
+        fzf = TTY::Which.which('fzf')
+        return fzf_menu(fzf, res, title, query, filename) unless fzf.empty?
+
+        gum = TTY::Which.which('gum')
+        return gum_menu(gum, res, title, query, filename) unless gum.empty?
+
+        console_menu(res, title, filename, query: query)
+      end
+    end
+  end
+end

data/lib/snibbets/os.rb

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+
+module Snibbets
+  module OS
+    class << self
+      ##
+      ## Platform-agnostic copy command
+      ##
+      ## @param      text  The text to copy
+      ##
+      def copy(text)
+        os = RbConfig::CONFIG['target_os']
+        case os
+        when /darwin.*/i
+          `echo #{Shellwords.escape(text)} | pbcopy`
+        else
+          if TTY::Which.exist?('xclip')
+            `echo #{Shellwords.escape(text)} | xclip -sel c`
+          elsif TTY::Which.exist('xsel')
+            `echo #{Shellwords.escape(text)} | xsel -ib`
+          else
+            puts 'Copy not supported on this system, please install xclip or xsel.'
+          end
+        end
+      end
+
+      ##
+      ## Platform-agnostic paste command
+      ##
+      def paste
+        os = RbConfig::CONFIG['target_os']
+        case os
+        when /darwin.*/i
+          `pbpaste -pboard general -Prefer txt`
+        else
+          if TTY::Which.exist?('xclip')
+            `xclip -o -sel c`
+          elsif TTY::Which.exist('xsel')
+            `xsel -ob`
+          else
+            puts 'Paste not supported on this system, please install xclip or xsel.'
+          end
+        end
+      end
+
+      ##
+      ## Platform-agnostic open command
+      ##
+      ## @param      file  [String] The file to open
+      ##
+      def open(file, app: nil)
+        os = RbConfig::CONFIG['target_os']
+        case os
+        when /darwin.*/i
+          darwin_open(file, app: app)
+        when /mingw|mswin/i
+          win_open(file)
+        else
+          linux_open(file)
+        end
+      end
+
+      ##
+      ## macOS open command
+      ##
+      ## @param      file  The file
+      ## @param      app   The application
+      ##
+      def darwin_open(file, app: nil)
+        if app
+          `open -a "#{app}" #{Shellwords.escape(file)}`
+        else
+          `open #{Shellwords.escape(file)}`
+        end
+      end
+
+      ##
+      ## Windows open command
+      ##
+      ## @param      file  The file
+      ##
+      def win_open(file)
+        `start #{Shellwords.escape(file)}`
+      end
+
+      ##
+      ## Linux open command
+      ##
+      ## @param      file  The file
+      ##
+      def linux_open(file)
+        if TTY::Which.exist?('xdg-open')
+          `xdg-open #{Shellwords.escape(file)}`
+        else
+          notify('{r}Unable to determine executable for `xdg-open`.')
+        end
+      end
+    end
+  end
+end

data/lib/snibbets/string.rb

@@ -0,0 +1,153 @@
+# frozen_string_literal: true
+
+module Snibbets
+  # String helpers
+  class ::String
+    def remove_spotlight_tags
+      words = Shellwords.shellsplit(self)
+      words.delete_if do |word|
+        word =~ /^\w+:/
+      end
+
+      words.join(' ').strip
+    end
+
+    def remove_spotlight_tags!
+      replace remove_spotlight_tags
+    end
+
+    def remove_meta
+      input = dup
+      lines = input.split(/\n/)
+      loop do
+        line = lines.shift
+        next if line =~ /^\s*[A-Z\s]+\w:\s*\S+/i || line =~ /^-{3,}\s*$/
+
+        break
+      end
+      lines.join("\n")
+    end
+
+    # Are there multiple snippets (indicated by ATX headers)
+    def multiple?
+      gsub(/(`{3,}).*?\n\1/m, '').scan(/^#+/).length > 1
+    end
+
+    # Is the snippet in this block fenced?
+    def fenced?
+      count = scan(/^```/).length
+      count > 1 && count.even?
+    end
+
+    def indented?
+      self =~ /^( {4,}|\t+)/
+    end
+
+    def rx
+      ".*#{gsub(/\s+/, '.*')}.*"
+    end
+
+    # remove outside comments, fences, and indentation
+    def clean_code
+      block = dup
+
+      # if it's a fenced code block, just discard the fence and everything
+      # outside it
+      if block.fenced?
+        code_blocks = block.scan(/(`{3,})(\w+)?\s*\n(.*?)\n\1/m)
+        code_blocks.map! { |b| b[2].strip }
+        return code_blocks.join("\n\n")
+      end
+
+      # assume it's indented code, discard non-indented lines and outdent
+      # the rest
+      block = block.outdent if block.indented?
+
+      block
+    end
+
+    def outdent
+      lines = split(/\n/)
+
+      incode = false
+      code = []
+      lines.each do |line|
+        next if line =~ /^\s*$/ && !incode
+
+        incode = true
+        code.push(line)
+      end
+
+      return self unless code[0]
+
+      indent = code[0].match(/^( {4,}|\t+)(?=\S)/)
+
+      if indent
+        code.map! { |line| line.sub(/(?mi)^#{indent[1]}/, '') }.join("\n")
+      else
+        self
+      end
+    end
+
+    # Returns an array of snippets. Single snippets are returned without a
+    # title, multiple snippets get titles from header lines
+    def snippets
+      content = dup.remove_meta
+      # If there's only one snippet, just clean it and return
+      # return [{ 'title' => '', 'code' => content.clean_code.strip }] unless multiple?
+
+      # Split content by ATX headers. Everything on the line after the #
+      # becomes the title, code is gleaned from text between that and the
+      # next ATX header (or end)
+      sections = []
+      counter = 0
+      code_blocks = {}
+
+      sans_blocks = content.gsub(/^(`{3,})(\w+)?\s*\n(.*?)\n\1/m) do
+        counter += 1
+        code_blocks["block#{counter}"] = Regexp.last_match(3)
+        "<block#{counter}>\n"
+      end
+
+      sans_blocks = sans_blocks.gsub(/(?mi)^((?:\s{4,}|\t+)\S[\S\s]*?)(?=\n\S|\Z)/) do
+        counter += 1
+        code = Regexp.last_match(1).split(/\n/)
+
+        code_blocks["block#{counter}"] = code.join("\n").outdent
+
+        "<block#{counter}>\n"
+      end
+
+      content = []
+      if sans_blocks =~ /<block\d+>/
+        sans_blocks.each_line do |line|
+          content << line if line =~ /^#/ || line =~ /<block\d+>/
+        end
+
+        parts = content.join("\n").split(/^#+/)
+      else
+        parts = sans_blocks.gsub(/\n{2,}/, "\n\n").split(/^#+/)
+      end
+
+      parts.shift if parts.count > 1
+
+      parts.each do |p|
+        lines = p.split(/\n/)
+
+        title = lines.count > 1 ? lines.shift.strip.sub(/[.:]$/, '') : 'Default snippet'
+        block = lines.join("\n").gsub(/<(block\d+)>/) { code_blocks[Regexp.last_match(1)] }
+
+        code = block.clean_code
+
+        next unless code && !code.empty?
+
+        sections << {
+          'title' => title,
+          'code' => code
+        }
+      end
+
+      sections
+    end
+  end
+end

data/lib/snibbets/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Snibbets
+  VERSION = '2.0.7'
+end

data/lib/snibbets/which.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module TTY
+  # Additional app-bundle-specific routines for TTY::Which
+  module Which
+    def app_bundle(cmd)
+      app = cmd.sub(/(\.app)?$/, '.app')
+      command = cmd.dup
+      command.sub!(/\.app$/, '')
+      app_dirs = %w[/Applications /Applications/Setapp ~/Applications]
+      return command if ::File.exist?(app)
+
+      return command if app_dirs.any? { |dir| ::File.exist?(::File.join(dir, app)) }
+
+      false
+    end
+    module_function :app_bundle
+
+    def bundle_id?(cmd)
+      cmd =~ /^\w+(\.\w+){2,}/
+    end
+    module_function :bundle_id?
+
+    def app?(cmd)
+      if file_with_path?(cmd)
+        return cmd if app_bundle(cmd)
+      else
+        app = app_bundle(cmd)
+        return app if app
+      end
+
+      false
+    end
+    module_function :app?
+  end
+end

data/lib/snibbets.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'optparse'
+require 'readline'
+require 'json'
+require 'cgi'
+require 'shellwords'
+require 'yaml'
+require 'fileutils'
+require 'tty-which'
+require_relative 'snibbets/version'
+require_relative 'snibbets/config'
+require_relative 'snibbets/which'
+require_relative 'snibbets/string'
+require_relative 'snibbets/hash'
+require_relative 'snibbets/menu'
+require_relative 'snibbets/os'
+require_relative 'snibbets/highlight'
+require_relative 'snibbets/lexers'
+
+# Top level module
+module Snibbets
+  class << self
+    def config
+      @config ||= Config.new
+    end
+
+    def options
+      @options = config.options
+    end
+  end
+end

data/snibbets.gemspec

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require_relative "lib/snibbets/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "snibbets"
+  spec.version = Snibbets::VERSION
+  spec.author = "Brett Terpstra"
+  spec.email = "me@brettterpstra.com"
+
+  spec.summary = "Snibbets"
+  spec.description = "A plain text code snippet manager"
+  spec.homepage = "https://github.com/ttscoff/snibbets"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 3.0.0"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = spec.homepage
+  spec.metadata["bug_tracker_uri"] = "#{spec.metadata["source_code_uri"]}/issues"
+  spec.metadata["changelog_uri"] = "#{spec.metadata["source_code_uri"]}/blob/main/CHANGELOG.md"
+  spec.metadata["github_repo"] = "git@github.com:ttscoff/snibbets.git"
+
+  spec.require_paths << 'lib'
+  spec.extra_rdoc_files = ['README.md']
+  spec.rdoc_options << '--title' << 'na' << '--main' << 'README.md' << '--markup' << 'markdown'
+
+  spec.bindir = "bin"
+  spec.executables << 'snibbets'
+
+  spec.files = Dir["lib/**/*.rb"].reject { |f| f.end_with?("_spec.rb") }
+  spec.files += Dir["[A-Z]*"]
+
+  spec.add_development_dependency "bundler", "~> 2.0"
+  spec.add_development_dependency "gem-release", "~> 2.2"
+  spec.add_development_dependency "parse_gemspec-cli", "~> 1.0"
+  spec.add_development_dependency "rake", "~> 13.0"
+  spec.add_development_dependency 'rdoc', '~> 4.3'
+  spec.add_development_dependency 'yard', '~> 0.9', '>= 0.9.26'
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "simplecov", "~> 0.21"
+  spec.add_development_dependency "simplecov-console", "~> 0.9"
+  spec.add_development_dependency "standard", "~> 1.3"
+  spec.add_runtime_dependency 'tty-which', '~> 0.5', '>= 0.5.0'
+end