joli.js

var joli = {
    each: function(collection, iterator, bind) {
        var i, l, property;

        switch (joli.getType(collection)) {
            case "array":
                for ( i = 0, l = collection.length; i < l; i++) {
                    iterator.call(bind, collection[i], i);
                }
                break;
            case "object":
                for (property in collection) {
                    if (collection.hasOwnProperty(property)) {
                        iterator.call(bind, collection[property], property);
                    }
                }
                break;
        }
    },
    extend: function(baseClass, options) {
        var opt, prop;

        if (!this.options) {
            this.options = {};
        }

        this.parent = new baseClass(options);

        for (prop in this.parent) {
            this[prop] = this[prop] || this.parent[prop];
        }

        // copy base options over
        for (opt in this.parent.options) {
            this.options[opt] = this.options[opt] || this.parent.options[opt];
        }
    },
    getType: function(obj) {
        if (typeof obj === "undefined" || obj === null || (typeof obj === "number" && isNaN(obj))) {
            return false;
        } else if (obj.constructor === Array || (Array.isArray && Array.isArray(obj))) {
            return "array";
        } else {
            return typeof obj;
        }
    },
    jsonParse: function(json) {
        return JSON.parse(json);
    },
    merge: function() {
        var i, l, prop;
        var mergedObject = {};

        for (i = 0, l = arguments.length; i < l; i++) {
            var object = arguments[i];

            if (joli.getType(object) !== "object") {
                continue;
            }

            for (prop in object) {
                if (object.hasOwnProperty(prop)) {
                    var objectProp = object[prop], mergedProp = mergedObject[prop];

                    if (mergedProp && joli.getType(objectProp) === "object" && joli.getType(mergedProp) === "object") {
                        mergedObject[prop] = joli.merge(mergedProp, objectProp);
                    } else {
                        mergedObject[prop] = objectProp;
                    }
                }
            }
        }
        return mergedObject;
    },
    setOptions: function(options, defaults) {
        var opt;

        if (!options) {
            options = {};
        }

        if (!this.options) {
            this.options = {};
        }

        var mergedOptions = joli.merge(defaults, options);

        for (opt in defaults) {
            this.options[opt] = mergedOptions[opt];
        }
    },
    toQueryString: function(obj) {
        var queryStringComponents = [];

        if (!obj) {
            obj = {};
        }

        joli.each(obj, function(val, key) {
            var result = null;

            switch (joli.getType(val)) {
                case "object":
                    result = joli.toQueryString(val);
                    break;
                case "array":
                    result = joli.toQueryString(val);
                    break;
                default:
                    result = encodeURIComponent(val);
            }

            if (result) {
                queryStringComponents.push(key + '=' + result);
            }
        });
        return '[' + queryStringComponents.join("&") + ']';
    },
    typeValue: function(val) {
        if (!joli.getType(val)) {
            return "NULL";
        } else {
            if (joli.getType(val) === "string") {
                // escape single quotes and dollar signs.
                // quotes are escaped for SQLite
                val = val.replace(/'/g, "''");
                // dollar signs are escaped for use in calling str.replace in
                // JazzRecord.replaceAndClean()
                val = val.replace(/\$/g, "$$$$");
                val = "'" + val + "'";
            } else if (joli.getType(val) === "boolean") {
                if (val) {
                    return "1";
                } else {
                    return "0";
                }
            }
        }

        return val;
    }
};

/**
 * Connection
 */
joli.Connection = function(database) {
    this.dbname = database;
    this.database = Titanium.Database.open(this.dbname);
    this.database.execute('PRAGMA read_uncommitted=true');
};

joli.Connection.prototype = {
    execute: function(query) {
        // Titanium.API.log('info', query);
        return this.database.execute(query);
    },
    lastInsertRowId: function() {
        return this.database.lastInsertRowId;
    }
};

/**
 * Migration description
 */
joli.migration = function(options) {
    var defaults = {
        tableName: 'migration'
    };

    joli.setOptions.call(this, options, defaults);
    this.table = this.options.tableName;
};

joli.migration.prototype = {
    getVersion: function() {
        var q = new joli.query().select().from(this.table).order('version desc');
        var version = q.execute();

        if (version.length > 0) {
            return version[0].version;
        } else {
            q = new joli.query().insertInto(this.table).values({
                version: 0
            });
            q.execute();
            return 0;
        }
    },
    setVersion: function(version) {
        var q = new joli.query().update(this.table).set({
            version: version
        });
        q.execute();
    }
};

/**
 * Model description
 */
joli.model = function(options) {
    var defaults = {
        table: '',
        columns: {},
        objectMethods: {}
    };

    if (options.methods) {
        joli.each(options.methods, function(method, name) {
            this[name] = method;
        }, this);
    }

    joli.setOptions.call(this, options, defaults);
    this.table = this.options.table;

    if (!joli.models.has(this.table)) {
        joli.models.set(this.table, this);
    }
};

joli.model.prototype = {
    all: function(constraints) {
        var q = new joli.query().select().from(this.table);

        if (!constraints) {
            constraints = {};
        }

        if (constraints.where) {
            joli.each(constraints.where, function(value, field) {
                q.where(field, value);
            });
        }

        if (constraints.order) {
            q.order(constraints.order);
        }

        if (constraints.limit) {
            q.limit(constraints.limit);
        }

        return q.execute();
    },
    count: function(constraints) {
        var q = new joli.query().count().from(this.table);

        if (!constraints) {
            constraints = {};
        }

        if (constraints.where) {
            joli.each(constraints.where, function(value, field) {
                q.where(field, value);
            });
        }

        return parseInt(q.execute(), 10);
    },
    // no callbacks, more efficient
    deleteRecords: function(id) {
        var q = new joli.query().destroy().from(this.table);

        if (joli.getType(id) === 'number') {
            q.where('id = ?', id);
        } else if (joli.getType(id) === 'array') {
            q.whereIn('id', id);
        }

        return q.execute();
    },
    exists: function(id) {
        var count = new joli.query().count().from(this.table).where('id = ?', id).execute();
        return (count > 0);
    },
    findBy: function(field, value) {
        return new joli.query().select().from(this.table).where(field + ' = ?', value).execute();
    },
    findById: function(value) {
        return this.findBy('id', value);
    },
    findOneBy: function(field, value) {
        var result = new joli.query().select().from(this.table).where(field + ' = ?', value).limit(1).execute();

        if (result.length === 0) {
            return false;
        } else {
            return result[0];
        }
    },
    findOneById: function(value) {
        return this.findOneBy('id', value);
    },
    getColumns: function() {
        return this.options.columns;
    },
    newRecord: function(values) {
        if (!values) {
            values = {};
        }

        var data = {};

        joli.each(this.options.columns, function(colType, colName) {
            data[colName] = (values[colName] === undefined) ? null : values[colName];
        });
        var record = new joli.record(this).fromArray(data);

        record.isNew = function() {
            return true;
        };
        // add object methods
        if (this.options.objectMethods) {
            joli.each(this.options.objectMethods, function(method, name) {
                record[name] = method;
            });
        }

        return record;
    },
    save: function(data) {
        if (data.data.length === 0) {
            return;
        }

        var q = new joli.query();

        if (data.originalData) {
            // existing record
            q.update(this.table).set(data.data).where('id = ?', data.originalData.id);
        } else {
            // new record
            q.insertInto(this.table).values(data.data);
        }

        return q.execute();
    },
    truncate: function() {
        new joli.query().destroy().from(this.table).execute();
    }
};

joli.Models = function() {
    this.models = {};
    this.migration = new joli.migration({
        tableName: 'migration'
    });
};

joli.Models.prototype = {
    get: function(table) {
        return this.models[table];
    },
    has: function(table) {
        if (this.models[table]) {
            return true;
        } else {
            return false;
        }
    },
    initialize: function() {
        joli.each(this.models, function(model, modelName) {
            var columns = [];

            joli.each(model.options.columns, function(type, name) {
                columns.push(name + ' ' + type);
            });
            var query = 'CREATE TABLE IF NOT EXISTS ' + modelName + ' (' + columns.join(', ') + ')';
            joli.connection.execute(query);
        });
    },
    migrate: function(version, migrationCallback) {
        // create migration table
        var query = 'CREATE TABLE IF NOT EXISTS ' + this.migration.table + ' (version)';
        joli.connection.execute(query);

        if (this.migration.getVersion() < version) {
            joli.each(this.models, function(model, modelName) {
                var query = 'DROP TABLE IF EXISTS ' + modelName;
                joli.connection.execute(query);
            });
            // optional migration callback
            if (migrationCallback && 'function' === joli.getType(migrationCallback)) {
                migrationCallback({
                    table: this.migration.table,
                    newVersion: version
                });
            }

            // insert migration
            this.migration.setVersion(version);
        }
    },
    set: function(table, model) {
        this.models[table] = model;
    }
};

joli.models = new joli.Models();

joli.query = function() {
    this.data = {
        as: null,
        from: null,
        join: [],
        limit: null,
        operation: null,
        order: [],
        select_columns: '*',
        set: [],
        values: [],
        where: null
    };
};

joli.query.prototype = {
    count: function() {
        this.data.operation = 'count';
        return this;
    },
    destroy: function() {
        this.data.operation = 'delete';
        return this;
    },
    execute: function(hydratationMode) {
        return this.executeQuery(this.getQuery(), hydratationMode);
    },
    executeQuery: function(query, hydratationMode) {
        var rows;

        switch (this.data.operation) {
            case 'count':
                rows = joli.connection.execute(query);
                return this.getCount(rows);
            case 'insert_into':
                joli.connection.execute(query);
                return joli.connection.lastInsertRowId();
            case 'select':
                if (typeof hydratationMode === 'undefined') {
                    hydratationMode = 'object';
                }
                rows = joli.connection.execute(query);
                return this.hydrate(rows, hydratationMode);
            default:
                return joli.connection.execute(query);
        }
    },
    from: function(table) {
        this.data.from = table;
        return this;
    },
    as: function(table) {
        this.data.as = table;
        return this;
    },
    getCount: function(rows) {
        var result;

        if (null === rows) {
            return 0;
        }

        if (0 === rows.rowCount) {
            result = 0;
        } else {
            result = rows.fieldByName('total');
        }

        rows.close();
        return result;
    },
    getOperation: function() {
        switch (this.data.operation) {
            case 'count':
                return 'select count(*) as total from ' + this.data.from;
            case 'delete':
                return 'delete from ' + this.data.from;
            case 'insert_into':
                return 'insert into ' + this.data.from + ' (' + this.data.set.join(', ') + ') values (' + this.data.values.join(', ') + ')';
            case 'select':
                var join = '';

                if (this.data.join.length > 0) {
                    joli.each(this.data.join, function(value, key) {
                        if (-1 === value[1].indexOf('.')) {
                            value[1] = value[0] + '.' + value[1];
                        }
                        join = join + ' left outer join ' + value[0] + ' on ' + value[1] + ' = ' + value[2];
                    });
                }

                return 'select ' + this.data.select_columns + ' from ' + this.data.from + join;
            case 'update':
                return 'update ' + this.data.from + ' set ' + this.data.set.join(', ');
            default:
                throw ("Operation type Error. joli.query operation type must be an insert, a delete, a select or an update.");
        }
    },
    getQuery: function() {
        var query = this.getOperation();

        if (this.data.where) {
            query += ' where ' + this.data.where;
        }

        if (this.data.groupBy) {
            query += ' group by ' + this.data.groupBy.join(', ');
        }

        if (this.data.order.length > 0) {
            query += ' order by ' + this.data.order.join(', ');
        }

        if (this.data.limit) {
            query += ' limit ' + this.data.limit;
        }

        return query;
    },
    groupBy: function(group) {
        if ('string' === joli.getType(group)) {
            group = [group];
        }

        this.data.groupBy = group;
        return this;
    },
    hydrate: function(rows, hydratationMode) {
        var result = [];

        if (null === hydratationMode) {
            hydratationMode = 'object';
        }

        if (!rows) {
            return result;
        }

        var fieldCount;

        if (Titanium.Platform.name !== 'android') {
            fieldCount = rows.fieldCount();
        } else {
            fieldCount = rows.fieldCount;
        }

        switch (hydratationMode) {
            case 'array':
                result = this.hydrateArray(rows, fieldCount);
                break;
            case 'object':
                result = this.hydrateObject(rows, fieldCount);
                break;
            default:
                throw ('Unknown hydratation mode "' + hydratationMode + '". hydratationMode must be "object" or "array"');
        }

        rows.close();
        return result;
    },
    hydrateArray: function(rows, fieldCount) {
        var result = [];
        var i;
        var rowData;

        while (rows.isValidRow()) {
            i = 0;
            rowData = {};

            while (i < fieldCount) {
                rowData[rows.fieldName(i)] = rows.field(i);
                i++;
            }

            result.push(rowData);
            rows.next();
        }

        return result;
    },
    hydrateObject: function(rows, fieldCount) {
        var result = [];
        var i;
        var record;
        var rowData;

        // use the model specified by as() first, then from()
        var model = joli.models.get(this.data.as || this.data.from);

        while (rows.isValidRow()) {
            i = 0;
            rowData = {};

            while (i < fieldCount) {
                rowData[rows.fieldName(i)] = rows.field(i);
                i++;
            }

            result.push(model.newRecord().fromArray(rowData));
            rows.next();
        }

        return result;
    },
    insertInto: function(table) {
        this.data.operation = 'insert_into';
        this.data.from = table;
        return this;
    },
    join: function(table, local_id, foreign_id) {
        this.data.join.push([table, local_id, foreign_id]);
        return this;
    },
    limit: function(limit) {
        this.data.limit = limit;
        return this;
    },
    order: function(order) {
        if ('string' === joli.getType(order)) {
            order = [order];
        }

        this.data.order = order;
        return this;
    },
    select: function(columns) {
        this.data.operation = 'select';

        if (columns) {
            this.data.select_columns = columns;
        }

        return this;
    },
    set: function(values) {
        joli.each(values, function(expression, value) {
            if (-1 === value.indexOf('=')) {
                this.data.set.push(value + ' = ' + joli.typeValue(expression));
            } else {
                // some particular expression containing "="
                this.data.set.push(value);
            }
        }, this);
        return this;
    },
    update: function(table) {
        this.data.operation = 'update';
        this.data.from = table;
        return this;
    },
    values: function(values) {
        joli.each(values, function(expression, value) {
            this.data.set.push(value);
            this.data.values.push(joli.typeValue(expression));
        }, this);
        return this;
    },
    where: function(expression, value) {
        if (null !== this.data.where) {
            this.data.where += ' and ';
        } else {
            this.data.where = '';
        }

        // handle replacing multiple values
        if ('array' === joli.getType(value)) {
            var i = 0;

            // replace question marks one at a time from the array
            while (expression.indexOf('?') !== -1 && value[i] !== undefined) {
                expression = expression.replace(/\?/i, '"' + value[i] + '"');
                i++;
            }

            this.data.where += expression;
        } else {
            this.data.where += expression.replace(/\?/gi, '"' + value + '"');
        }

        return this;
    },
    whereIn: function(expression, value) {
        if (null !== this.data.where) {
            this.data.where += ' and ';
        } else {
            this.data.where = '';
        }

        if ('array' === joli.getType(value)) {
            if (0 === value.length) {
                return this;
            }
            value = '(\'' + value.join('\', \'') + '\')';
        }

        this.data.where += expression + ' in ' + value;
        return this;
    }
};

joli.record = function(table) {
    this._options = {
        table: table,
        columns: table.getColumns()
    };
    this._data = {};
    this._metadata = {};
};

joli.record.prototype = {
    destroy: function() {
        if (!this.id) {
            throw ("Unsaved record cannot be destroyed");
        } else {
            this._options.table.deleteRecords(this.id);
        }
    },
    fromArray: function(data) {
        if (data.id) {
            this._originalData = {
                id: data.id
            };
            this.isNew = function() {
                return false;
            };
        } else {
            this._originalData = null;
            this.isNew = function() {
                return true;
            };
        }

        joli.each(this._options.columns, function(colType, colName) {
            this[colName] = null;
            this[colName] = data[colName];
            this._data[colName] = null;
            this._data[colName] = data[colName];

            if (this._originalData && this.isNew()) {
                this._originalData[colName] = data[colName];
            }
        }, this);
        return this;
    },
    get: function(key) {
        return this[key];
    },
    isChanged: function() {
        if (this.isNew()) {
            return false;
        }

        return !(this.id && joli.toQueryString(this._data) === joli.toQueryString(this._originalData));
    },
    save: function() {
        var data = {
            data: this._data,
            metadata: this._metadata
        };

        if (this.isChanged()) {
            data.originalData = this._originalData;
            this._options.table.save(data);
        } else if (this.isNew()) {
            var rowid = this._options.table.save(data);
            
            if (!data.id) {
                this._data.id = rowid;
            }
        }

        // overwrite original data so it is no longer "dirty" OR so it is no
        // longer new
        this._originalData = {};
        var newData = {};

        joli.each(this._options.columns, function(colType, colName) {
            this._originalData[colName] = this._data[colName];
            newData[colName] = this._data[colName];
            this[colName] = this._data[colName];
        }, this);

        this._data = newData;

        this.isNew = function() {
            return false;
        };
        return true;
    },
    set: function(key, value) {
        this[key] = value;
        this._data[key] = value;
    },
    toArray: function(data) {
        var result = [];

        joli.each(this._options.columns, function(colType, colName) {
            result[colName] = this._data[colName];
        }, this);
        return result;
    }
};

/**
 * In case joli.js is loaded as a CommonJS module
 */
if (typeof exports === 'object' && exports) {
    exports = function() {
        var api = {
            connect: function(database) {
    
                if (database) {
                    joli.connection = new joli.Connection(database);
                }
    
                return joli;
            }
        };
    
        return api;
    }();
}