code_review_notifier 0.1.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e426f5dccd98b195d1c7293883395783661f4d2f7415bbef49dec86366c359ed
-  data.tar.gz: 1bc90b0595fef06e71915280ae60abccb4b4f0210e5f5989b108f00c531f1d09
+  metadata.gz: 5760e7d91fa308531d0ab4c0da163525a0b4cf2ed169822c01ad5fedd5c6e28a
+  data.tar.gz: 86a8c7f51968a37bca25eaa0406a6c3b7af9f33cf9702b4c03a349ba4f0cc577
 SHA512:
-  metadata.gz: 3d04f2041ece20693f1111669c23d995173a7f463bbe40663de23542285c3610243aa099ad65a8f1dd9d14b45e930f332a74de89e25ec5dbb85a0cd6d644f05a
-  data.tar.gz: 4b7f3b4e1193931f975f350d51f18b14e57094ca82991a9ebd1f759901d6fc1cfd23b3d46b4f86e6402b3183c6aaaef737673f7c7c3f4827829b65b54bc7133c
+  metadata.gz: b0f8c25880aa149870bc5c39e13d081b35d9f95b5fcb3efea47ed272b2e7acadb860c7283d3efa9b29ec114b552ae92f9910bedaf5b57c003af0017cdb2fb3b1
+  data.tar.gz: b0ed3ddba99bff4fe392847c7f8a5430f68fde44e32f02fdcbfb8deb35edc24526691b3150dc8b4102896667ebf56aa530c7b0647844025522050fba4156f7c1

data/Gemfile

@@ -1,6 +1,5 @@
-source 'https://rubygems.org'
+source "https://rubygems.org"
 
 gemspec
 
-gem "sqlite3"
-gem "httparty"
+gem "rubiclifier", "1.0.0"

data/bin/code_review_notifier

@@ -1,5 +1,5 @@
 #!/usr/bin/env ruby
 
-require 'code_review_notifier'
+require "code_review_notifier"
 
-CodeReviewNotifier.call(ARGV)
+CodeReviewNotifier.new(ARGV).call

data/lib/code_change_notification.rb

@@ -0,0 +1,28 @@
+require "rubiclifier"
+require_relative "./api.rb"
+
+class CodeChangeNotification
+  attr_reader :code_change_activity
+  private :code_change_activity
+
+  def initialize(code_change_activity)
+    @code_change_activity = code_change_activity
+  end
+
+  def send
+    code_change = code_change_activity.code_change
+    id = code_change.id
+    owner = code_change.owner
+    subject = code_change.subject
+
+    message = code_change_activity.message
+    author = code_change_activity.author
+    Rubiclifier::Notification.new(
+      author,
+      message,
+      "#{owner}: #{subject}",
+      Api.current_api.favicon,
+      Api.current_api.code_change_url(code_change)
+    ).send
+  end
+end

data/lib/code_review_notifier.rb

@@ -1,36 +1,24 @@
-require "io/console"
+require "rubiclifier"
 require_relative "./api.rb"
-require_relative "./notifier.rb"
+require_relative "./code_change_notification.rb"
 
-SECONDS_BETWEEN_RUNS = 120
+SECONDS_BETWEEN_RUNS = 90
 SECONDS_BETWEEN_NOTIFICATIONS = 5
 
-class CodeReviewNotifier
-  def self.setup(args)
-    system("mkdir -p ~/.code_review_notifier")
-    system("brew bundle --file #{File.expand_path(File.dirname(__FILE__) + "/..")}/Brewfile")
-
-    if args[0] == "--setup" || !Api.current_api.is_setup?
-      print("What's the base URL? (i.e. https://gerrit.google.com) ")
-      DB.save_setting("base_api_url", STDIN.gets.chomp, is_secret: false)
-
-      print("What's the account username? ")
-      DB.save_setting("username", STDIN.gets.chomp, is_secret: false)
-
-      print("What's the account password? (hiding input) ")
-      DB.save_setting("password", STDIN.noecho(&:gets).chomp, is_secret: true)
-      puts
-
-      print("What's the account id? (check #{Api.current_api.base_api_url}/settings/) ")
-      DB.save_setting("account_id", STDIN.gets.chomp, is_secret: false)
-
-      puts("All setup!")
-    end
+class CodeReviewNotifier < Rubiclifier::BaseApplication
+  def show_help
+    puts
+    puts("This polls for updates to patch sets/pull requests and notifies you about any relevant changes.")
+    puts
+    puts("Usage:")
+    puts("  code_review_notifier --help  | Shows this help menu")
+    puts("  code_review_notifier --setup | Runs setup")
+    puts("  code_review_notifier         | Start listening for changes (should be run in background)")
+    puts
+    exit
   end
 
