From 425328b4af914cd18feefc9d0adc25ef07995d32 Mon Sep 17 00:00:00 2001
From: Hannah Wolfe <erisds@gmail.com>
Date: Tue, 17 Sep 2013 16:55:55 +0100
Subject: [PATCH] Standard Importer

closes #734

- use models instead of knex so that we know we get correct data
- preprocesses post_tags, adds the right tags back to the posts

TODO

- deduplicate posts (if they have the same title and slug?)
---
 core/server/data/import/000.js | 158 +++++++++++++++++++++++++++++++++
 1 file changed, 158 insertions(+)
 create mode 100644 core/server/data/import/000.js

diff --git a/core/server/data/import/000.js b/core/server/data/import/000.js
new file mode 100644
index 0000000000..0751495246
--- /dev/null
+++ b/core/server/data/import/000.js
@@ -0,0 +1,158 @@
+var when = require("when"),
+    _ = require("underscore"),
+    models = require('../../models'),
+    errors = require('../../errorHandling'),
+    Importer000;
+
+
+Importer000 = function () {
+    _.bindAll(this, "basicImport");
+
+    this.version = "000";
+
+    this.importFrom = {
+        "000": this.basicImport
+    };
+};
+
+Importer000.prototype.importData = function (data) {
+    return this.canImport(data)
+        .then(function (importerFunc) {
+            return importerFunc(data);
+        }, function (reason) {
+            return when.reject(reason);
+        });
+};
+
+Importer000.prototype.canImport = function (data) {
+    if (data.meta && data.meta.version && this.importFrom[data.meta.version]) {
+        return when.resolve(this.importFrom[data.meta.version]);
+    }
+
+    return when.reject("Unsupported version of data");
+};
+
+
+function stripProperties(properties, data) {
+    _.each(data, function (obj) {
+        _.each(properties, function (property) {
+            delete obj[property];
+        });
+    });
+    return data;
+}
+
+function preProcessPostTags(tableData) {
+    var postTags,
+        postsWithTags = {};
+
+
+    postTags = tableData.posts_tags;
+    _.each(postTags, function (post_tag) {
+        if (!postsWithTags.hasOwnProperty(post_tag.post_id)) {
+            postsWithTags[post_tag.post_id] = [];
+        }
+        postsWithTags[post_tag.post_id].push(post_tag.tag_id);
+    });
+
+    _.each(postsWithTags, function (tag_ids, post_id) {
+        var post, tags;
+        post = _.find(tableData.posts, function (post) {
+            return post.id === parseInt(post_id, 10);
+        });
+        if (post) {
+            tags = _.filter(tableData.tags, function (tag) {
+                return _.indexOf(tag_ids, tag.id) !== -1;
+            });
+            post.tags = [];
+            _.each(tags, function (tag) {
+                // names are unique.. this should get the right tags added
+                // as long as tags are added first;
+                post.tags.push({name: tag.name});
+            });
+        }
+    });
+
+    return tableData;
+}
+
+function importTags(ops, tableData) {
+    tableData = stripProperties(['id'], tableData);
+    _.each(tableData, function (tag) {
+        ops.push(models.Tag.add(tag));
+    });
+}
+
+function importPosts(ops, tableData) {
+    tableData = stripProperties(['id'], tableData);
+    _.each(tableData, function (post) {
+        ops.push(models.Post.add(post));
+    });
+}
+
+function importUsers(ops, tableData) {
+    tableData = stripProperties(['id'], tableData);
+    tableData[0].id = 1;
+    ops.push(models.User.edit(tableData[0]));
+}
+
+function importSettings(ops, tableData) {
+    // for settings we need to update individual settings, and insert any missing ones
+    // the one setting we MUST NOT update is the databaseVersion settings
+    var blackList = ['databaseVersion'];
+    tableData = stripProperties(['id'], tableData);
+    tableData = _.filter(tableData, function (data) {
+        return blackList.indexOf(data.key) === -1;
+    });
+
+    ops.push(models.Settings.edit(tableData));
+}
+
+// No data needs modifying, we just import whatever tables are available
+Importer000.prototype.basicImport = function (data) {
+    var ops = [],
+        tableData = data.data;
+
+    // Do any pre-processing of relationships (we can't depend on ids)
+    if (tableData.posts_tags && tableData.posts && tableData.tags) {
+        tableData = preProcessPostTags(tableData);
+    }
+
+    // Import things in the right order:
+    if (tableData.tags && tableData.tags.length) {
+        importTags(ops, tableData.tags);
+    }
+
+    if (tableData.posts && tableData.posts.length) {
+        importPosts(ops, tableData.posts);
+    }
+
+    if (tableData.users && tableData.users.length) {
+        importUsers(ops, tableData.users);
+    }
+
+    if (tableData.settings && tableData.settings.length) {
+        importSettings(ops, tableData.settings);
+    }
+
+    /** do nothing with these tables, the data shouldn't have changed from the fixtures
+     *   permissions
+     *   roles
+     *   permissions_roles
+     *   permissions_users
+     *   roles_users
+     */
+
+    return when.all(ops).then(function (results) {
+        return when.resolve(results);
+    }, function (err) {
+        return when.reject("Error importing data: " + err.message || err, err.stack);
+    });
+};
+
+module.exports = {
+    Importer000: Importer000,
+    importData: function (data) {
+        return new Importer000().importData(data);
+    }
+};
\ No newline at end of file