bas 1.1.2 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f98a741f079b8f2b6a3456f7c9ace7082288174f2a3bc37743e698742c3a42a1
-  data.tar.gz: f849ecf2d96fd6247c9e9f90085ba2aab63ba160b52c9f99b791964e8f5be97a
+  metadata.gz: 595076237d4e4862f9aa5c615bae73312ba412f953549d2e426b3166c36608cc
+  data.tar.gz: 221686b0a213c00a3bf1973963c2b3871ad24861f8f90f6095fb498c51de287c
 SHA512:
-  metadata.gz: bc67ff566802b434362cade800161931c80c4b5f612867abc3d55b622dd269dabc24ddf7872b20686bf4c859d60736d5bd0ed22030d0e7c4388427c1b6872bf2
-  data.tar.gz: e43c519d0b73512148e30e9069e63e16aa792aec7ceacf639d5ba403784efca83d5399187455783426d40c17689d162b668379d0e664b3d17208213a72b6967c
+  metadata.gz: '083d7414310b342bdc27c5be5bd44fa7ba347e070081c47bd784c92bbb8a524181aa086b1912ef876bd90c6679203e7818b5a58205ca352a5f127e85435e75e9'
+  data.tar.gz: d1f183178454b2b2a8883862a1dbaff9a6be69f83520d45ae02f0a35ae9027647118f89cbff85287c4597639b1942b7454b45c9e4b2efc84085de53990b8893a

