vibecode 0.0.1

data/lib/vibecode/ollama_client.rb

@@ -0,0 +1,111 @@
+require "httparty"
+require "json"
+
+module Vibecode
+  class OllamaClient
+    include HTTParty
+    base_uri "http://localhost:11434"
+
+    def initialize
+      @headers = { "Content-Type" => "application/json" }
+    end
+
+    # -----------------------------
+    # Chat with Model
+    # -----------------------------
+    def chat(model, system_prompt, user_input)
+      body = {
+        model: model,
+        stream: false,
+        messages: [
+          { role: "system", content: system_prompt },
+          { role: "user", content: user_input }
+        ]
+      }
+
+      response = self.class.post("/api/chat", headers: @headers, body: body.to_json)
+
+      unless response.success?
+        return "Error talking to Ollama: #{response.code} #{response.body}"
+      end
+
+      parsed = JSON.parse(response.body)
+      parsed.dig("message", "content") || "(No response from model)"
+    rescue Errno::ECONNREFUSED
+      "Cannot connect to Ollama. Is it running? Try: `ollama serve`"
+    rescue => e
+      "Ollama error: #{e.message}"
+    end
+
+    # -----------------------------
+    # List Installed Models
+    # -----------------------------
+    def list_models
+      response = self.class.get("/api/tags")
+
+      return [] unless response.success?
+
+      parsed = JSON.parse(response.body)
+      parsed.fetch("models", []).map { |m| m["name"] }
+    rescue
+      []
+    end
+
+    def model_installed?(model_name)
+      list_models.include?(model_name)
+    end
+
+    # -----------------------------
+    # Pull Model
+    # -----------------------------
+    def pull_model(model_name)
+      uri = URI("#{self.class.base_uri}/api/pull")
+
+      req = Net::HTTP::Post.new(uri, @headers)
+      req.body = { name: model_name, stream: true }.to_json
+
+      Net::HTTP.start(uri.hostname, uri.port) do |http|
+        http.request(req) do |res|
+          unless res.is_a?(Net::HTTPSuccess)
+            puts "Failed to start model pull"
+            return false
+          end
+
+          res.read_body do |chunk|
+            begin
+              data = JSON.parse(chunk)
+              show_pull_progress(data)
+            rescue JSON::ParserError
+              # ignore incomplete chunks
+            end
+          end
+        end
+      end
+
+      true
+    rescue => e
+      puts "Error pulling model: #{e.message}"
+      false
+    end
+
+    def show_pull_progress(data)
+      if data["status"]
+        print "\r#{data['status'].ljust(60)}"
+      elsif data["completed"] && data["total"]
+        percent = (data["completed"].to_f / data["total"] * 100).round(1)
+        print "\rDownloading... #{percent}%".ljust(60)
+      end
+    end
+
+    # -----------------------------
+    # Health Check
+    # -----------------------------
+    def server_alive?
+      response = self.class.get("/")
+      response.code == 200 || response.code == 404
+    rescue
+      false
+    end
+  end
+end
+

data/lib/vibecode/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Vibecode
+  VERSION = "0.0.1"
+end

data/lib/vibecode/workspace.rb

