pr_ai_gen 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 9d88c164cc0b3c276e43759573219cc064e5d141cd264d46e184bca12f02926b
+  data.tar.gz: 10041eea9b1b4465539cf00d260782692f84db5ea2347b8cbe5cc2bbdd725c57
+SHA512:
+  metadata.gz: dc88eee40aed632629dc2bdb1eb2fc2fa9ee5017b95a4f6c0dba9d2c9ca70d998563c020c6640766af70d57bdf6d74bf4412abe6d532730788e04e204f75d2fa
+  data.tar.gz: 553d015be904145afab53f9dde76bf605fe12d6d875ee5341522c18f637c84879c061e777c307edef44daba0a3c84dabff949153f37c29ede5c724812b1f5dce

data/.gitignore

@@ -0,0 +1 @@
+*.gem

data/Gemfile

@@ -0,0 +1,9 @@
+source "https://rubygems.org"
+
+gem 'git'
+gem 'httparty'
+gem 'octokit'
+gem 'capybara'
+gem 'selenium-webdriver'
+gem 'optparse'
+gem 'pry'

data/Gemfile.lock

@@ -0,0 +1,82 @@
+GEM
+  remote: https://rubygems.org/
+  specs:
+    addressable (2.8.6)
+      public_suffix (>= 2.0.2, < 6.0)
+    base64 (0.2.0)
+    capybara (3.39.2)
+      addressable
+      matrix
+      mini_mime (>= 0.1.3)
+      nokogiri (~> 1.8)
+      rack (>= 1.6.0)
+      rack-test (>= 0.6.3)
+      regexp_parser (>= 1.5, < 3.0)
+      xpath (~> 3.2)
+    coderay (1.1.3)
+    faraday (2.8.1)
+      base64
+      faraday-net_http (>= 2.0, < 3.1)
+      ruby2_keywords (>= 0.0.4)
+    faraday-net_http (3.0.2)
+    git (1.19.1)
+      addressable (~> 2.8)
+      rchardet (~> 1.8)
+    httparty (0.20.0)
+      mime-types (~> 3.0)
+      multi_xml (>= 0.5.2)
+    matrix (0.4.2)
+    method_source (1.0.0)
+    mime-types (3.4.1)
+      mime-types-data (~> 3.2015)
+    mime-types-data (3.2022.0105)
+    mini_mime (1.1.5)
+    multi_xml (0.6.0)
+    nokogiri (1.15.5-arm64-darwin)
+      racc (~> 1.4)
+    nokogiri (1.15.5-x86_64-linux)
+      racc (~> 1.4)
+    octokit (8.1.0)
+      base64
+      faraday (>= 1, < 3)
+      sawyer (~> 0.9)
+    optparse (0.4.0)
+    pry (0.14.1)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    public_suffix (5.0.4)
+    racc (1.7.3)
+    rack (3.0.9.1)
+    rack-test (2.1.0)
+      rack (>= 1.3)
+    rchardet (1.8.0)
+    regexp_parser (2.9.0)
+    rexml (3.2.6)
+    ruby2_keywords (0.0.5)
+    rubyzip (2.3.2)
+    sawyer (0.9.2)
+      addressable (>= 2.3.5)
+      faraday (>= 0.17.3, < 3)
+    selenium-webdriver (4.9.0)
+      rexml (~> 3.2, >= 3.2.5)
+      rubyzip (>= 1.2.2, < 3.0)
+      websocket (~> 1.0)
+    websocket (1.2.10)
+    xpath (3.2.0)
+      nokogiri (~> 1.8)
+
+PLATFORMS
+  arm64-darwin-22
+  x86_64-linux
+
+DEPENDENCIES
+  capybara
+  git
+  httparty
+  octokit
+  optparse
+  pry
+  selenium-webdriver
+
+BUNDLED WITH
+   2.4.21

data/exe/pr-ai-gen

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+require_relative '../lib/pull_request_ai_generator'
+
+generator = PullRequestAIGenerator.new(ARGV)
+
+case generator.options[:command]
+when :init
+  generator.init
+else
+  generator.run
+end

data/lib/pull_request_ai_generator.rb

