using InABox.Core;
using System;

namespace Comal.Classes
{
    public class DigitalFormAssignmentLookup : LookupDefinitionGenerator<DigitalForm, Assignment>
    {
        public override Filter<DigitalForm> DefineFilter(Assignment[] items)
        {
            var filter = LookupFactory.DefineLookupFilter<AssignmentForm, DigitalForm, DigitalFormLink>(x => x.Form, Array.Empty<AssignmentForm>())
                ?? new Filter<DigitalForm>().All();

            foreach (var assignment in items)
            {
                filter.And(x => x.ID).InQuery(
                    new Filter<EmployeeDigitalForm>(x => x.Employee.ID).IsEqualTo(assignment.EmployeeLink.ID),
                    x => x.Form.ID);
            }

            return filter;
        }

        public override Columns<Assignment> DefineFilterColumns()
            => Columns.None<Assignment>().Add(x => x.EmployeeLink.ID);
    }
    
    public class DigitalFormAssignmentFormLookup : LookupDefinitionGenerator<DigitalForm, AssignmentForm>
    {
        public override Filter<DigitalForm> DefineFilter(AssignmentForm[] items)
        {
            return new Filter<DigitalForm>(x => x.Active).IsEqualTo(true).And(x => x.AppliesTo).IsEqualTo("Assignment");
        }

        public override Columns<AssignmentForm> DefineFilterColumns()
            => Columns.None<AssignmentForm>();
    }

    public class AssignmentFormLookups : DigitalFormInstanceLookups<Assignment,AssignmentLink,AssignmentForm> { }
}