autodimension.gradle

import org.w3c.dom.Document
import org.w3c.dom.Element
import org.w3c.dom.Node
import org.w3c.dom.NodeList

import javax.xml.parsers.DocumentBuilder
import javax.xml.parsers.DocumentBuilderFactory
import java.util.regex.Matcher
import java.util.regex.Pattern

public class AutoFactory extends DefaultTask {
  @TaskAction
  def create() {}
}

public enum DimenType {
  FROM_LAYOUT, FROM_DIMEN_FILE, AUTO_CREATE
}

public class DimenFactory extends DefaultTask {
  @Input
  int[] dimens = [320, 360, 384, 411, 480, 540, 600, 720, 800, 1024, 1080, 1280, 1440, 2560, 3840];
  @Input
  int fromDimen = 300;
  @Input
  double positiveMaxDP = 200
  @Input
  double positiveMaxSP = 60
  @Input
  double negativeMaxDP = 60
  @Input
  double negativeMaxSP = 20
  int maxSize = 810;
  @Input
  DimenType type = DimenType.FROM_LAYOUT
  @Input
  String dimenFileName = "values/dimen.xml"

  String resFolder = project.getProjectDir().getPath() + "/src/main/res/"
  String layoutFolder = project.getProjectDir().getPath() + "/src/main/res/layout"
  String layoutLandFolder = project.getProjectDir().getPath() + "/src/main/res/layout-land"

  @TaskAction
  def create() {
    switch (type) {
      case DimenType.FROM_LAYOUT:
        createDimenFromLayout();
        break;
      case DimenType.AUTO_CREATE:
        autoCreateDimen();
        break;
      case DimenType.FROM_DIMEN_FILE:
        createDimenFromDimenFile();
        break;
      default:
        break;
    }
  }

  def createDimenFromDimenFile() {
    println "Start convert dimen from value file to other screen size";
    String path = resFolder + dimenFileName;
    Map<String, String> pairs = new HashMap<>();
    try {
      File fXmlFile = new File(path);
      DocumentBuilderFactory dbFactory = DocumentBuilderFactory.newInstance();
      DocumentBuilder dBuilder = dbFactory.newDocumentBuilder();
      Document doc = dBuilder.parse(fXmlFile);
      doc.getDocumentElement().normalize();

      NodeList nList = doc.getElementsByTagName("dimen");
      for (int temp = 0; temp < nList.getLength(); temp++) {
        Node nNode = nList.item(temp);
        if (nNode.getNodeType() == Node.ELEMENT_NODE) {
          Element eElement = (Element) nNode;
          if(eElement.getTextContent().contains("@dimen")) continue;
          pairs.put(eElement.getAttribute("name"), eElement.getTextContent())
        }
      }
      Map<String, Double> dps = new HashMap<>();
      Map<String, Double> sps = new HashMap<>();

      for (Map.Entry<String, String> entry : pairs) {
        String mVal = entry.getValue();
        mVal = mVal.replaceAll("dp", "").replaceAll("sp", "");
        double val = java.lang.Double.valueOf(mVal);
        if (entry.getValue().contains("dp")) {
          dps.put(entry.getKey(), val);
        } else {
          sps.put(entry.getKey(), val);
        }
      }
      //Sort
      dps = new TreeMap<>(dps);
      sps = new TreeMap<>(sps);
      for (int dimen : dimens) {
        String folder = resFolder + "values-sw" + (int) dimen + "dp";
        String fileName = folder + "/auto_dimens.xml";
        new File(folder).mkdir();
        new File(fileName).createNewFile();
        PrintWriter printWriter = new PrintWriter(fileName);
        printWriter.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>");
        printWriter.println("<resources>");
        for (Map.Entry<String, Double> entry : dps) {
          double ratio = entry.getValue() / fromDimen;
          double sdp = ratio * dimen;
          printWriter.printf("\t<dimen name=\"%s\">%.2fdp</dimen>\r\n", entry.getKey(), sdp);
        }
        for (Map.Entry<String, Double> entry : sps) {
          double ratio = entry.getValue() / fromDimen;
          double sdp = ratio * dimen;
          printWriter.printf("\t<dimen name=\"%s\">%.2fsp</dimen>\r\n", entry.getKey(), sdp);
        }
        printWriter.println("</resources>");
        printWriter.close();
      }
    } catch (Exception e) {
//      println e.getMessage();
      e.printStackTrace();
    }
  }

