activerecord-dbt 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c7948bb8cc4c14d07c3fca2855b20957a2871c54dd67149fb178029fa93f6f09
-  data.tar.gz: c8146bfcd3ffa762f16889a79750807ab1e72986e49f34069ac043fe8c89d5f0
+  metadata.gz: affd8c9f9864d11cbaf5c35a759ea603c72784b9eae777987debad89fd8cd509
+  data.tar.gz: 6b8ca507c1e413143f3d0a7330451767190b3acb731e70c8b2beb87bff9df64b
 SHA512:
-  metadata.gz: a2d3ce1aa15cafbd0bb6f282d5faa64f6319d669a38a8dde20b04af3cccf12a21e16fa22cf6f7b87b245dd094bad296de2dd22bd027db047b2ed845cc7de2b8a
-  data.tar.gz: 6097b5f0b25324679e012d4a16833afba17514f2ebb82eabe778ae510bc8f283c280dd3d6916b2a3220092c63c7827c56b4b0660f14caa6f1b210a496dff7cdc
+  metadata.gz: 2798392f8cdcb454a64957a005d7d16e3eeec7e1648234ba5c598cbacb6b683198493337619c9e0a9f17f67a231f6a3fc6f34658651671de3776c7caf28093c9
+  data.tar.gz: d417dfdd69d8a0c41baee03a41a0880ecba05abbbb61c86b80ed1ef48021bcdda16a4e2b43dd414a29d7f56ad9cf92a0e3afe0a693b5ae4aa3a38e0bbeb9f03d

data/README.md

@@ -58,6 +58,7 @@ dwh_platform | Specify the data warehouse platform to which dbt connects. The de
 data_sync_delayed | Indicates whether there is a data delay. If set to `true`, `severity: warn` is applied to the `relationships` test. The default is `false`.
 logger | The destination for log output. The default is `Logger.new('./log/active_record_dbt.log')`.
 used_dbt_package_names | An array of `dbt` package names to use.
+locale | I18n locale. The default is `I18n.locale`.
 
 List of platforms that can currently be set with `dwh_platform`.
 
@@ -165,10 +166,10 @@ table_overrides:
 
 ##### defaults
 
-Set default values for the `name` and `description` of `tables`.
+Set the default value for the `description`(`logical_name`, `description`) of `tables`.
 
-In `logical_name` and `description` of `table_descriptions`, you can refer to the table name with `{{ table_name }}`.
-In the `description` of `columns`, you can refer to the table name with `{{ table_name }}` and the column name with `{{ column_name }}`.
+In the `logical_name` and `description` of `table_descriptions`, you can refer to the table name with `{{ table_name }}`.
+In the `description` of `table_descriptions.columns`, you can refer to the table name with `{{ table_name }}` and the column name with `{{ column_name }}`.
 
 Example:
 
@@ -831,6 +832,10 @@ from final
 
 Example:
 
+> [!NOTE]
+>
+> The output will be as shown below. It is recommended to indent the YAML file with a tool of your choice.
+
 ```yaml
 ---
 version: 2
@@ -883,6 +888,116 @@ models:
 
 ```
 
