284 lines
8.3 KiB
C#
284 lines
8.3 KiB
C#
using System.Reflection;
|
||
using AntJob.Data;
|
||
using AntJob.Data.Entity;
|
||
using AntJob.Models;
|
||
using NewLife;
|
||
using NewLife.Log;
|
||
using NewLife.Remoting;
|
||
using NewLife.Remoting.Models;
|
||
using NewLife.Security;
|
||
using NewLife.Web;
|
||
|
||
namespace AntJob.Server.Services;
|
||
|
||
public class AppService
|
||
{
|
||
private readonly IPasswordProvider _passwordProvider;
|
||
private readonly AntJobSetting _setting;
|
||
private readonly ILog _log;
|
||
|
||
public AppService(IPasswordProvider passwordProvider, AntJobSetting setting, ILog log)
|
||
{
|
||
_passwordProvider = passwordProvider;
|
||
_setting = setting;
|
||
_log = log;
|
||
}
|
||
|
||
#region 登录
|
||
/// <summary>应用登录</summary>
|
||
/// <param name="model">模型</param>
|
||
/// <returns></returns>
|
||
public (App, AppOnline, LoginResponse) Login(LoginModel model, String ip)
|
||
{
|
||
if (model.Code.IsNullOrEmpty()) throw new ArgumentNullException(nameof(model.Code));
|
||
|
||
_log.Info("[{0}]从[{1}]登录[{2}]", model.Code, ip, model.ClientId);
|
||
|
||
// 找应用
|
||
var autoReg = false;
|
||
var app = App.FindByName(model.Code);
|
||
|
||
// 登录密码未设置或者未提交，则执行动态注册
|
||
var secret = model.Secret;
|
||
if (app == null || !app.Secret.IsNullOrEmpty() && secret.IsNullOrEmpty())
|
||
{
|
||
app = CheckApp(app, model.Code, model.Secret, ip);
|
||
if (app == null) throw new ArgumentOutOfRangeException(nameof(model.Code));
|
||
|
||
autoReg = true;
|
||
}
|
||
|
||
if (app == null) throw new Exception($"应用[{model.Code}]不存在！");
|
||
if (!app.Enable) throw new Exception("已禁用！");
|
||
|
||
// 核对密码
|
||
if (!autoReg && !app.Secret.IsNullOrEmpty())
|
||
{
|
||
if (secret.IsNullOrEmpty() || !_passwordProvider.Verify(app.Secret, secret)) throw new Exception("密码错误！");
|
||
}
|
||
|
||
// 版本和编译时间
|
||
if (app.Version.IsNullOrEmpty() || app.Version.CompareTo(model.Version) < 0) app.Version = model.Version;
|
||
if (app.CompileTime < model.Compile) app.CompileTime = model.Compile;
|
||
if (app.DisplayName.IsNullOrEmpty()) app.DisplayName = model.DisplayName;
|
||
|
||
app.Save();
|
||
|
||
// 应用上线
|
||
var online = CreateOnline(app, ip, model.ClientId);
|
||
online.Name = model.Machine;
|
||
online.ProcessId = model.ProcessId;
|
||
online.Version = model.Version;
|
||
online.CompileTime = model.Compile;
|
||
online.Save();
|
||
|
||
WriteHistory(app, autoReg ? "注册" : "登录", true, $"[{model.Code}/{model.Secret}]在[{model.ClientId}]登录[{app}]成功", ip);
|
||
|
||
var rs = new LoginResponse { Name = app.Name };
|
||
if (autoReg) rs.Secret = app.Secret;
|
||
|
||
return (app, online, rs);
|
||
}
|
||
|
||
protected virtual App CheckApp(App app, String user, String pass, String ip)
|
||
{
|
||
// 本地账号不存在时
|
||
var name = user;
|
||
if (app == null)
|
||
{
|
||
// 是否支持自动注册
|
||
//var set = AntJobSetting.Current;
|
||
if (!_setting.AutoRegistry) throw new Exception($"找不到应用[{name}]");
|
||
|
||
app = new App
|
||
{
|
||
//Secret = Rand.NextString(16)
|
||
};
|
||
}
|
||
else if (app.Secret.MD5() != pass)
|
||
{
|
||
// 是否支持自动注册
|
||
//var set = AntJobSetting.Current;
|
||
if (!_setting.AutoRegistry) throw new Exception($"应用[{name}]申请重新激活，但服务器设置禁止自动注册");
|
||
|
||
//if (app.Secret.IsNullOrEmpty()) app.Secret = Rand.NextString(16);
|
||
}
|
||
|
||
if (app.ID == 0)
|
||
{
|
||
app.Name = name;
|
||
app.CreateIP = ip;
|
||
app.CreateTime = DateTime.Now;
|
||
|
||
// 首次打开
|
||
app.Enable = true;
|
||
}
|
||
|
||
app.UpdateIP = ip;
|
||
app.UpdateTime = DateTime.Now;
|
||
|
||
//app.Save();
|
||
|
||
return app;
|
||
}
|
||
|
||
public ILogoutResponse Logout(App app, AppOnline online, String reason, String ip)
|
||
{
|
||
if (app != null)
|
||
{
|
||
online ??= GetOnline(app, ip);
|
||
if (online != null)
|
||
{
|
||
WriteHistory(app, "注销", true, reason, ip);
|
||
|
||
online.Delete();
|
||
}
|
||
}
|
||
|
||
return new LogoutResponse { Name = app?.Name };
|
||
}
|
||
#endregion
|
||
|
||
#region 在线状态
|
||
public IPingResponse Ping(App app, AppOnline online, IPingRequest request, String ip)
|
||
{
|
||
if (app != null)
|
||
{
|
||
online ??= GetOnline(app, ip);
|
||
if (online != null)
|
||
{
|
||
if (request is PingRequest req)
|
||
{
|
||
}
|
||
online.UpdateIP = ip;
|
||
|
||
online.Save();
|
||
}
|
||
}
|
||
|
||
return new PingResponse
|
||
{
|
||
Time = request.Time,
|
||
ServerTime = DateTime.UtcNow.ToLong(),
|
||
};
|
||
}
|
||
|
||
/// <summary>获取当前应用的所有在线实例</summary>
|
||
/// <returns></returns>
|
||
public PeerModel[] GetPeers(App app)
|
||
{
|
||
var olts = AppOnline.FindAllByAppID(app.ID);
|
||
|
||
return olts.Select(e => e.ToModel()).ToArray();
|
||
}
|
||
|
||
AppOnline CreateOnline(App app, String ip, String clientId)
|
||
{
|
||
var online = GetOnline(app, ip);
|
||
online.Client = clientId;
|
||
online.UpdateIP = ip;
|
||
|
||
online.Server = Environment.MachineName;
|
||
|
||
return online;
|
||
}
|
||
|
||
public AppOnline GetOnline(App app, String ip)
|
||
{
|
||
var ins = $"{app.Name}@{ip}";
|
||
var online = AppOnline.FindByInstance(ins) ?? new AppOnline { Enable = true, CreateIP = ip };
|
||
online.AppID = app.ID;
|
||
online.Instance = ins;
|
||
|
||
return online;
|
||
}
|
||
|
||
public void UpdateOnline(App app, JobTask ji, String ip)
|
||
{
|
||
var online = GetOnline(app, ip);
|
||
online.Total += ji.Total;
|
||
online.Success += ji.Success;
|
||
online.Error += ji.Error;
|
||
online.Cost += ji.Cost;
|
||
online.Speed = ji.Speed;
|
||
online.LastKey = ji.Key;
|
||
online.SaveAsync();
|
||
}
|
||
#endregion
|
||
|
||
#region 写历史
|
||
public void WriteHistory(App app, String action, Boolean success, String remark, String ip = null) =>
|
||
AppHistory.Create(app, action, success, remark, Environment.MachineName, ip);
|
||
#endregion
|
||
|
||
#region 辅助
|
||
public TokenModel IssueToken(String name, AntJobSetting set)
|
||
{
|
||
// 颁发令牌
|
||
var ss = set.TokenSecret.Split(':');
|
||
var jwt = new JwtBuilder
|
||
{
|
||
Issuer = Assembly.GetEntryAssembly().GetName().Name,
|
||
Subject = name,
|
||
Id = Rand.NextString(8),
|
||
Expire = DateTime.Now.AddSeconds(set.TokenExpire),
|
||
|
||
Algorithm = ss[0],
|
||
Secret = ss[1],
|
||
};
|
||
|
||
return new TokenModel
|
||
{
|
||
AccessToken = jwt.Encode(null),
|
||
TokenType = jwt.Type ?? "JWT",
|
||
ExpireIn = set.TokenExpire,
|
||
RefreshToken = jwt.Encode(null),
|
||
};
|
||
}
|
||
|
||
public (App, Exception) DecodeToken(String token, String tokenSecret)
|
||
{
|
||
if (token.IsNullOrEmpty()) throw new ArgumentNullException(nameof(token));
|
||
//if (token.IsNullOrEmpty()) throw new ApiException(401, $"节点未登录[ip={UserHost}]");
|
||
|
||
// 解码令牌
|
||
var ss = tokenSecret.Split(':');
|
||
var jwt = new JwtBuilder
|
||
{
|
||
Algorithm = ss[0],
|
||
Secret = ss[1],
|
||
};
|
||
|
||
var rs = jwt.TryDecode(token, out var message);
|
||
var app = App.FindByName(jwt.Subject);
|
||
|
||
Exception ex = null;
|
||
if (!rs || app == null)
|
||
{
|
||
if (app != null)
|
||
ex = new ApiException(403, $"[{app.Name}/{app.DisplayName}]非法访问 {message}");
|
||
else
|
||
ex = new ApiException(403, $"[{jwt.Subject}]非法访问 {message}");
|
||
}
|
||
|
||
return (app, ex);
|
||
}
|
||
|
||
public TokenModel ValidAndIssueToken(String deviceCode, String token, AntJobSetting set)
|
||
{
|
||
if (token.IsNullOrEmpty()) return null;
|
||
//var set = Setting.Current;
|
||
|
||
// 令牌有效期检查，10分钟内过期者，重新颁发令牌
|
||
var ss = set.TokenSecret.Split(':');
|
||
var jwt = new JwtBuilder
|
||
{
|
||
Algorithm = ss[0],
|
||
Secret = ss[1],
|
||
};
|
||
var rs = jwt.TryDecode(token, out var message);
|
||
return !rs || jwt == null ? null : DateTime.Now.AddMinutes(10) > jwt.Expire ? IssueToken(deviceCode, set) : null;
|
||
}
|
||
#endregion
|
||
}
|