active_storage_quota 0.1.0

data/lib/tasks/active_storage_quota_reconcile.rake

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+
+namespace :active_storage_quota do
+  # The safe name does the safe thing. Applying repairs has its own task rather
+  # than a flag on this one, so a mistyped environment variable can never turn a
+  # plan into a write.
+  desc "Show what reconciliation would repair (dry run, never writes)"
+  task reconcile: :environment do
+    ActiveStorageQuota::RakeSupport.run(dry_run: true)
+  end
+
+  namespace :reconcile do
+    desc "Apply reconciliation repairs (writes)"
+    task apply: :environment do
+      ActiveStorageQuota::RakeSupport.run(dry_run: false)
+    end
+  end
+end
+
+module ActiveStorageQuota
+  module RakeSupport # :nodoc:
+    class << self
+      def run(dry_run:)
+        limits = destructive_limits
+        owners = resolve_owners(dry_run: dry_run)
+
+        owners.each do |owner|
+          result = ActiveStorageQuota.reconcile!(owner: owner, dry_run: dry_run, **limits)
+          puts "#{owner.class.name}##{owner.id}: #{result}"
+        end
+
+        puts "Destructive removal disabled (no MAX_CHARGE_DELETIONS or " \
+             "MAX_REMOVED_FRACTION given)." if limits.empty?
+      end
+
+      private
+        # Refuses to guess. An unscoped apply is the most dangerous operation
+        # this gem offers, so it has to be asked for by name.
+        def resolve_owners(dry_run:)
+          owner_type = ENV["OWNER_TYPE"].presence
+          owner_id = ENV["OWNER_ID"].presence
+
+          if owner_type && owner_id
+            klass = owner_type.safe_constantize
+            abort "OWNER_TYPE #{owner_type.inspect} could not be resolved" if klass.nil?
+
+            return [ klass.find(owner_id) ]
+          end
+
+          if owner_type || owner_id
+            abort "OWNER_TYPE and OWNER_ID must be given together"
+          end
+
+          unless ENV["ALL"] == "1"
+            abort "Refusing to run across every owner. Pass OWNER_TYPE and OWNER_ID, or ALL=1."
+          end
+
+          all_owners
+        end
+
+        def all_owners
+          ActiveStorageQuota::Account
+            .where(scope_name: ActiveStorageQuota::DEFAULT_SCOPE)
+            .pluck(:owner_type, :owner_id)
+            .filter_map do |owner_type, owner_id|
+              klass = owner_type.safe_constantize
+              next warn("skipping unresolvable owner type #{owner_type}") if klass.nil?
+
+              klass.find_by(klass.primary_key => owner_id)
+            end
+        end
+
+        # A malformed value must never become zero or unlimited.
+        def destructive_limits
+          limits = {}
+          limits[:max_charge_deletions] = integer_env("MAX_CHARGE_DELETIONS")
+          limits[:max_removed_fraction] = float_env("MAX_REMOVED_FRACTION")
+          limits.compact
+        end
+
+        def integer_env(name)
+          raw = ENV[name].presence
+          return nil if raw.nil?
+
+          Integer(raw)
+        rescue ArgumentError, TypeError
+          abort "#{name} must be a positive Integer, got #{raw.inspect}"
+        end
+
+        def float_env(name)
+          raw = ENV[name].presence
+          return nil if raw.nil?
+
+          Float(raw)
+        rescue ArgumentError, TypeError
+          abort "#{name} must be a number greater than 0 and at most 1, got #{raw.inspect}"
+        end
+    end
+  end
+end

metadata

@@ -0,0 +1,150 @@
+--- !ruby/object:Gem::Specification
+name: active_storage_quota
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Mubasher Khan
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2026-09-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
+- !ruby/object:Gem::Dependency
+  name: activestorage
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
+description: |
+  Enforce per-owner storage quotas in Rails applications. Provides an atomic
+  reservation system so concurrent uploads across multiple processes cannot
+  overshoot a quota, with usage accounted in O(1) rather than by summing blob
+  sizes on every request.
+email:
+- mubasherkhan32@outlook.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- app/controllers/active_storage_quota/direct_uploads_controller.rb
+- db/migrate/20260917000001_create_active_storage_quota_tables.rb
+- db/migrate/20260918000001_create_active_storage_quota_charges.rb
+- lib/active_storage_quota.rb
+- lib/active_storage_quota/account.rb
+- lib/active_storage_quota/attachable.rb
+- lib/active_storage_quota/attachment_accounting.rb
+- lib/active_storage_quota/audit.rb
+- lib/active_storage_quota/audit/counter_audit.rb
+- lib/active_storage_quota/audit/finding.rb
+- lib/active_storage_quota/audit/ledger_audit.rb
+- lib/active_storage_quota/audit/result.rb
+- lib/active_storage_quota/backfill.rb
+- lib/active_storage_quota/backfill/cursor.rb
+- lib/active_storage_quota/backfill/result.rb
+- lib/active_storage_quota/blob_accounting.rb
+- lib/active_storage_quota/charge.rb
+- lib/active_storage_quota/configuration.rb
+- lib/active_storage_quota/definition.rb
+- lib/active_storage_quota/engine.rb
+- lib/active_storage_quota/errors.rb
+- lib/active_storage_quota/owner.rb
+- lib/active_storage_quota/owner_resolver.rb
+- lib/active_storage_quota/owner_source.rb
+- lib/active_storage_quota/reconciliation.rb
+- lib/active_storage_quota/reconciliation/budget.rb
+- lib/active_storage_quota/reconciliation/counter_reconciler.rb
+- lib/active_storage_quota/reconciliation/ledger_reconciler.rb
+- lib/active_storage_quota/reconciliation/result.rb
+- lib/active_storage_quota/record.rb
+- lib/active_storage_quota/reservation.rb
+- lib/active_storage_quota/version.rb
+- lib/tasks/active_storage_quota.rake
+- lib/tasks/active_storage_quota_audit.rake
+- lib/tasks/active_storage_quota_backfill.rake
+- lib/tasks/active_storage_quota_reconcile.rake
+homepage: https://github.com/mubasher-khan/active_storage_quota
+licenses:
+- MIT
+metadata:
+  source_code_uri: https://github.com/mubasher-khan/active_storage_quota
+  changelog_uri: https://github.com/mubasher-khan/active_storage_quota/blob/main/CHANGELOG.md
+  bug_tracker_uri: https://github.com/mubasher-khan/active_storage_quota/issues
+  rubygems_mfa_required: 'true'
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.1'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.11
+signing_key:
+specification_version: 4
+summary: Race-safe storage quotas for Rails Active Storage.
+test_files: []