Module: OllamaChat::InputContent

Included in:
Chat
Defined in:
lib/ollama_chat/input_content.rb

Overview

A module that provides input content processing functionality for OllamaChat.

The InputContent module encapsulates methods for reading and returning content from selected files, selecting files from a list of matching files, and collecting project context using the context_spook library. It supports interactive file selection and context collection for enhancing chat interactions with local or remote content.

Instance Method Summary collapse

Instance Method Details

#choose_filename(pattern, chosen: nil) ⇒ String?

The choose_filename method selects a file from a list of matching files. It searches for files matching the given pattern, excludes already chosen files, and presents them in an interactive chooser menu.

Parameters:

  • pattern (String) —

    the glob pattern to search for files

  • chosen (Set) (defaults to: nil) —

    a set of already chosen filenames to exclude from selection

Returns:

  • (String, nil) —

    the selected filename or nil if no file was chosen or user exited



42
43
44
45
46
47
48
49
50
51
52
53
 
# File 'lib/ollama_chat/input_content.rb', line 42

def choose_filename(pattern, chosen: nil)
  files = Dir.glob(pattern).reject { chosen&.member?(_1) }.
    select { File.file?(_1) }
  files.unshift('[EXIT]')
  case chosen = OllamaChat::Utils::Chooser.choose(files)
  when '[EXIT]', nil
    STDOUT.puts "Exiting chooser."
    return
  else
    chosen
  end
end

#compose ⇒ String?

The compose method opens an editor to compose content.

This method checks for a configured editor and opens a temporary file in that editor for the user to compose content. Upon successful editing, it reads the content from the temporary file and returns it. If the editor fails or no editor is configured, appropriate error messages are displayed and nil is returned.

Returns:

  • (String, nil) —

    the composed content if successful, nil otherwise



109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
 
# File 'lib/ollama_chat/input_content.rb', line 109

def compose
  unless editor = OllamaChat::EnvConfig::EDITOR?
    STDERR.puts "Editor required for compose, set env var "\
      "#{OllamaChat::EnvConfig::EDITOR!.env_var.inspect}."
    return
  end
  Tempfile.open do |tmp|
    result = system %{#{editor} #{tmp.path.inspect}}
    if result
      return File.read(tmp.path)
    else
      STDERR.puts "Editor failed to edit #{tmp.path.inspect}."
    end
  end
  nil
end

#context_spook(patterns) ⇒ String?

The context_spook method collects and returns project context using the context_spook library.

This method generates structured project context that can be used to provide AI models with comprehensive information about the codebase. It supports both:

  • On-the-fly pattern matching for specific file patterns

  • Loading context from predefined definition files in ./.contexts/

When patterns are provided, it collects files matching the glob patterns and generates context data including file contents, sizes, and metadata. When no patterns are provided, it loads the default context definition file.

Examples:

Collect context for Ruby files only

context_spook(['lib/**/*.rb'])

Collect context for multiple patterns

context_spook(['lib/**/*.rb', 'spec/**/*.rb'])

Load default context

context_spook(nil)

Parameters:

  • patterns (Array<String>, nil) —

    Optional array of glob patterns to filter files

Returns:

  • (String, nil) —

    JSON string of context data or nil if no context could be generated



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/ollama_chat/input_content.rb', line 82

def context_spook(patterns)
  format = config.context.format
  if patterns
    ContextSpook::generate_context(verbose: true, format:) do |context|
      context do
        Dir.glob(patterns).each do |filename|
          File.file?(filename) or next
          file filename
        end
      end
    end.to_json
  else
    if context_filename = choose_filename('.contexts/*.rb')
      ContextSpook.generate_context(context_filename, verbose: true, format:).to_json
    end
  end
end

#input(pattern) ⇒ String

The input method selects and reads content from files matching a pattern.

This method prompts the user to select files matching the given glob pattern, reads their content, and returns a concatenated string with each file’s content preceded by its filename.

Parameters:

  • pattern (String) —

    the glob pattern to search for files (defaults to ‘*/’)

Returns:

  • (String) —

    a concatenated string of file contents with filenames as headers



20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/ollama_chat/input_content.rb', line 20

def input(pattern)
  pattern ||= '**/*'
  files = Set[]
  while filename = choose_filename(pattern, chosen: files)
    files << filename
  end
  result = ''
  files.each do |filename|
    result << ("%s:\n\n%s\n\n" % [ filename, File.read(filename) ])
  end
  result.full?
end