TEAMModelOS/TEAMModelBI/Controllers/BINormal/BusinessUsersController.cs

using Azure.Cosmos;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Text.Json;
using System.Threading.Tasks;
using TEAMModelBI.Filter;
using TEAMModelBI.Tool.Extension;
using TEAMModelOS.Models;
using TEAMModelOS.SDK.Context.BI;
using TEAMModelOS.SDK.Context.Constant;
using TEAMModelOS.SDK.DI;
using TEAMModelOS.SDK.Extension;
using TEAMModelOS.SDK.Models.Cosmos.BI.BINormal;

namespace TEAMModelBI.Controllers.BINormal
{
    [Route("bizuser")]
    [ApiController]
    public class BusinessUsersController : ControllerBase
    {
        public readonly AzureCosmosFactory _azureCosmos;
        public readonly AzureStorageFactory _azureStorage;
        public readonly DingDing _dingDing;
        public readonly Option _option;
        public BusinessUsersController(AzureCosmosFactory azureCosmos, AzureStorageFactory azureStorage, DingDing dingDing, IOptionsSnapshot<Option> option)
        {
            _azureCosmos = azureCosmos;
            _azureStorage = azureStorage;
            _dingDing = dingDing;
            _option = option?.Value;
        }

        /// <summary>
        /// 获取第三方用户信息
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("get-infos")]
        public async Task<IActionResult> GetInfos(JsonElement jsonElement) 
        {
            jsonElement.TryGetProperty("id", out JsonElement id);
            //jsonElement.TryGetProperty("site", out JsonElement site);//分开部署，就不需要，一站多用时，取消注释
            var cosmosClient = _azureCosmos.GetCosmosClient();
            //if ($"{site}".Equals(BIConst.Global))
            //    cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.Global);
            StringBuilder sqlTxt = new("select value(c) from c");
            if (!string.IsNullOrEmpty($"{id}"))
            {
                sqlTxt.Append($" where c.id='{id}'");
            }

            List<BizUsers> bizUsers = new();
            await foreach (var items in cosmosClient.GetContainer("TEAMModelOS", "Normal").GetItemQueryIterator<BizUsers>(queryText: sqlTxt.ToString(), requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey("BizUsers") }))
            {
                bizUsers.Add(items);
            }

            return Ok(new { state = RespondCode.Ok, bizUsers });
        }

        /// <summary>
        /// 用户信息企业信息
        /// </summary>
        /// <param name="bizUsers"></param>
        /// <param name="site"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [AuthToken(Roles = "admin,rdc,assist")]
        [HttpPost("set-info")]
        public async Task<IActionResult> SetInfos([FromBody] BizUsers bizUsers, [FromHeader] string site)
        {
            var cosmosClient = _azureCosmos.GetCosmosClient();
            var tableClient = _azureStorage.GetCloudTableClient();
            var blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public");
            if ($"{site}".Equals(BIConst.Global))
            {
                cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.Global);
                tableClient = _azureStorage.GetCloudTableClient(BIConst.Global);
                blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public", BIConst.Global);
            }
            var (tmdId, tmdName, pic, _, _, _) = HttpJwtAnalysis.JwtXAuthBI(HttpContext.GetXAuth("AuthToken"), _option);
            StringBuilder strMsg = new($"{tmdName}[{tmdId}]操作:");
            string salt = Utils.CreatSaltString(8);
            string type = "";
            if (string.IsNullOrEmpty(bizUsers.id))
            {

                bizUsers.id = Guid.NewGuid().ToString();
                bizUsers.code = "BizUsers";
                bizUsers.name = string.IsNullOrEmpty(bizUsers.name)? bizUsers.mobile.ToString(): bizUsers.name;

                bizUsers.salt = salt;
                bizUsers.pwd = string.IsNullOrEmpty(bizUsers.pwd) ? Utils.HashedPassword(bizUsers.mobile.ToString(), salt) : Utils.HashedPassword(bizUsers.pwd, salt);
                
                bizUsers = await cosmosClient.GetContainer("TEAMModelOS", "Normal").CreateItemAsync<BizUsers>(bizUsers, new PartitionKey("BizUsers"));
                strMsg.Append($"{bizUsers.name}【{bizUsers.id}】新增第三方用户信息基础信息。");
                type = "bizuser-add";
            }
            else 
            {
                var response = await cosmosClient.GetContainer("TEAMModelOS", "Normal").ReadItemStreamAsync(bizUsers.id, new PartitionKey("BizUsers"));
                if (response.Status == RespondCode.Ok)
                {
                    using var json = await JsonDocument.ParseAsync(response.ContentStream);
                    BizUsers tempbizUsers = json.ToObject<BizUsers>();

                    bizUsers.pk = "BizUsers";
                    bizUsers.code = "BizUsers";
                    bizUsers.ttl = -1;
                    bizUsers.relation = tempbizUsers.relation;
                    bizUsers.salt = tempbizUsers.salt;
                    bizUsers.pwd = tempbizUsers.pwd;
                    bizUsers = await cosmosClient.GetContainer("TEAMModelOS", "Normal").ReplaceItemAsync<BizUsers>(bizUsers, bizUsers.id, new PartitionKey("BizUsers"));

                    strMsg.Append($"{bizUsers.name}【{bizUsers.id}】修改第三方用户信息基础信息。");
                    type = "bizuser-add";
                }
                else Ok(new { state = RespondCode.NotFound ,msg="未找到id用户。"});
            }

            //保存操作记录
            await AzureStorageBlobExtensions.SaveBILog(blobClient, tableClient, type, strMsg.ToString(), _dingDing, httpContext: HttpContext);
            return Ok(new { state = RespondCode.Ok, bizUsers });
        }

        /// <summary>
        /// 重置密码
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [AuthToken(Roles = "admin,rdc,assist")]
        [HttpPost("reset-pwd")]
        public async Task<IActionResult> ResetPassWord(JsonElement jsonElement)
        {
            if (!jsonElement.TryGetProperty("id", out JsonElement id)) return BadRequest();
            //jsonElement.TryGetProperty("site", out JsonElement site);  //分开部署，就不需要，一站多用时，取消注释
            var (tmdId, tmdName, pic, _, _, _) = HttpJwtAnalysis.JwtXAuthBI(HttpContext.GetXAuth("AuthToken"), _option);

            var cosmosClient = _azureCosmos.GetCosmosClient();
            var tableClient = _azureStorage.GetCloudTableClient();
            var blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public");
            ////分开部署，就不需要，一站多用时，取消注释
            //if ($"{site}".Equals(BIConst.Global))
            //{
            //    cosmosClient = _azureCosmos.GetCosmosClient(BIConst.Global);
            //    tableClient = _azureStorage.GetCloudTableClient(BIConst.Global);
            //    blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public", BIConst.Global);
            //}
            BizUsers bizUsers = new();
            StringBuilder strMsg = new($"{tmdName}[{tmdId}]操作:");
            string salt = Utils.CreatSaltString(8);
            var response = await cosmosClient.GetContainer("TEAMModelOS", "Normal").ReadItemStreamAsync($"{id}", new PartitionKey("BizUsers"));
            if (response.Status == 200)
            {
                using var json = await JsonDocument.ParseAsync(response.ContentStream);
                bizUsers = json.ToObject<BizUsers>();
                bizUsers.salt = salt;
                bizUsers.pwd = Utils.HashedPassword(bizUsers.mobile.ToString(), salt);

                strMsg.Append($"重置{bizUsers.name}【{bizUsers.id}】的密码，重置成功！");
                bizUsers = await cosmosClient.GetContainer("TEAMModelOS", "Normal").ReplaceItemAsync<BizUsers>(bizUsers, bizUsers.id, new PartitionKey("BizUsers"));
            }else return Ok(new { state = RespondCode.NotFound, msg="未找到该用户信息" });

            //保存操作记录
            await AzureStorageBlobExtensions.SaveBILog(blobClient, tableClient, "bizuser-reset", strMsg.ToString(), _dingDing, httpContext: HttpContext);
            return Ok(new { state = RespondCode.Ok, bizUsers });
        }

        /// <summary>
        /// 用户关联/移除企业信息
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [AuthToken(Roles = "admin,rdc,assist")]
        [HttpPost("rel-biz")]
        public async Task<ActionResult> RelationBusiness(JsonElement jsonElement) 
        {
            if(!jsonElement.TryGetProperty("id", out JsonElement id)) return BadRequest();
            if (!jsonElement.TryGetProperty("bizs", out JsonElement _bizs)) return BadRequest();
            if (!jsonElement.TryGetProperty("type", out JsonElement type)) return BadRequest();
            //jsonElement.TryGetProperty("site", out JsonElement site);//分开部署，就不需要，一站多用时，取消注释
            var (tmdId, tmdName, pic, _, _, _) = HttpJwtAnalysis.JwtXAuthBI(HttpContext.GetXAuth("AuthToken"), _option);

            List<BizRel> bizRels = _bizs.ToObject<List<BizRel>>();

            var cosmosClient = _azureCosmos.GetCosmosClient();
            var tableClient = _azureStorage.GetCloudTableClient();
            var blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public");
            ////分开部署，就不需要，一站多用时，取消注释
            //if ($"{site}".Equals(BIConst.Global))
            //{
            //    cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.Global);
            //    tableClient = _azureStorage.GetCloudTableClient(BIConst.Global);
            //    blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public", BIConst.Global);
            //}

            StringBuilder strMsg = new($"{tmdName}[{tmdId}]给");
            if ($"{type}".Equals("add"))
            {
                strMsg.Append($"用户【{id}】关联企业信息：添加企业:");
            }
            else if ($"{type}".Equals("del"))
            {
                strMsg.Append($"用户【{id}】移除企业信息,移除企业:");
            }
            else { return Ok(new { state = RespondCode.ParamsError, msg = "类型错误" }); }

            List<BizRel> noBizRel = new();
            BizUsers bizUsers = new();

            var response = await cosmosClient.GetContainer("TEAMModelOS", "Normal").ReadItemStreamAsync($"{id}", new PartitionKey("BizUsers"));
            if (response.Status == RespondCode.Ok)
            {
                using var json = await JsonDocument.ParseAsync(response.ContentStream);
                bizUsers = json.ToObject<BizUsers>();
                foreach (var item in bizRels)
                {
                    var temp = bizUsers.relation.Find(f => f.bizId == item.bizId);
                    if ($"{type}".Equals("add"))
                    {
                        if (temp != null)
                            noBizRel.Add(item);
                        else
                        {
                            bizUsers.relation.Add(item);
                            strMsg.Append($"{item.name}[{item.bizId}]|");
                        }
                    }
                    else if ($"{type}".Equals("del"))
                    {
                        if (temp != null)
                        {
                            bizUsers.relation.Add(item);
                            strMsg.Append($"{item.name}[{item.bizId}]|");
                        }
                        else
                            noBizRel.Add(item);
                    }
                }

                bizUsers = await cosmosClient.GetContainer("TEAMModelOS", "Normal").ReplaceItemAsync<BizUsers>(bizUsers, bizUsers.id, new PartitionKey("BizUsers"));
            }
            else return Ok(new { state = RespondCode.NotFound, msg = "未找到该用户" });

            //保存操作记录
            await AzureStorageBlobExtensions.SaveBILog(blobClient, tableClient, "bizconfig-addSchool", strMsg.ToString(), _dingDing, httpContext: HttpContext);
            if (noBizRel.Count > 0)
                return Ok(new { state = RespondCode.Created, bizUsers, noBizRel });
            return Ok(new { state = RespondCode.Ok, bizUsers });
        }

        /// <summary>
        /// 通过企业Id查询用户信息
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("get-bizid")]
        public async Task<IActionResult> GetBizIdUsers(JsonElement jsonElement)
        {

            if (!jsonElement.TryGetProperty("id", out JsonElement id)) return BadRequest();
            //jsonElement.TryGetProperty("site", out JsonElement site);//分开部署，就不需要，一站多用时，取消注释
            var cosmosClient = _azureCosmos.GetCosmosClient();
            ////分开部署，就不需要，一站多用时，取消注释
            //if ($"{site}".Equals(BIConst.Global))
            //    cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.Global);

            List<BizUsers> businessUsers = new();
            string sql = $"select value(c) from c join s in c.relation  where c.code='BizUsers' and s.bizId = '{id}'";
            await foreach (var item in cosmosClient.GetContainer("TEAMModelOS", "Normal").GetItemQueryIterator<BizUsers>(queryText: sql, requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey("BizUsers") }))
            {
                businessUsers.Add(item);
            }

            return Ok(new { state = RespondCode.Ok, businessUsers });
        }

    }
}