RackHD/on-core

// Copyright 2016, EMC, Inc.

'use strict';

module.exports = TaskDependencyFactory;

TaskDependencyFactory.$provide = 'Models.TaskDependency';
TaskDependencyFactory.$inject = [
    'Model',
    'Constants',
    'Services.Configuration'
];

function TaskDependencyFactory (Model, Constants, configuration) {
    return Model.extend({
        connection: configuration.get('taskgraph-store', 'mongo'),
        identity: 'taskdependencies',
        attributes: {
            domain: {
                type: 'string',
                defaultsTo: Constants.Task.DefaultDomain
            },
            taskId: {
                type: 'string',
                required: true,
                unique: true,
                uuidv4: true
            },
            graphId: {
                type: 'string',
                required: true
            },
            state: {
                type: 'string',
                required: true
            },
            evaluated: {
                type: 'boolean',
                defaultsTo: false
            },
            reachable: {
                type: 'boolean',
                defaultsTo: true
            },
            taskRunnerLease: {
                type: 'string',
                defaultsTo: null
            },
            taskRunnerHeartbeat: {
                type: 'date',
                defaultsTo: null
            },
            dependencies: {
                type: 'json',
                required: true
            },
            terminalOnStates: {
                type: 'array',
                defaultsTo: []
            },
            context: {
                type: 'json',
                defaultsTo: null
            },
            toJSON: function() {
                // Remove waterline keys that we don't want in our dependency object
                var obj = this.toObject();
                delete obj.createdAt;
                delete obj.updatedAt;
                delete obj.id;
                return obj;
            }
        },

        $indexes: [
            {
                keys: { taskId: 1, graphId: 1 },
                options: { unique: true }
            }
        ]
    });
}