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.Filter;
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);
                var cosmosClient = _azureCosmos.GetCosmosClient();
                List<JoinAreaSchool> joinAreaSchools = new List<JoinAreaSchool>();
                string sqltxt = $"SELECT c.id,c.name,c.schoolCode,c.province,c.city,c.dist,c.picture,c.period 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.manyAreas FROM c WHERE c.areaId ='{_areaId}' and ARRAY_CONTAINS(c.manyAreas,'{_areaId}',true) or ARRAY_LENGTH(c.manyAreas) = 0 ";

                    //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()
                            };

                            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,assist")]
        [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);

                var (_tmdId, _tmdName, pic, did, dname, dpic) = HttpJwtAnalysis.JwtXAuthBI(HttpContext.GetXAuth("AuthToken"), _option);

                var cosmosClient = _azureCosmos.GetCosmosClient();
                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;
                }

                if (tempSchool.manyAreas.Count > 0)
                {
                    var temp = tempSchool.manyAreas.Find(ma => ma.areaId == $"{areaId}");
                    if (temp != null)
                        tempSchool.manyAreas.Remove(new ManyArea { areaId = $"{areaId}", standard = $"{standard}" });
                }
                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);

                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 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>();
        }

    }
}