package ai.d.ai18;

import drjava.util.Log;
import drjava.util.StringUtil;
import net.luaos.tb.tb15.CentralDatabase;
import net.luaos.tb.tb20.Database;
import net.luaos.tb.tb20.DatabaseAPI;
import net.luaos.tb.tb20.FloraUtil;
import net.luaos.tb.tb20.ListEntry;

import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class GetDiskStats2 {
  public static void main(String[] args) {
    DatabaseAPI db = CentralDatabase.copyToMemory("df", "GetDiskStats2");
    String dfOutputID1 = "#t1-kokjwjhpwuas-qbtsjoyahagl";
    ListEntry dfOutput1 = db.get(dfOutputID1);
    ListEntry rowMarking = db.oneOfType_flex("RowMarking");
    List<ListEntry> colMarkings = db.allOfType("ColMarking");

    DatabaseAPI db2 = CentralDatabase.copyToMemory("df2", "GetDiskStats2");
    ListEntry dfOutput2 = db2.oneOfType_flex("DFOutput");

  /*
(adapts to differing widths in a table. assumes spaces are used as field
separator. looks only at one row.)

function adaptColMarkings(text1, text2, rowMarking, colMarkings)
  row1 = text1.getRow(rowMarking)
  row2 = text2.getRow(rowMarking)
  cols1 = findColumns(row1)
  cols2 = findColumns(row2)
  newColMarkings = []
  for colMarking in colMarkings:
    idx = cols1.indexOf(colMarking)
    col2 = cols2.get(idx)
    add col2 to newColMarkings with label of colMarking
  end
  return newColMarkings
end   */

    String row1 = StringUtil.toLines(dfOutput1.desc).get(rowMarking.getInt("textRow")-1);
    String row2 = StringUtil.toLines(dfOutput2.desc).get(rowMarking.getInt("textRow")-1);
    List<String> cols1 = findColumns(row1);
    List<String> cols2 = findColumns(row2);
    Database dbOut = new Database();
    for (ListEntry colMarking : colMarkings) {
      int idx = findColumn(cols1, colMarking.getPointer("textCol"));
      if (idx < 0)
        Log.surprise("Column not found: " + colMarking.desc);
      else
        dbOut.emit("ColMarking", colMarking.desc, "onText", dfOutput2.id, "textCol", cols2.get(idx));
    }

    FloraUtil.copyNoRewriting(db2, dbOut);
    System.out.println("\nResult DB:");
    FloraUtil.copyNoRewriting(db, dbOut, rowMarking.id);
    FloraUtil.printDatabase(dbOut);

    GetDiskStats.doIt(dbOut, dfOutput2.id);
  }

  private static int findColumn(List<String> cols, String textCol) {
    System.out.println("findColumn " + cols + " " + textCol);
    String start = textCol.split("\\-")[0];
    for (int i = 0; i < cols.size(); i++) {
      String[] fromTo = cols.get(i).split("\\-");
      if (rangeContains(fromTo, start))
        return i;
    }
    return -1;
  }

  private static boolean rangeContains(String[] fromTo, String start) {
    if (Integer.parseInt(start) < Integer.parseInt(fromTo[0])) return false;
    if (fromTo[1].equals("end"))
      return true;
    else if (fromTo[1].endsWith(" (inc)"))
      return Integer.parseInt(start) <= Integer.parseInt(fromTo[1].substring(0, fromTo[1].length() - " (inc)".length()));
    else
      throw new RuntimeException("Unparsable range end: " + fromTo[1]);
  }

  private static List<String> findColumns(String row) {
    int i = 0;
    System.out.println("findColumns: " + row);
    Matcher matcher = Pattern.compile("[^ ]+").matcher(row);
    List<String> list = new ArrayList<String>();
    while (matcher.find()) {
      int start = matcher.start(), end = matcher.end();
      list.add((start+1) + "-" + end + " (inc)");
    }
    return list;
  }
}