HTCloud/HT.Cloud.Service/SystemSecurity/FilterIPService.cs

/*******************************************************************************
 * Copyright © 2020 HT.Cloud.Framework 版权所有
 * Author: HT.Cloud
 * Description: WaterCloud快速开发平台
 * Website：
*********************************************************************************/

using SqlSugar;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using HT.Cloud.Domain.SystemSecurity;

namespace HT.Cloud.Service.SystemSecurity
{
	public class FilterIPService : BaseService<FilterIPEntity>, IDenpendency
	{
		public FilterIPService(ISqlSugarClient context) : base(context)
		{
		}

		public async Task<List<FilterIPEntity>> GetList(string keyword)
		{
			var list = repository.IQueryable();
			if (!string.IsNullOrEmpty(keyword))
			{
				list = list.Where(a => a.F_StartIP.Contains(keyword) || a.F_EndIP.Contains(keyword));
			}
			return await list.Where(a => a.F_DeleteMark == false).OrderBy(a => a.F_Id).ToListAsync();
		}

		public async Task<List<FilterIPEntity>> GetLookList(string keyword)
		{
			var query = repository.IQueryable().Where(a => a.F_DeleteMark == false);
			if (!string.IsNullOrEmpty(keyword))
			{
				//此处需修改
				query = query.Where(a => a.F_StartIP.Contains(keyword) || a.F_EndIP.Contains(keyword));
			}
			query = GetDataPrivilege("a", "", query);
			return await query.OrderBy(a => a.F_Id).ToListAsync();
		}

		public async Task<FilterIPEntity> GetLookForm(string keyValue)
		{
			var data = await repository.FindEntity(keyValue);
			return GetFieldsFilterData(data);
		}

		public async Task<FilterIPEntity> GetForm(string keyValue)
		{
			var data = await repository.FindEntity(keyValue);
			return data;
		}

		public async Task DeleteForm(string keyValue)
		{
			var ids = keyValue.Split(",");
			await repository.Delete(a => ids.Contains(a.F_Id));
		}

		public async Task<bool> CheckIP(string ip)
		{
			return await Task.Run(() =>
			{
				var list = repository.IQueryable().Where(a => a.F_EnabledMark == true && a.F_DeleteMark == false && a.F_Type == false && a.F_EndTime > DateTime.Now).ToList();
				long ipAddress = IP2Long(ip);
				foreach (var item in list)
				{
					if (string.IsNullOrEmpty(item.F_EndIP))
					{
						long start = IP2Long(item.F_StartIP);
						bool inRange = ipAddress == start;
						if (inRange)
						{
							return false;
						}
					}
					else
					{
						long start = IP2Long(item.F_StartIP);
						long end = IP2Long(item.F_EndIP);
						bool inRange = (ipAddress >= start && ipAddress <= end);
						if (inRange)
						{
							return false;
						}
					}
				}
				return true;
			});
		}

		public static long IP2Long(string ip)
		{
			string[] ipBytes;
			double num = 0;
			if (!string.IsNullOrEmpty(ip))
			{
				ipBytes = ip.Split('.');
				for (int i = ipBytes.Length - 1; i >= 0; i--)
				{
					num += ((int.Parse(ipBytes[i]) % 256) * Math.Pow(256, (3 - i)));
				}
			}
			return (long)num;
		}

		public async Task SubmitForm(FilterIPEntity filterIPEntity, string keyValue)
		{
			filterIPEntity.F_Type = false;
			if (!string.IsNullOrEmpty(keyValue))
			{
				filterIPEntity.Modify(keyValue);
				await repository.Update(filterIPEntity);
			}
			else
			{
				filterIPEntity.Create();
				await repository.Insert(filterIPEntity);
			}
		}
	}
}