TEAMModelOS/TEAMModelBI/Controllers/BISchool/AreaRelevantController.cs

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

namespace TEAMModelBI.Controllers.BISchool
{
    [Route("area")]
    [ApiController]
    public class AreaRelevantController : ControllerBase
    {
        //数据容器
        private readonly AzureCosmosFactory _azureCosmos;
        private readonly AzureStorageFactory _azureStorage;
        //钉钉提示信息
        private readonly DingDing _dingDing;
        private readonly Option _option;
        private readonly IConfiguration _configuration;
        private readonly CoreAPIHttpService _coreAPIHttpService;

        public AreaRelevantController(AzureCosmosFactory azureCosmos, AzureStorageFactory azureStorage, DingDing dingDing, IOptionsSnapshot<Option> option, IConfiguration configuration, CoreAPIHttpService coreAPIHttpService)
        {
            _azureCosmos = azureCosmos;
            _azureStorage = azureStorage;
            _dingDing = dingDing;
            _option = option?.Value;
        }

        /// <summary>
        /// 查询区域已经有的学校
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [HttpPost("get-areaschools")]
        public async Task<IActionResult> GetAreaSchools(JsonElement jsonElement) 
        {
            try
            {
                jsonElement.TryGetProperty("areaId", out JsonElement _areaId);
                jsonElement.TryGetProperty("isDefault", out JsonElement _isDefalue);
                jsonElement.TryGetProperty("site", out JsonElement site);
                var cosmosClient = _azureCosmos.GetCosmosClient(); 
                if ($"{site}".Equals(BIConst.GlobalSite))
                    cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.GlobalSite);
                var isManyArea = false;
                if (!string.IsNullOrEmpty($"{_isDefalue}"))
                {
                    isManyArea = true;
                }

                List<JoinAreaSchool> joinAreaSchools = new();
                string sqltxt = $"SELECT c.id,c.name,c.schoolCode,c.province,c.city,c.dist,c.picture,c.period,c.areaId,c.standard,c.manyAreas FROM c WHERE c.areaId='{_areaId}'";

                if (isManyArea)
                {
                    sqltxt = $"SELECT c.id,c.name,c.schoolCode,c.province,c.city,c.dist,c.picture,c.period,c.areaId,c.standard,c.manyAreas FROM c join m in c.manyAreas WHERE c.areaId ='{_areaId}' or m.areaId='{_areaId}'";
                    //sqltxt = $"SELECT c.id,c.name,c.schoolCode,c.province,c.city,c.dist,c.picture,c.period FROM c join m in c.manyAreas where m.areaId='{_areaId}' or c.areaId='{_areaId}'";
                }

                await foreach (var item in cosmosClient.GetContainer(Constant.TEAMModelOS, "School").GetItemQueryStreamIterator(queryText: sqltxt, requestOptions:new QueryRequestOptions() { PartitionKey = new PartitionKey("Base")})) 
                {
                    using var json = await JsonDocument.ParseAsync(item.ContentStream);
                    if (json.RootElement.TryGetProperty("_count", out JsonElement count) && count.GetUInt16() > 0) 
                    {
                        foreach(var obj in json.RootElement.GetProperty("Documents").EnumerateArray())
                        {
                            JoinAreaSchool joinAreaSchool = new JoinAreaSchool
                            {
                                id = obj.GetProperty("id").GetString(),
                                name = obj.GetProperty("name").GetString(),
                                schoolCode = obj.GetProperty("schoolCode").GetString(),
                                province = obj.GetProperty("province").GetString(),
                                city = obj.GetProperty("city").GetString(),
                                dist = obj.GetProperty("dist").GetString(),
                                picture = obj.GetProperty("picture").GetString(),
                                period = obj.GetProperty("period").ToObject<List<Period>>().Select(x => x.name).ToList(),
                                areaId = obj.GetProperty("areaId").GetString(),
                                standard = obj.GetProperty("standard").GetString()
                            };

                            try
                            {
                                joinAreaSchool.areas = obj.GetProperty("areas").ToObject<List<SchoolArea>>();
                            }
                            catch { }
                            joinAreaSchools.Add(joinAreaSchool);
                        }                    
                    }
                }

                return Ok(new { state = 200, joinAreaSchools });
            }
            catch (Exception ex)
            {
                await _dingDing.SendBotMsg($"BI， {_option.Location} /area/get-areaschools  \n {ex.Message}\n{ex.StackTrace}", GroupNames.成都开发測試群組);
                return BadRequest();
            }        
        }

