discourse-custom-wizard/lib/builder.rb

class CustomWizard::Builder

  attr_accessor :wizard, :updater, :submission

  def initialize(user, wizard_id)
    data = PluginStore.get('custom_wizard', wizard_id)
    @template = CustomWizard::Template.new(data)
    @wizard = CustomWizard::Wizard.new(user,
      id: wizard_id,
      save_submissions: data['save_submissions'],
      multiple_submissions: data['multiple_submissions'],
      background: data["background"],
      name: data["name"]
    )
  end

  def self.sorted_handlers
    @sorted_handlers ||= []
  end

  def self.step_handlers
    sorted_handlers.map { |h| { wizard_id: h[:wizard_id], block: h[:block] } }
  end

  def self.add_step_handler(priority = 0, wizard_id, &block)
    sorted_handlers << { priority: priority, wizard_id: wizard_id, block: block }
    @sorted_handlers.sort_by! { |h| -h[:priority] }
  end

  def build
    unless (@wizard.completed? && !@template.respond_to?(:multiple_submissions)) ||
           !@template.steps
      @template.steps.each do |s|
        @wizard.append_step(s['id']) do |step|
          step.title = s['title'] if s['title']
          step.description = s['description'] if s['description']
          step.banner = s['banner'] if s['banner']
          step.key = s['key'] if s['key']

          if s['fields'] && s['fields'].length
            s['fields'].each do |f|
              params = {
                id: f['id'],
                type: f['type'],
                required: f['required']
              }

              params[:label] = f['label'] if f['label']
              params[:description] = f['description'] if f['description']
              params[:key] = f['key'] if f['key']

              submissions = Array.wrap(PluginStore.get("custom_wizard_submissions", @wizard.id))
              if submissions.last && submissions.last['completed'] === false
                @submission = submissions.last
                params[:value] = @submission[f['id']] if @submission[f['id']]
              end

              field = step.add_field(params)

              if f['type'] === 'dropdown'
                if f['choices'] && f['choices'].length > 0
                  f['choices'].each do |c|
                    field.add_choice(c['value'], label: c['label'])
                  end
                elsif f['choices_key'] && f['choices_key'].length > 0
                  choices = I18n.t(f['choices_key'])
                  if choices.is_a?(Hash)
                    choices.each do |k, v|
                      field.add_choice(k, label: v)
                    end
                  end
                elsif f['choices_preset'] && f['choices_preset'].length > 0
                  objects = []

                  if f['choices_preset'] === 'categories'
                    objects = Site.new(Guardian.new(@wizard.user)).categories
                  end

                  if f['choices_filters'] && f['choices_filters'].length > 0
                    f['choices_filters'].each do |f|
                      objects.reject! { |o| o[f['key']] != f['value'] }
                    end
                  end

                  if objects.length > 0
                    objects.each do |o|
                      field.add_choice(o.id, label: o.name)
                    end
                  end
                end
              end
            end
          end

          step.on_update do |updater|
            @updater = updater
            input = updater.fields
            user = @wizard.user

            if s['fields'] && s['fields'].length
              s['fields'].each do |f|
                value = input[f['id']]
                min_length = f['min_length']
                if min_length && value.is_a?(String) && value.length < min_length.to_i
                  label = f['label'] || I18n.t("#{f['key']}.label")
                  updater.errors.add(f['id'].to_s, I18n.t('wizard.field.too_short', label: label, min: min_length.to_i))
                end
              end
            end

            next if updater.errors.any?

            CustomWizard::Builder.step_handlers.each do |handler|
              if handler[:wizard_id] == @wizard.id
                handler[:block].call(self)
              end
            end

            next if updater.errors.any?

            if s['actions'] && s['actions'].length
              s['actions'].each do |a|
                if a['type'] === 'create_topic'
                  title = input[a['title']]
                  post = input[a['post']]

                  if title && post
                    params = {
                      title: title,
                      raw: post,
                      skip_validations: true
                    }
                    params[:category] = a['category_id'] if a['category_id']
                    params[:featured_link] = input[a['featured_link']] if input[a['featured_link']]

                    creator = PostCreator.new(user, params)
                    post = creator.create

                    if creator.errors.present?
                      updater.errors.add(:create_topic, creator.errors.full_messages.join(" "))
                    else
                      updater.result = { topic_id: post.topic.id }
                    end
                  end
                end

                if a['type'] === 'send_message'
                  title = input[a['title']]
                  post = input[a['post']]

                  if title && post
                    creator = PostCreator.new(user,
                                    title: title,
                                    raw: post,
                                    archetype: Archetype.private_message,
                                    target_usernames: a['username'])

                    post = creator.create

                    if creator.errors.present?
                      updater.errors.add(:send_message, creator.errors.full_messages.join(" "))
                    else
                      updater.result = { topic_id: post.topic.id }
                    end
                  end
                end

                if a['type'] === 'update_profile' && a['profile_updates'].length
                  updater = UserUpdater.new(user, user)
                  attributes = a['profile_updates'].map do |pu|
                    { pu['profile_field'].to_sym => input[pu['wizard_field']] }
                  end
                  updater.update(attributes)
                end
              end
            end

            if @wizard.save_submissions && updater.errors.empty?
              store_key = @wizard.id
              submissions = Array.wrap(PluginStore.get("custom_wizard_submissions", store_key))
              submission = {}

              if submissions.last && submissions.last['completed'] === false
                submission = submissions.last
                submissions.pop(1)
              end

              submission['user_id'] = @wizard.user.id
              submission['completed'] = updater.step.next.nil?

              if input
                input.each do |key, value|
                  submission[key] = value
                end
              end

              submissions.push(submission)
              PluginStore.set('custom_wizard_submissions', store_key, submissions)
            end
          end
        end
      end
    end

    puts "BUILDER: #{@wizard.respond_to?(:multiple_submissions)}"

    @wizard
  end
end