captive-stack-detector 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ece3c2210380284f962da264f7e16f96a649ddfdb55c110baf0f1065a6c0d5b9
-  data.tar.gz: 29e1ffa94c58e29b44a2e868195e76593592204cb2449e1e46b19c6fc6069b03
+  metadata.gz: db3c27a80a53adecc0eaacc45336dad897fa31334c9f6da5b36422fc012e8a94
+  data.tar.gz: a7407996c89d2facb523558b4091d07474b8f1c45caa028e33f3e5ffa35a540c
 SHA512:
-  metadata.gz: 976889c3c993bc6548ecdf0bf61074ca7113fa3457243d0c06e75bf74172d136a0b5f6f6d62377cb3a438f50ca9a0bcc14febc79f892c47ec1439a832fffef99
-  data.tar.gz: 214e33cbfeff3724630a54868902d4e10cc9681ec3a8454c89eb15386ad359757a33ef2d505d8e749610e90854bd737406e02be4166ef2346e4d15370d9d0d4e
+  metadata.gz: bf3a1c1030044bbf76c1e905a07a33edf81ec4b540aec91d443da4012242ccde8d79769acc5d69b843401ddc52556c387a6afd8087de95f22361d440971df994
+  data.tar.gz: 33e5ea4a5de2c91e463cb2afcb7885e1518450f698f7caf315f2c38b97b74752e134e3211e9e7ce45e35001763fd464ea878c0dd173ddf7f24cc9833392150e6

data/lib/captive_stack_detector/file_reader.rb

@@ -28,12 +28,17 @@ module CaptiveStackDetector
       full = File.join(@path, filename)
       File.read(full, encoding: "utf-8") if File.exist?(full)
     end
+
+    def list(dir)
+      Dir.glob("#{dir}/**/*", base: @path).select { |entry| File.file?(File.join(@path, entry)) }.sort
+    end
   end
 
   class GithubFileReader
     extend Forwardable
 
     def_delegators :@parser, :ruby_version, :node_version, :env_vars
+    def_delegators :@client, :list
 
     def initialize(token, repo)
       @client = GithubApiClient.new(token, repo)

data/lib/captive_stack_detector/gemfile_analyzer.rb

@@ -17,6 +17,7 @@ module CaptiveStackDetector
     def subtype  = ASSET_GEMS.any? { |g| gem?(g) } ? "app" : "api"
     def database = gem?("pg") ? "postgres" : nil
     def queue    = REDIS_GEMS.any? { |g| gem?(g) } ? "redis" : nil
+    def solid_queue? = gem?("solid_queue")
 
     def worker_command(procfile)
       if procfile

data/lib/captive_stack_detector/github_api_client.rb

@@ -4,6 +4,7 @@ require "net/http"
 require "uri"
 require "base64"
 require "json"
+require_relative "github_tree"
 
 module CaptiveStackDetector
   class GithubApiClient
@@ -13,17 +14,30 @@ module CaptiveStackDetector
     end
 
     def fetch(filename)
-      uri = URI("https://api.github.com/repos/#{@repo}/contents/#{filename}")
-      req = Net::HTTP::Get.new(uri)
-      req["Authorization"] = "Bearer #{@token}"
-      req["Accept"]        = "application/vnd.github+json"
+      body = get("contents/#{filename}")
+      return nil unless body
 
-      res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) { |h| h.request(req) }
-      return nil unless res.is_a?(Net::HTTPSuccess)
-
-      Base64.decode64(JSON.parse(res.body)["content"]).force_encoding("utf-8")
+      Base64.decode64(JSON.parse(body)["content"]).force_encoding("utf-8")
     rescue StandardError
       nil
     end
+
+    def list(dir)
+      body = get("git/trees/HEAD?recursive=1")
+      return [] unless body
+
+      GithubTree.new(body).files_in(dir)
+    rescue StandardError
+      []
+    end
+
+    private
+
+    def get(path)
+      uri = URI("https://api.github.com/repos/#{@repo}/#{path}")
+      req = Net::HTTP::Get.new(uri, "Authorization" => "Bearer #{@token}", "Accept" => "application/vnd.github+json")
+      res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) { |http| http.request(req) }
+      res.body if res.is_a?(Net::HTTPSuccess)
+    end
   end
 end

