prdigest 0.1.0

data/lib/prdigest/telegram.rb

@@ -0,0 +1,180 @@
+# frozen_string_literal: true
+
+require "json"
+require "net/http"
+require "openssl"
+require "timeout"
+require "uri"
+
+module Prdigest
+  class Telegram
+    MAX_ATTEMPTS = 3
+    MAX_RETRY_WAIT = 60
+    API_ORIGIN = "https://api.telegram.org"
+
+    class NetHTTPTransport
+      def call(uri:, request:, open_timeout:, read_timeout:, write_timeout:)
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true
+        http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+        http.verify_hostname = true if http.respond_to?(:verify_hostname=)
+        http.open_timeout = open_timeout
+        http.read_timeout = read_timeout
+        http.write_timeout = write_timeout if http.respond_to?(:write_timeout=)
+        http.request(request)
+      end
+    end
+
+    def initialize(token:, chat_id:, allowlist:, transport: NetHTTPTransport.new, sleeper: ->(seconds) { sleep(seconds) })
+      @token = token.to_s
+      @chat_id = Integer(chat_id)
+      @allowlist = Array(allowlist).map { |id| Integer(id) }.freeze
+      @transport = transport
+      @sleeper = sleeper
+      @delivered_chunk = false
+    end
+
+    def deliver(chunks, digest_date:, checkpoint_store:, scope:)
+      refuse_unlisted_chat! unless @allowlist.include?(@chat_id)
+      checkpoint_store.with_checkpoint(
+        date: digest_date,
+        chat_id: @chat_id,
+        scope: scope,
+        chunks: chunks
+      ) do |delivery|
+        @delivered_chunk ||= delivery.next_chunk.positive?
+        (delivery.next_chunk...delivery.chunks.length).each do |index|
+          @sleeper.call(1) if @delivered_chunk
+          deliver_chunk(delivery, index)
+          @delivered_chunk = true
+        end
+        delivery.delivery
+      end
+    end
+
+    def inspect
+      "#<#{self.class} chat_id=#{@chat_id} token=[REDACTED]>"
+    end
+
+    alias to_s inspect
+
+    private
+
+    def deliver_chunk(delivery, index)
+      attempts = 0
+      waited = 0
+      loop do
+        attempts += 1
+        delivery.begin_attempt(index)
+        begin
+          response = @transport.call(
+            uri: request_uri,
+            request: build_request(delivery.chunks.fetch(index)),
+            open_timeout: 10,
+            read_timeout: 30,
+            write_timeout: 30
+          )
+        rescue StandardError => e
+          fail_ambiguous!(delivery, index, e)
+        end
+
+        disposition, delay, kind = classify_response(response)
+        if disposition == :success
+          delivery.accept(index)
+          return true
+        end
+
+        permanent = disposition == :failure
+        delivery.reject(
+          index,
+          kind: permanent ? "telegram_permanent" : "telegram",
+          message: failure_message(kind),
+          permanent: permanent
+        )
+        fail_send!(
+          delivery,
+          index,
+          kind: permanent ? "telegram_permanent" : "telegram",
+          reason: kind
+        ) if permanent || attempts >= MAX_ATTEMPTS
+
+        waited = wait!(delay || attempts, kind, waited, delivery, index)
+      end
+    end
+
+    def build_request(text)
+      request = Net::HTTP::Post.new(request_uri.request_uri)
+      request["Content-Type"] = "application/json"
+      request.body = JSON.generate(
+        chat_id: @chat_id,
+        text: text,
+        parse_mode: "HTML",
+        link_preview_options: { is_disabled: true }
+      )
+      request
+    end
+
+    def request_uri
+      URI("#{API_ORIGIN}/bot#{@token}/sendMessage")
+    end
+
+    def classify_response(response)
+      code = Integer(response.code)
+      payload = JSON.parse(response.body.to_s)
+      return [:success, nil, nil] if (200..299).cover?(code) && payload["ok"] == true
+
+      if code == 429
+        delay = Integer(payload.dig("parameters", "retry_after"), exception: false)
+        return [:retry, delay, "rate_limited"]
+      end
+      return [:retry, nil, "server"] if (500..599).cover?(code)
+
+      [:failure, nil, "http_#{code}"]
+    rescue JSON::ParserError, TypeError, ArgumentError
+      code && (500..599).cover?(code) ? [:retry, nil, "server"] : [:failure, nil, "invalid_response"]
+    end
+
+    def wait!(seconds, kind, waited, delivery, index)
+      delay = Integer(seconds, exception: false)
+      total = waited + delay.to_i
+      fail_send!(delivery, index, reason: "retry_wait_exceeded") unless delay && total <= MAX_RETRY_WAIT
+      @sleeper.call(delay)
+      total
+    rescue SendError
+      raise
+    rescue StandardError
+      fail_send!(delivery, index, reason: kind)
+    end
+
+    def refuse_unlisted_chat!
+      raise SendError.new("Telegram delivery refused: chat is not allowlisted", kind: "telegram_refused")
+    end
+
+    def fail_ambiguous!(delivery, index, error)
+      reason = case error
+               when Timeout::Error then "timeout"
+               when IOError then "io"
+               when SocketError then "socket"
+               when SystemCallError then "system"
+               else "transport"
+               end
+      raise SendError.new(
+        "Telegram send outcome is ambiguous (#{reason})",
+        kind: "telegram_ambiguous",
+        delivery: delivery.delivery(failed_chunk: index)
+      )
+    end
+
+    def fail_send!(delivery, index, kind: "telegram", reason:)
+      raise SendError.new(
+        failure_message(reason),
+        kind: kind,
+        delivery: delivery.delivery(failed_chunk: index)
+      )
+    end
+
+    def failure_message(reason)
+      "Telegram send failed (#{reason})"
+    end
+  end
+end