-  def self.call(args)
-    setup(args)
-
+  def run_application
     while true
       is_first_run = is_first_run?
       puts
@@ -48,15 +36,51 @@ class CodeReviewNotifier
         code_change_activity.notified
         unless is_first_run
           puts("Notifying of change!")
-          Notifier.notify(code_change_activity)
-          sleep SECONDS_BETWEEN_NOTIFICATIONS
+          CodeChangeNotification.new(code_change_activity).send
+          sleep(SECONDS_BETWEEN_NOTIFICATIONS)
         end
       end
-      sleep SECONDS_BETWEEN_RUNS
+      sleep(SECONDS_BETWEEN_RUNS)
     end
   end
 
-  def self.is_first_run?
-    DB.query_single_row("SELECT id FROM code_change_activity_notified;").nil?
+  def not_setup
+    Rubiclifier::Notification.new(
+      "Missing Setup Info",
+      "Run `code_review_notifier --setup` to setup."
+    ).send
+  end
+
+  def features
+    [
+      Rubiclifier::Feature::BACKGROUND,
+      Rubiclifier::Feature::DATABASE,
+      Rubiclifier::Feature::NOTIFICATIONS
+    ]
+  end
+
+  def settings
+    @settings ||= [
+      Rubiclifier::Setting.new("base_api_url", "base URL", explanation: "e.g. https://gerrit.google.com"),
+      Rubiclifier::Setting.new("username", "account username"),
+      Rubiclifier::Setting.new("password", "account password", explanation: "input hidden", is_secret: true),
+      Rubiclifier::Setting.new("account_id", "account ID", explanation: -> {"check #{Api.current_api.base_api_url}/settings/"})
+    ]
+  end
+
+  def executable_name
+    "code_review_notifier"
+  end
+
+  def data_directory
+    "~/.code_review_notifier"
+  end
+
+  def migrations_location
+    "#{File.expand_path(File.dirname(__FILE__) + "/..")}/migrations.rb"
+  end
+
+  def is_first_run?
+    Rubiclifier::DB.query_single_row("SELECT id FROM code_change_activity_notified;").nil?
   end
 end

data/lib/gerrit_api.rb

@@ -1,9 +1,15 @@
-require_relative "./base_api.rb"
+require "rubiclifier"
 require_relative "./models/code_change.rb"
 require_relative "./models/code_change_activity.rb"
 
-class GerritApi < BaseApi
-  def self.authenticate
+class HTTParty::Parser
+  def json
+    JSON.parse(body.gsub(")]}'", ""))
+  end
+end
+
+class GerritApi < Rubiclifier::BaseApi
+  def self.login_and_get_api_token
     res = post("/login/", {
       body: "username=#{username}&password=#{URI.escape(password)}&rememberme=1",
       headers: {
@@ -11,15 +17,24 @@ class GerritApi < BaseApi
       },
       follow_redirects: false
     })
-    @token = res.headers["set-cookie"].match("GerritAccount=(.*?);")[1]
-    raise "Your username or password is incorrect. Trying running `code_review_notifier --setup` again." if @token.nil?
+    set_cookie_header = res.headers["set-cookie"] || ""
+    set_cookie_header.match("GerritAccount=(.*?);")&.to_a&.fetch(1)
+  end
+
+  def self.invalid_credentials_error
+    Rubiclifier::Notification.new(
+      "Incorrect Credentials",
+      "Trying running `code_review_notifier --setup` again."
+    ).send
+    sleep(120)
+    exit
   end
 
   def self.all_code_changes
     wrap_with_authentication do
       get("/changes/?S=0&q=is%3Aopen%20owner%3Aself%20-is%3Awip%20-is%3Aignored%20limit%3A25&q=is%3Aopen%20owner%3Aself%20is%3Awip%20limit%3A25&q=is%3Aopen%20-owner%3Aself%20-is%3Awip%20-is%3Aignored%20reviewer%3Aself%20limit%3A25&q=is%3Aclosed%20-is%3Aignored%20%28-is%3Awip%20OR%20owner%3Aself%29%20%28owner%3Aself%20OR%20reviewer%3Aself%29%20-age%3A4w%20limit%3A10&o=DETAILED_ACCOUNTS&o=MESSAGES", {
         headers: {
-          "Cookie" => "GerritAccount=#{@token};"
+          "Cookie" => "GerritAccount=#{api_token};"
         }
       })
     end.parsed_response.flat_map { |js| js.map { |j| code_change_from_json(j) } }
