oqtane.framework/Oqtane.Server/Repository/AliasRepository.cs
2021-05-10 17:45:39 -04:00

92 lines
2.6 KiB
C#

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Caching.Memory;
using Oqtane.Models;
namespace Oqtane.Repository
{
public class AliasRepository : IAliasRepository
{
private MasterDBContext _db;
private readonly IMemoryCache _cache;
public AliasRepository(MasterDBContext context, IMemoryCache cache)
{
_db = context;
_cache = cache;
}
public IEnumerable<Alias> GetAliases()
{
return _cache.GetOrCreate("aliases", entry =>
{
entry.SlidingExpiration = TimeSpan.FromMinutes(30);
return _db.Alias.ToList();
});
}
public Alias AddAlias(Alias alias)
{
_db.Alias.Add(alias);
_db.SaveChanges();
_cache.Remove("aliases");
return alias;
}
public Alias UpdateAlias(Alias alias)
{
_db.Entry(alias).State = EntityState.Modified;
_db.SaveChanges();
_cache.Remove("aliases");
return alias;
}
public Alias GetAlias(int aliasId)
{
return _db.Alias.Find(aliasId);
}
public Alias GetAlias(string name)
{
Alias alias = null;
List<Alias> aliases = GetAliases().ToList();
var segments = name.Split(new[] { '/' }, StringSplitOptions.RemoveEmptyEntries);
// iterate segments to find keywords
int start = segments.Length;
for (int i = 0; i < segments.Length; i++)
{
if (segments[i] == "api" || segments[i] == "pages" || segments[i] == "*")
{
start = i;
break;
}
}
// iterate segments in reverse order to find alias match
for (int i = start; i > 0; i--)
{
alias = aliases.Find(item => item.Name.Equals(string.Join("/", segments, 0, i), StringComparison.OrdinalIgnoreCase));
if (alias != null)
{
break; // found a matching alias
}
}
// return fallback alias if none found
return alias ?? aliases.Find(item => item.Name.Equals("*"));
}
public void DeleteAlias(int aliasId)
{
Alias alias = _db.Alias.Find(aliasId);
_db.Alias.Remove(alias);
_cache.Remove("aliases");
_db.SaveChanges();
}
}
}