Allow mappers and validators to be injected by the client (#38)

* Allow mappers and validators to be injected into CodeOwnership

* Bump version

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    code_ownership (1.31.1)
+    code_ownership (1.32.0)
       code_teams (~> 1.0)
       packs
       sorbet-runtime
@@ -10,7 +10,7 @@ GEM
   remote: https://rubygems.org/
   specs:
     ast (2.4.2)
-    code_teams (1.0.0)
+    code_teams (1.0.1)
       sorbet-runtime
     coderay (1.1.3)
     diff-lcs (1.4.4)

README.md

@@ -57,6 +57,17 @@ js_package_paths:
 
 This defaults `**/`, which makes it look for `package.json` files across your application.
 
+### Custom Ownership
+To enable custom ownership, you can inject your own custom classes into `code_ownership`.
+To do this, first create a class that adheres to the `CodeOwnership::Mapper` and/or `CodeOwnership::Validator` interface.
+Then, in `config/code_ownership.yml`, you can require that file:
+```yml
+require:
+  - ./lib/my_extension.rb
+```
+
+Now, `bin/codeownership validate` will automatically include your new mapper and/or validator. See [`spec/lib/code_ownership/private/extension_loader_spec.rb](spec/lib/code_ownership/private/extension_loader_spec.rb) for an example of what this looks like.
+
 ## Usage: Reading CodeOwnership
 ### `for_file`
 `CodeOwnership.for_file`, given a relative path to a file returns a `CodeTeams::Team` if there is a team that owns the file, `nil` otherwise.

code_ownership.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "code_ownership"
-  spec.version       = '1.31.1'
+  spec.version       = '1.32.0'
   spec.authors       = ['Gusto Engineers']
   spec.email         = ['[email protected]']
   spec.summary       = 'A gem to help engineering teams declare ownership of code'

lib/code_ownership.rb

@@ -7,8 +7,11 @@
 require 'sorbet-runtime'
 require 'json'
 require 'packs'
-require 'code_ownership/cli'
+require 'code_ownership/mapper'
+require 'code_ownership/validator'
 require 'code_ownership/private'
+require 'code_ownership/cli'
+require 'code_ownership/configuration'
 
 module CodeOwnership
   extend self
@@ -27,7 +30,7 @@ def for_file(file)
 
     owner = T.let(nil, T.nilable(CodeTeams::Team))
 
-    Private::OwnershipMappers::Interface.all.each do |mapper|
+    Mapper.all.each do |mapper|
       owner = mapper.map_file_to_owner(file)
       break if owner
     end
@@ -41,7 +44,7 @@ def for_team(team)
     ownership_information = T.let([], T::Array[String])
 
     ownership_information << "# Code Ownership Report for `#{team.name}` Team"
-    Private::OwnershipMappers::Interface.all.each do |mapper|
+    Mapper.all.each do |mapper|
       ownership_information << "## #{mapper.description}"
       codeowners_lines = mapper.codeowners_lines_to_owners
       ownership_for_mapper = []
@@ -172,6 +175,11 @@ def self.bust_caches!
     @for_file = nil
     @memoized_values = nil
     Private.bust_caches!
-    Private::OwnershipMappers::Interface.all.each(&:bust_caches!)
+    Mapper.all.each(&:bust_caches!)
+  end
+
+  sig { returns(Configuration) }
+  def self.configuration
+    Private.configuration
   end
 end

lib/code_ownership/configuration.rb

@@ -0,0 +1,44 @@
+# typed: strict
+
+module CodeOwnership
+  class Configuration < T::Struct
+    extend T::Sig
+    DEFAULT_JS_PACKAGE_PATHS = T.let(['**/'], T::Array[String])
+
+    const :owned_globs, T::Array[String]
+    const :unowned_globs, T::Array[String]
+    const :js_package_paths, T::Array[String]
+    const :unbuilt_gems_path, T.nilable(String)
+    const :skip_codeowners_validation, T::Boolean
+    const :raw_hash, T::Hash[T.untyped, T.untyped]
+
+    sig { returns(Configuration) }
+    def self.fetch
+      config_hash = YAML.load_file('config/code_ownership.yml')
+
+      if config_hash.key?("require")
+        config_hash["require"].each do |require_directive|
+          Private::ExtensionLoader.load(require_directive)
+        end
+      end
+
+      new(
+        owned_globs: config_hash.fetch('owned_globs', []),
+        unowned_globs: config_hash.fetch('unowned_globs', []),
+        js_package_paths: js_package_paths(config_hash),
+        skip_codeowners_validation: config_hash.fetch('skip_codeowners_validation', false),
+        raw_hash: config_hash
+      )
+    end
+
+    sig { params(config_hash: T::Hash[T.untyped, T.untyped]).returns(T::Array[String]) }
+    def self.js_package_paths(config_hash)
+      specified_package_paths = config_hash['js_package_paths']
+      if specified_package_paths.nil?
+        DEFAULT_JS_PACKAGE_PATHS.dup
+      else
+        Array(specified_package_paths)
+      end
+    end
+  end
+end

lib/code_ownership/mapper.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+# typed: strict
+
+module CodeOwnership
+  module Mapper
+    extend T::Sig
+    extend T::Helpers
+
+    interface!
+
+    class << self
+      extend T::Sig
+
+      sig { params(base: Class).void }
+      def included(base)
+        @mappers ||= T.let(@mappers, T.nilable(T::Array[Class]))
+        @mappers ||= []
+        @mappers << base
+      end
+
+      sig { returns(T::Array[Mapper]) }
+      def all
+        T.unsafe(@mappers).map(&:new)
+      end
+    end
+
+    #
+    # This should be fast when run with ONE file
+    #
+    sig do
+      abstract.params(file: String).
+        returns(T.nilable(::CodeTeams::Team))
+    end
+    def map_file_to_owner(file)
+    end
+
+    #
+    # This should be fast when run with MANY files
+    #
+    sig do
+      abstract.params(files: T::Array[String]).
+        returns(T::Hash[String, T.nilable(::CodeTeams::Team)])
+    end
+    def map_files_to_owners(files)
+    end
+
+    sig do
+      abstract.returns(T::Hash[String, T.nilable(::CodeTeams::Team)])
+    end
+    def codeowners_lines_to_owners
+    end
+
+    sig { abstract.returns(String) }
+    def description
+    end
+
+    sig { abstract.void }
+    def bust_caches!
+    end
+  end
+end

lib/code_ownership/private.rb

@@ -2,15 +2,13 @@
 
 # typed: strict
 
-require 'code_ownership/private/configuration'
+require 'code_ownership/private/extension_loader'
 require 'code_ownership/private/team_plugins/ownership'
 require 'code_ownership/private/team_plugins/github'
 require 'code_ownership/private/parse_js_packages'
-require 'code_ownership/private/validations/interface'
 require 'code_ownership/private/validations/files_have_owners'
 require 'code_ownership/private/validations/github_codeowners_up_to_date'
 require 'code_ownership/private/validations/files_have_unique_owners'
-require 'code_ownership/private/ownership_mappers/interface'
 require 'code_ownership/private/ownership_mappers/file_annotations'
 require 'code_ownership/private/ownership_mappers/team_globs'
 require 'code_ownership/private/ownership_mappers/package_ownership'
@@ -21,10 +19,10 @@ module CodeOwnership
   module Private
     extend T::Sig
 
-    sig { returns(Private::Configuration) }
+    sig { returns(Configuration) }
     def self.configuration
-      @configuration ||= T.let(@configuration, T.nilable(Private::Configuration))
-      @configuration ||= Private::Configuration.fetch
+      @configuration ||= T.let(@configuration, T.nilable(Configuration))
+      @configuration ||= Configuration.fetch
     end
 
     sig { void }
@@ -36,7 +34,7 @@ def self.bust_caches!
 
     sig { params(files: T::Array[String], autocorrect: T::Boolean, stage_changes: T::Boolean).void }
     def self.validate!(files:, autocorrect: true, stage_changes: true)
-      errors = Validations::Interface.all.flat_map do |validator|
+      errors = Validator.all.flat_map do |validator|
         validator.validation_errors(
           files: files,
           autocorrect: autocorrect,
@@ -87,7 +85,7 @@ def self.files_by_mapper(files)
       @files_by_mapper ||= begin
         files_by_mapper = files.map { |file| [file, []] }.to_h
 
-        Private::OwnershipMappers::Interface.all.each do |mapper|
+        Mapper.all.each do |mapper|
           mapper.map_files_to_owners(files).each do |file, _team|
             files_by_mapper[file] ||= []
             T.must(files_by_mapper[file]) << mapper.description

lib/code_ownership/private/extension_loader.rb

@@ -0,0 +1,24 @@
+# typed: strict
+# frozen_string_literal: true
+
+module CodeOwnership
+  module Private
+    # This class handles loading extensions to code_ownership using the `require` directive
+    # in the `code_ownership.yml` configuration.
+    module ExtensionLoader
+      class << self
+        extend T::Sig
+        sig { params(require_directive: String).void }
+        def load(require_directive)
+          # We want to transform the require directive to behave differently
+          # if it's a specific local file being required versus a gem
+          if require_directive.start_with?(".")
+            require File.join(Pathname.pwd, require_directive)
+          else
+            require require_directive
+          end
+        end
+      end
+    end
+  end
+end

lib/code_ownership/private/ownership_mappers/file_annotations.rb

@@ -16,7 +16,7 @@ module OwnershipMappers
       #   }
       class FileAnnotations
         extend T::Sig
-        include Interface
+        include Mapper
 
         @@map_files_to_owners = T.let({}, T.nilable(T::Hash[String, T.nilable(::CodeTeams::Team)])) # rubocop:disable Style/ClassVars
 

lib/code_ownership/private/ownership_mappers/js_package_ownership.rb

@@ -7,7 +7,7 @@ module Private
     module OwnershipMappers
       class JsPackageOwnership
         extend T::Sig
-        include Interface
+        include Mapper
 
         @@package_json_cache = T.let({}, T::Hash[String, T.nilable(ParseJsPackages::Package)]) # rubocop:disable Style/ClassVars