src/ScalaHelper.java

   1 // Copyright 2011 the Contributors, as shown in the revision logs.
   2 // Licensed under the Apache Public Source License 2.0 ("the License").
   3 // You may not use this file except in compliance with the License.
   4 import edu.berkeley.sbp.*;
   5 import edu.berkeley.sbp.misc.*;
   6 import edu.berkeley.sbp.util.*;
   7 import edu.berkeley.sbp.meta.*;
   8 import edu.berkeley.sbp.chr.*;
   9 import java.io.*;
  10
  11 public class ScalaHelper {
  12     public static boolean isNull(Object o) { return o==null; }
  13
  14     private static CharParser parser = null;
  15     static {
  16         synchronized(ScalaHelper.class) {
  17             if (parser == null) {
  18                 try {
  19                     // FIXME: bundle this into the jarfile
  20                     InputStream grammarFile = ScalaHelper.class.getClassLoader().getResourceAsStream("wix.g");
  21                     Tree<String> res = new CharParser(GrammarAST.getMetaGrammar())
  22                         .parse(grammarFile).expand1();
  23                     Union grammar = GrammarAST.buildFromAST(res, "s", new GrammarAST.ImportResolver() {
  24                             public InputStream getImportStream(String filename) {
  25                                 return this.getClass().getClassLoader().getResourceAsStream(filename);
  26                             }
  27                         });
  28                     parser = new CharParser(grammar);
  29                 } catch (Exception e) {
  30                     throw new RuntimeException(e);
  31                 }
  32             }
  33         }
  34     }
  35
  36     public static Tree parseFile(String targetFile) throws Throwable {
  37         Tree ret = null;
  38         try {
  39             Reader r = new InputStreamReader(new FileInputStream(targetFile));
  40             Input input = new CharInput(new IndentingReader(r, CharAtom.left, CharAtom.right));
  41             ret = parser.parse(input).expand1();
  42         } catch (Throwable e) {
  43             e.printStackTrace();
  44             throw e;
  45         }
  46         if (ret==null) throw new NullPointerException("CharParser returned null");
  47         return ret;
  48     }
  49
  50     public static void main(String[] argv, TreeToString t2s) throws Throwable {
  51         if (argv.length != 2) {
  52             System.out.println("usage: java -jar wix.jar [-v] <indir> <outdir>");
  53             // FIXME: implement this
  54             System.out.println("     | java -jar wix.jar [-v] <infile>.wix");
  55             System.out.println("");
  56             // FIXME: implement these
  57             System.out.println("   -v   print text as it is parsed (sbp.verbose=true)");
  58             System.out.println("   -vv  like -v, but also dump parse tree");
  59             System.out.println("   -vvv like -vv, but also dump wix tree");
  60             System.exit(-1);
  61             return;
  62         }
  63         File indir = new File(argv[0]);
  64         File outdir = new File(argv[1]);
  65         if (!indir.isDirectory()) {
  66             process(new File(indir.getParent()), indir.getName(), outdir, t2s);
  67         } else {
  68             process(indir, "", outdir, t2s);
  69         }
  70     }
  71
  72     private static void process(File indir, String suffix, File outdir, TreeToString t2s) throws Throwable {
  73         File f = new File(indir.getAbsolutePath()+File.separatorChar+suffix);
  74         //System.out.println(f+" "+indir + " " + suffix + " " + outdir);
  75         if (!f.exists()) return;
  76         if (f.isDirectory()) {
  77             for (String s : f.list())
  78                 process(indir, suffix + File.separatorChar + s, outdir, t2s);
  79             return;
  80         }
  81         if (!f.getPath().endsWith(".wix")) {
  82             boolean skip = false;
  83             if (f.getName().equals(".DS_Store")) skip = true;
  84             if (f.getName().equals("._.DS_Store")) skip = true;
  85             if (f.getName().endsWith("-")) skip = true;
  86             if (!skip) {
  87                 File dest = new File(outdir.getAbsolutePath()+File.separatorChar+suffix);
  88                 if (dest.exists() && dest.lastModified()==f.lastModified() && dest.length()==f.length()) {
  89                     System.out.println(ANSI.yellow("no change: "+f.getPath()));
  90                     return;
  91                 }
  92                 System.out.println(ANSI.green("copying: "+f.getPath()));
  93                 File dest_ = new File(outdir.getAbsolutePath()+File.separatorChar+suffix+"-");
  94                 new File(dest_.getParent()).mkdirs();
  95                 FileOutputStream fos = new FileOutputStream(dest_);
  96                 FileInputStream fis = new FileInputStream(f);
  97                 byte[] buf = new byte[1024];
  98                 while(true) {
  99                     int numread = fis.read(buf, 0, buf.length);
 100                     if (numread==-1) break;
 101                     fos.write(buf, 0, numread);
 102                 }
 103                 fos.close();
 104                 fis.close();
 105                 dest_.renameTo(dest);
 106                 dest.setLastModified(f.lastModified());
 107             }
 108         } else {
 109             String out = "== " + suffix + " ";
 110             while(out.length() < 75) out+="=";
 111             System.out.println(ANSI.yellow(out));
 112             //System.out.println();
 113             String outPath = outdir.getAbsolutePath()+File.separatorChar+suffix;
 114             outPath = outPath.substring(0, outPath.length()-".wix".length())+".html";
 115             if (new File(outPath).exists() && new File(outPath).lastModified() > f.lastModified()) return;
 116
 117             Tree tree = parseFile(f.getAbsolutePath());
 118             String ret = t2s.run(tree);
 119             try {
 120                 new File(new File(outPath).getParent()).mkdirs();
 121                 PrintWriter pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(outPath+"+")));
 122                 pw.println(ret);
 123                 pw.flush();
 124                 pw.close();
 125                 File dest = new File(outPath);
 126                 if (dest.exists()) {
 127                     Process p = Runtime.getRuntime().exec(new String[] {
 128                             "diff",
 129                             "-Bub",
 130                             dest.getAbsolutePath(),
 131                             new File(outPath+"+").getAbsolutePath()
 132                         });
 133                     BufferedReader br = new BufferedReader(new InputStreamReader(p.getInputStream()));
 134                     br.readLine();
 135                     br.readLine();
 136                     for(String s = br.readLine(); s != null; s = br.readLine()) {
 137                         if      (s.startsWith("+")) System.out.println(ANSI.green(s));
 138                         else if (s.startsWith("-")) System.out.println(ANSI.red(s));
 139                         /*else System.out.println(ANSI.blue(s));*/
 140                     }
 141                     p.waitFor();
 142                 }
 143                 new File(outPath+"+").renameTo(dest);
 144                 if (dest.lastModified() <= f.lastModified())
 145                     dest.setLastModified(f.lastModified()+1);
 146             } catch (Exception e) {
 147                 e.printStackTrace();
 148             }
 149         }
 150     }
 151
 152     public static Object ret;
 153     public static void putBack(String o) { ret = o; }
 154 }