data/lib/prdigest/version.rb

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

data/lib/prdigest.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module Prdigest
+  class Error < StandardError; end
+  class ConfigError < Error; end
+  class StateError < Error; end
+  class FetchError < Error
+    attr_reader :kind
+
+    def initialize(message, kind: "github")
+      @kind = kind
+      super(message)
+    end
+  end
+  class RenderError < Error
+    attr_reader :kind
+
+    def initialize(message, kind: "render")
+      @kind = kind
+      super(message)
+    end
+  end
+  class SendError < Error
+    attr_reader :kind, :delivery
+
+    def initialize(message, kind: "telegram", delivery: {})
+      @kind = kind
+      @delivery = delivery.transform_keys(&:to_sym).freeze
+      super(message)
+    end
+  end
+end
+
+require_relative "prdigest/version"
+require_relative "prdigest/clock"
+require_relative "prdigest/config"
+require_relative "prdigest/digest"
+require_relative "prdigest/github"
+require_relative "prdigest/renderer"
+require_relative "prdigest/delivery_checkpoint_store"
+require_relative "prdigest/telegram"
+require_relative "prdigest/result"
+require_relative "prdigest/schedule"
+require_relative "prdigest/state"
+require_relative "prdigest/runner"
+require_relative "prdigest/cli"

data/scripts/systemd/prdigest.service

@@ -0,0 +1,27 @@
+[Unit]
+Description=prdigest one-shot merged-PR digests (use with timer)
+After=network-online.target
+Wants=network-online.target
+
+[Service]
+Type=oneshot
+User=prdigest
+Group=prdigest
+EnvironmentFile=/etc/prdigest/.env
+ExecStart=/usr/local/bin/prdigest run --config /etc/prdigest/config.yml
+StateDirectory=prdigest
+StateDirectoryMode=0700
+UMask=0077
+TimeoutStartSec=1h
+Nice=10
+NoNewPrivileges=true
+ProtectSystem=strict
+ProtectHome=true
+ReadWritePaths=/var/lib/prdigest
+PrivateTmp=true
+PrivateDevices=true
+ProtectKernelTunables=true
+ProtectKernelModules=true
+ProtectControlGroups=true
+RestrictSUIDSGID=true
+LockPersonality=true

data/scripts/systemd/prdigest.timer

@@ -0,0 +1,12 @@
+[Unit]
+Description=Daily prdigest merged-PR digest
+
+[Timer]
+# Uses the host timezone. Match it to config.yml so 09:05 leaves a nine-hour
+# indexing buffer after the previous local day.
+OnCalendar=*-*-* 09:05:00
+Persistent=true
+Unit=prdigest.service
+
+[Install]
+WantedBy=timers.target

metadata

@@ -0,0 +1,186 @@
+--- !ruby/object:Gem::Specification
+name: prdigest
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Ivan Kuznetsov
+bindir: exe
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: octokit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '9.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '9.0'
+- !ruby/object:Gem::Dependency
+  name: faraday-retry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+- !ruby/object:Gem::Dependency
+  name: tzinfo
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: erb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.25'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.25'
+- !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'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.25'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.25'
+description: 'Config-driven GitHub merged-PR digests with Telegram delivery. Hive-independent:
+  repo list + GitHub token + chat allowlist.'
+email:
+- ivan@ikuznetsov.com
+executables:
+- prdigest
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE
+- README.md
+- SECURITY.md
+- configs/config.example.yml
+- exe/prdigest
+- lib/prdigest.rb
+- lib/prdigest/cli.rb
+- lib/prdigest/clock.rb
+- lib/prdigest/config.rb
+- lib/prdigest/delivery_checkpoint_store.rb
+- lib/prdigest/digest.rb
+- lib/prdigest/github.rb
+- lib/prdigest/renderer.rb
+- lib/prdigest/result.rb
+- lib/prdigest/runner.rb
+- lib/prdigest/schedule.rb
+- lib/prdigest/state.rb
+- lib/prdigest/telegram.rb
+- lib/prdigest/version.rb
+- scripts/systemd/prdigest.service
+- scripts/systemd/prdigest.timer
+homepage: https://github.com/ivankuznetsov/prdigest
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/ivankuznetsov/prdigest
+  source_code_uri: https://github.com/ivankuznetsov/prdigest/tree/main
+  bug_tracker_uri: https://github.com/ivankuznetsov/prdigest/issues
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: Standalone multi-repo daily merged-PR digests for Telegram
+test_files: []