using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace OrmTest 
{
    public class UOneManyMany2
    {

        public static void init()
        {

            var db = NewUnitTest.Db;
            db.CodeFirst.InitTables<Student_002, School_002, Room_002, Desk_002>();
            db.DbMaintenance.TruncateTable<Student_002, School_002, Room_002, Desk_002>();

            db.Insertable(new Student_002() { sid = 1, Name = "北大jack", SchoolId = 1 }).ExecuteCommand();
            db.Insertable(new Student_002() { sid = 2, Name = "青华jack", SchoolId = 2 }).ExecuteCommand();

            db.Insertable(new School_002() { scid = 1, schname = "北大" }).ExecuteCommand();
            db.Insertable(new School_002() { scid = 2, schname = "青华" }).ExecuteCommand();


            db.Insertable(new Room_002() { roomId = 1, schoolId = 1, roomName = "北大01室" }).ExecuteCommand();
            db.Insertable(new Room_002() { roomId = 2, schoolId = 1, roomName = "北大02室" }).ExecuteCommand();
            db.Insertable(new Room_002() { roomId = 3, schoolId = 2, roomName = "青华03室" }).ExecuteCommand();
            db.Insertable(new Room_002() { roomId = 4, schoolId = 2, roomName = "青华04室" }).ExecuteCommand();

            db.Insertable(new Desk_002() { roomId = 1, deskid = 1, deskName = "北大01室_01" }).ExecuteCommand();
            db.Insertable(new Desk_002() { roomId = 2, deskid = 2, deskName = "北大02室_01" }).ExecuteCommand();
            db.Insertable(new Desk_002() { roomId = 3, deskid = 3, deskName = "青华03室_01" }).ExecuteCommand();
            db.Insertable(new Desk_002() { roomId = 4, deskid = 4, deskName = "青华04室_01" }).ExecuteCommand();


            var list = db.Queryable<Student_002>()
                         .Includes(x => x.school_001, x => x.rooms, x => x.desk)
                .Where(x => x.school_001.rooms.Any(z => z.desk.Any())).ToList();

            if (list.Count() != 2)
            {
                throw new Exception("unit error");
            }

            var list2 = db.Queryable<Student_002>()
                         .Includes(x => x.school_001, x => x.rooms)
            .Where(x => x.school_001.rooms.Any(z =>
            z.roomName == "北大01室" &&
            z.desk.Any())).ToList();


            if (list2.Count() != 1)
            {
                throw new Exception("unit error");
            }

            var list3 = db.Queryable<Student_002>()
                .Includes(x => x.school_001, x => x.rooms)
         .Where(x => x.school_001.rooms.Any(z =>
         z.roomName == "青华03室" &&
         z.desk.Any(c => c.deskName == "青华03室_01"))).ToList();

            if (list3.Count != 1)
            {
                throw new Exception("unit error");
            }

            var list4 = db.Queryable<Student_002>()
        .Where(x => x.school_001.rooms.Any(z =>
        z.roomName == "青华03室" &&
        z.desk.Any(c => c.deskName == "青华04室_01"))).ToList();


            if (list4.Count != 0)
            {
                throw new Exception("unit error");
            }
            db.Deleteable<Student_002>().ExecuteCommand();
            db.Deleteable<School_002>().ExecuteCommand();
            db.Deleteable<Room_002>().ExecuteCommand();
            db.Deleteable<Desk_002>().ExecuteCommand(); 
            foreach (var item in list) 
            {
                item.sid = 0;
                item.SchoolId = 0;
                item.school_001.scid = 0;
                foreach (var r in item.school_001.rooms)
                {
                    r.roomId = 0;
                    r.schoolId = 0;
                    foreach (var z in r.desk)
                    {
                        z.deskid = 0;
                        z.roomId = 0;
                    }
                }
            }
            db.InsertNav(list.First())
                .Include(x => x.school_001)
                .ThenInclude(x => x.rooms)
                .ThenInclude(x => x.desk).ExecuteCommand();
            db.InsertNav(list.Last())
              .Include(x => x.school_001)
              .ThenInclude(x => x.rooms)
              .ThenInclude(x => x.desk).ExecuteCommand();

            if (db.Queryable<Desk_002>().Count() != 4 || db.Queryable<Room_002>().Count() != 4
                || db.Queryable<School_002>().Count() != 2 || db.Queryable<Student_002>().Count() != 2) 
            {
                throw new Exception("unit error");
            }

            db.DbMaintenance.TruncateTable<Student_002, School_002, Room_002, Desk_002>();

            db.InsertNav(list.First().school_001)
             .Include(x => x.rooms)
             .ThenInclude(x => x.desk).ExecuteCommand();
            db.InsertNav(list.Last().school_001)
              .Include(x => x.rooms)
              .ThenInclude(x => x.desk).ExecuteCommand();

            if (db.Queryable<Desk_002>().Count() != 4 || db.Queryable<Room_002>().Count() != 4
               || db.Queryable<School_002>().Count() != 2 || db.Queryable<Student_002>().Count() != 0)
            {
                throw new Exception("unit error");
            }
        }

        public class Student_002 
        {
            [SqlSugar.SugarColumn(IsPrimaryKey =true,IsIdentity =true)]
            public int sid { get; set; } 
            public string Name { get; set; }

            public int SchoolId { get; set; }
            [SqlSugar.Navigate(SqlSugar.NavigateType.OneToOne,nameof(SchoolId))]
            public School_002 school_001 { get; set; }  

        }

        public class School_002
        {
            [SqlSugar.SugarColumn(IsPrimaryKey = true, IsIdentity = true)]
            public int scid { get; set; }
            public string schname { get; set; }
            [SqlSugar.Navigate(SqlSugar.NavigateType.OneToMany, nameof(Room_002.schoolId))]
            public List<Room_002> rooms { get; set; }
        }

        public class Room_002 
        {
            [SqlSugar.SugarColumn(IsPrimaryKey = true, IsIdentity = true)]
            public int roomId { get; set; }
            public int schoolId { get; set; }
            public string roomName { get; set; }
            [SqlSugar.Navigate(SqlSugar.NavigateType.OneToMany, nameof(Desk_002.roomId))]
            public List<Desk_002> desk { get; set; }
        }

        public class Desk_002 
        {
            [SqlSugar.SugarColumn(IsPrimaryKey = true, IsIdentity = true)]
            public int deskid { get; set; }
            public string deskName { get; set; }
            public int roomId { get; set; }
        }
    }
}