Allow theme to be added by name && use template when loading wizard for controller

controllers/wizard.rb

@@ -7,15 +7,15 @@ class CustomWizard::WizardController < ::ApplicationController
   helper_method :theme_key
 
   def wizard
-    PluginStore.get('custom_wizard', params[:wizard_id].underscore)
+    CustomWizard::Template.new(PluginStore.get('custom_wizard', params[:wizard_id].underscore))
   end
 
   def wizard_page_title
-    wizard ? (wizard['name'] || wizard['id']) : I18n.t('wizard.custom_title')
+    wizard ? (wizard.name || wizard.id) : I18n.t('wizard.custom_title')
   end
 
   def theme_key
-    wizard ? wizard['theme_key'] : nil
+    wizard ? wizard.theme_key : nil
   end
 
   def index

lib/template.rb

@@ -29,5 +29,10 @@ class CustomWizard::Template
     @after_time_scheduled = data['after_time_scheduled']
     @required = data['required'] || false
     @theme_key = data['theme_key']
+
+    if data['theme']
+      theme = Theme.find_by(name: data['theme'])
+      @theme_key = theme.key if theme
+    end
   end
 end