@@ -42,4 +57,24 @@ class GerritApi < BaseApi
   def self.code_change_url(code_change)
     "#{base_api_url}/c/#{code_change.id}"
   end
+
+  def self.api_token_db_key
+    "api_token"
+  end
+
+  def self.base_api_url_db_key
+    "base_api_url"
+  end
+
+  def self.username
+    @username ||= Rubiclifier::DB.get_setting("username")
+  end
+
+  def self.password
+    @password ||= Rubiclifier::DB.get_setting("password")
+  end
+
+  def self.account_id
+    @account_id ||= Rubiclifier::DB.get_setting("account_id")
+  end
 end

data/lib/models/code_change.rb

@@ -1,5 +1,4 @@
 require "date"
-require_relative "../database.rb"
 
 class CodeChange
   attr_accessor :code_change_activity
@@ -10,7 +9,7 @@ class CodeChange
     @id = id
     @owner = owner
     @project = project
-    @subject = subject.gsub("'", "")
+    @subject = subject.gsub("'", "’")
     @updated_at = updated_at
   end
 

data/lib/models/code_change_activity.rb

@@ -1,3 +1,5 @@
+require "rubiclifier"
+
 MESSAGES_TO_IGNORE = [/Uploaded patch set 1/, /Build Started/, /owns \d+% of/]
 AUTHOR_TRANSLATIONS = {
   "Service Cloud Jenkins" => "Jenkins",
@@ -17,14 +19,14 @@ class CodeChangeActivity
   end
 
   def notified
-    DB.execute("INSERT INTO code_change_activity_notified (id, notified_at) VALUES('#{id}', CURRENT_TIMESTAMP);")
+    Rubiclifier::DB.execute("INSERT INTO code_change_activity_notified (id, notified_at) VALUES('#{id}', CURRENT_TIMESTAMP);")
   end
 
   def should_notify?
     !is_self &&
       code_change.activity_from_self_at && created_at > code_change.activity_from_self_at &&
       MESSAGES_TO_IGNORE.none? { |m| message.match(m) } &&
-      !DB.query_single_row("SELECT id FROM code_change_activity_notified WHERE id = '#{id}'")
+      !Rubiclifier::DB.query_single_row("SELECT id FROM code_change_activity_notified WHERE id = '#{id}'")
   end
 
   def self.translate_author(author)

data/migrations.rb

@@ -0,0 +1,8 @@
+migrations = []
+migrations << <<SQL
+  CREATE TABLE IF NOT EXISTS code_change_activity_notified (
+    id VARCHAR(50) PRIMARY KEY,
+    notified_at TIMESTAMP
+  );
+SQL
+migrations

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: code_review_notifier
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Kyle Grinstead
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-07-07 00:00:00.000000000 Z
+date: 2020-07-09 00:00:00.000000000 Z
 dependencies: []
 description: 
 email: kyleag@hey.com
@@ -17,24 +17,22 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
-- Brewfile
 - Gemfile
 - bin/code_review_notifier
 - lib/api.rb
-- lib/base_api.rb
-- lib/cipher.rb
+- lib/code_change_notification.rb
 - lib/code_review_notifier.rb
-- lib/database.rb
 - lib/gerrit_api.rb
 - lib/models/code_change.rb
 - lib/models/code_change_activity.rb
-- lib/notifier.rb
+- migrations.rb
 homepage: https://rubygems.org/gems/code_review_notifier
 licenses:
 - MIT
 metadata:
   source_code_uri: https://github.com/MrGrinst/code_review_notifier
-post_install_message: 
+post_install_message: "\n\e[32mThanks for installing code_review_notifier!\e[0m\n\e[32mSet
+  it up by running `\e[0mcode_review_notifier --setup\e[32m`\e[0m\n\n"
 rdoc_options: []
 require_paths:
 - lib

data/Brewfile

@@ -1,2 +0,0 @@
-brew 'sqlite'
-brew 'terminal-notifier'

data/lib/base_api.rb

@@ -1,62 +0,0 @@
-require "httparty"
-require_relative "./gerrit_api.rb"
-
-class HTTParty::Parser
-  def json
-    JSON.parse(body.gsub(")]}'", ""))
-  end
-end
-
-class BaseApi
-  include HTTParty
-
-  def self.authenticate
-    raise NotImplementedError
-  end
-
-  def self.all_code_changes
-    raise NotImplementedError
-  end
-
-  def self.favicon
-    raise NotImplementedError
-  end
-
-  def self.code_change_url(code_change)
-    raise NotImplementedError
-  end
-
-  def self.wrap_with_authentication(&block)
-    res = block.call
-    if res.code == 401 || res.code == 403
-      self.authenticate
-      block.call
-    else
-      res
-    end
-  end
-
-  def self.is_setup?
-    base_api_url && username && password && account_id
-  end
-
-  def self.base_api_url
-    @base_api_url ||= begin
-                        url = DB.get_setting("base_api_url")
-                        base_uri(url)
-                        url.chomp("/")
-                      end
-  end
-
-  def self.username
-    @username ||= DB.get_setting("username")
-  end
-
-  def self.password
-    @password ||= DB.get_setting("password")
-  end
-
-  def self.account_id
-    @account_id ||= DB.get_setting("account_id")
-  end
-end

data/lib/cipher.rb

@@ -1,31 +0,0 @@
-require "base64"
-require "openssl"
-require "digest/sha1"
-
-KEY = ""
-
-class Cipher
-  def self.encrypt(text)
-    cipher = OpenSSL::Cipher.new("aes-256-cbc")
-    cipher.encrypt
-    key = Digest::SHA1.hexdigest("yourpass")[0..31]
-    iv = cipher.random_iv
-    cipher.key = key
-    cipher.iv = iv
-    encrypted = cipher.update(text)
-    encrypted << cipher.final
-    ["#{key}:#{Base64.encode64(iv).encode('utf-8')}", Base64.encode64(encrypted).encode('utf-8')]
-  end
-
-  def self.decrypt(encrypted, salt)
-    key, iv = salt.split(":")
-    iv = Base64.decode64(iv.encode('ascii-8bit'))
-    cipher = OpenSSL::Cipher.new("aes-256-cbc")
-    cipher.decrypt
-    cipher.key = key
-    cipher.iv = iv
-    decrypted = cipher.update(Base64.decode64(encrypted.encode('ascii-8bit')))
-    decrypted << cipher.final
-    decrypted
-  end
-end

data/lib/database.rb

@@ -1,48 +0,0 @@
-require "sqlite3"
-require_relative "./cipher.rb"
-
-class DB
-  def self.db
-    @db ||= begin
-              db = SQLite3::Database.new(File.expand_path("~/.code_review_notifier/data.db"))
-              migrate_if_needed(db)
-              db
-            end
-  end
-
-  def self.migrate_if_needed(db)
-    db.execute("CREATE TABLE IF NOT EXISTS code_change_activity_notified (id VARCHAR(50) PRIMARY KEY, notified_at TIMESTAMP);")
-    db.execute("CREATE TABLE IF NOT EXISTS settings (key VARCHAR(50) PRIMARY KEY, value TEXT, salt TEXT);")
-  end
-
-  def self.execute(sql)
-    db.execute(sql)
-  end
-
-  def self.query_single_row(sql)
-    db.execute(sql) do |row|
-      return row
-    end
-    return nil
-  end
-
-  def self.save_setting(key, value, is_secret:)
-    salt = "NULL"
-    if is_secret
-      salt, encrypted = Cipher.encrypt(value)
-      salt = "'#{salt}'"
-      value = encrypted
-    end
-    db.execute("DELETE FROM settings WHERE key = '#{key}';")
-    db.execute("INSERT INTO settings (key, value, salt) VALUES('#{key}', '#{value}', #{salt});")
-  end
-
-  def self.get_setting(key)
-    row = query_single_row("SELECT value, salt FROM settings WHERE key = '#{key}'")
-    if row && row[1]
-      Cipher.decrypt(row[0], row[1])
-    elsif row
-      row[0]
-    end
-  end
-end

data/lib/notifier.rb

@@ -1,14 +0,0 @@
-require_relative './api.rb'
-
-class Notifier
-  def self.notify(code_change_activity)
-    code_change = code_change_activity.code_change
-    id = code_change.id
-    owner = code_change.owner
-    subject = code_change.subject
-
-    message = code_change_activity.message
-    author = code_change_activity.author
-    system("/usr/local/bin/terminal-notifier -title '#{author}' -subtitle '#{owner}: #{subject}' -message '#{message}' -appIcon #{Api.current_api.favicon} -open '#{Api.current_api.code_change_url(code_change)}'")
-  end
-end