  def autoCreateDimen() {
    println "Auto create dimension file and values";
    for (int dimen : dimens) {
      String folder = resFolder + "values-sw" + (int) dimen + "dp";
      String fileName = folder + "/auto_dimens_positive.xml";
      new File(folder).mkdir();
      new File(fileName).createNewFile();
      PrintWriter printWriter = new PrintWriter(fileName);
      printWriter.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>");
      printWriter.println("<resources>");
      for (int i = 1; i <= positiveMaxDP; i++) {
        double ratio = i / fromDimen;
        double sdp = ratio * dimen;
        printWriter.printf("\t<dimen name=\"dp_%d\">%.2fdp</dimen>\r\n", i, sdp);
      }
      for (int i = 1; i <= positiveMaxSP; i++) {
        double ratio = i / dimens[0];
        double sdp = ratio * dimen;
        printWriter.printf("\t<dimen name=\"sp_%d\">%.2fsp</dimen>\r\n", i, sdp);
      }
      printWriter.println("</resources>");
      printWriter.close();
    }
    for (int dimen : dimens) {
      String folder = resFolder + "values-sw" + (int) dimen + "dp";
      String fileName = folder + "/auto_dimens_negative.xml";
      new File(folder).mkdir();
      new File(fileName).createNewFile();
      PrintWriter printWriter = new PrintWriter(fileName);
      printWriter.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>");
      printWriter.println("<resources>");
      for (int i = 1; i <= negativeMaxDP; i++) {
        double ratio = i / fromDimen;
        double sdp = ratio * dimen;
        printWriter.printf("\t<dimen name=\"dp_minus%d\">%.2fdp</dimen>\r\n", i, -sdp);
      }
      for (int i = 1; i <= negativeMaxSP; i++) {
        double ratio = i / fromDimen;
        double sdp = ratio * dimen;
        printWriter.printf("\t<dimen name=\"sp_minus%d\">%.2fsp</dimen>\r\n", i, -sdp);
      }
      printWriter.println("</resources>");
      printWriter.close();
    }
  }

  def createDimenFromLayout() {
    println "Create dimen from res/layout dimension values";
    Map<String, String> map = new HashMap<>();
    addToMap(layoutFolder, map);
    //addToMap(layoutLandFolder)
    Set<String> holder = new HashSet<>();
    for (Map.Entry<String, String> entry : map.entrySet()) {
      Set<String> childHolder = new HashSet<>();
      String val = entry.getValue();
      getDpValues(val, childHolder);
      holder.addAll(childHolder);
      for (String s : childHolder) {
        int valInt = java.lang.Integer.parseInt(s.replaceAll("sp", "").replaceAll("dp", ""));
        if (s.contains("dp")) {
          if (valInt < 0) {
            val = val.replaceAll(s, "@dimen/dp_minus" + (-valInt));
          } else {
            val = val.replaceAll(s, "@dimen/dp_" + valInt);
          }
        } else if (s.contains("sp")) {
          if (valInt < 0) {
            val = val.replaceAll(s, "@dimen/sp_minute" + (-valInt));
          } else {
            val = val.replaceAll(s, "@dimen/sp_" + valInt);
          }
        }
      }
      saveToFile(entry.getKey(), val);
    }

    Set<Integer> dps = new HashSet<>();
    Set<Integer> sps = new HashSet<>();
    for (String s : holder) {
      if (s.contains("dp")) {
        dps.add(java.lang.Integer.valueOf(s.replaceAll("dp", "")));
      } else if (s.contains("sp")) {
        sps.add(java.lang.Integer.valueOf(s.replaceAll("sp", "")));
      }
    }
    LinkedList<Integer> listDps = new ArrayList<>(dps);
    Collections.sort(listDps, new Comparator<Integer>() {
      @Override
      int compare(Integer i1, Integer i2) {
        return i1 - i2;
      }
    });
    LinkedList<Integer> listSps = new ArrayList<>(sps);
    Collections.sort(listSps, new Comparator<Integer>() {
      @Override
      int compare(Integer i1, Integer i2) {
        return i1 - i2;
      }
    });
    for (int dimen : dimens) {
      String folder = resFolder + "values-sw" + (int) dimen + "dp";
      String fileName = folder + "/auto_dimens.xml";
      new File(folder).mkdirs();
      new File(fileName).createNewFile();
      PrintWriter printWriter = new PrintWriter(fileName);
      printWriter.println("<?xml version=\"1.0\" encoding=\"utf-8\"?>");
      printWriter.println("<resources>");
      for (int i = 0; i < listDps.size(); i++) {
        int iVal = listDps.get(i);
        double ratio = iVal / fromDimen;
        double sdp = ratio * dimen;
        if (iVal < 0) {
          printWriter.printf("\t<dimen name=\"dp_minus%d\">%.2fdp</dimen>\r\n", -iVal, sdp);
        } else {
          printWriter.printf("\t<dimen name=\"dp_%d\">%.2fdp</dimen>\r\n", iVal, sdp);
        }
      }
      for (int i = 0; i < listSps.size(); i++) {
        int iVal = listSps.get(i);
        double ratio = iVal / fromDimen;
        double sdp = ratio * dimen;
        if (iVal < 0) {
          printWriter.printf("\t<dimen name=\"sp_minus%d\">%.2fsp</dimen>\r\n", -iVal, sdp);
        } else {
          printWriter.printf("\t<dimen name=\"sp_%d\">%.2fsp</dimen>\r\n", iVal, sdp);
        }
      }
      printWriter.println("</resources>");
      printWriter.close();
    }
  }

