gitingest 0.4.0 → 0.6.0

data/lib/gitingest/generator.rb

@@ -96,6 +96,7 @@ module Gitingest
     # @option options [Logger] :logger Custom logger instance
     # @option options [Integer] :threads Number of threads to use (default: auto-detected)
     # @option options [Integer] :thread_timeout Seconds to wait for thread pool shutdown (default: 60)
+    # @option options [Boolean] :show_structure Show repository directory structure (default: false)
     def initialize(options = {})
       @options = options
       @repo_files = []
@@ -109,6 +110,12 @@ module Gitingest
     # Main execution method for command line
     def run
       fetch_repository_contents
+
+      if @options[:show_structure]
+        puts generate_directory_structure
+        return
+      end
+
       generate_file
     end
 
@@ -144,6 +151,21 @@ module Gitingest
       result
     end
 
+    # Generate a textual representation of the repository's directory structure
+    #
+    # @return [String] The directory structure as a formatted string
+    def generate_directory_structure
+      fetch_repository_contents if @repo_files.empty?
+
+      @logger.info "Generating directory structure for #{@options[:repository]}"
+
+      repo_name = @options[:repository].split("/").last
+      structure = DirectoryStructureBuilder.new(repo_name, @repo_files).build
+
+      @logger.info "\n"
+      structure
+    end
+
     private
 
     # Set up logging based on verbosity options
@@ -165,10 +187,11 @@ module Gitingest
       raise ArgumentError, "Repository is required" unless @options[:repository]
 
       @options[:output_file] ||= "#{@options[:repository].split("/").last}_prompt.txt"
-      @options[:branch] ||= "main"
+      @options[:branch] ||= :default
       @options[:exclude] ||= []
       @options[:threads] ||= DEFAULT_THREAD_COUNT
       @options[:thread_timeout] ||= DEFAULT_THREAD_TIMEOUT
+      @options[:show_structure] ||= false
       @excluded_patterns = DEFAULT_EXCLUDES + @options[:exclude]
     end
 
@@ -215,7 +238,8 @@ module Gitingest
     # Validate repository and branch access
     def validate_repository_access
       begin
-        @client.repository(@options[:repository])
+        repo = @client.repository(@options[:repository])
+        @options[:branch] = repo.default_branch if @options[:branch] == :default
       rescue Octokit::Unauthorized
         raise "Authentication error: Invalid or expired GitHub token"
       rescue Octokit::NotFound
@@ -441,4 +465,57 @@ module Gitingest
       end
     end
   end
+
+  # Helper class to build directory structure visualization
+  class DirectoryStructureBuilder
+    def initialize(root_name, files)
+      @root_name = root_name
+      @files = files.map(&:path)
+    end
+
+    def build
+      tree = { @root_name => {} }
+
+      @files.sort.each do |path|
+        parts = path.split("/")
+        current = tree[@root_name]
+
+        parts.each do |part|
+          if part == parts.last
+            current[part] = nil
+          else
+            current[part] ||= {}
+            current = current[part]
+          end
+        end
+      end
+
+      output = ["Directory structure:"]
+      render_tree(tree, "", output)
+      output.join("\n")
+    end
+
+    private
+
+    def render_tree(tree, prefix, output)
+      return if tree.nil?
+
+      tree.keys.each_with_index do |key, index|
+        is_last = index == tree.keys.size - 1
+        current_prefix = prefix
+
+        if prefix.empty?
+          output << "└── #{key}/"
+          current_prefix = "    "
+        else
+          connector = is_last ? "└── " : "├── "
+          item = tree[key].is_a?(Hash) ? "#{key}/" : key
+          output << "#{prefix}#{connector}#{item}"
+          current_prefix = prefix + (is_last ? "    " : "│   ")
+        end
+
+        render_tree(tree[key], current_prefix, output) if tree[key].is_a?(Hash)
+      end
+    end
+  end
 end

data/lib/gitingest/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Gitingest
-  VERSION = "0.4.0"
+  VERSION = "0.6.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gitingest
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Davide Santangelo
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-03 00:00:00.000000000 Z
+date: 2025-03-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby