using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Text;
using System.Reflection;
using Microsoft.Samples.Debugging.CorSymbolStore;
using System.Diagnostics.SymbolStore;
using System.IO;
using System.Diagnostics;

namespace Cosmos.Debug.Common {

  public class SourceInfos : SortedList<uint, SourceInfo> {
    public SourceInfo GetMapping(uint aValue) {
      for (int i = Count - 1; i >= 0; i--) {
        if (Keys[i] <= aValue) {
          return Values[i];
        }
      }
      return null;
    }
  }

  public class SourceInfo {
    public string MethodName { get; set; }
    public string SourceFile { get; set; }
    public int Line { get; set; }
    public int Column { get; set; }
    public int LineEnd { get; set; }
    public int ColumnEnd { get; set; }

    #region Old NASM Map parser
    //public static List<Label> ParseMapFile(String buildPath) {
    //  var xSourceStrings = File.ReadAllLines(Path.Combine(buildPath, "main.map"));
    //  var xSource = new List<Label>();
    //  uint xIndex = 0;
    //  for (xIndex = 0; xIndex < xSourceStrings.Length; xIndex++) {
    //    if (xSourceStrings[xIndex].StartsWith("Real ")) {
    //      // further check it:
    //      //Virtual   Name"))
    //      if (!xSourceStrings[xIndex].Substring(4).TrimStart().StartsWith("Virtual ")
    //          || !xSourceStrings[xIndex].EndsWith(" Name")) {
    //        continue;
    //      }
    //      xIndex++;
    //      break;
    //    }
    //  }
    //  for (; xIndex < xSourceStrings.Length; xIndex++) {
    //    string xLine = xSourceStrings[xIndex];
    //    var xLineParts = xLine.Split(new[] { ' ' }, StringSplitOptions.RemoveEmptyEntries);

    //    if (xLineParts.Length == 3) {
    //      uint xAddress = UInt32.Parse(xLineParts[0], System.Globalization.NumberStyles.HexNumber);
    //      xSource.Add(new Label() {
    //        LABELNAME = xLineParts[2],
    //        ADDRESS = xAddress
    //      });
    //    }
    //  }
    //  return xSource;
    //}
    #endregion

    public static int GetIndexClosestSmallerMatch(IList<int> aList, int aValue) {
      int xIdx = -1;
      for (int i = 0; i < aList.Count; i++) {
        if (aList[i] <= aValue) {
          xIdx = i;
        } else {
          break;
        }
      }
      return xIdx;
    }

    public static SourceInfos GetSourceInfo(DebugInfo aDbgInfo) {
      var xResult = new SourceInfos();
      List<MLSYMBOL> xSymbols;
      using (var xDB = aDbgInfo.DB()) {
        xSymbols = xDB.MLSYMBOLs.ToList();

        // Sort
        xSymbols.Sort(delegate(MLSYMBOL a, MLSYMBOL b) {
          if (a == null) {
            throw new ArgumentNullException("a");
          } else if (b == null) {
            throw new ArgumentNullException("b");
          }
          int xCompareResult = StringComparer.InvariantCultureIgnoreCase.Compare(a.ILASMFILE, b.ILASMFILE);
          if (xCompareResult == 0) {
            xCompareResult = a.TYPETOKEN.CompareTo(b.TYPETOKEN);
            if (xCompareResult == 0) {
              xCompareResult = a.METHODTOKEN.CompareTo(b.METHODTOKEN);
              if (xCompareResult == 0) {
                return a.ILOFFSET.CompareTo(b.ILOFFSET);
              }

            }
          }
          return xCompareResult;
        });

        string xOldAssembly = null;
        ISymbolReader xSymbolReader = null;
        int[] xCodeOffsets = null;
        ISymbolDocument[] xCodeDocuments = null;
        int[] xCodeLines = null;
        int[] xCodeColumns = null;
        int[] xCodeEndLines = null;
        int[] xCodeEndColumns = null;
        int? xOldMethodToken = null;
        ISymbolMethod xMethodSymbol = null;

        foreach (var xSymbol in xSymbols) {
          if (!xSymbol.ILASMFILE.Equals(xOldAssembly, StringComparison.InvariantCultureIgnoreCase)) {
            try {
              xMethodSymbol = null;
              xSymbolReader = SymbolAccess.GetReaderForFile(xSymbol.ILASMFILE);
            } catch {
              xSymbolReader = null;
              xMethodSymbol = null;
            }
            xOldAssembly = xSymbol.ILASMFILE;
          }
          if (xOldMethodToken != xSymbol.METHODTOKEN) {
            if (xSymbolReader != null) {
              try {
                xMethodSymbol = xSymbolReader.GetMethod(new SymbolToken(xSymbol.METHODTOKEN));
                if (xMethodSymbol != null) {
                  xCodeOffsets = new int[xMethodSymbol.SequencePointCount];
                  xCodeDocuments = new ISymbolDocument[xMethodSymbol.SequencePointCount];
                  xCodeLines = new int[xMethodSymbol.SequencePointCount];
                  xCodeColumns = new int[xMethodSymbol.SequencePointCount];
                  xCodeEndLines = new int[xMethodSymbol.SequencePointCount];
                  xCodeEndColumns = new int[xMethodSymbol.SequencePointCount];
                  xMethodSymbol.GetSequencePoints(xCodeOffsets, xCodeDocuments, xCodeLines, xCodeColumns, xCodeEndLines, xCodeEndColumns);
                }
              } catch {
                xMethodSymbol = null;
              }
            }
            xOldMethodToken = xSymbol.METHODTOKEN;
          }

          if (xMethodSymbol != null) {
            var xRow = xDB.Labels.SingleOrDefault(q => q.LABELNAME == xSymbol.LABELNAME);
            if (xRow != null) {
              UInt32 xAddress = (UInt32)xRow.ADDRESS;
              // Each address could have mult labels, but this wont matter for SourceInfo, its not tied to label.
              // So we just ignore duplicate addresses.
              if (!xResult.ContainsKey(xAddress)) {
                int xIdx = GetIndexClosestSmallerMatch(xCodeOffsets, xSymbol.ILOFFSET);
                var xSourceInfo = new SourceInfo() {
                  SourceFile = xCodeDocuments[xIdx].URL,
                  Line = xCodeLines[xIdx],
                  LineEnd = xCodeEndLines[xIdx],
                  Column = xCodeColumns[xIdx],
                  ColumnEnd = xCodeEndColumns[xIdx],
                  MethodName = xSymbol.METHODNAME
                };
                xResult.Add(xAddress, xSourceInfo);
              }
            }
          }
        }
      }
      return xResult;
    }
  }
}