trees 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bc7886d327dd2a8f6d2a3833a08ba2b555b198d441c0348e276670514fde2b5f
-  data.tar.gz: a7a6a657bb62fa25936a50b4c991e260e0d1daf6b421c063b4184a088f89b57a
+  metadata.gz: a1dc77a5635f0f3a070cad2a2de939adfd6d9e3550d4b4f3a25c34b567da0dca
+  data.tar.gz: 04abe90dbcabe052c78a56581bf906596f8b6eb2c057985df0c4f364c9c5c8a9
 SHA512:
-  metadata.gz: ddd1461e661cfc499e54d516f801ececa8fafbfd3613f9bbb119d2ee14b62701d6b3e05784f3539619cbcd75adbb3476209416e2e3fa64b2a09a42342f23e7c4
-  data.tar.gz: af7bb3374418c5b39151076c07fceb1c38d58bf44deb890bfffb4d88c5a0a2218477e40e830e29418fecb374bc6804c3c52e722ed575260be99e31e2a0928674
+  metadata.gz: 4743bc99029f3fc8f7a3c21c58c2757b3916855d1f02a66db14a70fbfa73a44176406e75569bfb22a69001f6f2ba08b39f843051ba2f21f8578ff72837a6eb74
+  data.tar.gz: 4dc45169e415cb0e705ae3fecdfd4f83b5652cdda893778ab05402f53339e638bda08f7d8fb86b142a1546a4231629d8446d71483701d739a72a5868f1eb3f34

data/lib/line.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require 'low_type'
+
+module Trees
+  class Line
+    include LowType
+
+    attr_reader :path, :params
+    attr_accessor :summary, :execute
+
+    def initialize(path: String, params: {})
+      @path = path
+      @params = params
+
+      @summary = nil
+      @execute = nil
+    end
+  end
+end

data/lib/trees.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require_relative 'line'
+require_relative 'trie'
+
+module Trees
+  attr_reader :lines, :trie
+
+  def line(path, &block)
+    current_path << path
+    @current_line = Line.new(path: current_path.join, params: block.parameters.to_h)
+
+    lines[current_path.join] = @current_line
+    trie.merge(line: @current_line)
+
+    block.call if block_given?
+
+    current_path.pop
+  end
+
+  def summary(&block)
+    @current_line.summary = block
+  end
+
+  def execute(&block)
+    @current_line.execute = block
+  end
+
+  def run(args)
+    results = trie.match(path: args.join(' '))
+
+    execute_block(results:)
+  end
+
+  private
+
+  def execute_block(results:)
+    while results.count > 0
+      result = results.pop
+
+      if result.line.execute
+        result.line.params.values.each do |param|
+          result.line.execute.binding.local_variable_set(param, result.params[param])
+        end
+
+        result.line.execute.call
+
+        return true
+      end
+    end
+
+    false
+  end  
+
+  def current_path
+    @current_path ||= []
+  end
+
+  def lines
+    @lines ||= {}
+  end
+
+  def trie
+    @trie ||= Trie.new
+  end
+end

data/lib/trie.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+require_relative 'line'
+require_relative 'trie_node'
+
+module Trees
+  class Trie
+    include LowType
+
+    PARAM_DELIMITERS = [' ', ':'].freeze
+    ARG_DELIMITERS = [' '].freeze
+
+    attr_reader :root_node
+
+    Result = Data.define(:line, :params)
+
+    def initialize
+      @root_node = TrieNode.new
+    end
+
+    def merge(line:, current_node: @root_node)
+      @current_index = 0
+      path = line.path
+
+      while @current_index < path.length
+        key = path[@current_index]
+
+        @current_index += 1
+
+        # Sometimes the key is an entire variable name.
+        key = capture_param(path:) if key == ':'
+
+        current_node = current_node.upsert_child(key:)
+      end
+
+      current_node.line = line
+    end
+
+    def match(path:, current_node: @root_node, current_index: 0, params: {})
+      return [] if (key = path[current_index]).nil?
+
+      results = []
+
+      # Static path segment.
+      if (child_node = current_node.child(key:))
+        results << Result.new(line: child_node.line, params:) if child_node.line
+        results = [*results, *match(path:, current_node: child_node, current_index: current_index + 1, params:)]
+      end
+
+      # Dynamic path segment.
+      current_node.params.each do |param|
+        child_node = current_node.child(key: param)
+
+        arg, next_index = capture_arg(start_index: current_index, path:)
+        params[param.delete_prefix(':').to_sym] = arg
+
+        results << Result.new(line: child_node.line, params:) if child_node.line
+        results = [*results, *match(path:, current_node: child_node, current_index: next_index, params:)]
+      end
+
+      results
+    end
+
+    private
+
+    def capture_param(path:)
+      param = [':']
+
+      path[@current_index...path.length].chars.each do |char|
+        break if PARAM_DELIMITERS.include?(char)
+
+        @current_index += 1
+        param << char
+      end
+
+      param.join
+    end
+
+    def capture_arg(start_index:, path:)
+      next_index = start_index
+      arg = []
+
+      path[start_index...path.length].chars.each do |char|
+        arg << char
+        next_index += 1
+        break if path[next_index].nil? || ARG_DELIMITERS.include?(path[next_index])
+      end
+
+      [arg.join, next_index]
+    end
+  end
+end

data/lib/trie_node.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require_relative 'line'
+
+module Trees
+  class TrieNode
+    include LowType
+    using LowType::Syntax
+
+    # TODO: Represent empty hash/array return value as {}/[] rather than Hash/Array (https://github.com/low-rb/low_type/issues/16)
+    type_reader nodes: Hash[String => TrieNode] | Hash
+    type_reader params: Array[String] | Array
+    type_accessor line: Line | nil
+
+    def initialize
+      @nodes = {}
+      @params = []
+      @line = nil
+    end
+
+    def child(key: String)
+      @nodes[key]
+    end
+
+    def upsert_child(key: String)
+      @params << key if key.start_with?(':')
+      @nodes[key] || @nodes[key] = TrieNode.new
+    end
+  end
+end

data/lib/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Trees
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,14 +1,28 @@
 --- !ruby/object:Gem::Specification
 name: trees
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - maedi
 bindir: exe
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: low_type
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: "A CLI framework where you write the literal terminal commands and \nTrees
   breaks them apart into a tree of options.\n"
 email:
@@ -17,7 +31,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/plugs.rb
+- lib/line.rb
+- lib/trees.rb
+- lib/trie.rb
+- lib/trie_node.rb
 - lib/version.rb
 homepage: https://github.com/raindeer-rb/trees
 licenses: []

data/lib/plugs.rb

@@ -1,5 +0,0 @@
-# frozen_string_literal: true
-
-module Trees
-
-end