exwiw 0.2.7 → 0.2.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9ad123bcfe1cadde34a031d1db34fb4592892b88765d3b979c2fb827b53578f7
-  data.tar.gz: 59b2263b51e51f95e9f8d79cbc214dc0f385202e7ca120d108f3f64e18832a52
+  metadata.gz: 7e5d922e740407599ecb3fa3992e0de402434bacfb38ca817a189a53acf16ab3
+  data.tar.gz: 2a8919778bb9395434587ebb69f49f4d8445cb1890e0999cb065b5577d802532
 SHA512:
-  metadata.gz: f79b7338d57d93eebbfba42d81ff6a3f3c232dbb9163118e42d4f67b73844c50234910d3add9eafe72ff4f854d5497191ac7cb37a4ec1da1647dfb1987c118be
-  data.tar.gz: 22c97b02f4cb68badb0bc3e404996af4a48f46f0c32f0366cf0845a6e683b5d4a6a1af3ed499dae425d9adafe131f5c86eca933d5f51eac3d3d4458e508f36a8
+  metadata.gz: e38a240087564c3e3909106268fba7ad3a8dce881924f8b210b1733d65f0bc12d3bb514af80be21e6f0ad39707f99e96edbaed3cc7c54e2d26677c3a0c7d203f
+  data.tar.gz: cca54067266034f8df074fce301623fd6c0c860e9fd1a13eac43d930b89f46a7b63d3e6230d155abcce3f6ab021c2299d518f8131d26ee5bcbe7c0a535093ee9

data/CHANGELOG.md

@@ -2,6 +2,24 @@
 
 ## [Unreleased]
 
+## [0.2.9] - 2026-05-31
+
+### Added
+
+- New `--ids-field=FIELD` CLI option matches `--ids` against an arbitrary field on the target collection instead of its primary key (e.g. `--target-collection=users --ids=a@example.com --ids-field=email`). Only the target collection's filter changes — downstream foreign-key propagation still keys off the primary key. Unlike the primary-key path, the supplied ids are **not** type-coerced (a custom field's stored type is unknown, so values are passed through as-is). Currently **mongodb-only**: the SQL adapters (mysql2/postgresql/sqlite3) reject the flag at validation time, and threading `ids_field` through `QueryAstBuilder` for them is left as a TODO.
+- New `--target-collection=COLLECTION` CLI option, a mongodb-only alias of `--target-table`. Specifying both, or using `--target-collection` with a non-mongodb adapter, is rejected at validation time.
+- New rake task `exwiw:schema:generate_mongoid` (backed by `Exwiw::MongoidSchemaGenerator`) generates `MongodbCollectionConfig` files by introspecting Mongoid document models — a separate task/class from the ActiveRecord `schema:generate` because the ORMs expose different metadata. It derives the collection name, the `_id` primary key, `fields` (including referenced `belongs_to` foreign keys), `belongs_tos` from referenced `belongs_to` associations, and `embedded_in` from `embedded_in` / `embeds_many` / `embeds_one` associations (each embedded config names its immediate parent collection and `store_as` document key; nested embedding is emitted as a chain — `comments` embedded_in `posts`, `posts` embedded_in `users` — so the adapter can recurse through both array and Hash subdocuments). Regeneration preserves hand-edited `replace_with` / `filter` / `skip` / `bulk_insert_chunk_size`. Polymorphic `belongs_to` is not yet expanded. Models in an inheritance hierarchy whose subclasses share the base's collection (Mongoid STI, `_type` discriminator) collapse into a single config: subclasses are discovered via `descendants` (Mongoid registers only the base in `Mongoid.models`) and every class's `fields` / `belongs_tos` are unioned, so subclass-only fields and associations are preserved. A referenced `belongs_to` declared on an *embedded* document (e.g. `Comment embedded_in :post, belongs_to :author`) is dropped from the embedded config's `belongs_tos` (cross-collection refs from inside embedded subdocuments are unsupported and rejected on load), while its foreign-key column is still kept as an ordinary field. A `has_and_belongs_to_many` association is likewise dropped from `belongs_tos` (its foreign keys are stored as an array field such as `tag_ids`, which exwiw cannot follow as a single-valued foreign key), while that `*_ids` array column is kept as an ordinary field. A *polymorphic* `embedded_in` (`embedded_in :addressable, polymorphic: true`) has no single embedding parent collection and cannot be expressed as an `embedded_in` config, so the generator raises a clear, actionable error rather than crashing on the unresolvable parent class. A *self-referential / cyclic* embedding (Mongoid's `recursively_embeds_many` / `recursively_embeds_one`) makes a collection both top-level and embedded inside documents of its own type; since exwiw represents a collection as either top-level or embedded (not both), the generator likewise raises a clear error rather than emit an `embedded_in` config that would silently make the collection undumpable. The `created_at` / `updated_at` columns added by `include Mongoid::Timestamps` are tracked as ordinary fields, and their BSON `ObjectId` / `Date` values (the shape a live `find` returns) serialize as MongoDB Extended JSON (`$oid` / `$date`) through the dump path — now covered end-to-end against the generated configs. An aliased field (`field :ctry, as: :country`) is emitted by its **stored** document key (`ctry`), never the Ruby accessor (`country`), so masking and projection target the key that actually appears in the document; the accessor is additionally surfaced as `mongoid_field_name` on that field so the otherwise cryptic short key stays understandable (association aliases such as `shop => shop_id` and the built-in `id => _id` are not field renames and are not annotated).
+
+### Fixed
+
+- MongoDB adapter: `--ids` filtering against an `ObjectId` `_id` now works. `--ids` arrives as text and MongoDB compares types strictly, so a 24-char hex id is coerced to `BSON::ObjectId` (a plain String would never match). Integer-looking ids are still coerced to `Integer` and other strings (e.g. a String/UUID `_id`) are left as-is. This makes the `MongoidSchemaGenerator`-emitted `"primary_key": "_id"` configs usable end-to-end for the common case where `_id` is Mongoid's default `ObjectId`.
+
+## [0.2.8] - 2026-05-31
+
+### Added
+
+- `schema:generate` now supports polymorphic `belongs_to` associations. Each polymorphic relation (`belongs_to :reviewable, polymorphic: true`) is expanded into one `belongs_to` entry per concrete target table — discovered from the other models' `has_many` / `has_one ..., as:` declarations — carrying `foreign_type` (the type column, e.g. `reviewable_type`) and `type_value` (the stored type, e.g. `"Product"`). Targets are ordered by table name so the generated output is stable across Ruby versions. At dump time, a polymorphic `belongs_to` on the path to the dump target is constrained by both the foreign key and the type column — in both the `SELECT` and the `delete-*.sql` bulk-delete subquery — so only rows of the matching type are extracted. ([#43](https://github.com/heyinc/exwiw/pull/43))
+
 ## [0.2.7] - 2026-05-30
 
 ### Added

