marco.pms.api/Marco.Pms.Services/Controllers/TaskController.cs

using Marco.Pms.DataAccess.Data;
using Marco.Pms.Model.Activities;
using Marco.Pms.Model.Dtos.Activities;
using Marco.Pms.Model.Employees;
using Marco.Pms.Model.Mapper;
using Marco.Pms.Model.Projects;
using Marco.Pms.Model.Utilities;
using Marco.Pms.Model.ViewModels.Activities;
using Marco.Pms.Model.ViewModels.Employee;
using MarcoBMS.Services.Helpers;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Microsoft.EntityFrameworkCore;

namespace MarcoBMS.Services.Controllers
{
    
    [Route("api/[controller]")]
    [ApiController]
    [Authorize]
    public class TaskController : ControllerBase
    {
        private readonly ApplicationDbContext _context;
        private readonly UserHelper _userHelper;


        public TaskController(ApplicationDbContext context, UserHelper userHelper)
        {
            _context = context;
             _userHelper = userHelper;
        }

        private int GetTenantId()
        {
            return _userHelper.GetTenantId();
        }

        [HttpGet]
        [Route("activities")]
        public async Task<IActionResult> GetActivitiesMaster()
        {
            var tenantId = GetTenantId();
            var activities = await _context.ActivityMasters.Where(c => c.TenantId == tenantId).ToListAsync();
            return Ok(ApiResponse<object>.SuccessResponse(activities, "Success.", 200));
        }



        [HttpPost("assign")]
        public async Task<IActionResult> AssignTask([FromBody] AssignTaskDto assignTask)
        {
            if (!ModelState.IsValid)
            {
                var errors = ModelState.Values
                    .SelectMany(v => v.Errors)
                    .Select(e => e.ErrorMessage)
                    .ToList();
                return BadRequest(ApiResponse<object>.ErrorResponse("Invalid data", errors, 400));

            }
            var tenantId = GetTenantId();
            var Employee = await _userHelper.GetCurrentEmployeeAsync();

            var taskAllocation = assignTask.ToTaskAllocationFromAssignTaskDto(Employee.Id,tenantId);
            _context.TaskAllocations.Add(taskAllocation);
            await _context.SaveChangesAsync();
            var response = taskAllocation.ToAssignTaskVMFromTaskAllocation();

            var teamMembers = new List<TaskMembers> { };
            foreach (var teamMember in assignTask.TaskTeam) {
                var result = new TaskMembers
                {
                    TaskAllocationId = taskAllocation.Id,
                    EmployeeId = teamMember,
                    TenantId = tenantId,
                };
                teamMembers.Add(result);
            }
            _context.TaskMembers.AddRange(teamMembers);
            await _context.SaveChangesAsync();

            var idList = teamMembers.Select(m => m.EmployeeId);
            List<Employee> employees = await _context.Employees.Where(e=> idList.Contains(e.Id)).ToListAsync();
            List<EmployeeVM> team = new List<EmployeeVM>();
            foreach (var employee in employees) {
                team.Add(employee.ToEmployeeVMFromEmployee());
            }
            response.teamMembers = team;
            return Ok(ApiResponse<object>.SuccessResponse(response, "Task assignned successfully", 200));
        }


        [HttpPost("report")]
        public async Task<IActionResult> ReportTaskProgress([FromBody]ReportTaskDto reportTask)
        {
            if (!ModelState.IsValid)
            {
                var errors = ModelState.Values
                    .SelectMany(v => v.Errors)
                    .Select(e => e.ErrorMessage)
                    .ToList();
                return BadRequest(ApiResponse<object>.ErrorResponse("Invalid data", errors, 400));

            }
            var tenantId = GetTenantId();
            var Employee = await _userHelper.GetCurrentEmployeeAsync();

            var taskAllocation = await _context.TaskAllocations.Include(t => t.WorkItem).FirstOrDefaultAsync(t => t.Id == reportTask.Id);
            if (taskAllocation == null) {
                return BadRequest(ApiResponse<object>.ErrorResponse("No such task has been allocated.", "No such task has been allocated.", 400));
            }
            if(taskAllocation.CompletedTask != 0)
            {
                taskAllocation.WorkItem.CompletedWork -= taskAllocation.CompletedTask;
            }
            taskAllocation.ReportedDate = reportTask.ReportedDate;
            taskAllocation.CompletedTask = reportTask.CompletedTask;
            taskAllocation.WorkItem.CompletedWork += reportTask.CompletedTask;

            var comment = reportTask.ToCommentFromReportTaskDto(tenantId,Employee.Id);

            _context.TaskComments.Add(comment);
            await _context.SaveChangesAsync();

            var response = taskAllocation.ToReportTaskVMFromTaskAllocation();
            List < TaskComment > comments = await _context.TaskComments.Where(c => c.TaskAllocationId == taskAllocation.Id).ToListAsync();
            List < CommentVM > resultComments = new List<CommentVM> { };
            foreach(var result in comments)
            {
                resultComments.Add(result.ToCommentVMFromTaskComment());
            }
            response.Comments = resultComments;
            return Ok(ApiResponse<object>.SuccessResponse(response, "Task reported successfully", 200));
        }

        [HttpPost("comment")]
        public async Task<IActionResult> AddCommentForTask([FromBody]CreateCommentDto createComment )
        {
            var tenantId = GetTenantId();
            var Employee = await _userHelper.GetCurrentEmployeeAsync();

            var comment = createComment.ToCommentFromCommentDto(tenantId, Employee.Id);
            _context.TaskComments.Add(comment);
            await _context.SaveChangesAsync();

            CommentVM response = comment.ToCommentVMFromTaskComment();
            return Ok(ApiResponse<object>.SuccessResponse(response, "Comment saved successfully", 200));
        }

        [HttpGet("list")]
        public async Task<IActionResult> GetTasksList([FromQuery] int projectId, [FromQuery] string? dateFrom = null, [FromQuery] string? dateTo = null)
        {
            int tenantId = GetTenantId();
            DateTime fromDate = new DateTime();
            DateTime toDate = new DateTime();

            if (dateFrom != null && DateTime.TryParse(dateFrom, out fromDate) == false)
            {
                return BadRequest(ApiResponse<object>.ErrorResponse("Invalid starting date.", "Invalid starting date.", 400));

            }
            if (dateTo != null && DateTime.TryParse(dateTo, out toDate) == false)
            {
                return BadRequest(ApiResponse<object>.ErrorResponse("Invalid ending date.", "Invalid ending date.", 400));

            }
            if (dateFrom == null) fromDate = DateTime.UtcNow.Date;
            if (dateTo == null) toDate = fromDate.AddDays(1);

            //var taskAllocations = await _context.TaskAllocations.Where(t => t.WorkItem.WorkArea.Floor.Building.ProjectId == projectId && t.AssignmentDate >= fromDate && t.AssignmentDate <= toDate && t.TenantId == tenantId).Include(t => t.WorkItemId).ToListAsync();
            List<Building> buildings = await _context.Buildings.Where(b => b.ProjectId == projectId && b.TenantId == tenantId).ToListAsync();
            List<int> idList = buildings.Select(b => b.Id).ToList();

            List<Floor> floors = await _context.Floor.Where(f => idList.Contains(f.BuildingId) && f.TenantId == tenantId).ToListAsync();
            idList = floors.Select(f => f.Id).ToList();

            List<WorkArea> workAreas = await _context.WorkAreas.Where(a => idList.Contains(a.FloorId) && a.TenantId == tenantId).ToListAsync();
            idList = workAreas.Select(a => a.Id).ToList();

            List<WorkItem> workItems = await _context.WorkItems.Where(i => idList.Contains(i.WorkAreaId) && i.TenantId == tenantId).Include(i => i.ActivityMaster).ToListAsync();
            idList = workItems.Select(i => i.Id).ToList();

            List<TaskAllocation> taskAllocations = await _context.TaskAllocations.Where(t => idList.Contains(t.WorkItemId) && t.AssignmentDate >= fromDate && t.AssignmentDate <= toDate && t.TenantId == tenantId).Include(t => t.WorkItem).Include(t=>t.Employee).ToListAsync();
            var taskIdList = taskAllocations.Select(t => t.Id).ToList();
            List<TaskMembers> teamMembers = await _context.TaskMembers.Where(t => taskIdList.Contains(t.TaskAllocationId)).Include(t => t.Employee).ToListAsync();

            List<ListTaskVM> tasks = new List<ListTaskVM>();
            //foreach (var workItem in workItems)
            //{
                foreach (var taskAllocation in taskAllocations)
                {
                    
                    var response = taskAllocation.ToListTaskVMFromTaskAllocation();

                    List<TaskComment> comments = await _context.TaskComments.Where(c => c.TaskAllocationId == taskAllocation.Id).ToListAsync();
                    List<EmployeeVM> team = new List<EmployeeVM>();
                    foreach (var taskMember in teamMembers)
                    {
                        team.Add(taskMember.Employee.ToEmployeeVMFromEmployee());
                    }
                    List<CommentVM> commentVM = new List<CommentVM> { };
                    foreach (var comment in comments)
                    {
                        commentVM.Add(comment.ToCommentVMFromTaskComment());
                    }
                    response.comments = commentVM;
                    response.teamMembers = team;
                    tasks.Add(response);
                }
                
            //}
            return Ok(ApiResponse<object>.SuccessResponse(tasks, "Success", 200));
        }

        [HttpGet("get/{taskId}")]
        public async Task<IActionResult> GetTask(long taskId)
        {
            if (taskId == 0) return BadRequest(ApiResponse<object>.ErrorResponse("Invalid data", "Invalid data", 400));

            var taskAllocation = await _context.TaskAllocations.Include(t => t.Tenant).Include(t => t.Employee).Include(t => t.WorkItem).FirstOrDefaultAsync(t => t.Id == taskId);
            if (taskAllocation != null && taskAllocation.Employee != null && taskAllocation.Tenant != null)
            {
                //var employee = await _context.Employees.FirstOrDefaultAsync(e => e.Id == taskAllocation.AssignedBy);
                string employeeName = System.String.Format("{0} {1}", taskAllocation.Employee.FirstName, taskAllocation.Employee.LastName);
                string tenantName = taskAllocation.Tenant.ContactName;

                if (taskAllocation == null) return NotFound(ApiResponse<object>.ErrorResponse("Task Not Found", "Task not found", 404));
                var taskVM = taskAllocation.TaskAllocationToTaskVM(employeeName,tenantName);
                var comments = await _context.TaskComments.Where(c => c.TaskAllocationId == taskAllocation.Id).ToListAsync();
                var team = await _context.TaskMembers.Where(m => m.TaskAllocationId == taskAllocation.Id).Include(m => m.Employee).ToListAsync();
                var teamMembers = new List<EmployeeVM> { };
                foreach(var member in team)
                {
                    var result = member.Employee.ToEmployeeVMFromEmployee();
                    teamMembers.Add(result);
                }
                List<CommentVM> Comments = new List<CommentVM> { };
                foreach (var comment in comments)
                {
                    Comments.Add(comment.ToCommentVMFromTaskComment());
                }
                taskVM.Comments = Comments;
                taskVM.TeamMembers = teamMembers;
                return Ok(ApiResponse<object>.SuccessResponse(taskVM, "Success", 200));
            }
                
            
            return NotFound(ApiResponse<object>.ErrorResponse("Task Not Found", "Task not Found", 404));
        }
    }
}