+### Generated dbt Seed Files
+
+#### dbt Seed Configuration
+
+In the `#{config_directory_path}/source_config.yml` file, describe the properties you want to set for the seed enum.
+You can configure `defaults` in this file.
+
+##### defaults
+
+Set the default value for the `description` of the `seeds` enum.
+
+In the `description` of `seed_descriptions.enum`, you can refer to the source name with `{{ source_name }}`, the translated table name with `{{ translated_table_name }}`, and the translated column name with `{{ translated_attribute_name }}`.
+
+Example:
+
+```yml
+defaults:
+  seed_descriptions:
+    enum:
+      description: "{{ source_name }} {{ translated_table_name }} {{ translated_attribute_name }} enum"
+
+```
+
+If nothing is set, it defaults to the following:
+
+```yml
+defaults:
+  seed_descriptions:
+    enum:
+      description: "{{ source_name }} {{ translated_table_name }} {{ translated_attribute_name }} enum"
+
+```
+
+#### Generate dbt Seed Enum Files
+
+Generate seed enum files for dbt:
+
+```bash
+$ bin/rails generate active_record:dbt:enum TABLE_NAME ENUM_COLUMN_NAME
+```
+
+Generate seed enum files for dbt from the specified `TABLE_NAME` and `ENUM_COLUMN_NAME`.
+
+File | Description
+--------- | ---------
+`#{export_directory_path}/seed_#{source_name}__#{table_name_singularize}_enum_#{enum_pluralized}.csv` | Seed enum file for dbt.
+`#{export_directory_path}/seed_#{source_name}__#{table_name_singularize}_enum_#{enum_pluralized}.yml` | Seed enum documentation file for dbt.
+
+Example:
+
+```bash
+$ bin/rails generate active_record:dbt:enum posts status
+```
+
+##### Generate `#{export_directory_path}/seed_#{source_name}__#{table_name_singularize}_enum_#{enum_pluralized}.csv`
+
+Example:
+
+```csv
+status_before_type_of_cast,status_key,status_en,status_ja
+0,draft,Draft,下書き
+1,published,Published,公開
+2,deleted,Deleted,削除
+
+```
+
+##### Generate `#{export_directory_path}/seed_#{source_name}__#{table_name_singularize}_enum_#{enum_pluralized}.yml`
+
+Example:
+
+> [!NOTE]
+>
+> The output will be as shown below. It is recommended to indent the YAML file with a tool of your choice.
+
+```yaml
+---
+version: 2
+seeds:
+- name: seed_dummy__post_enum_statuses
+  description: dummy Post Status enum
+  config:
+    column_types:
+      status_before_type_of_cast: int64
+      status_key: string
+      status_en: string
+      status_ja: string
+columns:
+- name: status_before_type_of_cast
+  description: Status
+  tests:
+  - unique
+  - not_null
+- name: status_key
+  description: Status(key)
+  tests:
+  - unique
+  - not_null
+- name: status_en
+  description: Status(en)
+  tests:
+  - unique
+  - not_null
+- name: status_ja
+  description: Status(ja)
+  tests:
+  - unique
+  - not_null
+
+```
+
 ## Contributing
 
 Contribution directions go here.

data/lib/active_record/dbt/column/column.rb

@@ -5,6 +5,7 @@ module ActiveRecord
     module Column
       class Column
         include ActiveRecord::Dbt::DataType::Mapper
+        include ActiveRecord::Dbt::I18nWrapper::Translate
 
         attr_reader :table_name, :column, :column_test, :primary_keys
 
@@ -23,7 +24,7 @@ module ActiveRecord
           {
             'name' => column_name,
             'description' => description,
-            'data_type' => data_type,
+            'data_type' => data_type(column.type),
             **column_overrides.except(:tests),
             'tests' => column_test.config
           }.compact
@@ -45,18 +46,6 @@ module ActiveRecord
           source_config.dig(:table_descriptions, table_name, :columns, column_name)
         end
 
-        def translated_attribute_name
-          translated_column_name || translated_default_column_name
-        end
-
-        def translated_column_name
-          I18n.t("activerecord.attributes.#{table_name.singularize}.#{column_name}", default: nil)
-        end
-
-        def translated_default_column_name
-          I18n.t("attributes.#{column_name}", default: nil)
-        end
-
         def key_column_name
           column_name if primary_key? || foreign_key?
         end

data/lib/active_record/dbt/column/testable/not_null_testable.rb

@@ -10,7 +10,13 @@ module ActiveRecord
           define_required_methods :column
 
           def not_null_test
-            column.null == true ? nil : 'not_null'
+            null? ? nil : 'not_null'
+          end
+
+          private
+
+          def null?
+            column.null == true
           end
         end
       end

data/lib/active_record/dbt/config.rb

@@ -9,38 +9,10 @@ module ActiveRecord
 
       include ActiveRecord::Dbt::Configuration::DataSync
       include ActiveRecord::Dbt::Configuration::DwhPlatform
+      include ActiveRecord::Dbt::Configuration::I18nConfiguration
       include ActiveRecord::Dbt::Configuration::Logger
-      include ActiveRecord::Dbt::Configuration::Parser
+      include ActiveRecord::Dbt::Configuration::Source
       include ActiveRecord::Dbt::Configuration::UsedDbtPackage
-
-      DEFAULT_CONFIG_DIRECTORY_PATH = 'lib/dbt'
-      DEFAULT_EXPORT_DIRECTORY_PATH = 'doc/dbt'
-
-      attr_writer :config_directory_path, :export_directory_path
-
-      def source_config_path
-        @source_config_path ||= "#{config_directory_path}/source_config.yml"
-      end
-
-      def source_config
-        @source_config ||= parse_yaml(source_config_path)
-      end
-
-      def source_name
-        @source_name ||= source_config.dig(:sources, :name).tap do |source_name|
-          raise SourceNameIsNullError, "'sources.name' is required in '#{source_config_path}'." if source_name.nil?
-        end
-      end
-
-      def config_directory_path
-        @config_directory_path ||= DEFAULT_CONFIG_DIRECTORY_PATH
-      end
-
-      def export_directory_path
-        @export_directory_path ||= DEFAULT_EXPORT_DIRECTORY_PATH
-      end
-
-      class SourceNameIsNullError < StandardError; end
     end
   end
 end

