using System;
using System.Collections.Generic;
using System.Data.Entity;
using System.Data.Entity.Validation;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace CAPI.Custom.Data.Repositories
{
    public class Repository<TContext, TEntity> : ReadOnlyRepository<TContext, TEntity>
        where TContext : DbContext, new()
        where TEntity : class
    {
        public virtual TContext Context => db;
        /// <summary>
        /// Mark entity as Added - Applies on Save()
        /// </summary>
        /// <param name="entity"></param>
        public virtual void Add(TEntity entity)
        {
            db.Set<TEntity>().Add(entity);
        }

        /// <summary>
        /// Mark entity as Deleted - Applies on Save()
        /// </summary>
        /// <param name="entity"></param>
        public virtual void Remove(TEntity entity)
        {
            db.Set<TEntity>().Remove(entity);
        }

        /// <summary>
        /// Mark entity as Edited - Applies on Save()
        /// </summary>
        /// <param name="entity"></param>
        public virtual void Edit(TEntity entity)
        {
            db.Entry(entity).State = EntityState.Modified;
        }

        /// <summary>
        /// Apply changes
        /// </summary>
        /// <returns></returns>
        public virtual int Save()
        {
            return db.SaveChanges();
        }
    }
}