data/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+# 1.2.0 (12.07.2024)
+- [Add bots to process digital ocean billing checker](https://github.com/kommitters/bas/issues/78)
+
+# 1.1.3 (08.07.2024)
+- [Update write media request to set in process](https://github.com/kommitters/bas/issues/76)
+
 # 1.1.2 (03.07.2024)
 - [Add gem dependencies to the specification file](https://github.com/kommitters/bas/issues/72)
 

data/Gemfile

@@ -29,4 +29,7 @@ gem "pg", "~> 1.5", ">= 1.5.4"
 
 gem "gmail_xoauth"
 
+gem "google-api-client"
+gem "googleauth"
+
 gem "md_to_notion", "~> 0.1.4"

data/lib/bas/bot/fetch_billing_from_digital_ocean.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+require_relative "./base"
+require_relative "../read/default"
+require_relative "../utils/digital_ocean/request"
+require_relative "../write/postgres"
+
+module Bot
+  ##
+  # The Bot::FetchBillingFromDigitalOcean class serves as a bot implementation to read digital
+  # ocean current billing using the DigitalOcean API
+  #
+  # <br>
+  # <b>Example</b>
+  #
+  #   options = {
+  #     process_options: {
+  #       secret: "digital_ocean_secret_key"
+  #     },
+  #     write_options: {
+  #       connection: {
+  #         host: "host",
+  #         port: 5432,
+  #         dbname: "bas",
+  #         user: "postgres",
+  #         password: "postgres"
+  #       },
+  #       db_table: "use_cases",
+  #       tag: "FetchBillingFromDigitalOcean"
+  #     }
+  #   }
+  #
+  #   bot = Bot::FetchBillingFromDigitalOcean.new(options)
+  #   bot.execute
+  #
+  class FetchBillingFromDigitalOcean < Bot::Base
+    # Read function to execute the default Read component
+    #
+    def read
+      reader = Read::Default.new
+
+      reader.execute
+    end
+
+    # Process function to execute the DigitalOcean utility to fetch bills
+    #
+    def process
+      response = Utils::DigitalOcean::Request.execute(params)
+
+      if response.code == 200
+
+        { success: { billing: response.parsed_response } }
+      else
+        { error: { message: response.parsed_response, status_code: response.code } }
+      end
+    end
+
+    # Write function to execute the PostgresDB write component
+    #
+    def write
+      write = Write::Postgres.new(write_options, process_response)
+
+      write.execute
+    end
+
+    private
+
+    def params
+      {
+        endpoint: "customers/my/balance",
+        secret: process_options[:secret],
+        method: "get",
+        body: {}
+      }
+    end
+  end
+end

data/lib/bas/bot/format_do_bill_alert.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+require_relative "./base"
+require_relative "../read/postgres"
+require_relative "../write/postgres"
+
+module Bot
+  ##
+  # The Bot::FormatDoBillAlert class serves as a bot implementation to format DigitalOcean bill
+  # alerts from a PostgresDB database, format them with a specific template, and write them on a
+  # PostgresDB table with a specific format.
+  #
+  # <br>
+  # <b>Example</b>
+  #
+  #   options = {
+  #     read_options: {
+  #       connection: {
+  #         host: "localhost",
+  #         port: 5432,
+  #         dbname: "bas",
+  #         user: "postgres",
+  #         password: "postgres"
+  #       },
+  #       db_table: "do_billing",
+  #       tag: "FetchBillingFromDigitalOcean"
+  #     },
+  #     process_options: {
+  #       threshold: 7
+  #     },
+  #     write_options: {
+  #       connection: {
+  #         host: "localhost",
+  #         port: 5432,
+  #         dbname: "bas",
+  #         user: "postgres",
+  #         password: "postgres"
+  #       },
+  #       db_table: "do_billing",
+  #       tag: "FormatDoBillAlert"
+  #     }
+  #   }
+  #
+  #   bot = Bot::FormatDoBillAlert.new(options)
+  #   bot.execute
+  #
+  class FormatDoBillAlert < Bot::Base
+    # read function to execute the PostgresDB Read component
+    #
+    def read
+      reader = Read::Postgres.new(read_options.merge(conditions))
+
+      reader.execute
+    end
+
+    # Process function to format the notification using a template
+    #
+    def process
+      return { success: { notification: "" } } if unprocessable_response || !threshold_exceeded
+
+      { success: { notification: message } }
+    end
+
+    # Write function to execute the PostgresDB write component
+    #
+    def write
+      write = Write::Postgres.new(write_options, process_response)
+
+      write.execute
+    end
+
+    private
+
+    def conditions
+      {
+        where: "archived=$1 AND tag=$2 AND stage=$3 ORDER BY inserted_at ASC",
+        params: [false, read_options[:tag], "unprocessed"]
+      }
+    end
+
+    def threshold_exceeded
+      daily_usage > process_options[:threshold]
+    end
+
+    def daily_usage
+      balance = read_response.data["billing"]["month_to_date_balance"].to_f
+      day_of_month = Time.now.utc.mday
+
+      balance / day_of_month
+    end
+
+    def message
+      balance = read_response.data["billing"]["month_to_date_balance"]
+      threshold = process_options[:threshold]
+
+      """The daily usage was exceeded.
+      Current balance: #{balance}
+      Threshold: #{threshold}
+      Expected daily usage: #{daily_usage.round(3)}
+      """
+    end
+  end
+end

data/lib/bas/bot/notify_do_bill_alert_email.rb

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+require_relative "./base"
+require_relative "../read/postgres"
+require_relative "../write/postgres"
+require_relative "../utils/google/send_email"
+
+module Bot
+  ##
+  # The Bot::NotifyDoBillAlertEmail class serves as a bot implementation to send digital
+  # ocean daily bill exceeded alert emails to a recipient using a google account
+  #
+  # <br>
+  # <b>Example</b>
+  #
+  #   options = {
+  #     read_options: {
+  #       connection: {
+  #         host: "localhost",
+  #         port: 5432,
+  #         dbname: "bas",
+  #         user: "postgres",
+  #         password: "postgres"
+  #       },
+  #       db_table: "do_billing",
+  #       tag: "FetchBillingFromDigitalOcean"
+  #     },
+  #     process_options: {
+  #       refresh_token: "email server refresh token",
+  #       client_id: "email server client it",
+  #       client_secret: "email server client secret",
+  #       user_email: "sender@mail.com",
+  #       recipient_email: "recipient@mail.com",
+  #       threshold: "THRESHOLD"
+  #     },
+  #     write_options: {
+  #       connection: {
+  #         host: "localhost",
+  #         port: 5432,
+  #         dbname: "bas",
+  #         user: "postgres",
+  #         password: "postgres"
+  #       },
+  #       db_table: "do_billing",
+  #       tag: "NotifyDoBillAlertEmail"
+  #     }
+  #   }
+  #
+  #   bot = Bot::NotifyDoBillAlertEmail.new(options)
+  #   bot.execute
+  #
+  class NotifyDoBillAlertEmail < Bot::Base
+    SUBJECT = "Digital Ocean Daily Threshold Alert"
+
+    # read function to execute the PostgresDB Read component
+    #
+    def read
+      reader = Read::Postgres.new(read_options.merge(conditions))
+
+      reader.execute
+    end
+
+    # process function to execute the Google send email utility
+    #
+    def process
+      return { success: {} } if unprocessable_response
+
+      response = Utils::GoogleService::SendEmail.new(params).execute
+
+      response[:error].nil? ? { success: {} } : { error: response }
+    end
+
+    # write function to execute the PostgresDB write component
+    #
+    def write
+      write = Write::Postgres.new(write_options, process_response)
+
+      write.execute
+    end
+
+    private
+
+    def conditions
+      {
+        where: "archived=$1 AND tag=$2 AND stage=$3 ORDER BY inserted_at ASC",
+        params: [false, read_options[:tag], "unprocessed"]
+      }
+    end
+
+    def params
+      process_options.merge({ subject: SUBJECT, message: read_response.data["notification"] })
+    end
+  end
+end

data/lib/bas/bot/write_media_review_in_notion.rb

@@ -5,6 +5,7 @@ require "json"
 require_relative "./base"
 require_relative "../read/postgres"
 require_relative "../utils/notion/request"
+require_relative "../utils/notion/update_db_state"
 require_relative "../write/postgres"
 
 module Bot
@@ -47,7 +48,7 @@ module Bot
   #   bot.execute
   #
   class WriteMediaReviewInNotion < Bot::Base
-    CHUNK_SIZE = "1000"
+    READY_STATE = "ready"
 
     # read function to execute the PostgresDB Read component
     #
@@ -65,6 +66,8 @@ module Bot
       response = Utils::Notion::Request.execute(params)
 
       if response.code == 200
+        update_state
+
         { success: { page_id: read_response.data["page_id"], property: read_response.data["property"] } }
       else
         { error: { message: response.parsed_response, status_code: response.code } }
@@ -128,5 +131,16 @@ module Bot
       when "paragraph" then "Text review results/"
       end
     end
+
+    def update_state
+      data = {
+        property: read_response.data["property"],
+        page_id: read_response.data["page_id"],
+        state: READY_STATE,
+        secret: process_options[:secret]
+      }
+
+      Utils::Notion::UpdateDbState.execute(data)
+    end
   end
 end

data/lib/bas/bot/write_media_review_requests.rb

@@ -2,6 +2,7 @@
 
 require_relative "./base"
 require_relative "../read/postgres"
+require_relative "../utils/notion/update_db_state"
 require_relative "../write/postgres"
 
 module Bot
@@ -53,6 +54,8 @@ module Bot
   #   bot.execute
   #
   class WriteMediaReviewRequests < Bot::Base
+    IN_PROCESS_STATE = "in process"
+
     # read function to execute the PostgresDB Read component
     #
     def read
@@ -97,7 +100,20 @@ module Bot
     def write_request(request)
       return { error: request } if request["media"].empty? || !request["error"].nil?
 
+      update_state(request)
+
       { success: request }
     end
+
+    def update_state(request)
+      data = {
+        property: request["property"],
+        page_id: request["page_id"],
+        state: IN_PROCESS_STATE,
+        secret: process_options[:secret]
+      }
+
+      Utils::Notion::UpdateDbState.execute(data)
+    end
   end
 end

data/lib/bas/utils/digital_ocean/request.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require "httparty"
+require "json"
+
+module Utils
+  module DigitalOcean
+    ##
+    # This module is a Notion utility for sending request to create, update, or delete
+    # Notion resources.
+    #
+    module Request
+      DIGITAL_OCEAN_BASE_URL = "https://api.digitalocean.com/v2"
+
+      # Implements the request process logic to Notion.
+      #
+      # <br>
+      # <b>Params:</b>
+      # * <tt>method</tt> HTTP request method: post, get, put, etc.
+      # * <tt>body</tt> Request body (Hash).
+      # * <tt>endpoint</tt> Notion resource endpoint.
+      # * <tt>secret</tt> Notion secret.
+      #
+      # <br>
+      # <b>returns</b> <tt>HTTParty::Response</tt>
+      #
+      def self.execute(params)
+        url = "#{DIGITAL_OCEAN_BASE_URL}/#{params[:endpoint]}"
+
+        headers = headers(params[:secret])
+
+        HTTParty.send(params[:method], url, { body: params[:body].to_json, headers: })
+      end
+
+      def self.headers(secret)
+        {
+          "Authorization" => "Bearer #{secret}",
+          "Content-Type" => "application/json"
+        }
+      end
+    end
+  end
+end

data/lib/bas/utils/google/send_email.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+require "google/apis/gmail_v1"
+require "googleauth"
+
+module Utils
+  module GoogleService
+    ##
+    # This module is a Google service utility to send emails from a google account
+    #
+    class SendEmail
+      SCOPES = ["https://mail.google.com/", "https://www.googleapis.com/auth/gmail.send"].freeze
+      CONTENT_TYPE = "message/rfc822"
+
+      def initialize(params)
+        @refresh_token = params[:refresh_token]
+        @client_id = params[:client_id]
+        @client_secret = params[:client_secret]
+        @user_email = params[:user_email]
+        @recipient_email = params[:recipient_email]
+        @subject = params[:subject]
+        @message = params[:message]
+      end
+
+      def execute
+        { send_email: }
+      rescue StandardError => e
+        { error: e.to_s }
+      end
+
+      private
+
+      def send_email
+        service = Google::Apis::GmailV1::GmailService.new
+
+        service.authorization = access_token
+
+        service.send_user_message(@user_email, upload_source:, content_type: CONTENT_TYPE)
+      end
+
+      def upload_source
+        message = <<~END_OF_MESSAGE
+          From: me
+          To: #{@recipient_email.join(",")}
+          Subject: #{@subject}
+          MIME-Version: 1.0
+          Content-Type: text/plain; charset=UTF-8
+
+          #{@message}
+        END_OF_MESSAGE
+
+        StringIO.new(message)
+      end
+
+      def access_token
+        client = Google::Auth::UserRefreshCredentials.new(auth_params)
+
+        client.fetch_access_token!
+        client.access_token
+      end
+
+      def auth_params
+        {
+          client_id: @client_id,
+          client_secret: @client_secret,
+          refresh_token: @refresh_token,
+          scope: SCOPES
+        }
+      end
+    end
+  end
+end

data/lib/bas/utils/notion/update_db_state.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require "httparty"
+require_relative "request"
+
+module Utils
+  module Notion
+    ##
+    # This module is a Notion utility for sending update status to notion databases.
+    #
+    module UpdateDbState
+      # Implements the request process logic to Notion.
+      #
+      # <br>
+      # <b>Params:</b>
+      # * <tt>property</tt> Name of the db property to be updated.
+      # * <tt>page_id</tt> Id of the page to be updated.
+      # * <tt>state</tt> State to be updated
+      # * <tt>secret</tt> Notion secret.
+      #
+      # <br>
+      # <b>returns</b> <tt>HTTParty::Response</tt>
+      #
+
+      def self.execute(data)
+        params = build_params(data)
+
+        Utils::Notion::Request.execute(params)
+      end
+
+      def self.build_params(data)
+        {
+          endpoint: "pages/#{data[:page_id]}",
+          secret: data[:secret],
+          method: "patch",
+          body: body(data)
+        }
+      end
+
+      def self.body(data)
+        { properties: { data[:property] => { select: { name: data[:state] } } } }
+      end
+    end
+  end
+end

data/lib/bas/version.rb

@@ -2,5 +2,5 @@
 
 module Bas
   # Gem version
-  VERSION = "1.1.2"
+  VERSION = "1.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bas
 version: !ruby/object:Gem::Version
-  version: 1.1.2
+  version: 1.2.0
 platform: ruby
 authors:
 - kommitters Open Source
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-07-03 00:00:00.000000000 Z
+date: 2024-07-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gmail_xoauth
@@ -16,14 +16,42 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.4.2
+        version: 0.4.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.4.2
+        version: 0.4.1
+- !ruby/object:Gem::Dependency
+  name: google-api-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.53'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.53'
+- !ruby/object:Gem::Dependency
+  name: googleauth
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
 - !ruby/object:Gem::Dependency
   name: httparty
   requirement: !ruby/object:Gem::Requirement
@@ -159,6 +187,7 @@ files:
 - lib/bas.rb
 - lib/bas/bot/base.rb
 - lib/bas/bot/compare_wip_limit_count.rb
+- lib/bas/bot/fetch_billing_from_digital_ocean.rb
 - lib/bas/bot/fetch_birthdays_from_notion.rb
 - lib/bas/bot/fetch_domains_wip_counts_from_notion.rb
 - lib/bas/bot/fetch_domains_wip_limit_from_notion.rb
@@ -168,11 +197,13 @@ files:
 - lib/bas/bot/fetch_next_week_ptos_from_notion.rb
 - lib/bas/bot/fetch_ptos_from_notion.rb
 - lib/bas/bot/format_birthdays.rb
+- lib/bas/bot/format_do_bill_alert.rb
 - lib/bas/bot/format_emails.rb
 - lib/bas/bot/format_wip_limit_exceeded.rb
 - lib/bas/bot/garbage_collector.rb
 - lib/bas/bot/humanize_pto.rb
 - lib/bas/bot/notify_discord.rb
+- lib/bas/bot/notify_do_bill_alert_email.rb
 - lib/bas/bot/review_media.rb
 - lib/bas/bot/update_review_media_state.rb
 - lib/bas/bot/write_media_review_in_notion.rb
@@ -181,11 +212,14 @@ files:
 - lib/bas/read/default.rb
 - lib/bas/read/postgres.rb
 - lib/bas/read/types/response.rb
+- lib/bas/utils/digital_ocean/request.rb
 - lib/bas/utils/discord/integration.rb
 - lib/bas/utils/exceptions/function_not_implemented.rb
 - lib/bas/utils/exceptions/invalid_process_response.rb
+- lib/bas/utils/google/send_email.rb
 - lib/bas/utils/imap/request.rb
 - lib/bas/utils/notion/request.rb
+- lib/bas/utils/notion/update_db_state.rb
 - lib/bas/utils/openai/run_assistant.rb
 - lib/bas/utils/postgres/request.rb
 - lib/bas/version.rb