data/lib/active_record/dbt/configuration/i18n_configuration.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Dbt
+    module Configuration
+      module I18nConfiguration
+        def locale=(locale = I18n.locale)
+          I18n.load_path += Dir[Rails.root.join('config/locales/**/*.{rb,yml}')]
+          I18n.locale = locale
+        end
+      end
+    end
+  end
+end

data/lib/active_record/dbt/configuration/source.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Dbt
+    module Configuration
+      module Source
+        include ActiveRecord::Dbt::Configuration::Parser
+
+        DEFAULT_CONFIG_DIRECTORY_PATH = 'lib/dbt'
+        DEFAULT_EXPORT_DIRECTORY_PATH = 'doc/dbt'
+
+        attr_writer :config_directory_path, :export_directory_path
+
+        def config_directory_path
+          @config_directory_path ||= DEFAULT_CONFIG_DIRECTORY_PATH
+        end
+
+        def export_directory_path
+          @export_directory_path ||= DEFAULT_EXPORT_DIRECTORY_PATH
+        end
+
+        def source_config_path
+          @source_config_path ||= "#{config_directory_path}/source_config.yml"
+        end
+
+        def source_config
+          @source_config ||= parse_yaml(source_config_path)
+        end
+
+        def source_name
+          @source_name ||= source_config.dig(:sources, :name).tap do |source_name|
+            raise SourceNameIsNullError, "'sources.name' is required in '#{source_config_path}'." if source_name.nil?
+          end
+        end
+
+        class SourceNameIsNullError < StandardError; end
+      end
+    end
+  end
+end

data/lib/active_record/dbt/data_type/mapper.rb

@@ -4,31 +4,98 @@ module ActiveRecord
   module Dbt
     module DataType
       module Mapper
-        include ActiveRecord::Dbt::DataType::DwhPlatform::ApacheSpark
-        include ActiveRecord::Dbt::DataType::DwhPlatform::Bigquery
-        include ActiveRecord::Dbt::DataType::DwhPlatform::PostgreSql
-        include ActiveRecord::Dbt::DataType::DwhPlatform::Redshift
-        include ActiveRecord::Dbt::DataType::DwhPlatform::Snowflake
-
         extend ActiveRecord::Dbt::RequiredMethods
 
         # [Platform-specific data types | dbt Developer Hub](https://docs.getdbt.com/reference/resource-properties/data-types)
         RUBY_TO_DWH_PLATFORM_TYPE_MAP = {
-          'bigquery' => RUBY_TO_BIGQUERY_TYPES,
-          'postgres' => RUBY_TO_POSTGRES_TYPES,
-          'redshift' => RUBY_TO_REDSHIFT_TYPES,
-          'snowflake' => RUBY_TO_SNOWFLAKE_TYPES,
-          'spark' => RUBY_TO_SPARK_TYPES
+          # [Data types  |  BigQuery  |  Google Cloud](https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#data_type_list)
+          'bigquery' => {
+            binary: 'bytes',
+            boolean: 'bool',
+            date: 'date',
+            datetime: 'datetime',
+            decimal: 'int64',
+            float: 'float64',
+            integer: 'int64',
+            json: 'json',
+            string: 'string',
+            text: 'string',
+            time: 'time'
+          },
+
+          # TODO: I have not tried it. I don't know if this is the correct data_type.
+          # [PostgreSQL: Documentation: 16: Chapter 8. Data Types](https://www.postgresql.org/docs/current/datatype.html)
+          'postgres' => {
+            binary: 'bytea',
+            boolean: 'boolean',
+            date: 'date',
+            datetime: 'timestamp',
+            decimal: 'bigint',
+            float: 'double precision',
+            integer: 'bigint',
+            json: 'json',
+            string: 'text',
+            text: 'text',
+            time: 'time'
+          },
+
+          # TODO: I have not tried it. I don't know if this is the correct data_type.
+          # [Data types - Amazon Redshift](https://docs.aws.amazon.com/redshift/latest/dg/c_Supported_data_types.html)
+          'redshift' => {
+            binary: 'varbyte',
+            boolean: 'bool',
+            date: 'date',
+            datetime: 'timestamp',
+            decimal: 'decimal',
+            float: 'double precision',
+            integer: 'integer',
+            json: 'super',
+            string: 'varchar',
+            text: 'varchar',
+            time: 'time'
+          },
+
+          # TODO: I have not tried it. I don't know if this is the correct data_type.
+          # [Summary of data types | Snowflake Documentation](https://docs.snowflake.com/en/sql-reference/intro-summary-data-types)
+          'snowflake' => {
+            binary: 'binary',
+            boolean: 'boolean',
+            date: 'date',
+            datetime: 'datetime',
+            decimal: 'decimal',
+            float: 'float',
+            integer: 'integer',
+            json: 'variant',
+            string: 'varchar',
+            text: 'varchar',
+            time: 'time'
+          },
+
+          # TODO: I have not tried it. I don't know if this is the correct data_type.
+          # [Data Types - Spark 3.5.1 Documentation](https://spark.apache.org/docs/latest/sql-ref-datatypes.html)
+          'spark' => {
+            binary: 'binary',
+            boolean: 'boolean',
+            date: 'date',
+            datetime: 'timestamp',
+            decimal: 'decimal',
+            float: 'float',
+            integer: 'integer',
+            json: 'string',
+            string: 'string',
+            text: 'string',
+            time: 'string'
+          }
         }.freeze
 
