From d837a7c9bc1600cc4e81b8cda558a4ef5f4ba09d Mon Sep 17 00:00:00 2001
From: Angus McLeod <angus@mcleod.org.au>
Date: Sun, 4 Feb 2018 15:52:53 +0800
Subject: [PATCH] Add custom user fields to actions

---
 .../components/wizard-custom-action.hbs       |  3 +-
 lib/builder.rb                                | 33 ++++++++++++++++---
 2 files changed, 31 insertions(+), 5 deletions(-)

diff --git a/assets/javascripts/discourse/templates/components/wizard-custom-action.hbs b/assets/javascripts/discourse/templates/components/wizard-custom-action.hbs
index dec404bc..d8130b8f 100644
--- a/assets/javascripts/discourse/templates/components/wizard-custom-action.hbs
+++ b/assets/javascripts/discourse/templates/components/wizard-custom-action.hbs
@@ -179,6 +179,7 @@
     {{wizard-custom-input inputs=action.profile_updates
                           valueContent=profileFields
                           keyContent=availableFields
-                          noneValue='admin.wizard.action.update_profile.profile_field'}}
+                          noneValue='admin.wizard.action.update_profile.profile_field'
+                          optionalCustom=true}}      
   </div>
 {{/if}}
diff --git a/lib/builder.rb b/lib/builder.rb
index 70d24314..54a51898 100644
--- a/lib/builder.rb
+++ b/lib/builder.rb
@@ -80,7 +80,11 @@ class CustomWizard::Builder
                   profile_actions.each do |action|
                     if update = action['profile_updates'].select { |u| u['key'] === f['id'] }.first
                       attribute = update['value']
-                      if UserProfile.column_names.include? attribute
+                      custom_field = update['value_custom']
+                      if custom_field
+                        name = custom_field.split('.').try(:last) || nil
+                        params[:value] = UserCustomField.where(user_id: @wizard.user.id, name: name).pluck(:value)
+                      elsif UserProfile.column_names.include? attribute
                         params[:value] = UserProfile.find_by(user_id: @wizard.user.id).send(attribute)
                       elsif User.column_names.include? attribute
                         params[:value] = User.find(@wizard.user.id).send(attribute)
@@ -285,12 +289,33 @@ class CustomWizard::Builder
                 end
 
                 if a['type'] === 'update_profile' && a['profile_updates'].length && data
-                  user_updater = UserUpdater.new(user, user)
                   attributes = {}
+                  custom_fields = {}
+
                   a['profile_updates'].each do |pu|
-                    attributes[pu['value'].to_sym] = data[pu['key']]
+                    value = pu['value']
+                    custom_value = pu['value_custom']
+                    key = pu['key']
+
+                    if custom_value
+                      name = custom_value.split('.').try(:last) || nil
+                      custom_fields[name] = data[key]
+                    else
+                      attributes[value.to_sym] = data[key]
+                    end
+                  end
+
+                  if custom_fields.present?
+                    custom_fields.each do |k, v|
+                      user.custom_fields[k] = v
+                    end
+                    user.save_custom_fields(true)
+                  end
+
+                  if attributes.present?
+                    user_updater = UserUpdater.new(user, user)
+                    user_updater.update(attributes)
                   end
-                  user_updater.update(attributes) if attributes.present?
                 end
               end
             end