mirror of https://github.com/Ombi-app/Ombi
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.6 KiB
95 lines
2.6 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Ombi.Helpers;
|
|
using Ombi.Store.Context;
|
|
using Ombi.Store.Entities;
|
|
|
|
namespace Ombi.Store.Repository
|
|
{
|
|
public class NotificationTemplatesRepository : INotificationTemplatesRepository
|
|
{
|
|
public NotificationTemplatesRepository(IOmbiContext ctx)
|
|
{
|
|
Db = ctx;
|
|
}
|
|
|
|
private IOmbiContext Db { get; }
|
|
|
|
public IQueryable<NotificationTemplates> All()
|
|
{
|
|
return Db.NotificationTemplates.AsQueryable();
|
|
}
|
|
|
|
public IQueryable<NotificationTemplates> GetAllTemplates()
|
|
{
|
|
return Db.NotificationTemplates;
|
|
}
|
|
|
|
public IQueryable<NotificationTemplates> GetAllTemplates(NotificationAgent agent)
|
|
{
|
|
return Db.NotificationTemplates.Where(x => x.Agent == agent);
|
|
}
|
|
|
|
public async Task<NotificationTemplates> GetTemplate(NotificationAgent agent, NotificationType type)
|
|
{
|
|
return await Db.NotificationTemplates.FirstOrDefaultAsync(x => x.Agent == agent && x.NotificationType == type);
|
|
}
|
|
|
|
public async Task Update(NotificationTemplates template)
|
|
{
|
|
if (Db.Entry(template).State == EntityState.Detached)
|
|
{
|
|
Db.Attach(template);
|
|
Db.Entry(template).State = EntityState.Modified;
|
|
}
|
|
await InternalSaveChanges();
|
|
}
|
|
|
|
public async Task UpdateRange(IEnumerable<NotificationTemplates> templates)
|
|
{
|
|
foreach (var t in templates)
|
|
{
|
|
|
|
Db.Attach(t);
|
|
Db.Entry(t).State = EntityState.Modified;
|
|
}
|
|
await InternalSaveChanges();
|
|
}
|
|
|
|
public async Task<NotificationTemplates> Insert(NotificationTemplates entity)
|
|
{
|
|
var settings = await Db.NotificationTemplates.AddAsync(entity).ConfigureAwait(false);
|
|
await InternalSaveChanges().ConfigureAwait(false);
|
|
return settings.Entity;
|
|
}
|
|
|
|
private async Task<int> InternalSaveChanges()
|
|
{
|
|
return await Db.SaveChangesAsync();
|
|
}
|
|
|
|
private bool _disposed;
|
|
// Protected implementation of Dispose pattern.
|
|
protected virtual void Dispose(bool disposing)
|
|
{
|
|
if (_disposed)
|
|
return;
|
|
|
|
if (disposing)
|
|
{
|
|
Db?.Dispose();
|
|
}
|
|
|
|
_disposed = true;
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
}
|
|
} |