using System.Collections.Generic;
using Newtonsoft.Json;
using JsonSubTypes;
using System.Linq.Expressions;
using System;
using System.Linq;
using MoreLinq;
using UnityEngine;
using static UnityEditor.PlayerSettings;

public static class SOMDocManager
{
    public static class MMT_OMS_URI
    {
        public static readonly string Point = "http://mathhub.info/MitM/core/geometry?3DGeometry?point";
        public static readonly string Tuple = "http://gl.mathhub.info/MMT/LFX/Sigma?Symbols?Tuple";
        public static readonly string LineType = "http://mathhub.info/MitM/core/geometry?Geometry/Common?line_type";
        public static readonly string LineOf = "http://mathhub.info/MitM/core/geometry?Geometry/Common?lineOf";

        public static readonly string OnLine = "http://mathhub.info/MitM/core/geometry?Geometry/Common?onLine";
        public static readonly string Ded = "http://mathhub.info/MitM/Foundation?Logic?ded";
        public static readonly string Eq = "http://mathhub.info/MitM/Foundation?Logic?eq";
        public static readonly string Metric = "http://mathhub.info/MitM/core/geometry?Geometry/Common?metric";
        public static readonly string Angle = "http://mathhub.info/MitM/core/geometry?Geometry/Common?angle_between";
        public static readonly string Sketch = "http://mathhub.info/MitM/Foundation?InformalProofs?proofsketch";
        public static readonly string RealLit = "http://mathhub.info/MitM/Foundation?RealLiterals?real_lit";

        public static readonly string ParallelLine = "http://mathhub.info/MitM/core/geometry?Geometry/Common?parallelLine";
        //public static readonly string RectangleFact = "http://mathhub.info/FrameIT/frameworld?FrameITRectangles?rectangleType";
        //public static readonly string RectangleFactmk = "http://mathhub.info/FrameIT/frameworld?FrameITRectangles?mkRectangle";

        public static readonly string CircleType3d = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?circleType3D";
        public static readonly string MkCircle3d = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?circle3D";
        public static readonly string TriangleMiddlePoint = "http://mathhub.info/FrameIT/frameworld?FrameITTriangles?triangleMidPointWrapper";
        public static readonly string RadiusCircleMetric = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?circleRadius";

        public static readonly string AreaCircle = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?areaCircle";
        public static readonly string VolumeCone = "http://mathhub.info/FrameIT/frameworld?FrameITCone?volumeCone";
        public static readonly string ConeOfCircleApex = "http://mathhub.info/FrameIT/frameworld?FrameITCone?circleConeOf";

        public static readonly string ParametrizedPlane = "http://mathhub.info/MitM/core/geometry?Planes?ParametrizedPlane";
        public static readonly string pointNormalPlane = "http://mathhub.info/MitM/core/geometry?Planes?pointNormalPlane";
        public static readonly string OnCircle = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?pointOnCircle";
        public static readonly string AnglePlaneLine = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?angleCircleLine";
        public static readonly string OrthoCircleLine = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?orthogonalCircleLine";

        public static readonly string TruncatedVolumeCone = "http://mathhub.info/FrameIT/frameworld?FrameITCone?truncatedConeVolume";
        public static readonly string CylinderVolume = "http://mathhub.info/FrameIT/frameworld?FrameITCylinder?cylinderVolume";
        public static readonly string EqualityCircles = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?equalityCircles";
        public static readonly string UnEqualityCircles = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?unequalityCircles";

        public static readonly string ParallelCircles = "http://mathhub.info/FrameIT/frameworld?FrameITCone?parallelCircles";
        public static readonly string RightAngle = "http://mathhub.info/FrameIT/frameworld?FrameITBasics?rightAngle";

        public static readonly string TestType = "http://mathhub.info/FrameIT/frameworld?FrameITCircle?xcircleType3D";

