codeowners 0.0.3

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 790787fbbbface03dca93e7ee9fb70818d736b55b5b46feba832bb5d1d1cacde
+  data.tar.gz: 5cf3d0d5b9612527ac8e3d5416442dcb0442e1d69180b85df1b25752871e9d9e
+SHA512:
+  metadata.gz: 9728df04569e092a74db19feec631cb0f4031da24ab1bf7e127b6a728206c21026fc7a2988ff0c81f66b894a895350521a9f956236859e682eaa8be2f79194a3
+  data.tar.gz: 594590f13fe02ea9c8e7f5831e2d716a48779e7309f063caeb194c884a42d69037fed8ca64b6703ad88db7765f25258fa58d9d45e12b3eabb00875a016b1fb3a

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+Gemfile.lock
+.rubocop-*
+codeowners.json

data/.rubocop.yml

@@ -0,0 +1 @@
+inherit_from: https://raw.githubusercontent.com/jodosha/dotfiles/master/rubocop.yml

data/CHANGELOG.md

@@ -0,0 +1,17 @@
+# Codeowners
+Simple CLI to interact with GitHub CODEOWNERS
+
+## v0.0.3 - 2020-07-22
+### Added
+- [Luca Guidi] Added `codeowners guess`
+- [Luca Guidi] Added `codeowners import org`
+
+## v0.0.2 - 2020-06-27
+### Added
+- [Luca Guidi] Added pattern support to `codeowners contributors`
+
+## v0.0.1 - 2020-06-12
+### Added
+- [Luca Guidi] Added `codeowners contributors`
+- [Luca Guidi] Added `codeowners list`
+- [Luca Guidi] Added `codeowners --help`

data/Gemfile

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in codeowners.gemspec
+gemspec
+
+gem "rake", "~> 12.0"

data/README.md

