lhj-tools 0.1.2 → 0.1.6

data/lib/lhj/tree/tree.rb

@@ -0,0 +1,112 @@
+# frozen_string_literal: true
+
+require 'lhj/tree/node'
+require 'lhj/tree/directory_renderer'
+require 'lhj/tree/number_renderer'
+require 'lhj/tree/hash_walker'
+require 'lhj/tree/path_walker'
+
+module Lhj
+  class Tree
+    # @api public
+    def self.[](data)
+      new(data)
+    end
+
+    # The list of nodes in this tree.
+    attr_reader :nodes
+
+    # Create a Tree
+    #
+    # @param [String,Dir,Hash] data
+    #
+    # @api public
+    def initialize(data = nil, options = {}, &block)
+      @data   = data ? data.dup.freeze : nil
+      @walker = select_walker.new(options)
+      @nodes  = []
+
+      if @data
+        @walker.traverse(data)
+        @nodes = @walker.nodes
+      end
+
+      @nodes_stack = []
+
+      instance_eval(&block) if block_given?
+
+      freeze
+    end
+
+    # Add node to this tree.
+    #
+    # @param [Symbol,String] name
+    #   the name for the node
+    #
+    # @param [Node, LeafNode] type
+    #   the type of node to add
+    #
+    # @example
+    #   TTY::Tree.new do
+    #     node '...' do
+    #       node '...'
+    #     end
+    #   end
+    #
+    # @api public
+    def node(name, type = Node, &block)
+      parent = @nodes_stack.empty? ? Node::ROOT : @nodes_stack.last
+      level = [0, @nodes_stack.size - 1].max
+      prefix = ':pipe' * level
+      if parent.class == LeafNode
+        prefix = ':space' * level
+      end
+      node = type.new(name, parent.full_path, prefix, @nodes_stack.size)
+      @nodes << node
+
+      return unless block_given?
+
+      @nodes_stack << node
+      if block.arity.zero?
+        instance_eval(&block)
+      else
+        instance_eval(&(->(*_args) { block[node] }))
+      end
+      @nodes_stack.pop
+    end
+
+    # Add leaf node
+    #
+    # @api public
+    def leaf(name, &block)
+      node(name, LeafNode, &block)
+    end
+
+    # @api public
+    def render(options = {})
+      as = options.delete(:as) || :dir
+      renderer = select_renderer(as).new(nodes, options)
+      renderer.render
+    end
+
+    private
+
+    # @api private
+    def select_walker
+      if @data.is_a?(Hash) || @data.nil?
+        HashWalker
+      else
+        @data ||= Dir.pwd
+        PathWalker
+      end
+    end
+
+    def select_renderer(as)
+      case as
+      when :dir, :directory then DirectoryRenderer
+      when :num, :number then NumberRenderer
+      end
+    end
+  end
+end
+

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lhj-tools
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.6
 platform: ruby
 authors:
 - lihaijian
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-01-20 00:00:00.000000000 Z
+date: 2022-01-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: claide
@@ -104,6 +104,40 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.8.0
+- !ruby/object:Gem::Dependency
+  name: addressable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+- !ruby/object:Gem::Dependency
+  name: tty-spinner
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
 - !ruby/object:Gem::Dependency
   name: faraday-cookie_jar
   requirement: !ruby/object:Gem::Requirement
@@ -212,6 +246,7 @@ files:
 - lib/lhj/command/local/local.rb
 - lib/lhj/command/local/local_upload.rb
 - lib/lhj/command/local/micro_service.rb
+- lib/lhj/command/oss.rb
 - lib/lhj/command/oss/del.rb
 - lib/lhj/command/oss/list.rb
 - lib/lhj/command/oss/upload.rb
@@ -229,6 +264,12 @@ files:
 - lib/lhj/helper/trans_helper.rb
 - lib/lhj/tools.rb
 - lib/lhj/tools/version.rb
+- lib/lhj/tree/directory_renderer.rb
+- lib/lhj/tree/hash_walker.rb
+- lib/lhj/tree/node.rb
+- lib/lhj/tree/number_renderer.rb
+- lib/lhj/tree/path_walker.rb
+- lib/lhj/tree/tree.rb
 homepage:
 licenses:
 - MIT