dorian-git-tree 0.1.2 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fb002166fb2e78caa5668ba160022eb5765d2d372ab811fff3041164d8726d7e
-  data.tar.gz: 1ee4d98c0b4c7aae9ea5bc26caa60deb0f53bd5ed48f032a6053eec2691c1bbe
+  metadata.gz: db9e5c485461690eb5d6c477b7ee19800310a7ed75cc344a75ce4e010a84f39a
+  data.tar.gz: 2754c9fb19a4ea50b1caf7d456fac5407964d94f572e560b68d41bad4eb49a3c
 SHA512:
-  metadata.gz: f308c7395bbac715d1697e32358be0a0260555f0fe30839bd12f96d6dbbc22690da1d0f6222643803e940835cfafd8eb8db74c0339773b1038b12900dccbe6c5
-  data.tar.gz: 8db11420cb797c0d458ffaaf4f75e7a00ce907251bcbddf8934c96487b49af23c3e452e16e1bb15ec8c2f9ef4c55877abc787f83185c239e1299b4c7a20225f0
+  metadata.gz: 5bf1564b380bb50e76d70772cc9d58a8bea580a57d7a15de717906a7da535f7a36d1398c4c5aa651a70f014ed37d8fc75a0542215e87dc6d81d9326025091604
+  data.tar.gz: d4fdd7e69ed9f1c0802a6c7c83368d6937ae19321560b5ee9616c896a49573e98f28422b07076122803ef0fcc3752f360848e027dce502a30efbd91a727008d4

data/lib/dorian/git/tree.rb

@@ -1,30 +1,73 @@
-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
+
+        key = ARGV.first || "."
+
+        files = git_ls_files(key).map do |file|
+          ARGV.first ? file.sub(ARGV.first, "") : file
+        end
+
+        values = group(files)
 
-      private
+        key = "#{key}/" if values.any? && key != "." && key[-1] != "/"
 
-      def self.list(dir, start: true)
-        puts dir if start
+        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.git_ls_files(path)
+        result = `#{["git", "ls-files", path].compact.shelljoin}`.split("\n")
+        exit $?.exitstatus if $?.exitstatus > 0
+        result
+      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.3.0
 platform: ruby
 authors:
 - Dorian Marié
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-10-10 00:00:00.000000000 Z
+date: 2021-11-20 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: