ai-commit-message 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: ef95bc52d9e0600b53c10e093a2d3c057e61f8746d59495f16e0f29c68699995
+  data.tar.gz: 531a062526261dd27b1927660b817509aaa0fd79490f82ca3d16883aeab4c736
+SHA512:
+  metadata.gz: e8306cabe049db41c31b8cf9568c8cb52771041e5d9c26671984de3bd9c3b16957a6de1178a5aebd6af35990faa0b634e7f888c61cbcc95aa3a4d49d6342c332
+  data.tar.gz: 6348572010d2362c88ec3e7a5bd53fa94233663b0887ab3f9201dc20803aabe11e434c9fb7ebbfa1c59ecc3afb195e18f6f8e6fbb9c035795b2675edd82c357e

data/README.md

@@ -0,0 +1,2 @@
+# ai-commit-message
+Git commit message suggested by AI

data/bin/ai-commit-message

@@ -0,0 +1,2 @@
+#!/usr/bin/env ruby
+require_relative '../lib/ai-commit-message/cli'

data/lib/ai-commit-message/cli.rb

@@ -0,0 +1,41 @@
+require 'thor'
+require 'tty-prompt'
+require_relative '../ai-commit-message/suggester'
+require_relative 'config_manager'
+
+module AiCommitMessage
+  class CLI < Thor
+    desc "commit", "Generate git commit message"
+    method_option :url, type: :string, default: 'http://localhost:11434/api/generate'
+    method_option :model, type: :string, default: 'qwen2.5-coder:7b'
+
+    def commit(options = {})
+      git_diff_output = `git diff --cached --no-color`
+      git_log_output = `git log --format=%s -n 30`
+      git_current_branch = `git branch --show-current`
+
+      url = options[:url] || ConfigManager.get_url || 'http://localhost:11434/api/generate'
+      model = options[:model] || ConfigManager.get_model || 'qwen2.5-coder:7b'
+
+      suggester = AiCommitMessage::Suggester.new(git_diff_output, git_log_output, git_current_branch)
+      commit_message = suggester.generate_commit_message(url:, model:)
+
+      puts commit_message
+    end
+
+    desc "config", "Set global configs. API URL and Model name"
+    def config
+      prompt = TTY::Prompt.new
+
+      url = prompt.ask("API URL:", default: ConfigManager.get_url || 'http://localhost:11434/api/generate')
+      model = prompt.ask("Model name:", default: ConfigManager.get_model || 'qwen2.5-coder:7b')
+
+      ConfigManager.set_url(url)
+      ConfigManager.set_model(model)
+
+      puts "Configuration updated successfully!"
+    end
+  end
+end
+
+AiCommitMessage::CLI.start

data/lib/ai-commit-message/config_manager.rb

@@ -0,0 +1,38 @@
+require 'fileutils'
+
+class ConfigManager
+  CONFIG_FILE = File.expand_path('~/.ai-commit-message.conf')
+
+  def self.get_url = config['url']
+
+  def self.set_url(url)
+    config['url'] = url
+    save_config
+  end
+
+  def self.get_model = config['model']
+
+  def self.set_model(model)
+    config['model'] = model
+    save_config
+  end
+
+  private_class_method
+
+  def self.config
+    return @config if defined?(@config)
+
+    @config ||= {}
+
+    if File.exist?(CONFIG_FILE)
+      File.read(CONFIG_FILE).each_line do |line|
+        key, value = line.chomp.split('=')
+        @config[key.to_sym] = value
+      end
+    end
+
+    @config
+  end
+
+  def self.save_config = File.write(CONFIG_FILE, config.map { |key, value| "#{key}=#{value}" }.join("\n"))
+end

data/lib/ai-commit-message/suggester.rb

@@ -0,0 +1,41 @@
+require 'net/http'
+require 'uri'
+require 'json'
+
+module AiCommitMessage
+  class Suggester
+    def initialize(git_diff_output, git_log_output, git_current_branch)
+      @git_diff_output = git_diff_output
+      @git_log_output = git_log_output
+      @git_current_branch = git_current_branch
+    end
+
+    def generate_commit_message(url:, model:)
+      url = URI(url)
+      body = {
+        model: model,
+        prompt: "Create a concise git commit message with no more than 250 characters. Exclude anything unnecessary such as translation, backticks characters or multiple suggestions, since your entire response will be passed directly into git commit. Consider the following messages as example to follow: #{@git_log_output}.
+            The First [#XXX] option is the branch name. The current branch name is: #{@git_current_branch}. Now Do it for the following git diff: #{@git_diff_output}",
+        stream: false,
+        options: { temperature: 0.8 }
+      }
+      json_body = body.to_json
+      headers = { 'Content-Type' => 'application/json' }
+
+      http = Net::HTTP.new(url.host, url.port)
+      request = Net::HTTP::Post.new(url)
+      request.body = json_body
+      headers.each { |key, value| request[key] = value }
+      response = http.request(request)
+
+      begin
+        json = JSON.parse(response.body)
+        return json['response'] if json
+      rescue JSON::ParserError => e
+        puts "Failed to parse API response as JSON: #{e.message}"
+      end
+
+      nil
+    end
+  end
+end

metadata

@@ -0,0 +1,105 @@
+--- !ruby/object:Gem::Specification
+name: ai-commit-message
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Afonso Neto
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2025-03-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: tty-prompt
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-http
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Git commit messages suggested by AI
+email:
+- afonso.pontesneto@gmail.com
+executables:
+- ai-commit-message
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- bin/ai-commit-message
+- lib/ai-commit-message/cli.rb
+- lib/ai-commit-message/config_manager.rb
+- lib/ai-commit-message/suggester.rb
+homepage: https://github.com/AfonsoNeto/ai-commit-message
+licenses:
+- 
+metadata: {}
+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.11
+signing_key: 
+specification_version: 4
+summary: Git commit messages suggested by AI
+test_files: []