lib/appydave/tools/gpt_context/file_collector.rb in appydave-tools-0.10.4 vs lib/appydave/tools/gpt_context/file_collector.rb in appydave-tools-0.11.0

@@ -4,46 +4,47 @@
   module Tools
     # Build GPT context from various sources
     module GptContext
       # Gathers file names and content based on include and exclude patterns
       class FileCollector
-        attr_reader :include_patterns, :exclude_patterns, :format, :working_directory, :line_limit
-
-        def initialize(include_patterns: [], exclude_patterns: [], format: 'tree,content', working_directory: nil, line_limit: nil)
-          @include_patterns = include_patterns
-          @exclude_patterns = exclude_patterns
-          @format = format
-          @working_directory = working_directory
-          @line_limit = line_limit
+        def initialize(options)
+          @options = options
+          @include_patterns = options.include_patterns
+          @exclude_patterns = options.exclude_patterns
+          @format = options.format
+          @working_directory = options.working_directory
+          @line_limit = options.line_limit
         end
 
         def build
-          FileUtils.cd(working_directory) if working_directory && Dir.exist?(working_directory)
+          FileUtils.cd(@working_directory) if @working_directory && Dir.exist?(@working_directory)
 
-          formats = format.split(',')
+          formats = @format.split(',')
           result = formats.map do |fmt|
             case fmt
             when 'tree'
               build_tree
             when 'content'
               build_content
+            when 'json'
+              build_json
             else
               ''
             end
           end.join("\n\n")
 
-          FileUtils.cd(Dir.home) if working_directory
+          FileUtils.cd(Dir.home) if @working_directory
 
           result
         end
 
         private
 
         def build_content
           concatenated_content = []
 
-          include_patterns.each do |pattern|
+          @include_patterns.each do |pattern|
             Dir.glob(pattern).each do |file_path|
               next if excluded?(file_path) || File.directory?(file_path)
 
               content = "# file: #{file_path}\n\n#{read_file_content(file_path)}"
               concatenated_content << content
@@ -53,19 +54,19 @@
           concatenated_content.join("\n\n")
         end
 
         def read_file_content(file_path)
           lines = File.readlines(file_path)
-          return lines.first(line_limit).join if line_limit
+          return lines.first(@line_limit).join if @line_limit
 
           lines.join
         end
 
         def build_tree
           tree_view = {}
 
-          include_patterns.each do |pattern|
+          @include_patterns.each do |pattern|
             Dir.glob(pattern).each do |file_path|
               next if excluded?(file_path)
 
               path_parts = file_path.split('/')
               insert_into_tree(tree_view, path_parts)
@@ -91,11 +92,38 @@
             build_tree_pretty(child, prefix: "#{prefix}#{next_prefix}", is_last: child.empty? || index == node.size - 1, output: output)
           end
           output
         end
 
+        def build_json
+          json_output = {
+            'tree' => {},
+            'content' => []
+          }
+
+          # Building tree structure in JSON
+          @include_patterns.each do |pattern|
+            Dir.glob(pattern).each do |file_path|
+              next if excluded?(file_path)
+
+              path_parts = file_path.split('/')
+              insert_into_tree(json_output['tree'], path_parts)
+
+              # Building content structure in JSON
+              next if excluded?(file_path) || File.directory?(file_path)
+
+              json_output['content'] << {
+                'file' => file_path,
+                'content' => read_file_content(file_path)
+              }
+            end
+          end
+
+          JSON.pretty_generate(json_output)
+        end
+
         def excluded?(file_path)
-          exclude_patterns.any? { |pattern| File.fnmatch(pattern, file_path, File::FNM_PATHNAME | File::FNM_DOTMATCH) }
+          @exclude_patterns.any? { |pattern| File.fnmatch(pattern, file_path, File::FNM_PATHNAME | File::FNM_DOTMATCH) }
         end
       end
     end
   end
 end