resquirrel 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c7e1ffeaf1f7ec104553dabc0f016a0ae3d67a6b34638a81d26d0cca0072c1a7
-  data.tar.gz: fd3b10b5c35ddaf4399541f99d828b321a68ca396b46128436cd8965a853bca5
+  metadata.gz: 231c1d6fb753b78c3998309549bd06276fd5a9768ebeac4300d75a0e5f2497c1
+  data.tar.gz: ca016974e8327e4e0294b89da4f837102d15f51fee19543e256a0af77ce9fdde
 SHA512:
-  metadata.gz: 20b0e5df45f8dad12622dc5b2ff09a6485ca502ff09867ac7930981b17ef51c1f25cfbf412b56a0129196ff116a844b0ef63181f14bda9252bb28f42bdfc63c2
-  data.tar.gz: 0b7c49d458c503666b44da267497ab279fd078e77d8d779a1060008b2c4aaa175654dc9980e4292195204bc5bdb7fb6b10e0ff228ae8668308c8bb36b4eb8a68
+  metadata.gz: 1c914d844624dc48c97c0df0edbdbe383d3b04d3e9d5f06707aafddcfe00ed6ec584457ca0cfd1448ca21af15b3628f3190d4dd1ebb9340b230437222513db49
+  data.tar.gz: 991150935129f67c224bf3edb1d5f1990d68003a1fb3978dc7b058a8d3fa5596992b79f3364393eec59768d9d1f052ac3e31f2815030e8a481e3f7126f2ae4a8

data/exe/resquirrel

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "resquirrel"
+
+Resquirrel.generate_release_note

data/lib/client/notion.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "json"
+
+class NotionClient
+  def initialize(api_key, database_id)
+    @api_key = api_key
+    @database_id = database_id
+  end
+
+  def create_release_note(summary, url)
+    uri = URI("https://api.notion.com/v1/pages")
+    request = Net::HTTP::Post.new(uri)
+    request["Authorization"] = "Bearer #{@api_key}"
+    request["Notion-Version"] = "2022-06-28"
+    request.content_type = "application/json"
+    request.body = {
+      parent: { type: "database_id", database_id: @database_id },
+      properties: {
+        title: {
+          title: [
+            {
+              type: "text",
+              text: {
+                content: summary
+              }
+            }
+          ]
+        },
+        URL: {
+          url: url
+        }
+      }
+    }.to_json
+
+    Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
+      http.request(request)
+    end
+  end
+end

data/lib/client/openai.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "json"
+
+class OpenAiClient
+  BASE_URL = "https://api.openai.com/v1"
+
+  def initialize(api_key, model: "gpt-3.5-turbo")
+    @api_key = api_key
+    @model = model
+  end
+
+  def summary_pr(title, body, max_tokens: 200)
+    uri = URI("#{BASE_URL}/chat/completions")
+    request = Net::HTTP::Post.new(uri)
+    request["Authorization"] = "Bearer #{@api_key}"
+    request.content_type = "application/json"
+    request.body = {
+      model: @model,
+      messages: [{ "role": "user",
+                   "content":
+                   "これからGitHub PRのタイトルと内容を送ります。
+                   この内容を、プログラミングやウェブ技術にそれほど詳しくない人のために要約してください。
+                   日本語で必ず返答してください。
+                   要約の文章はタイトル：本文：のようにせず、文章だけでお願いします。
+                   タイトルと本文を踏まえて、どのような変更があったのか文章にわかりやすくまとめてください。
+                   以下がPRの内容です。
+                   タイトル： #{title}
+                   本文： #{body}" }],
+      max_tokens: max_tokens
+    }.to_json
+
+    response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
+      http.request(request)
+    end
+
+    JSON.parse(response.body)
+  end
+end

data/lib/resquirrel/version.rb

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

data/lib/resquirrel.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require_relative "resquirrel/version"
+require_relative "client/openai"
+require_relative "client/notion"
+require "json"
+require "logger"
+
+module Resquirrel
+  def self.generate_release_note
+    openai_client = OpenAiClient.new(ENV["OPENAI_API_KEY"])
+    notion_client = NotionClient.new(ENV["NOTION_API_KEY"], ENV["NOTION_DATABASE_ID"])
+    logger = Logger.new($stdout)
+
+    logger.info "Getting PR information..."
+    pr_data = JSON.parse(File.read(ENV["GITHUB_EVENT_PATH"]))["pull_request"]
+
+    title = pr_data["title"]
+    body = pr_data["body"]
+
+    logger.info "Summarizing PR with OpenAI..."
+    response = openai_client.summary_pr(title, body)
+
+    summary = response["choices"].first["message"]["content"]
+    url = pr_data["html_url"]
+
+    logger.info "Updating Notion database..."
+    notion_client.create_release_note(summary, url)
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: resquirrel
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - yuki-snow1823
 autorequire:
-bindir: bin
+bindir: exe
 cert_chain: []
-date: 2024-05-04 00:00:00.000000000 Z
+date: 2024-05-25 00:00:00.000000000 Z
 dependencies: []
 description: RubyGems that automatically create release notes for Notion DB using
   Open AI API when PR merge.
@@ -25,6 +25,11 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- exe/resquirrel
+- lib/client/notion.rb
+- lib/client/openai.rb
+- lib/resquirrel.rb
+- lib/resquirrel/version.rb
 homepage: https://github.com/yuki-snow1823/resquirrel
 licenses:
 - MIT