dagable 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1105295586a21290d1fefcaec67cc0fb4e4e7dc40e737c6542526d829e5618ee
-  data.tar.gz: b2e61fd515503636536364a7bc3b4694f0dbfa03bedeb5b1d38f1163c24ba074
+  metadata.gz: 6659e16d0c901d24bf885a03eb19e340081420a651084fa7b87c5578dec42052
+  data.tar.gz: e6456fa8ee41246b382a0bbf298c647f5f7286f7e21ed3aab44d918b8073e983
 SHA512:
-  metadata.gz: ee09a550c55af660d7cf30e3ab45203c9a205f6a133525bb40cdf02dc65de9bca6a70ef0506c72f364b8e7bd29880029c16d00ccc004feecb16d2e7bd74087eb
-  data.tar.gz: fe634af62642fb0e0989fcdca6da2b0601be0e5271c225200944020e1c20acaa1d8eed1f6c70d8405c8fcd8e058a33da816890776b99b74c9dbdb67b4b713a86
+  metadata.gz: 3395f73b85288c51e2e04db6fad33d08e8e0de752c37ecb1dfdc9902746b2af1fa5a95251f01e6e8d110b9b3477e399d8b85a9b822432148a23dd8e8ad72c046
+  data.tar.gz: c042d70a98bdd693168e953a4ed964fcd4641d8654656ed237b6ada29cda2bf40102e0e773d179c8866b61e796da040bf3064ce649d7446d2557e7936408ade1

data/lib/dagable/migration_helpers.rb

@@ -34,10 +34,11 @@ module Dagable
 
     def create_dagable_edges_table(source_table_name)
       edges_table = "#{source_table_name}_edges"
+      id_type = source_primary_key_type(source_table_name)
 
       create_table edges_table do |t|
-        t.bigint :parent_id, null: false
-        t.bigint :child_id, null: false
+        t.column :parent_id, id_type, null: false
+        t.column :child_id, id_type, null: false
       end
 
       add_index edges_table, %i[parent_id child_id], unique: true
@@ -47,10 +48,11 @@ module Dagable
 
     def create_dagable_ancestries_table(source_table_name)
       ancestries_table = "#{source_table_name}_ancestries"
+      id_type = source_primary_key_type(source_table_name)
 
       create_table ancestries_table do |t|
-        t.bigint :predecessor_id, null: false
-        t.bigint :successor_id, null: false
+        t.column :predecessor_id, id_type, null: false
+        t.column :successor_id, id_type, null: false
         t.integer :depth, null: false, default: 0
       end
 
@@ -60,5 +62,13 @@ module Dagable
       add_foreign_key ancestries_table, source_table_name, column: :predecessor_id
       add_foreign_key ancestries_table, source_table_name, column: :successor_id
     end
+
+    # Edge and ancestry columns must reference the source table's records, so
+    # their type mirrors whatever the source primary key is (bigint, uuid,
+    # string, ...), read from the live schema.
+    def source_primary_key_type(source_table_name)
+      pk_name = primary_key(source_table_name)
+      columns(source_table_name).find { |column| column.name == pk_name }.sql_type
+    end
   end
 end

data/lib/dagable/migrations/helper.rb

@@ -35,16 +35,16 @@ module Dagable
       end
 
       # Resolves a reference to a dagable record. Accepts either a model
-      # instance or an integer ID.
+      # instance or an ID (integer or UUID/string).
       #
       # @param model [Class] the dagable ActiveRecord model class
-      # @param reference [ActiveRecord::Base, Integer] the record or its ID
+      # @param reference [ActiveRecord::Base, Integer, String] the record or its ID
       # @return [ActiveRecord::Base]
       # @raise [ArgumentError] if the reference type is unsupported
       def retrieve_item(model, reference)
         case reference
         when model then reference
-        when Integer then model.find(reference)
+        when Integer, String then model.find(reference)
         else raise ArgumentError, "Invalid dagable item reference"
         end
       end

data/lib/dagable/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Dagable
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dagable
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Leandro Maduro
@@ -138,7 +138,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 4.0.6
 specification_version: 4
 summary: DAG infrastructure for ActiveRecord models
 test_files: []