feat: use Rails commands to replace rake tasks with business logic + write 2 new commands to remove/rename section types

Author: did

app/services/maglev/remove_section_type.rb

@@ -36,6 +36,8 @@ def remove_resource_sections(resource)
     end
 
     def remove_sections_of_type(sections, type)
+      return 0 if sections.blank?
+
       original_size = sections.size
       sections.reject! { |section| section['type'] == type }
       original_size - sections.size

app/services/maglev/rename_section_type.rb

@@ -43,6 +43,8 @@ def rename_resource_sections(resource)
     end
 
     def change_section_type(sections, old_type, new_type)
+      return if sections.blank?
+
       sections.each do |section|
         section['type'] = new_type if section['type'] == old_type
       end

lib/commands/maglev/change_site_locales_command.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require 'rails'
+
+module Maglev
+  class ChangeSiteLocalesCommand < Rails::Command::Base
+    desc 'change_site_locales [LOCALES]', 'Change site locales (format: label:prefix)'
+
+    def self.banner
+      'bin/rails maglev:change_site_locales [label:prefix label2:prefix2 ...]'
+    end
+
+    def perform(*locale_args)
+      require File.expand_path('config/environment', Rails.root)
+
+      site = fetch_site
+      return if site.blank?
+
+      locales = build_locales(locale_args)
+      return if locales.empty?
+
+      update_site_locales(site, locales)
+    end
+
+    private
+
+    def fetch_site
+      Maglev::Site.first.tap do |site|
+        say("[Error] You don't seem to have an existing site. 🤔", :red) unless site
+      end
+    end
+
+    def build_locales(locale_args)
+      build_custom_locales(locale_args).tap do |locales|
+        validate_locales(locales)
+      end
+    end
+
+    def build_custom_locales(locale_args)
+      (locale_args || []).map do |arg|
+        label, prefix = arg.split(':')
+        Maglev::Site::Locale.new(label: label, prefix: prefix)
+      end
+    end
+
+    def validate_locales(locales)
+      return true if !locales.empty? && locales.all?(&:valid?)
+
+      say("[Error] make sure your locales follow the 'label:prefix' pattern. 🤓", :red)
+      false
+    end
+
+    def update_site_locales(site, locales)
+      Maglev::ChangeSiteLocales.call(
+        site: site,
+        locales: locales
+      )
+      say('Success! 🎉🎉🎉', :green)
+    end
+  end
+end

lib/commands/maglev/create_site_command.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'rails'
+
+module Maglev
+  class CreateSiteCommand < Rails::Command::Base
+    desc 'create_site', 'Create your site'
+
+    def self.banner
+      'bin/rails maglev:create_site'
+    end
+
+    def perform
+      require File.expand_path('config/environment', Rails.root)
+
+      if Maglev::Site.exists?
+        say '🤔 You already have a site. 🤔', :yellow
+        return
+      end
+
+      Maglev::GenerateSite.call(
+        theme: Maglev.local_themes.first
+      )
+
+      say '🎉 Your site has been created with success!'
+    end
+  end
+end

lib/commands/maglev/sections/remove_command.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require 'rails'
+
+module Maglev
+  module Sections
+    class RemoveCommand < Rails::Command::Base
+      desc 'remove TYPE', 'Remove a section type across the site and its pages'
+
+      def self.banner
+        'bin/rails maglev:sections:remove TYPE'
+      end
+
+      def perform(type)
+        require File.expand_path('config/environment', Rails.root)
+
+        site = fetch_site
+
+        return if site.blank?
+
+        removed_count = Maglev::RemoveSectionType.call(
+          site: site,
+          type: type
+        )
+
+        display_final_message(removed_count, type)
+      end
+
+      private
+
+      def fetch_site
+        Maglev::Site.first.tap do |site|
+          say("[Error] You don't seem to have an existing site. 🤔", :red) unless site
+        end
+      end
+
+      def display_final_message(removed_count, type)
+        if removed_count.zero?
+          say "No section of type '#{type}' found 🤔", :yellow
+          return
+        end
+
+        say "Successfully removed #{removed_count} #{'section'.pluralize(removed_count)} of type '#{type}' 🎉",
+            :green
+      end
+    end
+  end
+end

lib/commands/maglev/sections/rename_command.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module Maglev
+  module Sections
+    class RenameCommand < Rails::Command::Base
+      desc 'rename OLD_TYPE NEW_TYPE', 'Rename a section type across the site and its pages'
+
+      def self.banner
+        'bin/rails maglev:sections:rename OLD_TYPE NEW_TYPE'
+      end
+
+      def perform(old_type, new_type)
+        require File.expand_path('config/environment', Rails.root)
+
+        site = fetch_site
+        theme = fetch_theme
+
+        return if site.blank? || theme.blank?
+
+        rename_sections(site, theme, old_type, new_type)
+
+        say "Successfully renamed all '#{old_type}' sections to '#{new_type}' 🎉", :green
+      end
+
+      private
+
+      def fetch_site
+        Maglev::Site.first.tap do |site|
+          say("[Error] You don't seem to have an existing site. 🤔", :red) unless site
+        end
+      end
+
+      def fetch_theme
+        Maglev.local_themes&.first.tap do |theme|
+          say('[Error] No theme found. 🤔', :red) unless theme
+        end
+      end
+
+      def rename_sections(site, theme, old_type, new_type)
+        Maglev::RenameSectionType.call(
+          site: site,
+          theme: theme,
+          old_type: old_type,
+          new_type: new_type
+        )
+      end
+    end
+  end
+end

lib/maglev.rb

@@ -7,6 +7,7 @@
 require_relative 'maglev/i18n'
 require_relative 'maglev/preview_constraint'
 require_relative 'maglev/reserved_paths'
+require_relative 'maglev/commands_railtie'
 
 require 'injectable'
 require 'jbuilder'

lib/maglev/commands_railtie.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Maglev
+  class CommandsRailtie < Rails::Railtie
+    railtie_name :maglev_commands
+
+    initializer 'maglev.commands' do
+      # Require all command files
+      Dir[File.expand_path('../commands/**/*_command.rb', __dir__)].sort.each do |file|
+        require file
+      end
+    end
+  end
+end

lib/maglev/engine.rb

@@ -39,6 +39,15 @@ class Engine < ::Rails::Engine
       Rails.application.config.i18n.load_path += Dir["#{config.root}/config/locales/**/*.yml"]
     end
 
+    # initializer 'maglev.load_commands' do
+    #   if defined?(Rails::Command)
+    #     Dir.glob(File.join(__dir__, '../commands/**/*.rb')).each do |file|
+    #       puts "require #{file}"
+    #       require file
+    #     end
+    #   end
+    # end
+
     delegate :vite_ruby, to: :class
 
     def self.vite_ruby