        /// <summary>
        /// 学校移出区域
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [AuthToken(Roles = "admin,rdc")]
        [HttpPost("set-areashiftschool")]
        public async Task<IActionResult> SetAreaShiftSchool(JsonElement jsonElement)
        {
            try
            {
                if (!jsonElement.TryGetProperty("schoolId", out JsonElement schoolId)) return BadRequest();
                jsonElement.TryGetProperty("areaId", out JsonElement areaId);
                jsonElement.TryGetProperty("standard", out JsonElement standard);
                jsonElement.TryGetProperty("isDefault", out JsonElement isDefault);
                jsonElement.TryGetProperty("site", out JsonElement site);


                var (_tmdId, _tmdName, pic, did, dname, dpic) = 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.GlobalSite)) 
                {
                    cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.GlobalSite);
                    tableClient = _azureStorage.GetCloudTableClient(BIConst.GlobalSite);
                    blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public", BIConst.GlobalSite);
                }

                var responseSet = await cosmosClient.GetContainer(Constant.TEAMModelOS, "Normal").ReadItemStreamAsync($"{areaId}", new PartitionKey("AreaSetting"));
                if (responseSet.Status == 200)
                {
                    using var fileJson = await JsonDocument.ParseAsync(responseSet.ContentStream);
                    AreaSetting delSet = fileJson.ToObject<AreaSetting>();
                    if (!string.IsNullOrEmpty(delSet.accessConfig))
                        return Ok(new { state = 401, msg = "区域已经规定了，不能切换能能力" });
                }

                School tempSchool = await cosmosClient.GetContainer(Constant.TEAMModelOS, "School").ReadItemAsync<School>($"{schoolId}", new PartitionKey("Base"));
                if (bool.Parse($"{isDefault}") == true)
                {
                    tempSchool.areaId = null;
                    tempSchool.standard = null;

                    List<Teacher> teachers = new();
                    await foreach (var item in cosmosClient.GetContainer("TEAMModelOS", "Teacher").GetItemQueryIterator<Teacher>(queryText: $"SELECT value(c) FROM c join s in c.schools where s.schoolId ='{schoolId}'", requestOptions: new QueryRequestOptions() { PartitionKey = new PartitionKey("Base") }))
                    {
                        teachers.Add(item);
                    }

                    foreach (var item in teachers)
                    {
                        var tchSchool = item.schools.Find(f => f.schoolId.Equals($"{schoolId}") && f.areaId.Equals($"{areaId}"));
                        if (tchSchool != null)
                        {
                            item.schools.Remove(tchSchool);
                            await cosmosClient.GetContainer("TEAMModelOS", "Teacher").ReplaceItemAsync<Teacher>(item, item.id, new PartitionKey("Base"));
                        }
                    }
                }

                if (tempSchool.areas != null)
                {
                    if (bool.Parse($"{isDefault}") == true) 
                    {
                        tempSchool.areaId = "";
                        tempSchool.standard = "";
                    }

                    if (tempSchool.areas != null)
                    {
                        if (!string.IsNullOrEmpty($"{areaId}"))
                        {
                            var temp = tempSchool.areas.Find(ma => ma.areaId == $"{areaId}");
                            if (temp != null)
                                tempSchool.areas.Remove(temp);
                        }
                    }
                }
                School school = await cosmosClient.GetContainer(Constant.TEAMModelOS, "School").ReplaceItemAsync<School>(tempSchool, tempSchool.id, new PartitionKey("Base"));

                //保存操作记录
                //await _azureStorage.SaveBILog("school-update", $"{_tmdName}【{_tmdId}】已操作学校({schoolId})移除该区域", _dingDing, httpContext: HttpContext);
                await AzureStorageBlobExtensions.SaveBILog(blobClient, tableClient, "school-update", $"{_tmdName}【{_tmdId}】已操作学校(ID:{schoolId})移除已区域(ID:{areaId})", _dingDing, httpContext: HttpContext);
                return Ok(new { state = 200, school });
            }
            catch (Exception ex)
            {
                await _dingDing.SendBotMsg($"BI， {_option.Location}   /area/set-areashiftschool \n  {ex.Message}\n{ex.StackTrace}", GroupNames.成都开发測試群組);
                return BadRequest();
            }
        }

        /// <summary>
        /// 设置区级管理员
        /// </summary>
        /// <param name="jsonElement"></param>
        /// <returns></returns>
        [ProducesDefaultResponseType]
        [AuthToken(Roles = "admin,rdc")]
        [HttpPost("set-manage")]
        public async Task<IActionResult> SetAreaManage(JsonElement jsonElement) 
        {
            var (_tmdId, _tmdName, _, _, _, _) = HttpJwtAnalysis.JwtXAuthBI(HttpContext.GetXAuth("AuthToken"), _option);

            if (!jsonElement.TryGetProperty("tmdId", out JsonElement tmdId)) return BadRequest();
            if (!jsonElement.TryGetProperty("tmdName", out JsonElement tmdName)) return BadRequest();
            jsonElement.TryGetProperty("tmdPic", out JsonElement tmdPic);
            if (!jsonElement.TryGetProperty("areaId", out JsonElement areaId)) return BadRequest();
            if (!jsonElement.TryGetProperty("areaName", out JsonElement areaName)) return BadRequest();
            jsonElement.TryGetProperty("site", out JsonElement site);
            var cosmosClient = _azureCosmos.GetCosmosClient();
            var tableClient = _azureStorage.GetCloudTableClient();
            var blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public");
            if ($"{site}".Equals(BIConst.GlobalSite))
            {
                cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.GlobalSite);
                tableClient = _azureStorage.GetCloudTableClient(BIConst.GlobalSite);
                blobClient = _azureStorage.GetBlobContainerClient(containerName: "0-public", BIConst.GlobalSite);
            }

            Teacher teacher = null;
            var response = await cosmosClient.GetContainer("TEAMModelOS", "Teacher").ReadItemStreamAsync($"{tmdId}", new PartitionKey($"Base"));
            if (response.Status == 200)
            {
                using var json = await JsonDocument.ParseAsync(response.ContentStream);
                teacher = json.ToObject<Teacher>();
                var existArea = teacher.areas.Find(f => f.areaId.Equals($"{areaId}"));
                if (existArea == null)
                {
                    teacher.areas.Add(new Teacher.TeacherArea() { areaId = $"{areaId}", name = $"{areaName}", status = "join" });
                    teacher = await cosmosClient.GetContainer("TEAMModelOS", "Teacher").ReplaceItemAsync<Teacher>(teacher, teacher.id, new PartitionKey("Base"));
                }
                else
                    return Ok(new { state = RespondCode.Conflict, msg = "该账户已是管理员" });
            }
            else
            {
                teacher.id = $"{tmdId}";
                teacher.name = $"{tmdName}";
                teacher.picture = string.IsNullOrEmpty($"{tmdPic}") ? "" : $"{tmdPic}";
                teacher.pk = "Base";
                teacher.code = "Base";
                teacher.size = 1;
                teacher.createTime = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
                //教师存在，在该教师信息中添加要管理的学校信息
                teacher.areas.Add(new Teacher.TeacherArea { areaId = $"{areaId}", name = $"{areaName}", status = "join" });
                await cosmosClient.GetContainer(Constant.TEAMModelOS, "Teacher").CreateItemAsync<Teacher>(teacher, new PartitionKey("Base"));
            }

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

        /// <summary>
        /// 已加入区域的学校
        /// </summary>
        public record JoinAreaSchool
        {
            public string id { get; set; }
            public string name { get; set; }
            public string schoolCode { get; set; }
            public string picture { get; set; }
            public string areaId{ get; set; }
            public string standard { get; set; }
            public List<string> period { get; set; }
            public string province { get; set; }
            public string city { get; set; }
            public string dist { get; set; }
            public List<SchoolArea> areas { get; set; } = new List<SchoolArea>();
        }

    }
}