feat: new reset section command (#180)

* fix: the -h option wasn't working

* feat: new reset section content command

Author: did

app/services/maglev/reset_section_content.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+module Maglev
+  # Reset the content of a section type across a site and its pages
+  class ResetSectionContent
+    include Injectable
+
+    argument :site
+    argument :theme
+    argument :type
+
+    def call
+      validate_section_type!
+
+      @count = 0
+
+      reset_content
+
+      @count
+    end
+
+    private
+
+    def validate_section_type!
+      return if theme.sections.find(type)
+
+      raise Maglev::Errors::UnknownSection,
+            "Section type '#{type}' doesn't exist in the theme"
+    end
+
+    def reset_content
+      ActiveRecord::Base.transaction do
+        reset_resource_content(site)
+        site_pages.find_each do |page|
+          reset_resource_content(page)
+        end
+      end
+    end
+
+    def reset_resource_content(resource)
+      site.each_locale do
+        @count += reset_sections_content_of_type(resource.find_sections_by_type(type), type)
+      end
+
+      resource.save!
+    end
+
+    def reset_sections_content_of_type(sections, type)
+      return 0 if sections.blank?
+
+      sections.map do |section|
+        section.replace(build_default_content(type))
+      end.size
+    end
+
+    def build_default_content(type)
+      theme.sections.find(type).build_default_content
+    end
+
+    def site_pages
+      Maglev::Page
+    end
+  end
+end

lib/commands/maglev/change_site_locales_command.rb

@@ -6,7 +6,7 @@ module Maglev
   class ChangeSiteLocalesCommand < Rails::Command::Base
     desc 'change_site_locales [LOCALES]', 'Change site locales (format: label:prefix)'
 
-    def self.banner
+    def self.banner(_command = nil, *)
       'bin/rails maglev:change_site_locales [label:prefix label2:prefix2 ...]'
     end
 

lib/commands/maglev/create_site_command.rb

@@ -6,7 +6,7 @@ module Maglev
   class CreateSiteCommand < Rails::Command::Base
     desc 'create_site', 'Create your site'
 
-    def self.banner
+    def self.banner(_command = nil, *)
       'bin/rails maglev:create_site'
     end
 

lib/commands/maglev/sections/remove_command.rb

@@ -7,7 +7,7 @@ module Sections
     class RemoveCommand < Rails::Command::Base
       desc 'remove TYPE', 'Remove a section type across the site and its pages'
 
-      def self.banner
+      def self.banner(_command = nil, *)
         'bin/rails maglev:sections:remove TYPE'
       end
 

lib/commands/maglev/sections/rename_command.rb

@@ -1,17 +1,21 @@
 # frozen_string_literal: true
 
+require 'rails'
+
 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
+      def self.banner(_command = nil, *)
         'bin/rails maglev:sections:rename OLD_TYPE NEW_TYPE'
       end
 
-      def perform(old_type, new_type)
+      def perform(*args)
         require File.expand_path('config/environment', Rails.root)
 
+        old_type, new_type = args
+
         site = fetch_site
         theme = fetch_theme
 

lib/commands/maglev/sections/reset_command.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require 'rails'
+
+module Maglev
+  module Sections
+    class ResetCommand < Rails::Command::Base
+      desc 'reset TYPE', 'Reset the content of a section type across the site and its pages'
+
+      def self.banner(_command = nil, *)
+        'bin/rails maglev:sections:reset TYPE'
+      end
+
+      def perform(type)
+        require File.expand_path('config/environment', Rails.root)
+
+        site = fetch_site
+        theme = fetch_theme
+
+        return if site.blank? || theme.blank?
+
+        count = Maglev::ResetSectionContent.call(
+          site: site,
+          theme: theme,
+          type: type
+        )
+
+        display_final_message(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 fetch_theme
+        Maglev.local_themes&.first.tap do |theme|
+          say('[Error] No theme found. 🤔', :red) unless theme
+        end
+      end
+
+      def display_final_message(count, type)
+        if count.zero?
+          say "No section of type '#{type}' found 🤔", :yellow
+          return
+        end
+
+        say "Successfully reset content of #{count} #{'section'.pluralize(count)} of type '#{type}' 🎉",
+            :green
+      end
+    end
+  end
+end

spec/services/maglev/reset_section_content_spec.rb

@@ -0,0 +1,136 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+describe Maglev::ResetSectionContent do
+  subject { described_class.call(site: site, theme: theme, type: type) }
+
+  let(:site) { create(:site, locales: [{ label: 'English', prefix: 'en' }, { label: 'French', prefix: 'fr' }]) }
+  let(:theme) { build(:theme) }
+  let(:type) { 'jumbotron' }
+
+  describe 'when the site has sections of the specified type' do
+    before do
+      site.update!(sections_translations: {
+                     en: [{ type: 'jumbotron', settings: [{ id: 'title', value: 'Hello' }] }],
+                     fr: [{ type: 'jumbotron', settings: [{ id: 'title', value: 'Bonjour' }] }]
+                   })
+    end
+
+    it 'resets the section content in all locales and returns the count' do
+      # Two hero sections (one in each locale)
+      expect(subject).to eq(2)
+
+      # Check English locale
+      Maglev::I18n.with_locale(:en) do
+        expect(site.reload.sections.first['type']).to eq('jumbotron')
+        expect(site.sections.first['settings'].first['value']).to eq('Title')
+      end
+
+      # Check French locale
+      Maglev::I18n.with_locale(:fr) do
+        expect(site.reload.sections.first['type']).to eq('jumbotron')
+        expect(site.sections.first['settings'].first['value']).to eq('Title')
+      end
+    end
+  end
+
+  describe 'when the site has no sections of the specified type' do
+    before do
+      site.update!(sections_translations: {
+                     en: [{ type: 'footer', settings: [{ value: 'Footer' }] }],
+                     fr: [{ type: 'footer', settings: [{ value: 'Pied de page' }] }]
+                   })
+    end
+
+    it 'does not modify the sections and returns zero' do
+      expect(subject).to eq(0)
+
+      # Check English locale
+      Maglev::I18n.with_locale(:en) do
+        sections = site.reload.sections
+        expect(sections.length).to eq(1)
+        expect(sections.first['type']).to eq('footer')
+        expect(sections.first['settings'].first['value']).to eq('Footer')
+      end
+
+      # Check French locale
+      Maglev::I18n.with_locale(:fr) do
+        sections = site.reload.sections
+        expect(sections.length).to eq(1)
+        expect(sections.first['type']).to eq('footer')
+        expect(sections.first['settings'].first['value']).to eq('Pied de page')
+      end
+    end
+  end
+
+  describe 'when the section type does not exist in the theme' do
+    let(:type) { 'unknown' }
+
+    it 'raises an UnknownSection error' do
+      expect { subject }.to raise_error(
+        Maglev::Errors::UnknownSection,
+        "Section type 'unknown' doesn't exist in the theme"
+      )
+    end
+  end
+
+  describe 'when resetting sections across site and pages' do
+    let!(:page) { create(:page) }
+
+    before do
+      # Set up sections for both site and page
+      site.update!(sections_translations: {
+                     en: [{ type: 'jumbotron', settings: [{ id: 'title', value: 'Site Hero' }] }, { type: 'footer' }],
+                     fr: [{ type: 'jumbotron', settings: [{ id: 'title', value: 'Site Hero FR' }] }, { type: 'footer' }]
+                   })
+
+      page.update!(sections_translations: {
+                     en: [{ type: 'jumbotron', settings: [{ id: 'title', value: 'Page Hero' }] },
+                          { type: 'navigation' }],
+                     fr: [{ type: 'jumbotron', settings: [{ id: 'title', value: 'Page Hero FR' }] },
+                          { type: 'navigation' }]
+                   })
+    end
+
+    it 'resets the sections in both site and pages and returns total count' do
+      # Four hero sections total (one per locale in both site and page)
+      expect(subject).to eq(4)
+
+      # Check site sections
+      Maglev::I18n.with_locale(:en) do
+        sections = site.reload.sections
+        expect(sections.map { |s| s['type'] }).to eq(%w[jumbotron footer])
+        expect(sections.find { |s| s['type'] == 'jumbotron' }['settings'].first['value']).to eq('Title')
+      end
+
+      # Check page sections
+      Maglev::I18n.with_locale(:en) do
+        sections = page.reload.sections
+        expect(sections.map { |s| s['type'] }).to eq(%w[jumbotron navigation])
+        expect(sections.find { |s| s['type'] == 'jumbotron' }['settings'].first['value']).to eq('Title')
+      end
+    end
+  end
+
+  describe 'when sections are blank' do
+    before do
+      site.update!(sections_translations: {
+                     en: [],
+                     fr: []
+                   })
+    end
+
+    it 'returns zero and does not modify anything' do
+      expect(subject).to eq(0)
+
+      Maglev::I18n.with_locale(:en) do
+        expect(site.reload.sections).to be_empty
+      end
+
+      Maglev::I18n.with_locale(:fr) do
+        expect(site.reload.sections).to be_empty
+      end
+    end
+  end
+end