        public static readonly string Sin = "Sin";
        public static readonly string Cos = "Cos";
        public static readonly string Add = "Add";
        public static readonly string Divide = "Divide";
        public static readonly string Multiply = "Multiply";
        public static readonly string Subtract = "Subtract";
        public static readonly string SquareRoot = "SquareRoot";
        public static readonly string MakeUnityEngineVector3 = "UnityEngine.Vector3";
    }


    public static class SOMDoctoLambdaExpression<T>
    {
        // TODO: Populate Dictionaries
        #region ExpressionDictionaries

        delegate LambdaExpression CustomFunction(LambdaExpression[] args_lamda, ParameterExpression[] bound_params);
        private static readonly Dictionary<string, CustomFunction> MMTtoLambdaMaker = new()
        {
            { MMT_OMS_URI.Sin,
                MakeSin},
            { MMT_OMS_URI.Cos,
                MakeCos},
            { MMT_OMS_URI.SquareRoot,
                MakeCos},
            { MMT_OMS_URI.MakeUnityEngineVector3,
                MakeUnityEngineVector3},
        };

        private static readonly Dictionary<string, ExpressionType> MMTtoBinaryExpressionType = new()
        {
            { MMT_OMS_URI.Add,
                ExpressionType.Add},
            { "AddAssign",
                ExpressionType.AddAssign},
            { "AddAssignChecked",
                ExpressionType.AddAssignChecked},
            { "AddChecked",
                ExpressionType.AddChecked},
            { "And",
                ExpressionType.And},
            { "AndAlso",
                ExpressionType.AndAlso},
            { "AndAssign",
                ExpressionType.AndAssign},
            { "Assign",
                ExpressionType.Assign},
            { MMT_OMS_URI.Divide,
                ExpressionType.Divide},
            { "DivideAssign",
                ExpressionType.DivideAssign},
            { "Equal",
                ExpressionType.Equal},
            { "ExclusiveOr",
                ExpressionType.ExclusiveOr},
            { "ExclusiveOrAssign",
                ExpressionType.ExclusiveOrAssign},
            { "GreaterThan",
                ExpressionType.GreaterThan},
            { "GreaterThanOrEqual",
                ExpressionType.GreaterThanOrEqual},
            { "LeftShift",
                ExpressionType.LeftShift},
            { "LeftShiftAssign",
                ExpressionType.LeftShiftAssign},
            { "LessThan",
                ExpressionType.LessThan},
            { "LessThanOrEqual",
                ExpressionType.LessThanOrEqual},
            { "Modulo",
                ExpressionType.Modulo},
            { "ModuloAssign",
                ExpressionType.ModuloAssign},
            { MMT_OMS_URI.Multiply,
                ExpressionType.Multiply},
            { "MultiplyAssign",
                ExpressionType.MultiplyAssign},
            { "MultiplyAssignChecked",
                ExpressionType.MultiplyAssignChecked},
            { "MultiplyChecked",
                ExpressionType.MultiplyChecked},
            { "NotEqual",
                ExpressionType.NotEqual},
            { "Or",
                ExpressionType.Or},
            { "OrAssign",
                ExpressionType.OrAssign},
            { "OrElse",
                ExpressionType.OrElse},
            { "Power",
                ExpressionType.Power},
            { "PowerAssign",
                ExpressionType.PowerAssign},
            { "RightShift",
                ExpressionType.RightShift},
            { "RightShiftAssign",
                ExpressionType.RightShiftAssign},
            { MMT_OMS_URI.Subtract,
                ExpressionType.Subtract},
            { "SubtractAssign",
                ExpressionType.SubtractAssign},
            { "SubtractAssignChecked",
                ExpressionType.SubtractAssignChecked},
            { "SubtractChecked",
                ExpressionType.SubtractChecked},
        };

