referral_box 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b232d0c39b152774ce3e521e1bfd5b49e12f4403b913391fdbdc0ef91684c07a
-  data.tar.gz: 8623def044d02bcfe7aaac01382c25bcac0a78fc87c9cda64f24cba20825cd4f
+  metadata.gz: d8ba983787c512d738d87748d8aef25955fc322c491bdabfd47ad8fda6980984
+  data.tar.gz: 29a2e05f19614deee0edde3005fc56788efeea814c78bdd0eed3eee37953f054
 SHA512:
-  metadata.gz: 15df1bb9559552c3648c30b667c73649b3af655296eb2206e4519779685d0d56733580d4f7d65395245911c6799985a2d432f74ade66652eea35f7f9c65c0d98
-  data.tar.gz: 1831e708802b59d48cfa08611c0fcc444a96307f42d5e3cdd840052b855449d503e978931b6605d35a406cd0f987cb420e91d4bf7a04935c5881338113258f6f
+  metadata.gz: bddb9b69fa3af2ad73bbd3ff32ca2de283611f75853731bf704bcf98b976c5b61da069ff1ce4364fc5382c323fe63aefb9b4b45c5131e5fc556566dfdd3e0f5f
+  data.tar.gz: 76dd52bfcbff503d155135d408bb81d4ec1095c3021a642f42d7bbf2abb352ffcdc04519478765e7f0bfcc1291886dcda4b18113eb5054b6c19014cb224f1c54

data/CHANGELOG.md

@@ -7,6 +7,19 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.1.8] - 2025-01-XX
+
+### Added
+- **Interactive Generator**: Generator now asks for model name and confirms the choice
+- **Model Detection**: Automatically detects existing models in the app
+- **Safe Migrations**: Migration now checks if columns exist before adding them
+- **Duplicate Prevention**: Prevents adding duplicate columns and methods
+
+### Fixed
+- **Migration Errors**: Fixed "duplicate column name" errors by adding existence checks
+- **Generator Flow**: Improved user experience with interactive prompts
+- **Model Flexibility**: Better support for different model names (User, Customer, Account, etc.)
+
 ## [0.1.7] - 2025-01-XX
 
 ### Added

data/lib/generators/referral_box/install/install_generator.rb

@@ -11,37 +11,66 @@ module ReferralBox
         template "initializer.rb", "config/initializers/referral_box.rb"
       end
 
-      def add_model_migration
-        # Read the configuration to determine the model class name
-        model_class = get_model_class_name
+      def ask_for_model_name
+        puts "\n" + "="*60
+        puts "ReferralBox Model Configuration"
+        puts "="*60
+        
+        # Detect existing models
+        model_files = Dir.glob("app/models/*.rb")
+        existing_models = model_files.map { |f| File.basename(f, '.rb').classify }
+        
+        if existing_models.any?
+          puts "\nExisting models found: #{existing_models.join(', ')}"
+        end
+        
+        @model_class = ask("What is your user model class name? (e.g., User, Customer, Account, Member):", default: "User")
+        
+        # Confirm the choice
+        puts "\nYou selected: #{@model_class}"
+        confirm = ask("Is this correct? (y/n):", default: "y")
+        
+        unless confirm.downcase.start_with?('y')
+          puts "Please run the generator again with the correct model name."
+          exit
+        end
         
-        if model_class.present?
-          generate_migration = "AddReferralBoxTo#{model_class.pluralize}"
-          migration_content = generate_migration_content(model_class)
+        # Update the initializer with the selected model
+        update_initializer_with_model(@model_class)
+      end
+
+      def add_model_migration
+        if @model_class.present?
+          generate_migration = "AddReferralBoxTo#{@model_class.pluralize}"
+          migration_content = generate_migration_content(@model_class)
           
           create_file "db/migrate/#{Time.current.strftime('%Y%m%d%H%M%S')}_#{generate_migration.underscore}.rb", migration_content
           
-          puts "Migration generated for #{model_class} model!"
+          puts "\nMigration generated for #{@model_class} model!"
           puts "Run 'rails db:migrate' to apply it."
         else