data/README.md

@@ -148,6 +148,25 @@ Each database keeps its own Rails migration history, so a `schema_migrations` (a
 
 - The rails-managed table *names* are resolved from the global `ActiveRecord::Base.schema_migrations_table_name` / `internal_metadata_table_name` accessors, which are shared across all connections. A per-database override of these names is not detected, so such a table will be missing from that database's generated configs.
 
+#### Mongoid applications
+
+For MongoDB applications backed by [Mongoid](https://www.mongodb.com/docs/mongoid/), a separate rake task introspects Mongoid document models and emits `MongodbCollectionConfig` files (the `fields` / `_id` / `embedded_in` shape described under [MongoDB notes](#mongodb-notes)):
+
+```bash
+bundle exec rake exwiw:schema:generate_mongoid
+```
+
+It is a distinct task and class (`Exwiw::MongoidSchemaGenerator`) from the ActiveRecord generator because the two ORMs expose entirely different metadata. From each model it derives:
+
+- the collection name and the `_id` primary key,
+- `fields` from the declared Mongoid fields (referenced `belongs_to` foreign keys such as `shop_id`, and the `created_at` / `updated_at` columns added by `Mongoid::Timestamps`, are ordinary fields — their BSON `ObjectId` / `Date` values serialize as MongoDB Extended JSON at dump time). For an aliased field (`field :ctry, as: :country`), the generator emits the **stored** document key (`ctry`), never the Ruby accessor (`country`), so masking and projection target the key that actually appears in the document, and additionally records the accessor as `mongoid_field_name` on that field so the short key stays understandable (association aliases such as `shop => shop_id` and the built-in `id => _id` are not field renames and are not annotated),
+- `belongs_tos` from referenced `belongs_to` associations (`{ table_name, foreign_key }`). A referenced `belongs_to` declared on an *embedded* document is dropped (cross-collection refs from inside embedded subdocuments are unsupported — see [MongoDB notes](#mongodb-notes)), but its foreign-key column is still kept as an ordinary field. A `has_and_belongs_to_many` association is also dropped (its foreign keys are stored as an array field, e.g. `tag_ids`, which exwiw cannot follow as a single-valued foreign key), while that `*_ids` array column is kept as an ordinary field,
+- `embedded_in` from `embedded_in` / `embeds_many` / `embeds_one` associations. Each embedded config names its *immediate* parent collection and the document key it lives under (`store_as`, defaulting to the relation name); nested embedding is represented as a chain (`comments` → `embedded_in` `posts`, `posts` → `embedded_in` `users`) rather than a flattened dot-path, matching how the adapter recurses through array and Hash subdocuments. A *polymorphic* `embedded_in` (`embedded_in :addressable, polymorphic: true`) has no single embedding parent collection and so cannot be expressed as an `embedded_in` config; the generator raises a clear error pointing you to define that collection's config by hand. A *self-referential / cyclic* embedding (Mongoid's `recursively_embeds_many` / `recursively_embeds_one`) makes a collection both a top-level document and embedded inside documents of its own type; exwiw represents a collection as either top-level or embedded, not both, so the generator likewise raises a clear error rather than emit a config that would silently make the collection undumpable.
+
+Models in an inheritance hierarchy whose subclasses share the base's collection (Mongoid STI, distinguished by the auto-added `_type` discriminator) collapse into a single config: the generator discovers the subclasses via `descendants` (Mongoid registers only the base class in `Mongoid.models`) and unions every class's `fields` and `belongs_tos` into the collection config, so subclass-only fields and associations are not lost.
+
+Regeneration preserves hand-edited `replace_with`, `filter`, `skip`, and `bulk_insert_chunk_size` values, like the ActiveRecord generator. Indexes are not written to the config — they are introspected from the live database at dump time (see [MongoDB notes](#mongodb-notes)). Polymorphic `belongs_to` is not yet expanded by this task.
+
 ### Configuration
 
 This is an example of the one table schema:
@@ -241,6 +260,47 @@ Constraints:
 - Specifying a skipped table as `--target-table` raises `ArgumentError`.
 - `skip: true` is preserved by `exwiw:schema:generate` regenerations (the receiver value wins over the auto-generated config).
 
+### Polymorphic `belongs_to`
+
+A Rails polymorphic association (`belongs_to :reviewable, polymorphic: true`) does not point at a single table — the target row is selected at runtime by a type column. exwiw models this as **one `belongs_to` entry per concrete target table**, each carrying two extra fields:
+
+- `foreign_type` — the type column on *this* table (e.g. `reviewable_type`).
+- `type_value` — the value stored in that column for this target (e.g. `"Product"`), i.e. the target model's `polymorphic_name`.
+
+```json
+{
+  "name": "reviews",
+  "primary_key": "id",
+  "belongs_tos": [
+    {
+      "table_name": "products",
+      "foreign_key": "reviewable_id",
+      "foreign_type": "reviewable_type",
+      "type_value": "Product"
+    },
+    {
+      "table_name": "shops",
+      "foreign_key": "reviewable_id",
+      "foreign_type": "reviewable_type",
+      "type_value": "Shop"
+    }
+  ],
+  "columns": [{ "name": "id" }, { "name": "reviewable_type" }, { "name": "reviewable_id" }]
+}
+```
+
+`exwiw:schema:generate` expands a polymorphic `belongs_to` automatically: it finds every model that registers the association as a target via `has_many` / `has_one ..., as: :reviewable` and emits one entry per target table (ordered by table name so the output is stable across Ruby versions). A plain (non-polymorphic) `belongs_to` simply omits `foreign_type` / `type_value`.
+
+At dump time, when a polymorphic `belongs_to` lies on the path to the dump target, exwiw constrains **both** the foreign key and the type column, so only rows of the matching type are extracted. For example, dumping `products` pulls only reviews whose `reviewable_type = 'Product'`:
+
+```sql
+SELECT reviews.* FROM reviews
+WHERE reviews.reviewable_id IN (/* products subquery */)
+  AND reviews.reviewable_type = 'Product'
+```
+
+The same type filter is applied on the join path — and in the matching `delete-*.sql` bulk-delete subquery — when the polymorphic table is an intermediate hop rather than the directly-dumped table.
+
 ### Rails-managed tables (special `type` values)
 
 Some tables are owned by Rails itself rather than the application — they have no ActiveRecord model and Rails reserves the right to evolve their column shape between versions (e.g. `schema_migrations`, `ar_internal_metadata`). exwiw treats them as a distinct category via the `type` field on a table config:
@@ -350,6 +410,9 @@ The MongoDB adapter is experimental. To use it:
 - Add `gem "mongo"` to your Gemfile in addition to `exwiw` (it is not declared as a runtime dependency of the gem).
 - Set `--adapter=mongodb`. `--user` / `DATABASE_PASSWORD` are optional and only needed when your MongoDB requires authentication.
 - The MongoDB adapter consumes a separate config type, `MongodbCollectionConfig`, with MongoDB-native naming. Use `fields` (instead of the SQL adapters' `columns`), and set `"primary_key": "_id"`. Foreign keys (`shop_id`, `user_id`, ...) stay as ordinary fields.
+- `--ids` values are coerced to the type actually stored in `_id` before filtering: integer-looking ids become `Integer`, 24-char hex ids become `BSON::ObjectId` (Mongoid's default `_id` type — a plain String would never match an ObjectId), and any other string is left as-is.
+- `--target-collection=COLLECTION` is a mongodb-only alias of `--target-table` (use whichever reads better for MongoDB). Specifying both, or using `--target-collection` with a non-mongodb adapter, is an error.
+- `--ids-field=FIELD` matches `--ids` against `FIELD` on the target collection instead of its primary key (e.g. `--target-collection=users --ids=a@example.com --ids-field=email`). Downstream foreign-key propagation still keys off the primary key, so only the target collection's filter changes. Unlike the primary-key path, the supplied ids are **not** type-coerced (the stored type of a custom field is unknown), so pass values matching the field's actual type. This flag is currently **mongodb-only** (the SQL adapters reject it; supporting them is a TODO).
 - Output is JSON Lines (`insert-{idx}-{collection}.jsonl`) using MongoDB Extended JSON (relaxed mode). Import with `mongoimport`:
   ```bash
   mongoimport --db app_dev --collection users --file dump/insert-002-users.jsonl

data/lib/exwiw/adapter/mongodb_adapter.rb

@@ -47,7 +47,22 @@ module Exwiw
 
         filter =
           if config.name == dump_target.table_name
-            { config.primary_key => { "$in" => coerce_ids(dump_target.ids) } }
+            # `--ids-field` may override which field --ids is matched against;
+            # otherwise fall back to the primary key. Note this only changes the
+            # WHERE filter on the target collection — downstream foreign-key
+            # propagation still keys off `primary_key` (see #execute, which
+            # stashes doc[primary_key] into @state).
+            #
+            # Type coercion is only applied to the primary key (`_id`), whose
+            # stored type we know (Mongoid's default ObjectId). For a custom
+            # `ids_field` the stored type is unknown, so the textual --ids are
+            # left as Strings rather than guessed at — the caller passes values
+            # matching the field's actual type.
+            if dump_target.ids_field
+              { dump_target.ids_field => { "$in" => dump_target.ids } }
+            else
+              { config.primary_key => { "$in" => coerce_ids(dump_target.ids) } }
+            end
           else
             constrained = config.belongs_tos.select do |relation|
               @state.key?(relation.table_name) && !@state[relation.table_name].empty?
@@ -155,18 +170,42 @@ module Exwiw
       end
 
       # `--ids` from the CLI arrives as Strings. Mongo compares types strictly,
-      # so integer-looking ids are coerced to Integer. Other strings (e.g. ObjectId
-      # hex) are left as-is.
+      # so the textual ids must be coerced to the type actually stored in `_id`:
+      #
+      # - integer-looking ids -> Integer
+      # - 24-char hex ids -> BSON::ObjectId (Mongoid's default `_id` type; a
+      #   plain String would never match an ObjectId in a `$in` filter)
+      # - anything else (e.g. a String/UUID `_id`) is left as-is
+      #
+      # Only used for the primary-key filter; a custom `--ids-field` skips this
+      # because its stored type is unknown (see build_query).
       private def coerce_ids(ids)
-        Array(ids).map do |id|
-          if id.is_a?(String) && id.match?(/\A-?\d+\z/)
-            id.to_i
-          else
-            id
-          end
+        Array(ids).map { |id| coerce_id(id) }
+      end
+
+      private def coerce_id(id)
+        return id unless id.is_a?(String)
+
+        if id.match?(/\A-?\d+\z/)
+          id.to_i
+        elsif object_id_hex?(id)
+          BSON::ObjectId.from_string(id)
+        else
+          id
         end
       end
 
+      # True when `str` is a canonical 24-char hex ObjectId. `bson` ships with
+      # `mongo`/`mongoid` but may not be loaded yet when build_query runs before
+      # any db access, so require it lazily; if it is genuinely unavailable we
+      # fall back to leaving the id as a String.
+      private def object_id_hex?(str)
+        require 'bson' unless defined?(::BSON::ObjectId)
+        ::BSON::ObjectId.legal?(str)
+      rescue LoadError
+        false
+      end
+
       private def reject_filter!(config)
         return if config.filter.nil? || config.filter.to_s.empty?
 

data/lib/exwiw/adapter/mysql2_adapter.rb

@@ -136,7 +136,18 @@ module Exwiw
 
         foreign_key = first_join.foreign_key
         subquery_sql = compile_ast(subquery_ast)
-        sql += "\nWHERE #{select_query_ast.from_table_name}.#{foreign_key} IN (#{subquery_sql});"
+        sql += "\nWHERE #{select_query_ast.from_table_name}.#{foreign_key} IN (#{subquery_sql})"
+
+        # first_join.base_where_clauses は外側の削除対象テーブル
+        # (from_table_name) 上の条件 (polymorphic 型カラム等)。subquery には
+        # 含まれないため、外側の WHERE に追加する。これにより、別の
+        # polymorphic 型に属する行まで削除してしまうのを防ぐ。
+        first_join.base_where_clauses.each do |where|
+          next unless where.is_a?(Exwiw::QueryAst::WhereClause)
+
+          sql += " AND #{compile_where_condition(where, select_query_ast.from_table_name)}"
+        end
+        sql += ";"
 
         sql
       end
@@ -159,6 +170,13 @@ module Exwiw
             compiled_where_condition = compile_where_condition(where, join.join_table_name)
             sql += " AND #{compiled_where_condition}"
           end
+
+          # base_where_clauses は結合元テーブル (base_table_name) に対して
+          # コンパイルする。polymorphic な結合元テーブルの型カラム絞り込み等。
+          join.base_where_clauses.each do |where|
+            compiled_where_condition = compile_where_condition(where, join.base_table_name)
+            sql += " AND #{compiled_where_condition}"
+          end
         end
 
         if query_ast.where_clauses.any?

data/lib/exwiw/adapter/postgresql_adapter.rb

@@ -178,7 +178,18 @@ module Exwiw
 
         foreign_key = first_join.foreign_key
         subquery_sql = compile_ast(subquery_ast)
-        sql += "\nWHERE #{select_query_ast.from_table_name}.#{foreign_key} IN (#{subquery_sql});"
+        sql += "\nWHERE #{select_query_ast.from_table_name}.#{foreign_key} IN (#{subquery_sql})"
+
+        # first_join.base_where_clauses は外側の削除対象テーブル
+        # (from_table_name) 上の条件 (polymorphic 型カラム等)。subquery には
+        # 含まれないため、外側の WHERE に追加する。これにより、別の
+        # polymorphic 型に属する行まで削除してしまうのを防ぐ。
+        first_join.base_where_clauses.each do |where|
+          next unless where.is_a?(Exwiw::QueryAst::WhereClause)
+
+          sql += " AND #{compile_where_condition(where, select_query_ast.from_table_name)}"
+        end
+        sql += ";"
 
         sql
       end
@@ -201,6 +212,13 @@ module Exwiw
             compiled_where_condition = compile_where_condition(where, join.join_table_name)
             sql += " AND #{compiled_where_condition}"
           end
+
+          # base_where_clauses は結合元テーブル (base_table_name) に対して
+          # コンパイルする。polymorphic な結合元テーブルの型カラム絞り込み等。
+          join.base_where_clauses.each do |where|
+            compiled_where_condition = compile_where_condition(where, join.base_table_name)
+            sql += " AND #{compiled_where_condition}"
+          end
         end
 
         if query_ast.where_clauses.any?

data/lib/exwiw/adapter/sqlite3_adapter.rb

@@ -123,7 +123,18 @@ module Exwiw
 
         foreign_key = first_join.foreign_key
         subquery_sql = compile_ast(subquery_ast)
-        sql += "\nWHERE #{select_query_ast.from_table_name}.#{foreign_key} IN (#{subquery_sql});"
+        sql += "\nWHERE #{select_query_ast.from_table_name}.#{foreign_key} IN (#{subquery_sql})"
+
+        # first_join.base_where_clauses は外側の削除対象テーブル
+        # (from_table_name) 上の条件 (polymorphic 型カラム等)。subquery には
+        # 含まれないため、外側の WHERE に追加する。これにより、別の
+        # polymorphic 型に属する行まで削除してしまうのを防ぐ。
+        first_join.base_where_clauses.each do |where|
+          next unless where.is_a?(Exwiw::QueryAst::WhereClause)
+
+          sql += " AND #{compile_where_condition(where, select_query_ast.from_table_name)}"
+        end
+        sql += ";"
 
         sql
       end
@@ -146,6 +157,13 @@ module Exwiw
             compiled_where_condition = compile_where_condition(where, join.join_table_name)
             sql += " AND #{compiled_where_condition}"
           end
+
+          # base_where_clauses は結合元テーブル (base_table_name) に対して
+          # コンパイルする。polymorphic な結合元テーブルの型カラム絞り込み等。
+          join.base_where_clauses.each do |where|
+            compiled_where_condition = compile_where_condition(where, join.base_table_name)
+            sql += " AND #{compiled_where_condition}"
+          end
         end
 
         if query_ast.where_clauses.any?

data/lib/exwiw/belongs_to.rb

@@ -6,9 +6,22 @@ module Exwiw
 
     attribute :foreign_key, String
     attribute :table_name, String
+    # polymorphic 関連の場合のみ設定される。`foreign_type` は型を格納するカラム名
+    # (例: `reviewable_type`)、`type_value` はそのカラムに入る値 (例: `"Product"`)。
+    # 非 polymorphic の belongs_to では両方とも nil。
+    attribute :foreign_type, optional(String), skip_serializing_if_nil: true
+    attribute :type_value, optional(String), skip_serializing_if_nil: true
 
     def self.from_symbol_keys(hash)
       from(hash.transform_keys(&:to_s))
     end
+
+    def polymorphic?
+      !foreign_type.nil?
+    end
+
+    def to_hash
+      super.compact
+    end
   end
 end

data/lib/exwiw/cli.rb

@@ -37,7 +37,9 @@ module Exwiw
       @database_adapter = nil
       @database_name = nil
       @target_table_name = nil
+      @target_collection_name = nil
       @ids = []
+      @ids_field = nil
       @output_format = nil
       @insert_only = nil
       @after_insert_hook_path = nil
@@ -66,6 +68,7 @@ module Exwiw
       dump_target = DumpTarget.new(
         table_name: @target_table_name,
         ids: @ids,
+        ids_field: @ids_field,
       )
 
       logger = build_logger
@@ -95,6 +98,8 @@ module Exwiw
     end
 
     private def validate_options!
+      resolve_target_collection_alias!
+
       if @subcommand == "explain"
         validate_explain_only!
       end
@@ -167,6 +172,23 @@ module Exwiw
         exit 1
       end
 
+      if @ids_field
+        # --ids-field overrides the field --ids filters against on the target
+        # table; it is meaningless without a target table to constrain.
+        if !@target_table_name
+          $stderr.puts "--target-table is required when --ids-field is specified"
+          exit 1
+        end
+
+        # TODO: support --ids-field for the sql adapters (mysql2/postgresql/
+        # sqlite3) by threading dump_target.ids_field through QueryAstBuilder's
+        # WHERE clause on the target table. For now it is mongodb-only.
+        if @database_adapter != "mongodb"
+          $stderr.puts "--ids-field is currently only supported by the mongodb adapter"
+          exit 1
+        end
+      end
+
       if @after_insert_hook_path
         unless File.file?(@after_insert_hook_path)
           $stderr.puts "--after-insert-hook file not found: #{@after_insert_hook_path}"
@@ -181,6 +203,26 @@ module Exwiw
       end
     end
 
+    # `--target-collection` is a mongodb-only alias of `--target-table`. Fold it
+    # into @target_table_name (the single field the rest of the CLI/runner uses)
+    # after rejecting the misuses: combining it with --target-table, or using it
+    # with a non-mongodb adapter.
+    private def resolve_target_collection_alias!
+      return if @target_collection_name.nil?
+
+      if @target_table_name
+        $stderr.puts "Specify only one of --target-table and --target-collection"
+        exit 1
+      end
+
+      if @database_adapter != "mongodb"
+        $stderr.puts "--target-collection is only supported by the mongodb adapter (use --target-table)"
+        exit 1
+      end
+
+      @target_table_name = @target_collection_name
+    end
+
     private def validate_explain_only!
       if @database_adapter == "mongodb"
         $stderr.puts "mongodb adapter is not yet supported by 'explain' subcommand"
@@ -211,6 +253,7 @@ module Exwiw
         database_name: @database_name,
         target_table: @target_table_name,
         ids: @ids.dup.freeze,
+        ids_field: @ids_field,
         output_format: @output_format,
         insert_only: @insert_only,
         log_level: @log_level,
@@ -261,7 +304,9 @@ module Exwiw
         opts.on("-a", "--adapter=ADAPTER", "Database adapter") { |v| @database_adapter = v }
         opts.on("--database=DATABASE", "Target database name") { |v| @database_name = v }
         opts.on("--target-table=[TABLE]", "Target table for extraction. If omitted, dump all tables.") { |v| @target_table_name = v }
+        opts.on("--target-collection=[COLLECTION]", "Alias of --target-table for the mongodb adapter.") { |v| @target_collection_name = v }
         opts.on("--ids=[IDS]", "Comma-separated list of identifiers. Required when --target-table is given.") { |v| @ids = v.split(',') }
+        opts.on("--ids-field=[FIELD]", "Field on the target table that --ids is matched against. Defaults to the primary key. (mongodb adapter only)") { |v| @ids_field = v }
         opts.on("--output-format=[FORMAT]", "Output format: insert (default) or copy (PostgreSQL only, dump subcommand only)") { |v| @output_format = v }
         opts.on("--insert-only", "Do not generate DELETE SQL files (dump subcommand only)") { @insert_only = true }
         opts.on("--after-insert-hook=PATH", "Path to a .rb or .sh post-processing hook executed after all insert/delete files are written (dump subcommand only)") do |v|

data/lib/exwiw/mongodb_collection_config.rb

@@ -35,6 +35,39 @@ module Exwiw
       !embedded_in.nil?
     end
 
+    # Merge an auto-generated config (`passed`) into this user-maintained one so
+    # that `MongoidSchemaGenerator` regenerations preserve hand-edited values.
+    #
+    # - structural facts come from the freshly generated config: primary_key,
+    #   belongs_tos, embedded_in.
+    # - user customizations are kept from the receiver: filter, skip,
+    #   bulk_insert_chunk_size, and each field's `replace_with` masking rule.
+    # - generated fields drive the field list (so added/removed fields track the
+    #   model), but a matching receiver field wins to retain its masking.
+    def merge(passed)
+      return passed if passed.to_hash == to_hash
+
+      MongodbCollectionConfig.new.tap do |merged|
+        merged.name = name
+        merged.primary_key = passed.primary_key
+        merged.filter = filter
+        merged.belongs_tos = passed.belongs_tos
+        merged.bulk_insert_chunk_size = bulk_insert_chunk_size
+        merged.skip = skip
+        merged.embedded_in = passed.embedded_in
+
+        # Take each field from the freshly generated config (so structural facts
+        # like `mongoid_field_name` track the model) but carry over the user's
+        # hand-edited `replace_with` masking when the field still exists.
+        receiver_field_by_name = fields.each_with_object({}) { |f, h| h[f.name] = f }
+        merged.fields = passed.fields.map do |pf|
+          receiver = receiver_field_by_name[pf.name]
+          pf.replace_with = receiver.replace_with if receiver&.replace_with
+          pf
+        end
+      end
+    end
+
     private def validate_embedded!
       return unless embedded?
       return if belongs_tos.empty?

data/lib/exwiw/mongodb_field.rb

@@ -6,6 +6,11 @@ module Exwiw
 
     attribute :name, String
     attribute :replace_with, optional(String), skip_serializing_if_nil: true
+    # The Mongoid model's Ruby accessor when the stored document key (`name`)
+    # was renamed via `field :ctry, as: :country`. Purely informational — exwiw
+    # masks/projects by `name` (the storage key) — but surfacing the accessor
+    # keeps an otherwise cryptic short key understandable in the config.
+    attribute :mongoid_field_name, optional(String), skip_serializing_if_nil: true
 
     def self.from_symbol_keys(hash)
       from(hash.transform_keys(&:to_s))

data/lib/exwiw/mongoid_schema_generator.rb

@@ -0,0 +1,240 @@
+# frozen_string_literal: true
+
+require "fileutils"
+require "json"
+
+module Exwiw
+  # Generates exwiw `MongodbCollectionConfig` files by introspecting Mongoid
+  # document models. This is the MongoDB/Mongoid counterpart of
+  # `SchemaGenerator` (which targets ActiveRecord); it is intentionally a
+  # separate class and rake task because the two ORMs expose entirely
+  # different metadata APIs.
+  #
+  # Introspection relies only on class-level Mongoid metadata
+  # (`fields`, `relations`, `collection_name`), so it does not require a live
+  # MongoDB connection.
+  class MongoidSchemaGenerator
+    def self.from_rails_application(output_dir:)
+      Rails.application.eager_load!
+      new(models: ::Mongoid.models, output_dir: output_dir)
+    end
+
+    def initialize(models:, output_dir:)
+      @models = models
+      @output_dir = output_dir
+    end
+
+    def generate!
+      collections = build_collections
+      write_files(@output_dir, collections)
+      collections
+    end
+
+    # Returns an array of `MongodbCollectionConfig`, one per *collection*
+    # (top-level collections and embedded subdocument configs alike).
+    #
+    # Models are grouped by `collection_name` so an inheritance hierarchy whose
+    # subclasses share the base's collection (Mongoid STI, discriminated by the
+    # auto-added `_type` field) collapses into a single config that aggregates
+    # every class's fields and associations. See `expand_with_descendants`.
+    def build_collections
+      models = expand_with_descendants(concrete_models)
+      models
+        .group_by { |model| model.collection_name.to_s }
+        .map { |collection_name, group| build_collection_for(collection_name, group) }
+    end
+
+    def write_files(dir, collections)
+      FileUtils.mkdir_p(dir)
+
+      collections.each do |collection|
+        path = File.join(dir, "#{collection.name}.json")
+        config_to_write =
+          if File.exist?(path)
+            MongodbCollectionConfig.from(JSON.parse(File.read(path))).merge(collection)
+          else
+            collection
+          end
+        File.write(path, JSON.pretty_generate(config_to_write.to_hash) + "\n")
+      end
+    end
+
+    # Builds one config for the collection shared by `models` (usually a single
+    # model, but an inheritance hierarchy contributes several). Fields and
+    # belongs_tos are unioned across the group; processing least-derived first
+    # keeps the base's fields leading the list and the output deterministic
+    # regardless of input order or sibling subclasses.
+    private def build_collection_for(collection_name, models)
+      ordered = models.sort_by { |model| [model.fields.size, model.name] }
+
+      attrs = {
+        name: collection_name,
+        primary_key: "_id",
+        fields: aggregate_fields(ordered),
+      }
+
+      if ordered.any?(&:embedded?)
+        # Cross-collection references from inside an embedded array are not
+        # supported (MongodbCollectionConfig rejects them), so embedded configs
+        # always carry an empty belongs_tos and instead declare where they live.
+        attrs[:belongs_tos] = []
+        attrs[:embedded_in] = embedded_in_for(ordered.find(&:embedded?))
+      else
+        attrs[:belongs_tos] = aggregate_belongs_tos(ordered)
+      end
+
+      MongodbCollectionConfig.from_symbol_keys(attrs)
+    end
+
+    # Mongoid registers only the base class of an inheritance hierarchy in
+    # `Mongoid.models`; subclasses that store into the base's collection
+    # (STI-style, distinguished by the auto-added `_type` discriminator) are
+    # reachable only via `descendants`, and the base's own metadata does NOT
+    # include subclass-only fields or associations. Expand the model set with
+    # descendants so each collection's config aggregates every class that
+    # stores into it. (A subclass that overrides `store_in` to a different
+    # collection naturally falls into its own group via the `collection_name`
+    # grouping in `build_collections`.)
+    private def expand_with_descendants(models)
+      concrete((models + models.flat_map(&:descendants)).uniq)
+    end
+
+    # Mongoid registers internal helper classes (e.g. the discriminator key
+    # host) in `Mongoid.models`; those have no usable `collection_name`. Keep
+    # only application documents.
+    private def concrete_models
+      concrete(@models)
+    end
+
+    private def concrete(models)
+      models.select do |model|
+        model.respond_to?(:collection_name) &&
+          model.name &&
+          !model.name.start_with?("Mongoid::")
+      end
+    end
+
+    # Unions the declared field names across `models`, preserving first-seen
+    # order. A subclass's `fields` already includes everything it inherits, so
+    # the base's fields lead and each subclass appends only its own.
+    private def aggregate_fields(models)
+      seen = {}
+      models.each_with_object([]) do |model, fields|
+        accessor_by_storage = aliased_field_accessors(model)
+        model.fields.keys.each do |name|
+          next if seen[name]
+
+          seen[name] = true
+          field = { name: name }
+          # When `field :ctry, as: :country` renamed the storage key, surface the
+          # Ruby accessor so the short key is not cryptic in the config.
+          accessor = accessor_by_storage[name]
+          field[:mongoid_field_name] = accessor if accessor
+          fields << field
+        end
+      end
+    end
+
+    # Maps a stored document key -> its Mongoid Ruby accessor, but ONLY for
+    # genuine `field ..., as:` renames. `Model.aliased_fields` also contains the
+    # built-in `id => _id` and one entry per association (e.g. `shop => shop_id`,
+    # `profile => user_profile`); those are not field renames, so exclude any
+    # accessor that names a relation, the `_id` storage key, or a no-op alias.
+    private def aliased_field_accessors(model)
+      relation_names = model.relations.keys
+      model.aliased_fields.each_with_object({}) do |(accessor, storage), acc|
+        next if accessor == storage
+        next if storage == "_id"
+        next if relation_names.include?(accessor)
+        next unless model.fields.key?(storage)
+
+        acc[storage] = accessor
+      end
+    end
+
+    private def aggregate_belongs_tos(models)
+      belongs_to_assocs = models.flat_map do |model|
+        model.relations.values.select do |assoc|
+          assoc.is_a?(::Mongoid::Association::Referenced::BelongsTo)
+        end
+      end
+
+      # polymorphic belongs_to (`belongs_to :reviewable, polymorphic: true`) は
+      # 単一の対象コレクションを持たないため現状未対応。誤った FK を出力しないよう
+      # ここでは除外する (将来 ActiveRecord 版と同様に展開する余地を残す)。
+      #
+      # 継承階層では基底クラスとサブクラスが同じ belongs_to を二重に持つため uniq する。
+      belongs_to_assocs
+        .reject(&:polymorphic?)
+        .map { |assoc| { table_name: assoc.klass.collection_name.to_s, foreign_key: assoc.foreign_key } }
+        .uniq
+    end
+
+    # Resolves the `embedded_in` config for an embedded model. Each embedded
+    # model points at its *immediate* embedding parent: the parent's collection
+    # name plus the single document key (`store_as`, defaulting to the relation
+    # name) the subdocuments live under within that parent.
+    #
+    # Multi-level nesting is represented one link at a time, NOT flattened into
+    # a dot-separated path. For `User embeds_many :posts` and
+    # `Post embeds_many :comments`, the Post config resolves to
+    # `{ collection_name: "users", path: "posts" }` and the Comment config to
+    # `{ collection_name: "posts", path: "comments" }`. `MongodbAdapter` walks
+    # this chain recursively (masking each `posts` subdocument, then its
+    # `comments`), which is the only form that correctly traverses both array
+    # (`embeds_many`) and Hash (`embeds_one`) intermediates — a flattened
+    # `posts.comments` path would stop at the `posts` array boundary.
+    private def embedded_in_for(model)
+      assoc = embedded_in_association(model)
+
+      # A polymorphic `embedded_in` (`embedded_in :addressable, polymorphic: true`)
+      # can live inside several different parent collections, so it has no single
+      # embedding parent and `assoc.klass` would raise a cryptic NameError
+      # (uninitialized constant) trying to resolve one. exwiw's `embedded_in`
+      # names exactly one parent collection + path, so this shape cannot be
+      # represented; fail loudly with an actionable message instead of crashing.
+      if assoc.polymorphic?
+        raise ArgumentError,
+              "MongoidSchemaGenerator: '#{model.name}' (collection '#{model.collection_name}') " \
+              "declares a polymorphic `embedded_in :#{assoc.name}`, which has no single embedding " \
+              "parent collection and cannot be expressed as an exwiw `embedded_in` config. " \
+              "Define the collection's config by hand, or make the relation non-polymorphic."
+      end
+
+      parent = assoc.klass
+
+      # A self-referential / cyclic `embedded_in` — Mongoid's
+      # `recursively_embeds_many` / `recursively_embeds_one` (which declare a
+      # `cyclic: true` `embedded_in`/`embeds_*` pair pointing at the same model),
+      # or any hand-rolled self-embedding — makes a collection BOTH a top-level
+      # document AND embedded inside documents of its own type. exwiw represents
+      # a collection as either top-level (dumpable on its own) or embedded
+      # (masked through its parent at `path`), never both: emitting an
+      # `embedded_in` here would mark the whole collection embedded, so
+      # `MongodbAdapter#dumpable?` (`!embedded?`) would silently never dump the
+      # collection's root documents. Fail loudly instead.
+      if parent.collection_name.to_s == model.collection_name.to_s
+        raise ArgumentError,
+              "MongoidSchemaGenerator: '#{model.name}' (collection '#{model.collection_name}') " \
+              "declares a self-referential (cyclic) `embedded_in :#{assoc.name}` that embeds the " \
+              "collection inside documents of its own type (e.g. `recursively_embeds_many` / " \
+              "`recursively_embeds_one`). " \
+              "exwiw represents a collection as either top-level or embedded, not both, so this " \
+              "cannot be expressed as an exwiw `embedded_in` config. Define the collection's config " \
+              "by hand."
+      end
+
+      # `store_as` defaults to the relation name and is the actual document key
+      # the subdocuments are stored under inside the immediate parent.
+      parent_relation = parent.relations[assoc.inverse.to_s]
+
+      { collection_name: parent.collection_name.to_s, path: parent_relation.store_as }
+    end
+
+    private def embedded_in_association(model)
+      model.relations.values.find do |assoc|
+        assoc.is_a?(::Mongoid::Association::Embedded::EmbeddedIn)
+      end
+    end
+  end
+end

data/lib/exwiw/query_ast.rb

@@ -3,14 +3,20 @@
 module Exwiw
   module QueryAst
     class JoinClause
-      attr_reader :base_table_name, :foreign_key, :join_table_name, :primary_key, :where_clauses
+      # `where_clauses` はこの join の join_table_name (= 結合先テーブル) に対して
+      # コンパイルされる。一方 `base_where_clauses` は base_table_name (= 結合元
+      # テーブル) に対してコンパイルされる。後者は、結合元テーブルが結合先へ
+      # polymorphic belongs_to していて型カラム (foreign_type) が結合元テーブル
+      # 側に存在するケースのために使う。
+      attr_reader :base_table_name, :foreign_key, :join_table_name, :primary_key, :where_clauses, :base_where_clauses
 
-      def initialize(base_table_name:, foreign_key:, join_table_name:, primary_key:, where_clauses: [])
+      def initialize(base_table_name:, foreign_key:, join_table_name:, primary_key:, where_clauses: [], base_where_clauses: [])
         @base_table_name = base_table_name
         @foreign_key = foreign_key
         @join_table_name = join_table_name
         @primary_key = primary_key
         @where_clauses = where_clauses
+        @base_where_clauses = base_where_clauses
       end
 
       def to_h
@@ -23,6 +29,9 @@ module Exwiw
         if where_clauses.size.positive?
           hash[:where_clauses] = where_clauses.map { |wc| wc.is_a?(String) ? wc : wc.to_h }
         end
+        if base_where_clauses.size.positive?
+          hash[:base_where_clauses] = base_where_clauses.map { |wc| wc.is_a?(String) ? wc : wc.to_h }
+        end
         hash
       end
     end

data/lib/exwiw/query_ast_builder.rb

@@ -54,8 +54,23 @@ module Exwiw
           foreign_key: relation.foreign_key,
           join_table_name: to_table.name,
           primary_key: to_table.primary_key,
-          where_clauses: []
+          where_clauses: [],
+          base_where_clauses: []
         )
+
+        # この hop 自体が polymorphic belongs_to の場合 (例: comments が
+        # commentable として posts へ polymorphic belongs_to)、型カラム
+        # (foreign_type) は結合元テーブル (from_table = base_table_name) 側に
+        # 存在する。外部キーだけでは reviewable_id=1 のような値が別モデルの
+        # 行と衝突しうるため、base_where_clauses に型条件を追加して結合元
+        # テーブルを絞り込む。
+        if relation.polymorphic?
+          join_clause.base_where_clauses.push QueryAst::WhereClause.new(
+            column_name: relation.foreign_type,
+            operator: :eq,
+            value: [relation.type_value]
+          )
+        end
         relation_to_dump_target = to_table.belongs_to(dump_target.table_name)
         if relation_to_dump_target
           join_clause.where_clauses.push QueryAst::WhereClause.new(
@@ -63,6 +78,18 @@ module Exwiw
             operator: :eq,
             value: dump_target.ids
           )
+
+          # 中間テーブルが dump target へ polymorphic belongs_to している場合は、
+          # 型カラム (foreign_type) も join 条件に追加する。型カラムは to_table
+          # (= join_table_name) 上に存在するため、JoinClause の where_clauses が
+          # join_table_name に対してコンパイルされる仕組みにそのまま乗せられる。
+          if relation_to_dump_target.polymorphic?
+            join_clause.where_clauses.push QueryAst::WhereClause.new(
+              column_name: relation_to_dump_target.foreign_type,
+              operator: :eq,
+              value: [relation_to_dump_target.type_value]
+            )
+          end
         end
 
         # Add filter from intermediate table to join clause
@@ -80,6 +107,10 @@ module Exwiw
       clauses = []
 
       if table.name == dump_target.table_name
+        # TODO: honor dump_target.ids_field here so `--ids` can match a non
+        # primary-key column on the target table (currently mongodb-only; the
+        # CLI rejects --ids-field for the sql adapters). When implemented, use
+        # `dump_target.ids_field || table.primary_key` as the column_name.
         clauses.push Exwiw::QueryAst::WhereClause.new(
           column_name: table.primary_key,
           operator: :eq,
@@ -98,6 +129,17 @@ module Exwiw
         value: dump_target.ids
       )
 
+      # polymorphic belongs_to の場合は外部キーだけでは型を区別できないため
+      # (例: reviewable_id=1 が Product なのか別モデルなのか判別できない)、
+      # 型カラム (foreign_type) を type_value で絞り込む条件を追加する。
+      if belongs_to.polymorphic?
+        clauses.push Exwiw::QueryAst::WhereClause.new(
+          column_name: belongs_to.foreign_type,
+          operator: :eq,
+          value: [belongs_to.type_value]
+        )
+      end
+
       if table.filter
         clauses.push table.filter
       end

data/lib/exwiw/schema_generator.rb

@@ -145,15 +145,44 @@ module Exwiw
     end
 
     private def aggregate_belongs_tos(models)
-      pairs = models
-        .flat_map { |m| m.reflect_on_all_associations(:belongs_to) }
-        .reject(&:polymorphic?) # XXX: Support polymorphic
-        .map { |assoc| [assoc.table_name, assoc.foreign_key] }
-        .uniq
-
-      pairs.map do |table_name, foreign_key|
-        { table_name: table_name, foreign_key: foreign_key }
-      end
+      belongs_to_assocs = models.flat_map { |m| m.reflect_on_all_associations(:belongs_to) }
+
+      non_polymorphic = belongs_to_assocs
+        .reject(&:polymorphic?)
+        .map { |assoc| { table_name: assoc.table_name, foreign_key: assoc.foreign_key } }
+
+      # polymorphic な belongs_to (`belongs_to :reviewable, polymorphic: true`) は
+      # 単一の対象テーブルを持たない。対象になりうるテーブルは、他モデルで
+      # `has_many/has_one ..., as: <association_name>` と宣言されている側から逆引き
+      # する。各候補テーブルごとに、型カラム (`foreign_type`) と格納される型の値
+      # (`type_value`) を添えた belongs_to を 1 件ずつ展開する。
+      polymorphic = belongs_to_assocs
+        .select(&:polymorphic?)
+        .flat_map do |assoc|
+          polymorphic_target_models(assoc.name).map do |target_model|
+            {
+              table_name: target_model.table_name,
+              foreign_key: assoc.foreign_key,
+              foreign_type: assoc.foreign_type,
+              type_value: target_model.polymorphic_name,
+            }
+          end
+        end
+
+      (non_polymorphic + polymorphic).uniq
+    end
+
+    # polymorphic 関連 `association_name` の対象となりうる具象モデルを、全モデルの
+    # `has_many` / `has_one` の `as:` オプションから逆引きして列挙する。
+    # `concrete_models` の並びは `ActiveRecord::Base.descendants` の順に依存し、
+    # Ruby バージョンによって変わりうるため、生成される belongs_to の並びが安定する
+    # よう `table_name` でソートして決定的に返す。
+    private def polymorphic_target_models(association_name)
+      concrete_models.select do |model|
+        (model.reflect_on_all_associations(:has_many) +
+         model.reflect_on_all_associations(:has_one))
+          .any? { |reflection| reflection.options[:as] == association_name }
+      end.sort_by(&:table_name)
     end
 
     # Identifies which database a model belongs to. With Rails multi-DB

data/lib/exwiw/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Exwiw
-  VERSION = "0.2.7"
+  VERSION = "0.2.9"
 end

data/lib/exwiw.rb

@@ -25,6 +25,7 @@ require_relative "exwiw/after_insert_hook"
 require_relative "exwiw/runner"
 require_relative "exwiw/explain_runner"
 require_relative "exwiw/schema_generator"
+require_relative "exwiw/mongoid_schema_generator"
 
 begin
   require 'rails'
@@ -34,6 +35,9 @@ else
 end
 
 module Exwiw
-  DumpTarget = Struct.new(:table_name, :ids, keyword_init: true)
+  # `ids_field` optionally overrides which field `--ids` is matched against on
+  # the target table. When nil the table's primary key is used (the historical
+  # behavior). Currently only honored by the mongodb adapter.
+  DumpTarget = Struct.new(:table_name, :ids, :ids_field, keyword_init: true)
   ConnectionConfig = Struct.new(:adapter, :host, :port, :user, :password, :database_name, keyword_init: true)
 end

data/lib/tasks/exwiw.rake

@@ -10,5 +10,14 @@ namespace :exwiw do
         output_dir: ENV["OUTPUT_DIR_PATH"] || "exwiw",
       ).generate!
     end
+
+    desc "Generate schema from a Mongoid application"
+    task generate_mongoid: :environment do
+      require "exwiw"
+
+      Exwiw::MongoidSchemaGenerator.from_rails_application(
+        output_dir: ENV["OUTPUT_DIR_PATH"] || "exwiw",
+      ).generate!
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: exwiw
 version: !ruby/object:Gem::Version
-  version: 0.2.7
+  version: 0.2.9
 platform: ruby
 authors:
 - Shia
@@ -57,6 +57,7 @@ files:
 - lib/exwiw/mongo_query.rb
 - lib/exwiw/mongodb_collection_config.rb
 - lib/exwiw/mongodb_field.rb
+- lib/exwiw/mongoid_schema_generator.rb
 - lib/exwiw/query_ast.rb
 - lib/exwiw/query_ast_builder.rb
 - lib/exwiw/railtie.rb