-        define_required_methods :column, :@config
+        define_required_methods :@config
 
         delegate :dwh_platform, to: :@config
 
         private
 
-        def data_type
-          RUBY_TO_DWH_PLATFORM_TYPE_MAP[dwh_platform].fetch(column.type, 'unknown')
+        def data_type(type)
+          RUBY_TO_DWH_PLATFORM_TYPE_MAP[dwh_platform].fetch(type, 'unknown')
         end
       end
     end

data/lib/active_record/dbt/dbt_package/dbterd/column/testable/relationships_meta_relationship_type.rb

@@ -14,13 +14,12 @@ module ActiveRecord
               delegate :used_dbterd?, :add_log, to: :@config
 
               def relationships_meta_relationship_type
-                return nil unless used_dbterd?
-                return nil if no_relationship?
+                return nil if !used_dbterd? || no_relationship?
 
                 {
                   'relationship_type' => relationship_type
                 }
-              rescue NotSpecifiedOrNotInvalidIdError, StandardError => e
+              rescue NotSpecifiedOrNotInvalidIdError, NameError => e
                 relationships_meta_relationship_type_with_active_record_dbt_error(e)
               end
 

data/lib/active_record/dbt/factory/model/staging_factory.rb

@@ -5,14 +5,14 @@ module ActiveRecord
     module Factory
       module Model
         module StagingFactory
-          def self.yml_build(table_name)
+          def self.build(table_name)
             table_factory = ActiveRecord::Dbt::Factory::TableFactory.build(table_name)
             yml = ActiveRecord::Dbt::Model::Staging::Yml.new(table_factory)
-            struct = Struct.new(:export_path, :yml_dump, keyword_init: true)
+            struct = Struct.new(:export_path, :dump, keyword_init: true)
 
             struct.new(
               export_path: yml.export_path,
-              yml_dump: YAML.dump(yml.config.deep_stringify_keys)
+              dump: yml.dump
             )
           end
         end

data/lib/active_record/dbt/factory/source_factory.rb

@@ -6,9 +6,8 @@ module ActiveRecord
       module SourceFactory
         def self.build
           tables_factory = ActiveRecord::Dbt::Factory::TablesFactory.build
-          config = ActiveRecord::Dbt::Source::Yml.new(tables_factory).config
 
-          YAML.dump(config.deep_stringify_keys)
+          ActiveRecord::Dbt::Source::Yml.new(tables_factory).dump
         end
       end
     end

data/lib/active_record/dbt/i18n_wrapper/translate.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Dbt
+    module I18nWrapper
+      module Translate
+        extend ActiveRecord::Dbt::RequiredMethods
+
+        define_required_methods :table_name
+
+        delegate :singularize, to: :table_name, prefix: true
+
+        def translated_table_name
+          I18n.t("activerecord.models.#{table_name_singularize}", default: nil)
+        end
+
+        def translated_attribute_name
+          @translated_attribute_name ||=
+            translated_column_name || translated_default_column_name
+        end
+
+        private
+
+        def translated_column_name
+          I18n.t("activerecord.attributes.#{table_name_singularize}.#{column_name}", default: nil)
+        end
+
+        def translated_default_column_name
+          I18n.t("attributes.#{column_name}", default: nil)
+        end
+      end
+    end
+  end
+end

