✨ Implement DTO + mapper profile in controller

cat_cafe/Controllers/CatsController.cs

@@ -7,6 +7,8 @@ using Microsoft.AspNetCore.Mvc;
 using Microsoft.EntityFrameworkCore;
 using cat_cafe.Entities;
 using cat_cafe.Repositories;
+using AutoMapper;
+using cat_cafe.Dto;
 
 namespace cat_cafe.Controllers
 {
@@ -15,22 +17,26 @@ namespace cat_cafe.Controllers
     public class CatsController : ControllerBase
     {
         private readonly CatContext _context;
+        private readonly IMapper _mapper;
 
-        public CatsController(CatContext context)
+        public CatsController(CatContext context, IMapper mapper)
         {
+            _mapper = mapper;
             _context = context;
         }
 
         // GET: api/Cats
         [HttpGet]
-        public async Task<ActionResult<IEnumerable<Cat>>> GetCats()
+        public async Task<ActionResult<IEnumerable<CatDto>>> GetCats()
         {
-            return await _context.Cats.ToListAsync();
+            var cats = await _context.Cats.ToListAsync();
+            return _mapper.Map<List<CatDto>>(cats);
+
         }
 
         // GET: api/Cats/5
         [HttpGet("{id}")]
-        public async Task<ActionResult<Cat>> GetCat(long id)
+        public async Task<ActionResult<CatDto>> GetCat(long id)
         {
             var cat = await _context.Cats.FindAsync(id);
 
@@ -39,19 +45,20 @@ namespace cat_cafe.Controllers
                 return NotFound();
             }
 
-            return cat;
+            return _mapper.Map<CatDto>(cat);
         }
 
         // PUT: api/Cats/5
         // To protect from overposting attacks, see https://go.microsoft.com/fwlink/?linkid=2123754
         [HttpPut("{id}")]
-        public async Task<IActionResult> PutCat(long id, Cat cat)
+        public async Task<IActionResult> PutCat(long id, CatDto catDto)
         {
-            if (id != cat.Id)
+            if (id != catDto.Id)
             {
                 return BadRequest();
             }
 
+            Cat cat = _mapper.Map<Cat>(catDto);
             _context.Entry(cat).State = EntityState.Modified;
 
             try
@@ -76,12 +83,13 @@ namespace cat_cafe.Controllers
         // POST: api/Cats
         // To protect from overposting attacks, see https://go.microsoft.com/fwlink/?linkid=2123754
         [HttpPost]
-        public async Task<ActionResult<Cat>> PostCat(Cat cat)
+        public async Task<ActionResult<CatDto>> PostCat(CatDto catDto)
         {
+            Cat cat = _mapper.Map<Cat>(catDto);
             _context.Cats.Add(cat);
             await _context.SaveChangesAsync();
 
-            return CreatedAtAction("GetCat", new { id = cat.Id }, cat);
+            return CreatedAtAction("GetCat", new { id = catDto.Id }, _mapper.Map<CatDto>(cat));
         }
 
         // DELETE: api/Cats/5

cat_cafe/Dto/CatDto.cs

@@ -5,6 +5,5 @@ namespace cat_cafe.Dto
     {
         public long Id { get; set; }
         public string? Name { get; set; }
-        public int Age { get; set; } = 0;
     }
 }

cat_cafe/Mappers/CatMapper.cs

@@ -1,7 +1,14 @@
-namespace cat_cafe.Mappers
+using AutoMapper;
+using cat_cafe.Dto;
+using cat_cafe.Entities;
+
+namespace cat_cafe.Mappers
 {
-    public class CatMapper
+    public class CatMapper : Profile
     {
-
+        public CatMapper()
+        {
+            CreateMap<Cat, CatDto>().ReverseMap();
+        }
     }
 }

cat_cafe/Program.cs

@@ -14,6 +14,9 @@ opt.UseInMemoryDatabase("CatCafe"));
 builder.Services.AddEndpointsApiExplorer();
 builder.Services.AddSwaggerGen();
 
+builder.Services.AddAutoMapper(typeof(Program));
+builder.Services.AddControllersWithViews();
+
 var app = builder.Build();
 
 // Configure the HTTP request pipeline.