-          puts "Warning: Could not determine model class name from configuration."
+          puts "Warning: Could not determine model class name."
           puts "Please manually create migration for your model."
         end
       end
 
       def add_model_methods
-        model_class = get_model_class_name
-        
-        if model_class.present?
-          model_file = "app/models/#{model_class.underscore}.rb"
+        if @model_class.present?
+          model_file = "app/models/#{@model_class.underscore}.rb"
           
           if File.exist?(model_file)
-            inject_into_file model_file, after: "class #{model_class} < ApplicationRecord" do
-              "\n  " + generate_model_methods(model_class)
+            # Check if methods already exist
+            content = File.read(model_file)
+            if content.include?('has_many :referral_box_transactions')
+              puts "ReferralBox methods already exist in #{@model_class} model."
+            else
+              inject_into_file model_file, after: "class #{@model_class} < ApplicationRecord" do
+                "\n  " + generate_model_methods(@model_class)
+              end
+              puts "Added ReferralBox methods to #{@model_class} model."
             end
-            puts "Added ReferralBox methods to #{model_class} model."
           else
-            puts "Warning: #{model_file} not found. Please manually add ReferralBox methods to your #{model_class} model."
+            puts "Warning: #{model_file} not found. Please manually add ReferralBox methods to your #{@model_class} model."
           end
         end
       end
@@ -60,19 +89,19 @@ module ReferralBox
 
       private
 
-      def get_model_class_name
-        # Try to read from existing initializer first
+      def update_initializer_with_model(model_class)
         initializer_path = "config/initializers/referral_box.rb"
         
         if File.exist?(initializer_path)
           content = File.read(initializer_path)
-          if match = content.match(/config\.reference_class_name\s*=\s*['"]([^'"]+)['"]/)
-            return match[1]
-          end
+          updated_content = content.gsub(
+            /config\.reference_class_name\s*=\s*['"][^'"]*['"]/,
+            "config.reference_class_name = '#{model_class}'"
+          )
+          
+          File.write(initializer_path, updated_content)
+          puts "Updated initializer with #{model_class} model."
         end
-        
-        # Default to 'User' if not found
-        'User'
       end
 
       def generate_migration_content(model_class)
@@ -81,11 +110,25 @@ module ReferralBox
 
           class AddReferralBoxTo#{model_class.pluralize} < ActiveRecord::Migration[#{get_rails_version}]
             def change
-              add_column :#{model_class.underscore.pluralize}, :referral_code, :string
-              add_column :#{model_class.underscore.pluralize}, :tier, :string
-              add_reference :#{model_class.underscore.pluralize}, :referrer, null: true, foreign_key: { to_table: :#{model_class.underscore.pluralize} }
+              # Add referral_code column if it doesn't exist
+              unless column_exists?(:#{model_class.underscore.pluralize}, :referral_code)
+                add_column :#{model_class.underscore.pluralize}, :referral_code, :string
+              end
+              
+              # Add tier column if it doesn't exist
+              unless column_exists?(:#{model_class.underscore.pluralize}, :tier)
+                add_column :#{model_class.underscore.pluralize}, :tier, :string
+              end
+              
+              # Add referrer reference if it doesn't exist
+              unless column_exists?(:#{model_class.underscore.pluralize}, :referrer_id)
+                add_reference :#{model_class.underscore.pluralize}, :referrer, null: true, foreign_key: { to_table: :#{model_class.underscore.pluralize} }
+              end
               
-              add_index :#{model_class.underscore.pluralize}, :referral_code, unique: true
+              # Add unique index on referral_code if it doesn't exist
+              unless index_exists?(:#{model_class.underscore.pluralize}, :referral_code)
+                add_index :#{model_class.underscore.pluralize}, :referral_code, unique: true
+              end
             end
           end
         MIGRATION

data/lib/referral_box/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ReferralBox
-  VERSION = "0.1.7"
+  VERSION = "0.1.8"
 end 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: referral_box
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - Kapil Dev Pal