@@ -0,0 +1,225 @@
+require "fileutils"
+require "pathname"
+require "tty-prompt"
+require "pastel"
+require "diffy"
+require "open3"
+
+module Vibecode
+  class Workspace
+    attr_reader :root
+
+    def initialize(root = Dir.pwd)
+      @root = File.expand_path(root)
+      @prompt = TTY::Prompt.new
+      @pastel = Pastel.new
+    end
+
+    # --------------------------------------------------
+    # File Reading
+    # --------------------------------------------------
+
+    def read_file(path)
+      full_path = safe_path(path)
+      return error("File does not exist: #{path}") unless File.exist?(full_path)
+
+      File.read(full_path)
+    rescue => e
+      error("Failed to read file: #{e.message}")
+    end
+
+    def file_exists?(path)
+      full_path = safe_path(path)
+      File.exist?(full_path)
+    rescue
+      false
+    end
+
+    def list_files(limit: 200)
+      files = Dir.glob("**/*", base: @root)
+                 .reject { |f| File.directory?(File.join(@root, f)) }
+                 .first(limit)
+
+      files.join("\n")
+    end
+
+    # --------------------------------------------------
+    # File Writing
+    # --------------------------------------------------
+
+    def diff_for(path, new_content)
+      full_path = safe_path(path)
+      old_content = File.exist?(full_path) ? File.read(full_path) : ""
+      Diffy::Diff.new(old_content, new_content, context: 3).to_s(:color)
+    rescue => e
+      error("Failed to diff file: #{e.message}")
+      ""
+    end
+
+    def write_file(path, new_content, show_diff: true)
+      full_path = safe_path(path)
+
+      if show_diff
+        diff = diff_for(path, new_content)
+        puts @pastel.yellow("\nProposed changes to #{path}:\n")
+        puts diff.empty? ? @pastel.dim("(No changes)") : diff
+      end
+
+      FileUtils.mkdir_p(File.dirname(full_path))
+      File.write(full_path, new_content)
+
+      puts @pastel.green("Updated #{path}")
+      true
+    rescue => e
+      error("Failed to write file: #{e.message}")
+    end
+
+    # --------------------------------------------------
+    # Ruby Execution
+    # --------------------------------------------------
+
+    def ruby_executable_content?(content)
+      return false if content.nil? || content.strip.empty?
+
+      return true if content.match?(/if\s+__FILE__\s*==\s*\$0/)
+      return true if content.match?(/\bputs\b/)
+
+      line = last_significant_line(content)
+      return false unless line
+
+      stripped = line.strip
+      return false if stripped.match?(/^(def|class|module|end)\b/)
+
+      stripped.match?(/[A-Za-z_]\w*(\s*\(|\b)/)
+    end
+
+    def run_ruby(path)
+      full_path = safe_path(path)
+      return {
+        stdout: "",
+        stderr: "File does not exist: #{path}",
+        status: nil,
+        command: "ruby #{path}",
+        skipped: true
+      } unless File.exist?(full_path)
+
+      content = File.read(full_path)
+      unless ruby_executable_content?(content)
+        return {
+          stdout: "",
+          stderr: "",
+          status: nil,
+          command: "ruby #{path}",
+          skipped: true
+        }
+      end
+
+      stdout, stderr, status = Open3.capture3("ruby #{full_path}", chdir: @root)
+      {
+        stdout: stdout,
+        stderr: stderr,
+        status: status,
+        command: "ruby #{path}",
+        skipped: false
+      }
+    rescue => e
+      {
+        stdout: "",
+        stderr: e.message,
+        status: nil,
+        command: "ruby #{path}",
+        skipped: false
+      }
+    end
+
+    # --------------------------------------------------
+    # Filename Suggestions
+    # --------------------------------------------------
+
+    def suggest_filename(task_description)
+      prompt = task_description.to_s.downcase
+
+      return "hello_world.rb" if prompt.include?("hello world")
+      return "greet.rb" if prompt.match?(/\bgreet\b/)
+
+      preferred = %w[
+        greet hello world user server client parser json api http config file data
+      ]
+      stopwords = %w[
+        a an the to for of and in on with from into is are be create build make write
+        ruby method function class module script program app code that this
+      ]
+
+      words = prompt.scan(/[a-z0-9]+/)
+      words = words.reject { |w| stopwords.include?(w) }
+
+      if words.include?("hello") && words.include?("world")
+        return "hello_world.rb"
+      end
+
+      picked = []
+      preferred.each do |w|
+        picked << w if words.include?(w)
+        break if picked.size >= 3
+      end
+
+      if picked.empty?
+        words.each do |w|
+          picked << w
+          break if picked.size >= 3
+        end
+      end
+
+      return "main.rb" if picked.empty?
+
+      "#{picked.uniq.first(3).join("_")}.rb"
+    end
+
+    # --------------------------------------------------
+    # Directory Tree Snapshot
+    # --------------------------------------------------
+
+    def tree(max_depth: 3)
+      output = []
+
+      Dir.glob("**/*", base: @root).each do |path|
+        depth = path.count(File::SEPARATOR)
+        next if depth > max_depth
+        next if path.start_with?(".git")
+
+        output << path
+      end
+
+      output.sort.join("\n")
+    end
+
+    # --------------------------------------------------
+    # Safety Helpers
+    # --------------------------------------------------
+
+    private
+
+    def last_significant_line(content)
+      content.lines.reverse_each do |line|
+        stripped = line.strip
+        next if stripped.empty?
+        next if stripped.start_with?("#")
+        return line
+      end
+      nil
+    end
+
+    def safe_path(path)
+      expanded = File.expand_path(path, @root)
+      unless expanded.start_with?(@root)
+        raise "Access outside project root is not allowed"
+      end
+      expanded
+    end
+
+    def error(message)
+      puts @pastel.red(message)
+      nil
+    end
+  end
+end

data/lib/vibecode.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require_relative "vibecode/version"
+require_relative "vibecode/cli"
+require_relative "vibecode/ollama_client"
+require_relative "vibecode/workspace"
+require_relative "vibecode/git"
+require_relative "vibecode/agent"
+
+module Vibecode
+  class Error < StandardError; end
+  # Your code goes here...
+end

data/sig/vibecode.rbs

@@ -0,0 +1,4 @@
+module Vibecode
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

data/test/test_helper.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+$LOAD_PATH.unshift File.expand_path("../lib", __dir__)
+require "vibecode"
+
+require "minitest/autorun"

data/test/test_vibecode.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require "test_helper"
+
+class TestVibecode < Minitest::Test
+  def test_that_it_has_a_version_number
+    refute_nil ::Vibecode::VERSION
+  end
+
+  def test_it_does_something_useful
+    assert false
+  end
+end

data/vibecode.gemspec

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative "lib/vibecode/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "vibecode"
+  spec.version = Vibecode::VERSION
+  spec.authors = ["hackliteracy"]
+  spec.email = ["hackliteracy@gmail.com"]
+
+  spec.summary = "A local-first “Codex-style” CLI but powered by Ollama"
+  spec.description = "Local AI coding agent with abilities like: File editing with diffs, Git command approvals, Model switching, Repo awareness. All on your machine available offline"
+  spec.homepage = "https://github.com/ktamulonis/vibecode"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 3.0"
+
+  spec.metadata["allowed_push_host"] = "https://rubygems.org/"
+
+  spec.metadata["source_code_uri"] = "https://github.com/ktamulonis/vibecode"
+  spec.metadata["changelog_uri"] = "https://github.com/ktamulonis/vibecode/blob/main/CHANGELOG.md"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(__dir__) { `git ls-files -z`.split("\x0") }
+  spec.bindir = "exe"
+  spec.executables = ["vibecode"]
+
+  spec.require_paths = ["lib"]
+
+  # Uncomment to register a new dependency of your gem
+  spec.add_dependency "tty-prompt", "~> 0.23"
+  spec.add_dependency "tty-spinner", "~> 0.9"
+  spec.add_dependency "pastel", "~> 0.8"
+  spec.add_dependency "httparty", "~> 0.21"
+  spec.add_dependency "diffy", "~> 3.4"
+
+  # For more information and examples about making a new gem, check out our
+  # guide at: https://bundler.io/guides/creating_gem.html
+end

metadata

@@ -0,0 +1,140 @@
+--- !ruby/object:Gem::Specification
+name: vibecode
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- hackliteracy
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2026-02-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: tty-prompt
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.23'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.23'
+- !ruby/object:Gem::Dependency
+  name: tty-spinner
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: pastel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.8'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.21'
+- !ruby/object:Gem::Dependency
+  name: diffy
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.4'
+description: 'Local AI coding agent with abilities like: File editing with diffs,
+  Git command approvals, Model switching, Repo awareness. All on your machine available
+  offline'
+email:
+- hackliteracy@gmail.com
+executables:
+- vibecode
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- CHANGELOG.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/vibecode
+- lib/vibecode.rb
+- lib/vibecode/agent.rb
+- lib/vibecode/cli.rb
+- lib/vibecode/git.rb
+- lib/vibecode/ollama_client.rb
+- lib/vibecode/version.rb
+- lib/vibecode/workspace.rb
+- sig/vibecode.rbs
+- test/test_helper.rb
+- test/test_vibecode.rb
+- vibecode.gemspec
+homepage: https://github.com/ktamulonis/vibecode
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org/
+  source_code_uri: https://github.com/ktamulonis/vibecode
+  changelog_uri: https://github.com/ktamulonis/vibecode/blob/main/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.22
+signing_key:
+specification_version: 4
+summary: A local-first “Codex-style” CLI but powered by Ollama
+test_files: []