data/lib/active_record/dbt/model/staging/yml.rb

@@ -21,7 +21,13 @@ module ActiveRecord
             "#{basename}.yml"
           end
 
-          def config
+          def dump
+            YAML.dump(model_config.deep_stringify_keys)
+          end
+
+          private
+
+          def model_config
             {
               'version' => 2,
               'models' => [
@@ -34,8 +40,6 @@ module ActiveRecord
             }
           end
 
-          private
-
           def columns
             @columns ||= sort_columns(table.config['columns'])
           end

data/lib/active_record/dbt/seed/enum/base.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Dbt
+    module Seed
+      module Enum
+        module Base
+          attr_reader :table_name, :enum_column_name
+
+          delegate :source_name, :export_directory_path, to: :@config
+          delegate :singularize, to: :table_name, prefix: true
+
+          def initialize(table_name, enum_column_name)
+            @table_name = table_name
+            @enum_column_name = enum_column_name
+            @config = ActiveRecord::Dbt::Config.instance
+          end
+
+          private
+
+          def basename
+            "#{export_directory_path}/#{seed_name}"
+          end
+
+          def seed_name
+            "seed_#{source_name}__#{table_name_singularize}_enum_#{enum_pluralized}"
+          end
+
+          def locales
+            @locales ||= I18n.available_locales
+          end
+
+          def application_record_klass
+            @application_record_klass ||= table_name_singularize.classify.constantize
+          rescue NameError => _e
+            raise DoesNotExistTableError, "#{table_name} table does not exist."
+          end
+
+          def enum_pluralized
+            enum_column_name.pluralize
+          end
+
+          class DoesNotExistTableError < StandardError; end
+        end
+      end
+    end
+  end
+end

data/lib/active_record/dbt/seed/enum/csv.rb

@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+require 'csv'
+
+module ActiveRecord
+  module Dbt
+    module Seed
+      module Enum
+        class Csv
+          include ActiveRecord::Dbt::Seed::Enum::Base
+
+          delegate :singularize, to: :table_name, prefix: true
+
+          def export_path
+            "#{basename}.csv"
+          end
+
+          def dump
+            CSV.generate(headers: true) do |csv|
+              rows.each { |row| csv << row }
+            end
+          end
+
+          private
+
+          def rows
+            [
+              header,
+              *enum_rows
+            ]
+          end
+
+          def header
+            [
+              "#{enum_column_name}_before_type_of_cast",
+              "#{enum_column_name}_key",
+              *locale_header
+            ]
+          end
+
+          def locale_header
+            locales.map { |locale| "#{enum_column_name}_#{locale}" }
+          end
+
+          def enum_rows
+            enums.map do |enum_key, enum_before_type_of_cast|
+              [
+                enum_before_type_of_cast,
+                enum_key,
+                *enum_values(enum_key)
+              ]
+            end
+          end
+
+          def enums
+            application_record_klass.defined_enums.fetch(enum_column_name)
+          rescue KeyError => _e
+            raise NotEnumColumnError, "#{table_name}.#{enum_column_name} column is not an enum." if enum_column?
+
+            raise DoesNotExistColumnError, "The #{enum_column_name} column in the #{table_name} table does not exist."
+          end
+
+          def enum_column?
+            application_record_klass.column_names.include?(enum_column_name)
+          end
+
+          def enum_values(enum_key)
+            locales.map { |locale| translated_enum_value(enum_key, locale) }
+          end
+
+          def translated_enum_value(enum_key, locale)
+            I18n.t(
+              "activerecord.enum.#{table_name_singularize}.#{enum_column_name}.#{enum_key}",
+              locale: locale,
+              default: nil
+            )
+          end
+
+          class DoesNotExistColumnError < StandardError; end
+          class NotEnumColumnError < StandardError; end
+        end
+      end
+    end
+  end
+end

data/lib/active_record/dbt/seed/enum/yml.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Dbt
+    module Seed
+      module Enum
+        class Yml
+          include ActiveRecord::Dbt::Column::Testable::UniqueTestable
+          include ActiveRecord::Dbt::Column::Testable::NotNullTestable
+          include ActiveRecord::Dbt::DataType::Mapper
+          include ActiveRecord::Dbt::I18nWrapper::Translate
+          include ActiveRecord::Dbt::Seed::Enum::Base
+
+          delegate :source_config, to: :@config
+
+          alias column_name enum_column_name
+
+          def export_path
+            "#{basename}.yml"
+          end
+
+          def dump
+            YAML.dump(seed_config.deep_stringify_keys)
+          end
+
+          private
+
+          def seed_config
+            {
+              'version' => 2,
+              'seeds' => [
+                'name' => seed_name,
+                'description' => seed_description,
+                'config' => {
+                  'column_types' => column_types
+                }
+              ],
+              'columns' => columns
+            }
+          end
+
+          def seed_description
+            default_seed_description ||
+              "#{source_name} #{translated_table_name} #{translated_attribute_name} enum".strip
+          end
+
+          def default_seed_description
+            source_config.dig(:defaults, :seed_descriptions, :enum, :description)
+                         &.gsub(/{{\s*source_name\s*}}/, source_name)
+                         &.gsub(/{{\s*translated_table_name\s*}}/, translated_table_name)
+                         &.gsub(/{{\s*translated_attribute_name\s*}}/, translated_attribute_name)
+          end
+
+          def column_types
+            {
+              "#{enum_column_name}_before_type_of_cast" => data_type(before_type_of_cast_type),
+              "#{enum_column_name}_key" => data_type(:string),
+              **enum_column_types
+            }
+          end
+
+          def before_type_of_cast_type
+            application_record_klass.columns_hash[enum_column_name].type
+          end
+
+          def enum_column_types
+            locales.each_with_object({}) do |locale, hash|
+              hash["#{enum_column_name}_#{locale}"] = data_type(:string)
+            end
+          end
+
+          def columns
+            [
+              before_type_of_cast_column,
+              enum_key_column,
+              *enum_columns
+            ].compact
+          end
+
+          def before_type_of_cast_column
+            {
+              'name' => "#{enum_column_name}_before_type_of_cast",
+              'description' => translated_attribute_name,
+              'tests' => tests
+            }.compact
+          end
+
+          def enum_key_column
+            {
+              'name' => "#{enum_column_name}_key",
+              'description' => "#{translated_attribute_name}(key)",
+              'tests' => tests
+            }.compact
+          end
+
+          def enum_columns
+            locales.each_with_object([]) do |locale, array|
+              array.push(
+                {
+                  'name' => "#{enum_column_name}_#{locale}",
+                  'description' => "#{translated_attribute_name}(#{locale})",
+                  'tests' => tests
+                }.compact
+              )
+            end
+          end
+
+          def tests
+            [
+              unique_test,
+              not_null_test
+            ].compact.presence
+          end
+
+          # MEMO: I think all enums are unique.
+          def unique?
+            true
+          end
+
+          # MEMO: I think all enums are null.
+          def null?
+            false
+          end
+        end
+      end
+    end
+  end
+end

data/lib/active_record/dbt/source/yml.rb

@@ -13,6 +13,12 @@ module ActiveRecord
           @config = ActiveRecord::Dbt::Config.instance
         end
 
+        def dump
+          YAML.dump(config.deep_stringify_keys)
+        end
+
+        private
+
         def config
           {
             'version' => 2,
@@ -22,8 +28,6 @@ module ActiveRecord
           }
         end
 
-        private
-
         def source_properties
           source_config[:sources]
         end

data/lib/active_record/dbt/table/yml.rb

@@ -4,6 +4,7 @@ module ActiveRecord
   module Dbt
     module Table
       class Yml
+        include ActiveRecord::Dbt::I18nWrapper::Translate
         include ActiveRecord::Dbt::Table::Base
 
         attr_reader :table_test, :columns
@@ -51,10 +52,6 @@ module ActiveRecord
             "Write a logical_name of the '#{table_name}' table."
         end
 
-        def translated_table_name
-          I18n.t("activerecord.models.#{table_name.singularize}", default: nil)
-        end
-
         def default_logical_name
           source_config.dig(:defaults, :table_descriptions, :logical_name)
                        &.gsub(/{{\s*table_name\s*}}/, table_name)

data/lib/active_record/dbt/version.rb

@@ -2,6 +2,6 @@
 
 module ActiveRecord
   module Dbt
-    VERSION = '0.2.0'
+    VERSION = '0.3.0'
   end
 end

data/lib/generators/active_record/dbt/config/templates/source_config.yml.tt

@@ -40,6 +40,9 @@ defaults:
     logical_name: Write a logical_name of the '{{ table_name }}' table.
     columns:
       description: Write a description of the '{{ table_name }}.{{ column_name }}' column.
+  seed_descriptions:
+    enum:
+      description: "{{ source_name }} {{ translated_table_name }} {{ translated_attribute_name }} enum"
 
 table_descriptions:
   ar_internal_metadata:

data/lib/generators/active_record/dbt/enum/USAGE

@@ -0,0 +1,9 @@
+Description:
+    Generate seed enum files for dbt from the specified TABLE_NAME and ENUM_COLUMN_NAME.
+
+Example:
+    bin/rails generate active_record:dbt:enum TABLE_NAME ENUM_COLUMN_NAME
+
+    This will create:
+        #{export_directory_path}/seed_#{source_name}__#{table_name_singularize}_enum_#{enum_pluralized}.csv
+        #{export_directory_path}/seed_#{source_name}__#{table_name_singularize}_enum_#{enum_pluralized}.yml

data/lib/generators/active_record/dbt/enum/enum_generator.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Dbt
+    module Generators
+      class EnumGenerator < Rails::Generators::NamedBase
+        source_root File.expand_path('templates', __dir__)
+
+        argument :enum_column_name, type: :string, default: nil
+
+        def create_enum_seed_csv_file
+          create_file csv.export_path, csv.dump
+        end
+
+        def create_enum_seed_yml_file
+          create_file yml.export_path, yml.dump
+        end
+
+        private
+
+        def csv
+          @csv ||= ActiveRecord::Dbt::Seed::Enum::Csv.new(name, enum_column_name)
+        end
+
+        def yml
+          @yml ||= ActiveRecord::Dbt::Seed::Enum::Yml.new(name, enum_column_name)
+        end
+      end
+    end
+  end
+end

data/lib/generators/active_record/dbt/staging_model/staging_model_generator.rb

@@ -11,7 +11,7 @@ module ActiveRecord
         end
 
         def create_staging_model_yml_file
-          create_file yml.export_path, yml.yml_dump
+          create_file yml.export_path, yml.dump
         end
 
         private
@@ -21,7 +21,7 @@ module ActiveRecord
         end
 
         def yml
-          @yml ||= ActiveRecord::Dbt::Factory::Model::StagingFactory.yml_build(name)
+          @yml ||= ActiveRecord::Dbt::Factory::Model::StagingFactory.build(name)
         end
 
         def source_paths

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-dbt
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - yamotech
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-03 00:00:00.000000000 Z
+date: 2024-08-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -73,14 +73,11 @@ files:
 - lib/active_record/dbt/config.rb
 - lib/active_record/dbt/configuration/data_sync.rb
 - lib/active_record/dbt/configuration/dwh_platform.rb
+- lib/active_record/dbt/configuration/i18n_configuration.rb
 - lib/active_record/dbt/configuration/logger.rb
 - lib/active_record/dbt/configuration/parser.rb
+- lib/active_record/dbt/configuration/source.rb
 - lib/active_record/dbt/configuration/used_dbt_package.rb
-- lib/active_record/dbt/data_type/dwh_platform/apache_spark.rb
-- lib/active_record/dbt/data_type/dwh_platform/bigquery.rb
-- lib/active_record/dbt/data_type/dwh_platform/postgre_sql.rb
-- lib/active_record/dbt/data_type/dwh_platform/redshift.rb
-- lib/active_record/dbt/data_type/dwh_platform/snowflake.rb
 - lib/active_record/dbt/data_type/mapper.rb
 - lib/active_record/dbt/dbt_package/dbt_utils/table/testable/unique_combination_of_columns_testable.rb
 - lib/active_record/dbt/dbt_package/dbterd/column/testable/relationships_meta_relationship_type.rb
@@ -89,11 +86,15 @@ files:
 - lib/active_record/dbt/factory/source_factory.rb
 - lib/active_record/dbt/factory/table_factory.rb
 - lib/active_record/dbt/factory/tables_factory.rb
+- lib/active_record/dbt/i18n_wrapper/translate.rb
 - lib/active_record/dbt/model/staging/base.rb
 - lib/active_record/dbt/model/staging/sql.rb
 - lib/active_record/dbt/model/staging/yml.rb
 - lib/active_record/dbt/railtie.rb
 - lib/active_record/dbt/required_methods.rb
+- lib/active_record/dbt/seed/enum/base.rb
+- lib/active_record/dbt/seed/enum/csv.rb
+- lib/active_record/dbt/seed/enum/yml.rb
 - lib/active_record/dbt/source/yml.rb
 - lib/active_record/dbt/table/base.rb
 - lib/active_record/dbt/table/test.rb
@@ -102,6 +103,8 @@ files:
 - lib/generators/active_record/dbt/config/USAGE
 - lib/generators/active_record/dbt/config/config_generator.rb
 - lib/generators/active_record/dbt/config/templates/source_config.yml.tt
+- lib/generators/active_record/dbt/enum/USAGE
+- lib/generators/active_record/dbt/enum/enum_generator.rb
 - lib/generators/active_record/dbt/initializer/USAGE
 - lib/generators/active_record/dbt/initializer/initializer_generator.rb
 - lib/generators/active_record/dbt/initializer/templates/dbt.rb

data/lib/active_record/dbt/data_type/dwh_platform/apache_spark.rb

@@ -1,27 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecord
-  module Dbt
-    module DataType
-      module DwhPlatform
-        module ApacheSpark
-          # TODO: I have not tried it. I don't know if this is the correct data_type.
-          # [Data Types - Spark 3.5.1 Documentation](https://spark.apache.org/docs/latest/sql-ref-datatypes.html)
-          RUBY_TO_SPARK_TYPES = {
-            binary: 'binary',
-            boolean: 'boolean',
-            date: 'date',
-            datetime: 'timestamp',
-            decimal: 'decimal',
-            float: 'float',
-            integer: 'integer',
-            json: 'string',
-            string: 'string',
-            text: 'string',
-            time: 'string'
-          }.freeze
-        end
-      end
-    end
-  end
-end

data/lib/active_record/dbt/data_type/dwh_platform/bigquery.rb

@@ -1,26 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecord
-  module Dbt
-    module DataType
-      module DwhPlatform
-        module Bigquery
-          # [Data types  |  BigQuery  |  Google Cloud](https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#data_type_list)
-          RUBY_TO_BIGQUERY_TYPES = {
-            binary: 'bytes',
-            boolean: 'bool',
-            date: 'date',
-            datetime: 'datetime',
-            decimal: 'int64',
-            float: 'float64',
-            integer: 'int64',
-            json: 'json',
-            string: 'string',
-            text: 'string',
-            time: 'time'
-          }.freeze
-        end
-      end
-    end
-  end
-end

data/lib/active_record/dbt/data_type/dwh_platform/postgre_sql.rb

@@ -1,27 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecord
-  module Dbt
-    module DataType
-      module DwhPlatform
-        module PostgreSql
-          # TODO: I have not tried it. I don't know if this is the correct data_type.
-          # [PostgreSQL: Documentation: 16: Chapter 8. Data Types](https://www.postgresql.org/docs/current/datatype.html)
-          RUBY_TO_POSTGRES_TYPES = {
-            binary: 'bytea',
-            boolean: 'boolean',
-            date: 'date',
-            datetime: 'timestamp',
-            decimal: 'bigint',
-            float: 'double precision',
-            integer: 'bigint',
-            json: 'json',
-            string: 'text',
-            text: 'text',
-            time: 'time'
-          }.freeze
-        end
-      end
-    end
-  end
-end

data/lib/active_record/dbt/data_type/dwh_platform/redshift.rb

@@ -1,27 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecord
-  module Dbt
-    module DataType
-      module DwhPlatform
-        module Redshift
-          # TODO: I have not tried it. I don't know if this is the correct data_type.
-          # [Data types - Amazon Redshift](https://docs.aws.amazon.com/redshift/latest/dg/c_Supported_data_types.html)
-          RUBY_TO_REDSHIFT_TYPES = {
-            binary: 'varbyte',
-            boolean: 'bool',
-            date: 'date',
-            datetime: 'timestamp',
-            decimal: 'decimal',
-            float: 'double precision',
-            integer: 'integer',
-            json: 'super',
-            string: 'varchar',
-            text: 'varchar',
-            time: 'time'
-          }.freeze
-        end
-      end
-    end
-  end
-end

data/lib/active_record/dbt/data_type/dwh_platform/snowflake.rb

@@ -1,27 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecord
-  module Dbt
-    module DataType
-      module DwhPlatform
-        module Snowflake
-          # TODO: I have not tried it. I don't know if this is the correct data_type.
-          # [Summary of data types | Snowflake Documentation](https://docs.snowflake.com/en/sql-reference/intro-summary-data-types)
-          RUBY_TO_SNOWFLAKE_TYPES = {
-            binary: 'binary',
-            boolean: 'boolean',
-            date: 'date',
-            datetime: 'datetime',
-            decimal: 'decimal',
-            float: 'float',
-            integer: 'integer',
-            json: 'variant',
-            string: 'varchar',
-            text: 'varchar',
-            time: 'time'
-          }.freeze
-        end
-      end
-    end
-  end
-end