github_project.rb in licensee-9.17.0

- old
+ new

@@ -10,26 +10,28 @@
 autoload :Octokit, 'octokit'
 
 module Licensee
   module Projects
     class GitHubProject < Licensee::Projects::Project
+      attr_reader :ref, :repo
+
       # If there's any trailing data (e.g. `.git`) this pattern will ignore it:
       # we're going to use the API rather than clone the repo.
       GITHUB_REPO_PATTERN =
         %r{https://github.com/([^/]+/([^/]+(?=\.git)|[^/]+)).*}
 
       class RepoNotFound < StandardError; end
 
-      def initialize(github_url, **args)
+      def initialize(github_url, ref: nil, **args)
         @repo = github_url[GITHUB_REPO_PATTERN, 1]
         raise ArgumentError, "Not a github URL: #{github_url}" unless @repo
 
+        @ref = ref
+
         super(**args)
       end
 
-      attr_reader :repo
-
       private
 
       def files
         return @files if defined? @files_from_tree
 
@@ -38,17 +40,24 @@
 
         msg = "Could not load GitHub repo #{repo}, it may be private or deleted"
         raise RepoNotFound, msg
       end
 
+      def query_params
+        return { ref: @ref } unless @ref.nil?
+
+        {}
+      end
+
       def load_file(file)
         client.contents(@repo, path:   file[:path],
-                               accept: 'application/vnd.github.v3.raw').to_s
+                               accept: 'application/vnd.github.v3.raw',
+                               query:  query_params).to_s
       end
 
       def dir_files(path = nil)
         path = path.gsub('./', '') if path
-        files = client.contents(@repo, path: path)
+        files = client.contents(@repo, path: path, query: query_params)
         files = files.select { |data| data[:type] == 'file' }
         files.each { |data| data[:dir] = File.dirname(data[:path]) }
         files.map(&:to_h)
       rescue Octokit::NotFound
         []