TEAMModelOS/TEAMModelBI/Controllers/BISchool/AreaRelevantController.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.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.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;

        public AreaRelevantController(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-areaschools")]
        public async Task<IActionResult> GetAreaSchools(JsonElement jsonElement) 
        {
            try
            {
                jsonElement.TryGetProperty("areaId", out JsonElement _areaId);
                jsonElement.TryGetProperty("site", out JsonElement site);
                var cosmosClient = _azureCosmos.GetCosmosClient(); 
                if ($"{site}".Equals(BIConst.GlobalSite))
                    cosmosClient = _azureCosmos.GetCosmosClient(name: BIConst.GlobalSite);

                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 (!string.IsNullOrEmpty($"{_areaId}"))
                {
                    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.manyAreas = obj.GetProperty("manyAreas").ToObject<List<ManyArea>>();
                            }
                            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);
                }

                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)
                    {
                        item.schools.FindAll(f => f.schoolId == $"{schoolId}").ForEach(fe => fe.areaId = null);
                        await cosmosClient.GetContainer("TEAMModelOS", "Teacher").ReplaceItemAsync<Teacher>(item, item.id, new PartitionKey("Base"));
                    }
                }

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

                    if (tempSchool.manyAreas != null)
                    {
                        if (!string.IsNullOrEmpty($"{areaId}"))
                        {
                            var temp = tempSchool.manyAreas.Find(ma => ma.areaId == $"{areaId}");
                            if (temp != null)
                                tempSchool.manyAreas.Remove(temp);
                        }

                        if (!string.IsNullOrEmpty($"{standard}"))
                        {
                            var temp = tempSchool.manyAreas.Find(ma => ma.standard == $"{standard}");
                            if (temp != null)
                                tempSchool.manyAreas.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>
        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<ManyArea> manyAreas { get; set; } = new List<ManyArea>();
        }

    }
}