using System.Collections.Generic;
using System.Linq;
using Neo.Afx.ComponentModel;

namespace Neo.Afx.Admin.Wf
{
    ///<summary>
    /// Afx admin database
    ///</summary>
    public class AfxAdminWorkflowDatabase : Database<AfxAdminWorkflowContext>
    {
        ///<summary>
        /// Constructor
        ///</summary>
        public AfxAdminWorkflowDatabase()
            : base(new AfxAdminWorkflowContext())
        {
        }

        #region GetWorkflows

        /// <summary>
        /// Get all data from table.
        /// </summary>
        /// <returns>List of objects</returns>
        public IEnumerable<Workflow> GetWorkflows()
        {
            return (from item in Context.Workflows
                    select item).ToList();
        }

        /// <summary>
        /// Get data from table based on count.
        /// </summary>
        /// <param name="count">Restricts amount returned items in the collection.</param>
        /// <returns>List of objects.</returns>
        public IEnumerable<Workflow> GetWorkflows(int count)
        {
            return (from item in Context.Workflows
                    select item).Take(count).ToList();
        }

        #endregion

        #region GetWorkflowCategories

        /// <summary>
        /// Gets the workflow categories.
        /// </summary>
        /// <returns>List of objects.</returns>
        public IEnumerable<WorkflowCategory> GetWorkflowCategories()
        {
            return (from item in Context.WorkflowCategories
                    where !item.IsCancelled
                    select item).ToList();
        }

        #endregion

        #region GetWorkflow

        /// <summary>
        /// Gets the workflow.
        /// </summary>
        /// <param name="workflowId">The workflow id.</param>
        /// <returns></returns>
        public Workflow GetWorkflow(long? workflowId)
        {
            if(!workflowId.HasValue)
            {
                return null;
            }

            return (from item in Context.Workflows
                        .Include("WorkflowTaskHandlers")
                        .Include("WorkflowTemplates")
                    where item.WorkflowID == workflowId.Value
                    select item).FirstOrDefault();
        }

        #endregion

        #region GetWorkflowTemplate

        public WorkflowTemplate GetWorkflowTemplate(long? workflowTemplateId)
        {
            if(!workflowTemplateId.HasValue)
            {
                return null;
            }

            return (from item in Context.WorkflowTemplates
                        .Include("WorkflowTemplateDefaults")
                    where item.WorkflowTemplateID == workflowTemplateId.Value
                    select item).SingleOrDefault();
        }

        #endregion

        #region GetWorkflowTemplateScripts

        public IEnumerable<WorkflowTemplateScript> GetWorkflowTemplateScripts(long? workflowTemplateId)
        {
            if(!workflowTemplateId.HasValue)
            {
                return null;
            }

            return (from item in Context.WorkflowTemplateScripts
                    where item.WorkflowTemplateID == workflowTemplateId.Value
                    select item);
        }

        #endregion
    }
}