91 lines
3.4 KiB
C#
91 lines
3.4 KiB
C#
/*
|
|
*
|
|
* (c) Copyright Ascensio System Limited 2010-2018
|
|
*
|
|
* This program is freeware. You can redistribute it and/or modify it under the terms of the GNU
|
|
* General Public License (GPL) version 3 as published by the Free Software Foundation (https://www.gnu.org/copyleft/gpl.html).
|
|
* In accordance with Section 7(a) of the GNU GPL its Section 15 shall be amended to the effect that
|
|
* Ascensio System SIA expressly excludes the warranty of non-infringement of any third-party rights.
|
|
*
|
|
* THIS PROGRAM IS DISTRIBUTED WITHOUT ANY WARRANTY; WITHOUT EVEN THE IMPLIED WARRANTY OF MERCHANTABILITY OR
|
|
* FITNESS FOR A PARTICULAR PURPOSE. For more details, see GNU GPL at https://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
* You can contact Ascensio System SIA by email at sales@onlyoffice.com
|
|
*
|
|
* The interactive user interfaces in modified source and object code versions of ONLYOFFICE must display
|
|
* Appropriate Legal Notices, as required under Section 5 of the GNU GPL version 3.
|
|
*
|
|
* Pursuant to Section 7 § 3(b) of the GNU GPL you must retain the original ONLYOFFICE logo which contains
|
|
* relevant author attributions when distributing the software. If the display of the logo in its graphic
|
|
* form is not reasonably feasible for technical reasons, you must include the words "Powered by ONLYOFFICE"
|
|
* in every copy of the program you distribute.
|
|
* Pursuant to Section 7 § 3(e) we decline to grant you any rights under trademark law for use of our trademarks.
|
|
*
|
|
*/
|
|
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using ASC.Common.Data;
|
|
using ASC.Common.Data.Sql;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.DependencyInjection.Extensions;
|
|
|
|
namespace ASC.IPSecurity
|
|
{
|
|
public class IPRestrictionsRepository
|
|
{
|
|
private const string dbId = "core";
|
|
|
|
public DbOptionsManager DbOptions { get; }
|
|
|
|
public IPRestrictionsRepository(DbOptionsManager dbOptions)
|
|
{
|
|
DbOptions = dbOptions;
|
|
}
|
|
|
|
public List<IPRestriction> Get(int tenant)
|
|
{
|
|
var db = DbOptions.Get(dbId);
|
|
return db
|
|
.ExecuteList(new SqlQuery("tenants_iprestrictions").Select("id", "ip").Where("tenant", tenant))
|
|
.ConvertAll(r => new IPRestriction
|
|
{
|
|
Id = Convert.ToInt32(r[0]),
|
|
Ip = Convert.ToString(r[1]),
|
|
TenantId = tenant,
|
|
});
|
|
}
|
|
|
|
public List<string> Save(IEnumerable<string> ips, int tenant)
|
|
{
|
|
var db = DbOptions.Get(dbId);
|
|
using var tx = db.BeginTransaction();
|
|
var d = new SqlDelete("tenants_iprestrictions").Where("tenant", tenant);
|
|
db.ExecuteNonQuery(d);
|
|
|
|
var ipsList = ips.ToList();
|
|
foreach (var ip in ipsList)
|
|
{
|
|
var i = new SqlInsert("tenants_iprestrictions")
|
|
.InColumnValue("tenant", tenant)
|
|
.InColumnValue("ip", ip);
|
|
|
|
db.ExecuteNonQuery(i);
|
|
}
|
|
|
|
tx.Commit();
|
|
return ipsList;
|
|
}
|
|
}
|
|
public static class IPRestrictionsRepositoryExtension
|
|
{
|
|
public static IServiceCollection AddIPRestrictionsRepositoryService(this IServiceCollection services)
|
|
{
|
|
services.TryAddScoped<IPRestrictionsRepository>();
|
|
|
|
return services.AddDbManagerService();
|
|
}
|
|
}
|
|
} |