data/lib/captive_stack_detector/github_tree.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require "json"
+
+module CaptiveStackDetector
+  class GithubTree
+    def initialize(body)
+      @entries = JSON.parse(body)["tree"]
+    end
+
+    def files_in(dir)
+      prefix = "#{dir}/"
+      @entries.filter_map { |entry| entry["path"] if entry["type"] == "blob" && entry["path"].start_with?(prefix) }
+    end
+  end
+end

data/lib/captive_stack_detector/rails_stack_detector.rb

@@ -2,6 +2,7 @@
 
 require_relative "../captive_stack_detector"
 require_relative "system_package_detector"
+require_relative "solid_queue_worker"
 
 module CaptiveStackDetector
   class RailsStackDetector
@@ -27,7 +28,7 @@ module CaptiveStackDetector
     private
 
     def build_worker
-      command = @analyzer.worker_command(@reader.read("Procfile"))
+      command = @analyzer.worker_command(@reader.read("Procfile")) || SolidQueueWorker.new(@reader, @analyzer).command
       command ? Worker.new(command: command) : nil
     end
 

data/lib/captive_stack_detector/recurring_schedule.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require "yaml"
+
+module CaptiveStackDetector
+  class RecurringSchedule
+    def initialize(yml)
+      @yml = yml
+    end
+
+    def task_names
+      environments.values.grep(Hash).flat_map(&:keys)
+    end
+
+    private
+
+    def environments
+      parsed = YAML.safe_load(@yml.to_s)
+      parsed.is_a?(Hash) ? parsed : {}
+    rescue Psych::Exception
+      {}
+    end
+  end
+end

data/lib/captive_stack_detector/solid_queue_usage.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module CaptiveStackDetector
+  class SolidQueueUsage
+    DEFAULT_RECURRING_TASK = "clear_solid_queue_finished_jobs"
+    DEFAULT_JOB_FILE       = "application_job.rb"
+
+    def initialize(recurring_task_names:, job_files:)
+      @recurring_task_names = recurring_task_names
+      @job_files            = job_files
+    end
+
+    def used? = recurring_task? || application_job?
+
+    private
+
+    def recurring_task? = @recurring_task_names.any? { |task| task != DEFAULT_RECURRING_TASK }
+
+    def application_job?
+      @job_files.any? { |file| file.end_with?(".rb") && File.basename(file) != DEFAULT_JOB_FILE }
+    end
+  end
+end

data/lib/captive_stack_detector/solid_queue_worker.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require_relative "solid_queue_usage"
+require_relative "recurring_schedule"
+
+module CaptiveStackDetector
+  class SolidQueueWorker
+    COMMAND = "bin/jobs"
+
+    def initialize(reader, analyzer)
+      @reader   = reader
+      @analyzer = analyzer
+    end
+
+    def command
+      COMMAND if @analyzer.solid_queue? && usage.used?
+    end
+
+    private
+
+    def usage
+      SolidQueueUsage.new(
+        recurring_task_names: RecurringSchedule.new(@reader.read("config/recurring.yml")).task_names,
+        job_files:            @reader.list("app/jobs"),
+      )
+    end
+  end
+end

data/lib/captive_stack_detector/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CaptiveStackDetector
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: captive-stack-detector
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Captive Studio
@@ -26,11 +26,15 @@ files:
 - lib/captive_stack_detector/gem_package_map.rb
 - lib/captive_stack_detector/gemfile_analyzer.rb
 - lib/captive_stack_detector/github_api_client.rb
+- lib/captive_stack_detector/github_tree.rb
 - lib/captive_stack_detector/js_stack_detector.rb
 - lib/captive_stack_detector/node_version_detector.rb
 - lib/captive_stack_detector/package_json_analyzer.rb
 - lib/captive_stack_detector/rails_stack_detector.rb
+- lib/captive_stack_detector/recurring_schedule.rb
 - lib/captive_stack_detector/ruby_version_detector.rb
+- lib/captive_stack_detector/solid_queue_usage.rb
+- lib/captive_stack_detector/solid_queue_worker.rb
 - lib/captive_stack_detector/ssr_framework_detector.rb
 - lib/captive_stack_detector/system_package_detector.rb
 - lib/captive_stack_detector/types.rb