  void saveToFile(String path, String content) {
    try {
      File file = new File(path);
      FileWriter fw = new FileWriter(file.getAbsoluteFile());
      BufferedWriter bw = new BufferedWriter(fw);
      if (file.exists()) {
        bw.append(content);
      } else {
        bw.write(content);
      }
      bw.close();
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  void getDpValues(String string, Set<String> holder) {
    String[] vals = string.split("\n");
    Pattern pattern = Pattern.compile(".*\\\"(.*)\\\".*");
    for (String val : vals) {
      Matcher matcher = pattern.matcher(val);
      if (!matcher.find()) continue;
      for (int i = 1; i <= matcher.groupCount(); i++) {
        String mVal = matcher.group(i)
        if (mVal != null &&
            (mVal.contains("dp") || mVal.contains("sp")) &&
            !mVal.contains("@dimen")) {
          holder.add(mVal)
        }
      }
    }
  }

  void addToMap(String path, Map<String, String> map) {
    File file = new File(path);
    if (!file.exists()) return;
    if (file.isDirectory()) {
      File[] files = file.listFiles();
      for (File f : files) {
        addToMap(f.getAbsolutePath(), map);
      }
    } else {
      String content = fileToString(path);
      println path;
      map.put(path, content);
    }
  }

  String fileToString(String path) {
    try {
      File f = new File(path);
      FileInputStream inp = new FileInputStream(f);
      byte[] bf = new byte[(int) f.length()];
      inp.read(bf);
      return new String(bf, "UTF-8");
    } catch (FileNotFoundException e) {
      e.printStackTrace();
    } catch (IOException e) {
      e.printStackTrace();
    }
    return null;
  }
}

task createDimen(type: DimenFactory) {}

createDimen {
  dimens = [320, 360, 384, 411, 480, 540, 600, 720, 800, 1024, 1080, 1280, 1440, 2560, 3840];
  fromDimen = 360
  positiveMaxDP = 200 //change to 600 or any other value if needed
  positiveMaxSP = 60 //change to 600 or any other value if needed
  negativeMaxDP = 60 //change to 600 or any other value if needed
  negativeMaxSP = 20 //change to 600 or any other value if needed
  type = DimenType.FROM_DIMEN_FILE
  dimenFileName = 'values-sw360dp/auto_dimens.xml' // name of file, for type = DimenType.FROM_DIMEN_FILE only
}