        private static readonly Dictionary<string, ExpressionType> MMTtoUnaryExpressionType = new()
        {
            //{ "Constant", // Not Unary
            //    ExpressionType.Constant},
            { "Convert",
                ExpressionType.Convert},
            { "ConvertChecked",
                ExpressionType.ConvertChecked},
            { "Decrement",
                ExpressionType.Decrement},
            { "Increment",
                ExpressionType.Increment},
            { "Negate",
                ExpressionType.Negate},
            { "NegateChecked",
                ExpressionType.NegateChecked},
            { "Not",
                ExpressionType.Not},
            { "OnesComplement",
                ExpressionType.OnesComplement},
            { "PostDecrementAssign",
                ExpressionType.PostDecrementAssign},
            { "PostIncrementAssign",
                ExpressionType.PostIncrementAssign},
            { "PreDecrementAssign",
                ExpressionType.PreDecrementAssign},
            { "PreIncrementAssign",
                ExpressionType.PreIncrementAssign},
            { "UnaryPlus",
                ExpressionType.UnaryPlus},
        };

        #endregion ExpressionDictionaries

        public static LambdaExpression MakeLambdaExpression(string URI, LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
        {
            void ThrowArgumentException(ExpressionType expression_cast, int expected)
            {
                throw new ArgumentException(string.Format(
                    "\"Wrong number of Arguments. Required: {2}. Supplied: {3}.\\n\\tFor URI:\\\"{0}\\\"\\n\\tmapped to:\\\"{1}\\\"\"",
                    URI, expression_cast, expected, lambda_args.Count()
                ));
            }

            if (MMTtoUnaryExpressionType.TryGetValue(URI, out var unnary_type))
            {
                if (bound_params.Count() < 1)
                    ThrowArgumentException(unnary_type, 1);

                Type UnarySecondArgument = bound_params.Count() < 2 ? null : bound_params[1].Type;

                return Expression.Lambda(Expression.MakeUnary(unnary_type, lambda_args[0].Body, UnarySecondArgument), bound_params);
            }
            else if (MMTtoBinaryExpressionType.TryGetValue(URI, out var binary_type))
            {
                if (lambda_args.Count() != 2)
                    ThrowArgumentException(binary_type, 2);

                return Expression.Lambda(Expression.MakeBinary(binary_type, lambda_args[0].Body, lambda_args[1].Body), bound_params);
            }
            else if (MMTtoLambdaMaker.TryGetValue(URI, out var lamda_maker))
            {
                return lamda_maker(lambda_args, bound_params);
            }

            throw new NotImplementedException("Could not map URI: \"" + URI + "\"");
        }

        private static LambdaExpression ExpresionFuncToLambda(LambdaExpression func, string name, LambdaExpression[] args_lamda, ParameterExpression[] bound_params, uint nTargs_fallback)
            => Expression.Lambda(Expression.Invoke(func, args_lamda.Select(l => l.Body)), name, bound_params);

        private static LambdaExpression ParseFuncUUToExpression<U>(Func<U, U> func)
            => (Expression<Func<U, U>>)((U x) => func(x));

        private static LambdaExpression MakeSin(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
            => ExpresionFuncToLambda(
                    default(T) switch // TODO? cleaner switch
                    {
                        float => ParseFuncUUToExpression<float>(MathF.Sin),
                        double => ParseFuncUUToExpression<double>(Math.Sin),
                        _ => throw new NotImplementedException("Sinus for " + typeof(T))
                    },
                    "Sin", lambda_args, bound_params, 1
                );

        private static LambdaExpression MakeCos(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
            => ExpresionFuncToLambda(
                    default(T) switch // TODO? cleaner switch
                    {
                        float => ParseFuncUUToExpression<float>(MathF.Cos),
                        double => ParseFuncUUToExpression<double>(Math.Cos),
                        _ => throw new NotImplementedException("Cosinus for " + typeof(T))
                    },
                    "Cos", lambda_args, bound_params, 1
                );

        private static LambdaExpression MakeUnityEngineVector3(LambdaExpression[] args_lamda, ParameterExpression[] bound_params)
            => ExpresionFuncToLambda(
                    (Expression<Func<float, float, float, Vector3>>)((x, y, z) => new Vector3(x, y, z)),
                    "UnityEngineVector3", args_lamda, bound_params, 3
                );
    }


    [JsonConverter(typeof(JsonSubtypes), "kind")]
    [JsonSubtypes.KnownSubType(typeof(OMA), "OMA")]
    [JsonSubtypes.KnownSubType(typeof(OMBINDC), "OMBINDC")]
    [JsonSubtypes.KnownSubType(typeof(OMS), "OMS")]
    [JsonSubtypes.KnownSubType(typeof(OMSTR), "OMSTR")]
    [JsonSubtypes.KnownSubType(typeof(OMF), "OMF")]
    [JsonSubtypes.KnownSubType(typeof(OMV), "OMV")]
    //[JsonSubtypes.KnownSubType(typeof(OMC<T>), "OMC<" + typeof(T) + ">")]
    [JsonSubtypes.KnownSubType(typeof(OMC<Vector3>), "OMC<UnityEngine.Vector3>")]
    [JsonSubtypes.KnownSubType(typeof(OMC<float>), "OMC<System.Single>")]
    [JsonSubtypes.KnownSubType(typeof(OMC<double>), "OMC<System.Double>")]
    abstract public class SOMDoc
    {
        string kind;

        [JsonIgnore]
        protected static readonly Dictionary<string, Type> StringToType = new(){
            {"R",
                typeof(float)},
            {typeof(float).ToString(),
                typeof(float)},
            {typeof(double).ToString(),
                typeof(double)},
            {typeof(string).ToString(),
                typeof(string)},
            {typeof(UnityEngine.Vector3).ToString(),
                typeof(UnityEngine.Vector3)},
        };
        [JsonIgnore]
        protected static readonly Dictionary<Type, string> TypeToString;

        static SOMDoc()
        {
            TypeToString = new();
            foreach (KeyValuePair<string, Type> KeyVal in StringToType)
                TypeToString.TryAdd(KeyVal.Value, KeyVal.Key);
        }

        protected SOMDoc() { }

        public static bool Equivalent(SOMDoc sd1, SOMDoc sd2)
            => sd1.Equivalent(sd2);

        public abstract bool Equivalent(SOMDoc sd2);

        public LambdaExpression GetLambdaExpression()
            => GetLambdaExpression(new LambdaExpression[0], new ParameterExpression[0]);

        protected internal abstract LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params);

        public abstract override string ToString();

        public abstract SOMDoc MapURIs(Dictionary<string, string> old_to_new);

        #region MakeMMT_OMS_URItoSOMDoc

        public static SOMDoc MakeUnityEngineVector3(Vector3 vec)
            => new OMA(
                new OMS(MMT_OMS_URI.MakeUnityEngineVector3),
                    new() {
                        new OMF(vec.x),
                        new OMF(vec.y),
                        new OMF(vec.z),
                    }
                );

        #endregion MakeMMT_OMS_URItoSOMDoc
    }

    public abstract class SOMDocCRTP<T> : SOMDoc where T : SOMDocCRTP<T>
    {
        protected SOMDocCRTP() : base() { }

        /// \copydoc Fact.Equivalent(Fact)
        public override bool Equivalent(SOMDoc sd2)
            => this.GetType() == sd2.GetType() && (this as T).EquivalentWrapped(sd2 as T);

        protected abstract bool EquivalentWrapped(T sd2);

        public override SOMDoc MapURIs(Dictionary<string, string> old_to_new)
            => MapURIsWrapped(old_to_new);

        public abstract T MapURIsWrapped(Dictionary<string, string> old_to_new);
    }

    public class OMBINDC : SOMDocCRTP<OMBINDC>
    {
        public string kind = "OMBINDC";

        public string name;

        public SOMDoc lambdabody;

        /// <summary>@Type of this variable</summary>
        [JsonIgnore]
        public Type type;

        /// <summary>Enables (especially <see cref="JsonConverter"/>) to read and set <see cref="type"/> by its <c>string</c> representation.</summary>
        public string typeString
        {
            get => TypeToString[type];
            set => type = StringToType[value];
        }

        [JsonConstructor]
        public OMBINDC(string name, string typeString, SOMDoc lambdabody) : base()
        {
            this.name = name;
            this.typeString = typeString;
            this.lambdabody = lambdabody;
        }

        public OMBINDC(string name, Type type, SOMDoc lambdabody) : base()
        {
            this.name = name;
            this.type = type;
            this.lambdabody = lambdabody;
        }

        protected override bool EquivalentWrapped(OMBINDC sd2)
            => this.type == sd2.type && this.name.Equals(sd2.name);

        protected internal override LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
        {
            ParameterExpression[] bind_me = bound_params.ShallowCloneAppend(
                new[] { Expression.Parameter(type, name) }
            );

            return lambdabody.GetLambdaExpression(lambda_args, bind_me);
        }

        public override string ToString()
            => "OMBINDC(" + name + ", " + typeString + ")";

        public override OMBINDC MapURIsWrapped(Dictionary<string, string> old_to_new)
            => new(name, type, lambdabody.MapURIs(old_to_new));
    }

    public class OMA : SOMDocCRTP<OMA>
    {
        public string kind = "OMA";

        public SOMDoc applicant;
        public List<SOMDoc> arguments;

        [JsonConstructor]
        public OMA(SOMDoc applicant, List<SOMDoc> arguments) : base()
        {
            this.applicant = applicant;
            this.arguments = arguments;
        }

        protected override bool EquivalentWrapped(OMA sd2)
            => Equivalent(this.applicant, sd2.applicant)
            && this.arguments
                .Zip(sd2.arguments, (arg1, arg2) => Equivalent(arg1, arg2))
                .All(b => b);

        protected internal override LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
            => applicant.GetLambdaExpression(
                arguments.Select(arg => arg.GetLambdaExpression(lambda_args, bound_params)).ToArray(),
                bound_params
            );

        public override string ToString()
            => applicant.ToString() + "(" + string.Join(", ", arguments.Select(a => a.ToString())) + ")";

        public override OMA MapURIsWrapped(Dictionary<string, string> old_to_new)
            => new OMA(
                applicant.MapURIs(old_to_new),
                arguments.Select(arg => arg.MapURIs(old_to_new)).ToList()
            );
    }

    public class OMS : SOMDocCRTP<OMS> //OMSTR?
    {
        public string kind = "OMS";

        public string uri;

        [JsonConstructor]
        public OMS(string uri) : base()
        {
            this.uri = uri;
        }

        protected override bool EquivalentWrapped(OMS sd2)
            => this.uri == sd2.uri;

        protected internal override LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
            => SOMDoctoLambdaExpression<float>.MakeLambdaExpression(uri, lambda_args, bound_params);

        public override string ToString()
            => uri;

        public override OMS MapURIsWrapped(Dictionary<string, string> old_to_new)
        {
            if (!old_to_new.TryGetValue(uri, out string new_uri))
                new_uri = uri;

            return new OMS(new_uri);
        }
    }

    public class OMSTR : SOMDocCRTP<OMSTR>
    {
        public string kind = "OMSTR";

        [JsonProperty("float")]
        public string s;

        [JsonConstructor]
        public OMSTR(string s) : base()
        {
            this.s = s;
        }

        protected override bool EquivalentWrapped(OMSTR sd2)
            => this.s == sd2.s;

        protected internal override LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
            => Expression.Lambda(Expression.Constant(s, typeof(string)), null);

        public override string ToString()
            => s;

        public override OMSTR MapURIsWrapped(Dictionary<string, string> old_to_new)
            => (OMSTR)this.MemberwiseClone();
    }

    public class OMF : SOMDocCRTP<OMF>
    {
        public string kind = "OMF";

        [JsonProperty("float")]
        public float f;

        [JsonConstructor]
        public OMF(float f) : base()
        {
            this.f = f;
        }

        protected override bool EquivalentWrapped(OMF sd2)
            => this.f.IsApproximatelyEqual(sd2.f);

        protected internal override LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
            => Expression.Lambda(Expression.Constant(f, typeof(float)), null);

        public override string ToString()
            => f.ToString();

        public override OMF MapURIsWrapped(Dictionary<string, string> old_to_new)
            => (OMF)this.MemberwiseClone();
    }

    public class OMC<T> : SOMDocCRTP<OMC<T>>
    {
        public string kind = "OMC<" + typeof(T) + ">";

        public T value;

        [JsonConstructor]
        public OMC(T value) : base()
        {
            this.value = value;
        }

        protected override bool EquivalentWrapped(OMC<T> sd2)
        {
            Debug.LogWarning("Cannot check Equivalency for " + this.GetType() + "; only whether it's exact!");
            return this.value.Equals(value);
        }

        protected internal override LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
            => Expression.Lambda(Expression.Constant(value, typeof(T)), null);

        public override string ToString()
            => "C_" + typeof(T) + "(" + value.ToString() + ")";

        public override OMC<T> MapURIsWrapped(Dictionary<string, string> old_to_new)
            => (OMC<T>)this.MemberwiseClone();
    }

    public class OMV : SOMDocCRTP<OMV>
    {
        public string kind = "OMV";

        public string name;

        [JsonConstructor]
        public OMV(string name) : base()
        {
            this.name = name;
        }

        protected override bool EquivalentWrapped(OMV sd2)
            => this.name == sd2.name;

        protected internal override LambdaExpression GetLambdaExpression(LambdaExpression[] lambda_args, ParameterExpression[] bound_params)
        {
            ParameterExpression v = bound_params.FirstOrDefault(param => param.Name.Equals(name));
            if (v == null)
            {
                Debug.LogErrorFormat("Unable to find {0} for {1} with name: {2}", nameof(OMBINDC), nameof(OMV), name);
                return Expression.Lambda(Expression.Empty(), null);
            }
            else
                return Expression.Lambda(v, new[] { v });
        }

        public override string ToString()
            => "Variable_" + "(" + name + ")";

        public override OMV MapURIsWrapped(Dictionary<string, string> old_to_new)
            => (OMV)this.MemberwiseClone();
    }


    public class MMTDeclaration
    {
        public string label;

        public static MMTDeclaration FromJson(string json)
        {
            MMTDeclaration mmtDecl = JsonConvert.DeserializeObject<MMTDeclaration>(json);
            mmtDecl.label ??= string.Empty;

            return mmtDecl;
        }

        public static string ToJson(MMTDeclaration mmtDecl)
        {
            mmtDecl.label ??= string.Empty;

            string json = JsonConvert.SerializeObject(mmtDecl);
            return json;
        }
    }

    /**
     * MMTSymbolDeclaration: Class for facts without values, e.g. Points
     */
    public class MMTSymbolDeclaration : MMTDeclaration
    {
        public string kind = "general";
        public SOMDoc tp;
        public SOMDoc df;

        /**
         * Constructor used for sending new declarations to mmt
         */
        public MMTSymbolDeclaration(string label, SOMDoc tp, SOMDoc df)
        {
            this.label = label;
            this.tp = tp;
            this.df = df;
        }
    }

    /**
     * MMTValueDeclaration: Class for facts with values, e.g. Distances or Angles
     */
    public class MMTValueDeclaration : MMTDeclaration
    {
        public string kind = "veq";
        public SOMDoc lhs;
        public SOMDoc valueTp;
        public SOMDoc value;

        /**
         * Constructor used for sending new declarations to mmt
         */
        public MMTValueDeclaration(string label, SOMDoc lhs, SOMDoc valueTp, SOMDoc value)
        {
            this.label = label;
            this.lhs = lhs;
            this.valueTp = valueTp;
            this.value = value;
        }
    }
}