octostat 0.3.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9fb6c2b54bc742cfecb4c07dd77e511fcd288b49cb00155a9ccd8d0054bef48a
-  data.tar.gz: e85a9d972c1aaa65d912ab4bbf3ba8e88df494e75005bd3e4262e13a79c07600
+  metadata.gz: 5233f9bbf3702251a2914d07093caab07a8aeaf5461f970027237c6c6ea0f301
+  data.tar.gz: d9171098a6fdc823d91bb7254028f5092ca1c197d14002c69369551554d7c4a1
 SHA512:
-  metadata.gz: b6dd97af0de58bf4239d47533665be8104f7b6d88e6f15fee41c7bbf64d073bea145badb5edb4af64d6bfafe10e3635c45e6357c32897169c6038ff41c603036
-  data.tar.gz: 1b5779debb9f9a3fc5999c3731a15dca9d4e5af93ea611d67a2fc84f4447d58620b3af46ca1dff727d9e443ba53eeb1643b7ecd136bf2562dd18d58b893d4ff0
+  metadata.gz: 994ca4fe6a5531b57f22144dfb31eac32be8a05c4fc0b3b14a5ede3de5e15d8e9ab26bb75af6192cf590bc05e9d338a1f70ff295381b6439830344ba14b73a54
+  data.tar.gz: 5998dcefbd49b274acd2035e8214ecf1e21fa5b5fd05d8ee24d924fe9be7eb01be05bcfcd8703cbe8d3d940de12a01a5e3d3fffcd0ae3a80dce845ae613e9987

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 ## [Unreleased]
 
+## [0.4.0] 2025-03-15
+
+- Display progress
+- Automatically clone repo to tmp dir if repo is not a local directory
+
 ## [0.3.1] 2025-03-14
 
 - Fix changelog

data/lib/octostat/command.rb

@@ -6,18 +6,28 @@ module Octostat
       @db_path = "octostat.sqlite"
       args = parser.parse!(args)
       @path = args.first || Dir.pwd
+      @batch_size = 1000
     end
 
     def call
       db = Database.new(db_path)
-      Git.new(path).each do |commit|
-        db.insert_commit(**commit)
+      @git = Git.new(path)
+      git.each_slice(batch_size).with_index do |commits, batch|
+        puts_progress batch * batch_size
+        commits.each { |commit| db.insert_commit(**commit) }
       end
+      puts_progress git.count
+      puts "\nDone!"
     end
 
     private
 
-    attr_reader :db_path, :path
+    attr_reader :db_path, :path, :batch_size, :git
+
+    def puts_progress processed
+      print "\r#{(processed.to_f / git.count.to_f * 100).ceil}%"
+      $stdout.flush
+    end
 
     def parser
       OptionParser.new do |opts|

data/lib/octostat/git.rb

@@ -1,4 +1,5 @@
 require "open3"
+require "tmpdir"
 
 module Octostat
   class Git
@@ -15,19 +16,25 @@ module Octostat
 
     ENTRY_LENGTH = LOG_FORMAT.lines.size
 
-    COMMAND = ["git", "log", "--pretty=format:#{LOG_FORMAT}"]
+    LIST_COMMAND = ["git", "log", "--pretty=format:#{LOG_FORMAT}"]
+    COUNT_COMMAND = ["git", "rev-list", "--count", "HEAD"]
+    CLONE_COMMAND = ["git", "clone"]
 
     def initialize path
-      @path = path
+      @path = Dir.exist?(path) ? path : clone_repo(path)
     end
 
     def env
       {"GIT_DIR" => path}
     end
 
+    def count
+      @count ||= Open3.capture2(*COUNT_COMMAND, chdir: path).first.to_i
+    end
+
     def each
       return enum_for(:each) unless block_given?
-      Open3.popen2e(*COMMAND, chdir: path) do |input, output, wait_thr|
+      Open3.popen2e(*LIST_COMMAND, chdir: path) do |input, output, wait_thr|
         output.each_slice(ENTRY_LENGTH) do |commit|
           commit.each(&:strip!)
           hash, email, name, date, parents, subject = commit
@@ -46,6 +53,12 @@ module Octostat
 
     private
 
+    def clone_repo upstream
+      repo_path = Dir.mktmpdir
+      Open3.capture2(*CLONE_COMMAND, upstream, repo_path)
+      repo_path
+    end
+
     attr_reader :path
   end
 end

data/lib/octostat/version.rb

@@ -1,3 +1,3 @@
 module Octostat
-  VERSION = "0.3.1"
+  VERSION = "0.4.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: octostat
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
 platform: ruby
 authors:
 - Joé Dupuis