using System.Collections.Generic;

using System.Linq;

using Microsoft.EntityFrameworkCore;

​

​

//---------------------------- MarketInfo.cs------------------------------

​

​

namespace ReviewExercise

{

    public enum EntityType

    {

        Agent,

        Office,

        Firm

    }

    public class MarketInfo

    {

        string InfoCard(EntityType entityType, string entityId)

        {

            var context = new MarketContext();

            string infoCard;

            switch (entityType) {

                case EntityType.Agent:

                    var agent = context.Agents.Find(entityId);

                    infoCard = new string($"Agent: {agent.Name}\r\n" +

                        $"Phone: {agent.Phone}\r\n" +

                        $"Email: {agent.Email}");

                    break;

                case EntityType.Office:

                    var office = context.Offices.Find(entityId);

                    infoCard = new string($"Office: {office.Name}\r\n" +

                        $"Address: {office.Address}\r\n" +

                        $"Phone: {office.Phone}");

                    break;

                case EntityType.Firm:

                    var firm = context.Firms.Find(entityId);

                    infoCard = new string($"firm: {firm.Name}\r\n" +

                        $"Address: {firm.Address}\r\n" +

                        $"Phone: {firm.Phone}");

                    break;

                default:

                    infoCard = "";

                    break;

            }

            return infoCard;

        }

​

        List<string> Top5(EntityType entityType)

        {

            var context = new MarketContext();

            var top5List = new List<string>();

            switch (entityType)

            {

                case EntityType.Agent:

                    top5List = (List<string>)context.Agents

                        .OrderBy(a => a.Sales).Take(5).Select(a => a.Name);

                    break;

                case EntityType.Office:

                    top5List = (List<string>)context.Offices

                        .OrderBy(a => a.Sales).Take(5).Select(a => a.Name);

                    break;

                case EntityType.Firm:

                    top5List = (List<string>)context.Firms

                        .OrderBy(a => a.Sales).Take(5).Select(a => a.Name);

                    break;

            }

            return top5List;

        }

    }

}

​

​

​

​

​

//------------------------------ Model.cs --------------------------------

​

namespace ReviewExercise

{

    public class MarketContext : DbContext

    {

        public DbSet<Agent> Agents { get; set; }

        public DbSet<Office> Offices { get; set; }

        public DbSet<Firm> Firms { get; set; }

​

        protected override void OnConfiguring(DbContextOptionsBuilder options)

            => options.UseSqlServer("Data Source=Market.db");

    }

​

    public class Agent

    {

        public int AgentId { get; set; }

        public string Name { get; set; }

        public string Phone { get; set; }

        public string Email { get; set; }

        public string OfficeId { get; set; }

        public decimal Sales { get; set; }

    }

​

    public class Office

    {

        public int OfficeId { get; set; }

        public string Name { get; set; }

        public string Phone { get; set; }

        public string Address { get; set; }

        public string FirmId { get; set; }

        public decimal Sales { get; set; }

        public List<Agent> Agents { get; } = new List<Agent>();

    }

    public class Firm

    {

        public int FirmId { get; set; }

        public string Name { get; set; }

        public string Phone { get; set; }

        public string Address { get; set; }

        public decimal Sales { get; set; }

        public List<Agent> Offices { get; } = new List<Agent>();

    }

}