MTWorkHR/MTWorkHR.Application/Middlewares/LoggingMiddleware.cs

using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc.Controllers;
using Microsoft.Data.SqlClient;
using Microsoft.Extensions.Primitives;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Text.Json;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Mvc;
using MTWorkHR.Core.UnitOfWork;
using MTWorkHR.Application.Services;
using MTWorkHR.Core.Global;

namespace MTWorkHR.Application.Middlewares
{
    public class LoggingMiddleware
    {
        private readonly RequestDelegate _next;

        public LoggingMiddleware(RequestDelegate next)
        {
            _next = next;
        }

        public async Task Invoke(HttpContext context, IUnitOfWorkLog unitOfWork)
        {
            try
            {
                await _next(context);
            }
            catch (Exception error)
            {
                //Get target controller info
                var controllerActionDescriptor = context?
                    .GetEndpoint()?
                    .Metadata
                    .GetMetadata<ControllerActionDescriptor>();

                //get controllerName & actionName
                var controllerName = controllerActionDescriptor?.ControllerName;
                var actionName = controllerActionDescriptor?.ActionName;

                //get QueryString
                var req = context?.Request;
                var QueryString = req?.QueryString.Value?.ToString();

                // get request body
                string bodyStr;
                req.EnableBuffering();
                req.Body.Seek(0, SeekOrigin.Begin);
                req.Body.Position = 0;
                using (StreamReader reader
                          = new StreamReader(req.Body, Encoding.UTF8, true, 1024, true))
                {
                    bodyStr = await reader.ReadToEndAsync();
                }


                //**Get Log service and entity info by refliction

                //Get target log entity by the traget controller name
                Type? entityType = Type.GetType("MTWorkHR.Core.Entities." + controllerName + "Log, MTWorkHR.Core");

                var logServiceWithGenericType = typeof(LogService<>).MakeGenericType(entityType);

                dynamic service = Activator.CreateInstance(logServiceWithGenericType, new object[] { unitOfWork });

                var msg = error is AppException ? ((AppException)error).ErrorMessage : error.Message;
                var errorNo = error is AppException ? ((AppException)error).ErrorNumber : "0";
                int errorNum = 0;
                int.TryParse(errorNo, out errorNum); 
                dynamic logEnitity = Activator.CreateInstance(entityType, new object[] {
                    actionName, QueryString, bodyStr, DateTime.Now, "",GetLocalIPAddress(), GetServerIp(context), GetUserExternalIp(context), "", "", msg, error?.InnerException?.Message });


                //finally call service.create to insert the log
                await service.Create(logEnitity);
                //****

                switch (error)
                {
                    case AppException e:
                        {
                            context.Response.Clear();
                            context.Response.ContentType = "text/plain";
                            context.Response.StatusCode = StatusCodes.Status400BadRequest;
                            await context.Response.WriteAsJsonAsync(
                             new BadRequestResult
                             {
                                 ErrorMsg = "*_*" + ((AppException)error).ErrorMessage + "*_*",
                                 ErrorNo = errorNum
                             });
                            return;
                        }
                    default:
                        {
                            context.Response.Clear();
                            context.Response.ContentType = "text/plain";
                            context.Response.StatusCode = StatusCodes.Status500InternalServerError;
                            await context.Response.WriteAsync("*_*" + "Internal Server Error" + "*_*");
                            return;

                        }
                }
            }
        }
        public static string GetServerIp(HttpContext context)
        {
            try
            {
                IPAddress ipAddressString = context.Connection.LocalIpAddress;
                string REMOTE_ADDR = context.GetServerVariable("REMOTE_ADDR");
                string LOCAL_ADDR = context.GetServerVariable("LOCAL_ADDR");
                string SERVER_ADDR = context.GetServerVariable("SERVER_ADDR"); 
                string REMOTE_HOST = context.GetServerVariable("REMOTE_HOST"); 

                string result = "LocalIpAddress: "+ipAddressString.ToString();
                result += "  REMOTE_ADDR: " + REMOTE_ADDR + "  LOCAL_ADDR:" + LOCAL_ADDR
                    + "  SERVER_ADDR:" + SERVER_ADDR + "  REMOTE_HOST:" + REMOTE_HOST;

                return result;
            }
            catch (Exception e)
            {
                return "";
            }
        }
        public static string GetLocalIPAddress()
        {
            try
            {
                var host = Dns.GetHostEntry(Dns.GetHostName());
                foreach (var ip in host.AddressList)
                {

                    if (ip.AddressFamily == AddressFamily.InterNetwork)
                    {
                        return ip.ToString();
                    }
                }
                return "";
            }
            catch (Exception e)
            {
                return "";
            }
        }

        public static string GetUserExternalIp(HttpContext context)
        {
            try
            {
                IPAddress remoteIpAddress = context.Connection.RemoteIpAddress;
                string result = "";
                if (remoteIpAddress != null)
                {
                    // If we got an IPV6 address, then we need to ask the network for the IPV4 address 
                    // This usually only happens when the browser is on the same machine as the server.
                    if (remoteIpAddress.AddressFamily == System.Net.Sockets.AddressFamily.InterNetworkV6)
                    {
                        remoteIpAddress = System.Net.Dns.GetHostEntry(remoteIpAddress).AddressList
                            .First(x => x.AddressFamily == System.Net.Sockets.AddressFamily.InterNetwork);
                    }
                    result = remoteIpAddress.ToString();
                }
                return result;
            }
            catch (Exception e)
            {
                return "";
            }
        }
    }

    public class BadRequestResult
    {
        public string ErrorMsg { get; set; }
        public int ErrorNo { get; set; }

    }
}