From 695d7a9eccfbba75d344886c962b36d656a619fa Mon Sep 17 00:00:00 2001
From: Martin Edenhofer <me@edenhofer.de>
Date: Sat, 23 May 2015 03:53:43 +0200
Subject: [PATCH] Added signature support to auto wizard.

---
 contrib/auto_wizard_example.json |  7 +++++
 lib/auto_wizard.rb               | 46 ++++++++++++++------------------
 2 files changed, 27 insertions(+), 26 deletions(-)

diff --git a/contrib/auto_wizard_example.json b/contrib/auto_wizard_example.json
index 0987e2fbe..384078c57 100644
--- a/contrib/auto_wizard_example.json
+++ b/contrib/auto_wizard_example.json
@@ -35,6 +35,13 @@
       "name": "Demo Organization"
     }
   ],
+  "Signatures": [
+    {
+      "id": 1,
+      "name": "default",
+      "body": "\n  #{user.firstname} #{user.lastname}\n\n--\n Demo Organization\n--"
+    }
+  ],
   "Channels": [
     {
       "area": "Email::Inbound",
diff --git a/lib/auto_wizard.rb b/lib/auto_wizard.rb
index 6a1ddcdf8..3686625ed 100644
--- a/lib/auto_wizard.rb
+++ b/lib/auto_wizard.rb
@@ -92,39 +92,33 @@ returns
       }
     end
 
-    # create EmailAddresses
-    if auto_wizard_hash['EmailAddresses']
-      auto_wizard_hash['EmailAddresses'].each { |email_address_data|
-
-        email_address_data_symbolized = email_address_data.symbolize_keys.merge(
+    # create EmailAddresses/Channels/Signatures
+    model_map = {
+      'EmailAddresses' => 'EmailAddress',
+      'Channels'       => 'Channel',
+      'Signatures'     => 'Signature',
+    }
+    model_map.each {|map_name, model|
+      next if !auto_wizard_hash[map_name]
+      auto_wizard_hash[map_name].each {|data|
+        data_symbolized = data.symbolize_keys.merge(
           {
             updated_by_id: admin_user.id,
             created_by_id: admin_user.id
           }
         )
 
-        EmailAddress.create_or_update(
-          email_address_data_symbolized
-        )
+        if data_symbolized[:id] || data_symbolized[:name]
+          Kernel.const_get(model).create_or_update(
+            data_symbolized
+          )
+        else
+          Kernel.const_get(model).create(
+            data_symbolized
+          )
+        end
       }
-    end
-
-    # create Channels
-    if auto_wizard_hash['Channels']
-      auto_wizard_hash['Channels'].each { |channel_data|
-
-        channel_data_symbolized = channel_data.symbolize_keys.merge(
-          {
-            updated_by_id: admin_user.id,
-            created_by_id: admin_user.id
-          }
-        )
-
-        Channel.create(
-          channel_data_symbolized
-        )
-      }
-    end
+    }
 
     # remove auto wizard file
     FileUtils.rm auto_wizard_file_location