dorian-git-tree 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fb002166fb2e78caa5668ba160022eb5765d2d372ab811fff3041164d8726d7e
-  data.tar.gz: 1ee4d98c0b4c7aae9ea5bc26caa60deb0f53bd5ed48f032a6053eec2691c1bbe
+  metadata.gz: 04af5c2eb017066ef0c51c4e4a9be09fe0db93fe1910c21dd5d548193f2435c0
+  data.tar.gz: b5cb32c3cec0101276085e2b10497a5d8a0dacab5ec034229c470cc8afa7e5fd
 SHA512:
-  metadata.gz: f308c7395bbac715d1697e32358be0a0260555f0fe30839bd12f96d6dbbc22690da1d0f6222643803e940835cfafd8eb8db74c0339773b1038b12900dccbe6c5
-  data.tar.gz: 8db11420cb797c0d458ffaaf4f75e7a00ce907251bcbddf8934c96487b49af23c3e452e16e1bb15ec8c2f9ef4c55877abc787f83185c239e1299b4c7a20225f0
+  metadata.gz: 156fb05f12b601339ee98bcfa4ad6a208b607748b1743f4b5e123056d428eeb361a18e8b8182b5e0973b409a105319f3e5a0d1c11ceaba51d5e3c860290bdb08
+  data.tar.gz: 0c2f4230381f94549ec95145735f1816da1db669ef206371be10b6944c2f3076dbdac0c8db0219220ca7fe93ccfe949f7233daf2d7915f46e98714497cdacb30

data/lib/dorian/git/tree.rb

@@ -1,30 +1,64 @@
-require 'shellwords'
+require "shellwords"
 
 module Dorian
   module Git
     class Tree
+      SPACE = "    "
+      RIGHT = "└── "
+      DOWN = "│    "
+      DOWN_AND_RIGHT = "├── "
+
       def self.run
-        list(ARGV.first || '.')
-      end
+        if ARGV[0] == "--help" || ARGV[0] == "-h" || ARGV.size > 1
+          puts "USAGE: git tree [PATH]"
+          exit
+        end
 
-      private
+        key = ARGV.first || "."
+        files = git_ls_files(key)
+        values = group(files)
+        key = "#{key}/" if values.any? && key != "."
+        puts key
+        values.each.with_index do |(value_key, value_values), value_index|
+          print(
+            key: value_key,
+            values: value_values,
+            index: value_index,
+            size: values.size,
+          )
+        end
+      end
 
-      def self.list(dir, start: true)
-        puts dir if start
+      def self.git_ls_files(path)
+        `#{["git", "ls-files", path].compact.shelljoin}`.split("\n")
+      end
 
-        `git ls-tree HEAD #{Shellwords.escape(dir)} --name-only`.split("\n")
-          .sort
-          .each do |line|
-            split = line.split('/')
-            puts format(line)
-            list(line + '/', start: false) if File.directory?(line)
+      def self.group(files)
+        files
+          .group_by { |file| file.split("/").first }
+          .transform_values do |values|
+            group(
+              values
+                .map { |value| value.split("/")[1..-1].join("/") }
+                .reject(&:empty?)
+            )
           end
       end
 
-      def self.format(file)
-        split = file.split('/')
-        return file unless split.size > 1
-        '│  ' * (split.size - 2) + '└── ' + split.last
+      def self.print(key:, values:, index: 0, size: 1, prefix: "")
+        key = "#{key}/" if values.any?
+        last = index + 1 == size
+        right_prefix = last ? RIGHT : DOWN_AND_RIGHT
+        puts prefix + right_prefix + key
+        values.each.with_index do |(value_key, value_values), value_index|
+          print(
+            key: value_key,
+            values: value_values,
+            index: value_index,
+            size: values.size,
+            prefix: prefix + (last ? SPACE : DOWN)
+          )
+        end
       end
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dorian-git-tree
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Dorian Marié
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-10-10 00:00:00.000000000 Z
+date: 2021-11-19 00:00:00.000000000 Z
 dependencies: []
 description: |-
   Lists your files and directories in your git repository as a tree
@@ -25,7 +25,8 @@ files:
 homepage: https://github.com/dorianmariefr/git-tree
 licenses:
 - MIT
-metadata: {}
+metadata:
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths: