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
{
    [Authorize]
    [Route("api/[controller]")]
    [ApiController]
    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();

            foreach (var employee in employees) {
                response.teamMembers.Add(employee.ToEmployeeVMFromEmployee());
            }
            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));
            }

            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();
            response.Comments = await _context.TaskComments.Where(c => c.TaskAllocationId == taskAllocation.Id).ToListAsync();
            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();

            return Ok(ApiResponse<object>.SuccessResponse(comment, "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 (dateTo == null) toDate = DateTime.UtcNow;
            if (dateFrom == null) fromDate = toDate.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 = workAreas.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).ToListAsync();
            //var taskAllocations = await (from t in _context.TaskAllocations
            //                            join w in _context.WorkItems on t.WorkItemId equals w.Id
            //                            join a in _context.ActivityMasters on w.ActivityId equals a.Id
            //                            where t.WorkItem.WorkArea.Floor.Building.ProjectId == projectId && t.AssignmentDate >= fromDate && t.AssignmentDate <= toDate && t.TenantId == tenantId
            //                            select new {TaskAllocation = t, WorkItems = w, ActivityMasters = a }).ToListAsync();
            List<TasksVM> tasks = new List<TasksVM>();
            foreach (var workItem in workItems)
            {
                foreach (var taskAllocation in taskAllocations)
                {
                    var result = new TasksVM
                    {
                        Id = workItem.ActivityMaster.Id,
                        ActivityName = workItem.ActivityMaster.ActivityName,
                        TotalPlannedTask = workItem.PlannedWork,
                        PlannedTask = taskAllocation.PlannedTask,
                        CompletedTask = taskAllocation.CompletedTask,
                        TotalCopmletedTask = workItem.CompletedWork
                    };
                    tasks.Add(result);
                }
            }
            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);
                }
                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));
        }
    }
}