activerecord-dbt 0.1.0 → 0.3.0

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.1.0'
+    VERSION = '0.3.0'
   end
 end

data/lib/active_record/dbt.rb

@@ -11,7 +11,5 @@ module ActiveRecord
 
       config
     end
-
-    class Error < StandardError; end
   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/initializer/templates/dbt.rb

@@ -5,6 +5,7 @@ require 'active_record/dbt'
 ActiveRecord::Dbt.configure do |c|
   c.config_directory_path = 'lib/dbt'
   c.export_directory_path = 'doc/dbt'
+  c.dwh_platform = 'bigquery'
   c.data_sync_delayed = false
   c.used_dbt_package_names = []
 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.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - yamotech
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-07-20 00:00:00.000000000 Z
+date: 2024-08-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -72,9 +72,13 @@ files:
 - lib/active_record/dbt/column/testable/unique_testable.rb
 - 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/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
 - lib/active_record/dbt/factory/columns_factory.rb
@@ -82,10 +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
@@ -94,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