@@ -0,0 +1,75 @@
+# Codeowners
+
+Simple CLI to interact with GitHub CODEOWNERS.
+
+## Installation
+
+Install as:
+
+```shell
+$ gem install codeowners
+```
+
+## Usage
+
+### List
+
+List code owners for a file, if any.
+
+```shell
+$ codeowners list path/to/file
+@company/team-a @company/team-b
+```
+
+### Contributors
+
+List code contributors for a file.
+This is useful to guess who can be a candidate to own a file.
+
+```shell
+$ codeowners contributors path/to/file
+path/to/file
+
+Person One <person.one@company.com> / +106, -0
+Person Two <person.two@company.com> / +12, -2
+```
+
+The command accepts also a pattern to match files in bulk.
+
+```shell
+$ codeowners contributors 'path/to/**/*.rb'
+path/to/**/*.rb
+
+Person One <person.one@company.com> / +243, -438
+Person Three <person.three@company.com> / +104, -56
+Person Two <person.two@company.com> / +12, -2
+```
+
+### Help
+
+For a complete set of options, please run:
+
+```shell
+$ codeowners --help
+$ codeowners COMMAND --help
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To run `codeowners` executable during development:
+
+```shell
+$ bundle exec exe/codeowners contributors path/to/file --base-directory=/path/to/git/repository/to/analyze
+```
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/jodosha/codeowners.
+
+## Copyright
+
+&copy; 2020 - Luca Guidi - https://lucaguidi.com

data/Rakefile

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+task default: :spec

data/bin/console

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "codeowners"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/codeowners.gemspec

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require_relative "lib/codeowners/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "codeowners"
+  spec.version       = Codeowners::VERSION
+  spec.authors       = ["Luca Guidi"]
+  spec.email         = ["me@lucaguidi.com"]
+
+  spec.summary       = "GitHub Codeowners check and guess"
+  spec.description   = "Check GitHub Codeowners and guess which team should be assigned to a file"
+  spec.homepage      = "https://lucaguidi.com"
+  spec.required_ruby_version = Gem::Requirement.new(">= 2.5.0")
+
+  spec.metadata["allowed_push_host"] = "https://rubygems.org"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/jodosha/codeowners"
+  spec.metadata["changelog_uri"] = "https://github.com/jodosha/codeowners/blob/master/CHANGELOG.md"
+
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "dry-cli", "~> 0.6"
+  spec.add_runtime_dependency "excon", "~> 0.75"
+  spec.add_development_dependency "rubocop"
+  spec.add_development_dependency "byebug"
+end

data/exe/codeowners

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "codeowners/cli"
+
+Dry::CLI.new(Codeowners::CLI::Commands).call

data/lib/codeowners.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Codeowners
+  require "codeowners/version"
+  require "codeowners/result"
+  require "codeowners/storage"
+  require "codeowners/list_owners"
+  require "codeowners/list_contributors"
+  require "codeowners/guess"
+  require "codeowners/import/client"
+  require "codeowners/import/organization"
+
+  class Error < StandardError
+  end
+
+  class SystemCallError < Error
+  end
+end

data/lib/codeowners/cli.rb

@@ -0,0 +1,198 @@
+# frozen_string_literal: true
+
+require "codeowners"
+require "dry/cli"
+
+module Codeowners
+  module CLI
+    module Commands
+      extend Dry::CLI::Registry
+
+      class Command < Dry::CLI::Command
+        def initialize(out: $stdout)
+          @out = out
+        end
+
+        private
+
+        attr_reader :out
+      end
+
+      class Version < Command
+        desc "Print version"
+
+        def call(*)
+          out.puts "v#{Codeowners::VERSION}"
+        end
+      end
+
+      class List < Command
+        DEFAULT_BASE_DIRECTORY = Dir.pwd.dup.freeze
+        private_constant :DEFAULT_BASE_DIRECTORY
+
+        DEFAULT_CODEOWNERS_PATH = ::File.join(".github", "CODEOWNERS").freeze
+        private_constant :DEFAULT_CODEOWNERS_PATH
+
+        desc "List code owners for a file, if any"
+
+        argument :file, required: true, desc: "File to check"
+
+        option :base_directory, type: :string, default: DEFAULT_BASE_DIRECTORY,  desc: "Base directory"
+        option :codeowners,     type: :string, default: DEFAULT_CODEOWNERS_PATH, desc: "Path to CODEOWNERS file"
+
+        def call(file:, base_directory:, codeowners:, **)
+          result = Codeowners::ListOwners.new(base_directory, codeowners).call(file)
+          exit(1) unless result.successful?
+
+          out.puts result.to_s
+        end
+      end
+
+      class Contributors < Command
+        DEFAULT_BASE_DIRECTORY = Dir.pwd.dup.freeze
+        private_constant :DEFAULT_BASE_DIRECTORY
+
+        DEFAULT_CODEOWNERS_PATH = ::File.join(".github", "CODEOWNERS").freeze
+        private_constant :DEFAULT_CODEOWNERS_PATH
+
+        FORMAT_MAPPING = { "string" => "to_s", "csv" => "to_csv" }.freeze
+        private_constant :FORMAT_MAPPING
+
+        FORMAT_VALUES = FORMAT_MAPPING.keys.freeze
+        private_constant :FORMAT_VALUES
+
+        DEFAULT_FORMAT = FORMAT_VALUES.first
+        private_constant :DEFAULT_FORMAT
+
+        DEFAULT_DEBUG = false
+        private_constant :DEFAULT_DEBUG
+
+        desc "List code contributors for a file (or a pattern)"
+
+        argument :file, required: true, desc: "File (or pattern) to check"
+
+        option :base_directory, type: :string, default: DEFAULT_BASE_DIRECTORY, desc: "Base directory"
+        option :format, type: :string, default: DEFAULT_FORMAT, values: FORMAT_VALUES, desc: "Output format"
+        option :debug, type: :boolean, default: DEFAULT_DEBUG, desc: "Print debug information to stdout"
+
+        example [
+          "path/to/file.rb # file",
+          "'path/to/**/*.rb' # pattern"
+        ]
+
+        def call(file:, base_directory:, format:, debug:, **)
+          result = Codeowners::ListContributors.new(base_directory).call(file, debug)
+          exit(1) unless result.successful?
+
+          out.puts output(result, format)
+        end
+
+        private
+
+        def output(result, format)
+          method_name = FORMAT_MAPPING.fetch(format)
+          result.public_send(method_name.to_sym)
+        end
+      end
+
+      class Guess < Command
+        DEFAULT_BASE_DIRECTORY = Dir.pwd.dup.freeze
+        private_constant :DEFAULT_BASE_DIRECTORY
+
+        DEFAULT_CODEOWNERS_PATH = ::File.join(".github", "CODEOWNERS").freeze
+        private_constant :DEFAULT_CODEOWNERS_PATH
+
+        FORMAT_MAPPING = { "string" => "to_s", "csv" => "to_csv" }.freeze
+        private_constant :FORMAT_MAPPING
+
+        FORMAT_VALUES = FORMAT_MAPPING.keys.freeze
+        private_constant :FORMAT_VALUES
+
+        DEFAULT_FORMAT = FORMAT_VALUES.first
+        private_constant :DEFAULT_FORMAT
+
+        DEFAULT_STORAGE_PATH = ::File.join(Dir.pwd, "codeowners.json").freeze
+        private_constant :DEFAULT_STORAGE_PATH
+
+        DEFAULT_DEBUG = false
+        private_constant :DEFAULT_DEBUG
+
+        desc "Check CODEOWNERS for a file (or a pattern), if missing, it tries to guess, by looking at the git commit history"
+
+        argument :file, required: true, desc: "File (or pattern) to check"
+
+        option :base_directory, type: :string, default: DEFAULT_BASE_DIRECTORY, desc: "Base directory"
+        option :codeowners,     type: :string, default: DEFAULT_CODEOWNERS_PATH, desc: "Path to CODEOWNERS file"
+        option :storage, type: :string, default: DEFAULT_STORAGE_PATH, desc: "Storage path (default: #{DEFAULT_STORAGE_PATH})"
+
+        option :format, type: :string, default: DEFAULT_FORMAT, values: FORMAT_VALUES, desc: "Output format"
+        option :debug, type: :boolean, default: DEFAULT_DEBUG, desc: "Print debug information to stdout"
+
+        example [
+          "path/to/file.rb # file",
+          "'path/to/**/*.rb' # pattern"
+        ]
+
+        def call(file:, base_directory:, codeowners:, storage:, format:, debug:, **)
+          owners = Codeowners::ListOwners.new(base_directory, codeowners)
+          contributors = Codeowners::ListContributors.new(base_directory)
+          storage = Codeowners::Storage.new(storage)
+
+          result = Codeowners::Guess.new(owners, contributors, storage, base_directory, out).call(file, debug)
+          # exit(1) unless result.successful?
+
+          out.puts output(result, format)
+        end
+
+        private
+
+        def output(result, format)
+          # method_name = FORMAT_MAPPING.fetch(format)
+          return unless format == "csv"
+
+          result.map do |file, data|
+            "#{file},#{data.fetch(:teams).join('/')},#{data.fetch(:codeowners)}"
+          end.join("\n")
+        end
+      end
+
+      module Import
+        class Org < Command
+          DEFAULT_STORAGE_PATH = ::File.join(Dir.pwd, "codeowners.json").freeze
+          private_constant :DEFAULT_STORAGE_PATH
+
+          DEFAULT_DEBUG = false
+          private_constant :DEFAULT_DEBUG
+
+          desc "Import teams and members for a GitHub organization"
+
+          argument :org, required: true, desc: "GitHub organization login"
+          argument :token, required: true, desc: "GitHub APIv3 token"
+
+          option :storage, type: :string, default: DEFAULT_STORAGE_PATH, desc: "Storage path (default: #{DEFAULT_STORAGE_PATH})"
+          option :debug, type: :boolean, default: DEFAULT_DEBUG, desc: "Print debug information to stdout"
+
+          example [
+            "hanami s3cr374p1t0k3n"
+          ]
+
+          def call(org:, token:, storage:, debug:, **)
+            client = Codeowners::Import::Client.new(token, out)
+            storage = Codeowners::Storage.new(storage)
+
+            Codeowners::Import::Organization.new(client, storage).call(org, debug)
+          end
+        end
+      end
+
+      register "version",      Version, aliases: ["v", "-v", "--version"]
+      register "list",         List
+      register "contributors", Contributors
+      register "guess",        Guess
+
+      register "import" do |prefix|
+        prefix.register "org", Import::Org
+      end
+    end
+  end
+end

data/lib/codeowners/git.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require "pathname"
+require "shellwords"
+
+module Codeowners
+  class Git
+    def initialize(base_directory)
+      @base_directory = Pathname.new(::File.expand_path(base_directory))
+    end
+
+    def contributors(file, debug = false)
+      require "codeowners/git/contributors"
+      output = git(["log", "--numstat", %(--pretty=format:"author:%aN email:%ae"), "--no-color", "--", escape(file)])
+      print_debug(output, debug)
+
+      Contributors.call(file, output)
+    end
+
+    private
+
+    def git(command_and_args)
+      execute(["git", "--git-dir=#{git_directory}", "--work-tree=#{work_tree}", "-c", "'color.ui=false'"] + command_and_args)
+    end
+
+    def work_tree
+      escape(@base_directory.to_s)
+    end
+
+    def git_directory
+      escape(@base_directory.join(".git").to_s)
+    end
+
+    def escape(string)
+      Shellwords.shellescape(string)
+    end
+
+    def execute(command, env: {}, error: ->(err) { raise Codeowners::SystemCallError.new(err) })
+      require "open3"
+
+      Open3.popen3(env, command.join(" ")) do |_, stdout, stderr, wait_thr|
+        error.call(stderr.read) unless wait_thr.value.success?
+        return stdout.read
+      end
+    end
+
+    def print_debug(output, debug)
+      return unless debug
+
+      puts output
+      puts "\n" * 10
+    end
+  end
+end

data/lib/codeowners/git/contributor.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Codeowners
+  class Git
+    class Contributor
+      attr_reader :email, :name, :file, :insertions, :deletions
+
+      def initialize(email, name, file, insertions, deletions)
+        @email = email
+        @name = name
+        @file = file
+        @insertions = insertions
+        @deletions = deletions
+
+        freeze
+      end
+
+      def to_s
+        "#{name} <#{email}> / +#{insertions}, -#{deletions}"
+      end
+
+      def to_csv
+        "#{name}, #{email}, #{insertions}, #{deletions}"
+      end
+    end
+  end
+end

data/lib/codeowners/git/contributors.rb

@@ -0,0 +1,89 @@
+# frozen_string_literal: true
+
+require "English"
+require "codeowners/git/contributor"
+
+module Codeowners
+  class Git
+    class Contributors
+      # rubocop:disable Metrics/AbcSize
+      # rubocop:disable Metrics/MethodLength
+      def self.call(file, output)
+        lines = output.split($INPUT_RECORD_SEPARATOR)
+
+        result = {}
+        each_commit(lines) do |authors, insertions, deletions|
+          authors.each do |author|
+            author_email = author.fetch("email")
+            author_name  = author.fetch("name")
+
+            result[author_email] ||= {}
+            result[author_email]["name"] = author_name
+            result[author_email]["file"] = file
+            result[author_email]["insertions"] ||= 0
+            result[author_email]["deletions"] ||= 0
+            result[author_email]["insertions"] += insertions
+            result[author_email]["deletions"] += deletions
+          end
+        end
+
+        new(result)
+      end
+      # rubocop:enable Metrics/MethodLength
+      # rubocop:enable Metrics/AbcSize
+
+      def self.each_commit(lines)
+        while lines.any?
+          commit = lines.take_while { |line| line != "" }
+          yield parse(commit.dup) unless commit.empty?
+          lines.shift(commit.size + 1)
+        end
+      end
+
+      def self.parse(commit)
+        authors, stats = commit.partition { |line| line.match?(/author:/) }
+
+        [extract_authors(authors), *calculate_stats(stats)]
+      end
+
+      def self.extract_authors(authors)
+        authors.map do |author|
+          {
+            "name" => scan(author, /author:(.*)email:/).chop,
+            "email" => scan(author, /email:(.*)/)
+          }
+        end.uniq
+      end
+
+      def self.calculate_stats(stats)
+        stats.each_with_object([0, 0]) do |stat, result|
+          stat = stat.split(/[[:space:]]+/)
+
+          insertions, deletions, = *stat
+          result[0] += Integer(insertions)
+          result[1] += Integer(deletions)
+        end
+      end
+
+      def self.scan(string, pattern)
+        string.scan(pattern).flatten.first
+      end
+
+      def initialize(data)
+        @contributors = data.map do |email, stats|
+          Contributor.new(email, *stats.values)
+        end
+      end
+
+      def each(&blk)
+        return enum_for(:each) unless block_given?
+
+        @contributors.each(&blk)
+      end
+
+      def empty?
+        @contributors.empty?
+      end
+    end
+  end
+end

data/lib/codeowners/guess.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+module Codeowners
+  class Guess
+    def initialize(owners, contributors, storage, base_directory, out)
+      @owners = owners
+      @contributors = contributors
+      @storage = storage
+      @base_directory = ::File.expand_path(base_directory)
+      @out = out
+    end
+
+    def call(file, debug)
+      result = {}
+
+      Dir.chdir(base_directory) do
+        Dir.glob(file).sort.each do |f|
+          *teams, codeowners = list_code_owners(f, debug)
+          *teams, codeowners = guess_code_owners(f, debug) unless codeowners
+          teams ||= []
+
+          result[f] = { teams: teams, codeowners: codeowners }
+        end
+      end
+
+      result
+    end
+
+    private
+
+    attr_reader :owners
+    attr_reader :contributors
+    attr_reader :storage
+    attr_reader :base_directory
+    attr_reader :out
+
+    def list_code_owners(file, _debug)
+      result = owners.call(file)
+
+      if result.successful?
+        teams = result.to_a.find_all do |team|
+          storage.team_exist?(team) &&
+            !storage.blacklisted_team?(team)
+        end
+
+        return [result, true] if teams.any?
+      end
+
+      [nil, false]
+    end
+
+    def guess_code_owners(file, debug)
+      result = contributors.call(file, debug)
+      return [nil, false] unless result.successful?
+
+      contributors = result.to_a
+      result = contributors.each_with_object({}) do |contributor, memo|
+        teams = storage.teams_for(contributor)
+        teams.each do |team|
+          slug = team.fetch("slug")
+
+          memo[slug] ||= {}
+          memo[slug][:insertions] ||= 0
+          memo[slug][:deletions] ||= 0
+          memo[slug][:insertions] += contributor.insertions
+          memo[slug][:deletions] += contributor.deletions
+        end
+      end
+
+      team = result.sort do |a, b|
+        -a.last.fetch(:insertions) <=> -b.last.fetch(:insertions)
+      end&.first&.first
+
+      [team, false]
+    end
+  end
+end

data/lib/codeowners/import/client.rb

@@ -0,0 +1,136 @@
+# frozen_string_literal: true
+
+require "json"
+require "excon"
+
+module Codeowners
+  module Import
+    class Client
+      BASE_URL = "https://api.github.com"
+      private_constant :BASE_URL
+
+      USER_AGENT = "codeowners v#{Codeowners::VERSION}"
+      private_constant :USER_AGENT
+
+      def initialize(token, out, base_url = BASE_URL, user_agent = USER_AGENT, client = Excon, sleep_time: 3)
+        @base_url = base_url
+        @user_agent = user_agent
+        @token = token
+        @client = client
+        @out = out
+        @sleep_time = sleep_time
+      end
+
+      def org(login, debug = false)
+        result = get("/orgs/#{login}", debug: debug)
+
+        {
+          id: result.fetch("id"),
+          login: result.fetch("login")
+        }
+      end
+
+      def org_members(org, debug = false)
+        result = get_paginated("/orgs/#{org.fetch(:login)}/members", debug: debug)
+        result.map do |user|
+          {
+            id: user.fetch("id"),
+            login: user.fetch("login")
+          }
+        end
+      end
+
+      def teams(org, debug = false)
+        result = get_paginated("/orgs/#{org.fetch(:login)}/teams", debug: debug)
+        result.map do |team|
+          {
+            id: team.fetch("id"),
+            org_id: org.fetch(:id),
+            name: team.fetch("name"),
+            slug: team.fetch("slug")
+          }
+        end
+      end
+
+      def team_members(org, teams, debug = false)
+        teams.each_with_object([]) do |team, memo|
+          result = get_paginated("/orgs/#{org.fetch(:login)}/teams/#{team.fetch(:slug)}/members", debug: debug)
+          result.each do |member|
+            team_id = team.fetch(:id)
+            user_id = member.fetch("id")
+
+            memo << {
+              id: [team_id, user_id],
+              team_id: team_id,
+              user_id: user_id
+            }
+          end
+
+          sleep_for_a_while
+        end
+      end
+
+      def users(users, debug)
+        users.each do |user|
+          remote_user = get("/users/#{user.fetch(:login)}", debug: debug)
+          user.merge!(
+            name: remote_user.fetch("name"),
+            email: remote_user.fetch("email")
+          )
+
+          sleep_for_a_while
+        end
+      end
+
+      private
+
+      attr_reader :base_url
+      attr_reader :user_agent
+      attr_reader :token
+      attr_reader :client
+      attr_reader :out
+      attr_reader :sleep_time
+
+      def get(path, debug: false)
+        out.puts "requesting GET #{path}" if debug
+
+        response = client.get(base_url + path, query: query, headers: headers)
+        return {} unless response.status == 200
+
+        JSON.parse(response.body)
+      end
+
+      def get_paginated(path, result = [], debug: false, page: 1)
+        out.puts "requesting GET #{path}, page: #{page}" if debug
+
+        response = client.get(base_url + path, query: query(page: page), headers: headers)
+        return [] unless response.status == 200
+
+        parsed = JSON.parse(response.body)
+        result.push(parsed)
+
+        if parsed.any?
+          sleep_for_a_while
+          get_paginated(path, result, debug: debug, page: page + 1)
+        else
+          result.flatten
+        end
+      end
+
+      def query(options = {})
+        { page: 1, per_page: 100 }.merge(options)
+      end
+
+      def headers
+        {
+          "Authorization" => "token #{token}",
+          "User-Agent" => user_agent
+        }
+      end
+
+      def sleep_for_a_while
+        sleep(sleep_time)
+      end
+    end
+  end
+end

data/lib/codeowners/import/organization.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Codeowners
+  module Import
+    class Organization
+      def initialize(client, storage)
+        @client = client
+        @storage = storage
+      end
+
+      def call(org, debug)
+        org = client.org(org, debug)
+        users = client.org_members(org, debug)
+        users = client.users(users, debug)
+        teams = client.teams(org, debug)
+        memberships = client.team_members(org, teams, debug)
+
+        storage.transaction do |db|
+          db[:orgs].upsert(org)
+          db[:users].upsert(users)
+          db[:teams].upsert(teams)
+          db[:memberships].upsert(memberships)
+        end
+      end
+
+      private
+
+      attr_reader :client
+      attr_reader :storage
+    end
+  end
+end

data/lib/codeowners/list_contributors.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require "codeowners/git"
+
+module Codeowners
+  class ListContributors
+    class Result < ::Codeowners::Result
+      def initialize(file = nil, contributors = [])
+        @file = file
+        @contributors = contributors
+      end
+
+      def successful?
+        !@file.nil?
+      end
+
+      def to_s
+        [@file, "", *@contributors.map(&:to_s)].join("\n")
+      end
+
+      def to_a
+        @contributors.dup
+      end
+
+      def to_csv
+        @contributors.map(&:to_csv).join("\n")
+      end
+    end
+
+    def initialize(base_directory, git: Git.new(base_directory))
+      @git = git
+    end
+
+    def call(file, debug = false)
+      contributors = @git.contributors(file, debug)
+      return Result.new if contributors.empty?
+
+      contributors = contributors.each.lazy.sort_by { |c| -c.insertions }
+
+      Result.new(file, contributors)
+    end
+  end
+end

data/lib/codeowners/list_owners.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require "pathname"
+
+module Codeowners
+  class ListOwners
+    def initialize(base_directory, codeowners)
+      @base_directory = Pathname.new(::File.expand_path(base_directory))
+      @codeowners = @base_directory.join(codeowners)
+    end
+
+    def call(file)
+      ::File.open(@codeowners, "r").each_line do |line|
+        line = line.chomp
+        next if line.empty? || line.match?(/[[:space:]]*#/)
+
+        pattern, *owners = line.split(/[[:space:]]+/)
+
+        return Result.new(pattern, owners) if File.fnmatch(pattern, file)
+      end
+
+      Result.new
+    end
+  end
+end

data/lib/codeowners/result.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Codeowners
+  class Result
+    attr_reader :owners
+
+    def initialize(pattern = nil, owners = [])
+      @pattern = pattern
+      @owners = owners
+    end
+
+    def successful?
+      !@pattern.nil?
+    end
+
+    def to_s
+      "#{@pattern}\n\n#{@owners.join('\n')}"
+    end
+
+    def to_a
+      @owners.dup.flatten
+    end
+  end
+end

data/lib/codeowners/storage.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+require "json"
+require "tempfile"
+require "fileutils"
+
+module Codeowners
+  class Storage
+    require "codeowners/storage/data"
+
+    def initialize(path)
+      @path = path
+      @data = Data.new(load_data)
+      @mutex = Mutex.new
+    end
+
+    def transaction
+      @mutex.synchronize do
+        tmp_data = data.dup
+        yield tmp_data
+
+        Tempfile.open("codeowners-storage") do |tmp|
+          tmp.binmode
+          tmp.write(JSON.generate(data.dump))
+          tmp.close
+
+          FileUtils.mv(tmp.path, path, force: true)
+        end
+      end
+    end
+
+    def team_exist?(team)
+      data[:teams].find do |record|
+        record.fetch("slug") == team
+      end
+    end
+
+    def blacklisted_team?(team)
+      data[:teams].find do |record|
+        record.fetch("slug") == team &&
+          record.fetch("blacklisted")
+      end
+    end
+
+    def teams_for(user)
+      found = data[:users].find do |record|
+        record.fetch("email") == user.email ||
+          record.fetch("name") == user.name
+      end
+
+      return [] unless found
+
+      memberships = data[:memberships].find_all do |record|
+        record.fetch("user_id") == found.fetch("id")
+      end
+      memberships.map! { |hash| hash.fetch("team_id") }
+
+      return [] if memberships.empty?
+
+      teams = data[:teams].find_all do |record|
+        !record.fetch("blacklisted") &&
+          memberships.include?(record.fetch("id"))
+      end.flatten
+
+      teams
+    end
+
+    private
+
+    attr_reader :path
+    attr_reader :data
+
+    def load_data
+      return {} unless File.exist?(path)
+
+      JSON.parse(File.read(path))
+    end
+  end
+end

data/lib/codeowners/storage/collection.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Codeowners
+  class Storage
+    class Collection
+      def initialize(collection)
+        @collection = collection.each_with_object({}) do |record, memo|
+          memo[record.fetch("id")] = record
+        end
+      end
+
+      def find(&blk)
+        collection.values.find(&blk)
+      end
+
+      def find_all(&blk)
+        collection.values.find_all(&blk)
+      end
+
+      def upsert(*records)
+        records = Array(records).flatten
+
+        records.each do |record|
+          collection[record.fetch(:id)] = record
+        end
+      end
+
+      def dump
+        collection.values.dup
+      end
+
+      private
+
+      attr_reader :collection
+    end
+  end
+end

data/lib/codeowners/storage/data.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require "codeowners/storage/collection"
+
+module Codeowners
+  class Storage
+    class Data
+      COLLECTIONS = %w[orgs users teams memberships].freeze
+      private_constant :COLLECTIONS
+
+      def initialize(data, collections: COLLECTIONS)
+        @data = collections.each_with_object({}) do |name, memo|
+          memo[name] = Collection.new(data.fetch(name, []))
+        end
+      end
+
+      def [](name)
+        data.fetch(name.to_s)
+      end
+
+      def dump
+        data.transform_values(&:dump)
+      end
+
+      private
+
+      attr_reader :data
+    end
+  end
+end

data/lib/codeowners/version.rb

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

metadata

@@ -0,0 +1,129 @@
+--- !ruby/object:Gem::Specification
+name: codeowners
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+platform: ruby
+authors:
+- Luca Guidi
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2020-07-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: dry-cli
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: excon
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.75'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.75'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Check GitHub Codeowners and guess which team should be assigned to a
+  file
+email:
+- me@lucaguidi.com
+executables:
+- codeowners
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rubocop.yml"
+- CHANGELOG.md
+- Gemfile
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- codeowners.gemspec
+- exe/codeowners
+- lib/codeowners.rb
+- lib/codeowners/cli.rb
+- lib/codeowners/git.rb
+- lib/codeowners/git/contributor.rb
+- lib/codeowners/git/contributors.rb
+- lib/codeowners/guess.rb
+- lib/codeowners/import/client.rb
+- lib/codeowners/import/organization.rb
+- lib/codeowners/list_contributors.rb
+- lib/codeowners/list_owners.rb
+- lib/codeowners/result.rb
+- lib/codeowners/storage.rb
+- lib/codeowners/storage/collection.rb
+- lib/codeowners/storage/data.rb
+- lib/codeowners/version.rb
+homepage: https://lucaguidi.com
+licenses: []
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://lucaguidi.com
+  source_code_uri: https://github.com/jodosha/codeowners
+  changelog_uri: https://github.com/jodosha/codeowners/blob/master/CHANGELOG.md
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.5.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.3
+signing_key: 
+specification_version: 4
+summary: GitHub Codeowners check and guess
+test_files: []