@@ -0,0 +1,161 @@
+require 'optparse'
+require 'git'
+require 'httparty'
+require 'octokit'
+require 'capybara/dsl'
+require 'pry'
+
+# Define the CLI class
+class PullRequestAIGenerator
+  OPEN_AI_URL = "https://api.openai.com/v1"
+
+  include Capybara::DSL
+
+  attr_reader :options
+
+  def initialize(args)
+    @options = parse_options(args)
+    @diff = nil
+    @pr_content = nil
+    @git_repo = nil
+  end
+
+  def init
+    puts 'Initializing pr-ai-gen...'
+    credentials_path = File.expand_path('~/.pr-gem/credentials')
+    pr_gem_path = File.expand_path('~/.pr-gem')
+
+    unless File.exist?(pr_gem_path)
+      Dir.mkdir(pr_gem_path)
+      puts '.pr-gem repository created successfully!'
+    end
+
+    if File.exist?(credentials_path)
+      puts 'Credentials file already exists!'
+      return
+    end
+
+    puts 'Please enter your OpenAI API key:'
+    openai_token = STDIN.gets.chomp
+    File.write(credentials_path, "OPENAI_TOKEN=#{openai_token}\n")
+    puts 'Credentials file created successfully!'
+  end
+
+  def run
+    puts 'Running pr-ai-gen...'
+    fetch_diff
+    load_template
+    get_openai_token
+    generate_pr_content
+    print_content
+  end
+
+  private
+
+  def parse_options(args)
+    options = { target_branch: 'main', directory_location: File.expand_path("."), command: :run }
+
+    if args[0] == 'init'
+      options[:command] = :init
+    elsif args[0] == 'generate'
+      options[:command] = :run
+
+      if args[1]
+        options[:directory_location] = File.expand_path(args[1])
+      end
+
+      if args[2]
+        local, target = args[2].split(':')
+        options[:local_branch] = local
+        options[:target_branch] = target || options[:target_branch]
+      end
+    else
+      raise Exception.new "Invalid command!"
+    end
+
+    OptionParser.new do |opts|
+      opts.banner = "Usage: pr-ai-gen <directory_location> <branch>:<target-branch=main>"
+
+      opts.on("-h", "--help", "Prints this help") do
+        puts opts
+        exit
+      end
+
+    end.parse!(args)
+
+    options
+  end
+
+  def fetch_diff
+    # Start the ssh-agent in the background
+    `eval $(ssh-agent -s)`
+
+    # Add your SSH private key to the ssh-agent
+    `ssh-add ~/.ssh/id_ed25519`
+    ssh_keys = `ssh-add -l`
+
+    if ssh_keys.include?('No identities')
+      puts 'No SSH keys are added to the ssh-agent.'
+      exit
+    end
+
+    @git_repo = Git.open(@options[:directory_location])
+    @git_repo.fetch
+    @git_repo.checkout(@options[:local_branch])
+    @diff = @git_repo.diff(@options[:target_branch], @options[:local_branch])
+  end
+
+  def load_template
+    template_path = File.join(@options[:directory_location], '.github', 'PULL_REQUEST_TEMPLATE.md')
+    if File.exist?(template_path)
+      @template = File.read(template_path)
+    else
+      @template = "# Default PR Template\n\n## Changes Made\n\nDescribe your changes in detail here."
+    end
+  end
+
+  def get_openai_token
+    credentials_path = File.expand_path('~/.pr-gem/credentials')
+    if File.exist?(credentials_path)
+      credentials = File.readlines(credentials_path).map(&:strip)
+      @openai_token = credentials.find { |line| line.start_with?('OPENAI_TOKEN=') }
+      raise "OpenAI credentials not found!" unless @openai_token
+      @openai_token = @openai_token.split('=')[1]
+    else
+      raise "OpenAI credentials file not found!"
+    end
+  end
+
+  def generate_pr_content
+    puts "Generating PR content..."
+
+    response = HTTParty.post(
+      "#{OPEN_AI_URL}/chat/completions",
+      headers: {
+        "Content-Type" => "application/json",
+        "Authorization" => "Bearer #{@openai_token}"
+      },
+      body: {
+        model: "gpt-4-turbo-preview",
+        messages: [
+          {
+            role: "user",
+            content: "fill this templete:\n #{@template}\n with the following changes:\n #{@diff.to_s}"
+          },
+        ],
+        max_tokens: 1024
+      }.to_json
+    )
+
+    @pr_content = response.dig("choices", 0, "message", "content").strip
+
+    puts "PR content generated successfully!"
+  end
+
+  def print_content
+    puts "PR CONTENT FOLLOWING:"
+    puts "BEGIN-------------------------------------------------------------------------"
+    puts @pr_content
+    puts "---------------------------------------------------------------------------END"
+  end
+end

data/pr_ai_gen.gemspec

@@ -0,0 +1,23 @@
+Gem::Specification.new do |spec|
+  spec.name          = "pr_ai_gen"
+  spec.version       = "0.1.0"
+  spec.authors       = ["Lucas Stoller"]
+  spec.email         = ["l.s.stoller@gmail.com"]
+  spec.summary       = "A CLI tool to generate pull requests using OpenAI"
+  spec.description   = "This tool automates the process of creating pull requests by using OpenAI to generate the content based on the differences between two git branches."
+  spec.homepage      = "https://theright.dev"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.bindir        = "exe"
+  spec.executables   = spec.executables = ['pr-ai-gen']
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "git"
+  spec.add_dependency "openai"
+  spec.add_dependency "octokit"
+  spec.add_dependency "capybara"
+  spec.add_dependency "selenium-webdriver"
+
+  spec.add_development_dependency "bundler", "~> 2.0"
+  spec.add_development_dependency "rake", "~> 13.0"
+end

metadata

@@ -0,0 +1,148 @@
+--- !ruby/object:Gem::Specification
+name: pr_ai_gen
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Lucas Stoller
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-03-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: git
+  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: openai
+  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: octokit
+  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: capybara
+  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: selenium-webdriver
+  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: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+description: This tool automates the process of creating pull requests by using OpenAI
+  to generate the content based on the differences between two git branches.
+email:
+- l.s.stoller@gmail.com
+executables:
+- pr-ai-gen
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- exe/pr-ai-gen
+- lib/pull_request_ai_generator.rb
+- pr_ai_gen.gemspec
+homepage: https://theright.dev
+licenses: []
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.7
+signing_key:
+specification_version: 4
+summary: A CLI tool to generate pull requests using OpenAI
+test_files: []