using HospitalContracts.BindingModels;
using HospitalContracts.ViewModels;
using HospitalDataModels.Models;
using System;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.ComponentModel.DataAnnotations.Schema;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace HospitalDatabaseImplement.Models
{
    public class Treatment : ITreatmentModel
    {
        public int Id { get; private set; }
        [Required]
        [MaxLength(55)]
        public string Name { get; private set; } = string.Empty;
        
        private Dictionary<int, IProcedureModel>? _treatmentProcedures = null;
        [NotMapped]
        public Dictionary<int, IProcedureModel> TreatmentProcedures { 
            get
            {
                if (_treatmentProcedures == null)
                {
                    _treatmentProcedures = Procedures
                    .ToDictionary(rec => rec.ProcedureId, rec =>
                   rec.Procedure as IProcedureModel);
                }
                return _treatmentProcedures;
            }
        }
        [ForeignKey("TreatmentId")]
        public virtual List<Patient> Patients { get; set; } = new();

        [ForeignKey("TreatmentId")]
        public virtual List<TreatmentProcedure> Procedures { get; set; } = new();

        [ForeignKey("TreatmentId")]
        public virtual List<RecipeTreatment> Recipes { get; set; } = new();

        public static Treatment Create(HospitalDatabase context, TreatmentBindingModel model)
        {
            return new Treatment()
            {
                Id = model.Id,
                Name = model.Name,
                Procedures = model.TreatmentProcedures.Select(x => new TreatmentProcedure
                {
                    Procedure = context.Procedures.First(y => y.Id == x.Key),
                }).ToList()
            };
        }

        public void Update(TreatmentBindingModel model)
        {
            Name = model.Name;
        }
        public TreatmentViewModel GetViewModel => new()
        {
            Id = Id,
            Name = Name      
        };

        public void UpdateProcedures(HospitalDatabase context, TreatmentBindingModel model)
        {
            var treatmentProcedures = context.TreatmentProcedures.Where(rec => rec.TreatmentId == model.Id).ToList();
            if (treatmentProcedures != null && treatmentProcedures.Count > 0)
            { 
                context.TreatmentProcedures.RemoveRange(treatmentProcedures.Where(rec
               => !model.TreatmentProcedures.ContainsKey(rec.ProcedureId)));
                context.SaveChanges();
            }
            var treatment = context.Treatments.First(x => x.Id == Id);
            foreach (var pc in model.TreatmentProcedures)
            {
                context.TreatmentProcedures.Add(new TreatmentProcedure
                {
                    Treatment = treatment,
                    Procedure = context.Procedures.First(x => x.Id == pc.Key),
                });
                context.SaveChanges();
            }
            _treatmentProcedures = null;
        }
    }
}