/*
 * Copyright 2003-2005 Sun Microsystems, Inc.  All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 *  
 */

import java.io.*;
import java.util.*;

class jvmtiEnvFill {

    public static void main(String[] args) throws IOException {
	if (args.length != 3) {
	    System.err.println("usage: <filledFile> <stubFile> <resultFile>");
	    System.exit(1);
	}
	String filledFN = args[0];
	String stubFN = args[1];
	String resultFN = args[2];

	SourceFile filledSF = new SourceFile(filledFN);
	SourceFile stubSF = new SourceFile(stubFN);


	stubSF.fill(filledSF);

	PrintWriter out = new PrintWriter(new FileWriter(resultFN));
	stubSF.output(out);
	out.close();
    }
}

class SourceFile {

    static final String endFilePrefix = "// end file prefix";
    static final String functionPrefix = "JvmtiEnv::";

    final String fn;
    LineNumberReader in;
    String line;
    List<String> top = new ArrayList<String>();
    List<String> before = new ArrayList<String>();
    boolean inFilePrefix = true;
    List<Function> functions = new ArrayList<Function>();
    Map<String, Function> functionMap = new HashMap<String, Function>();

    class Function {
      String name;
      String args;
      String compareArgs;
      List comment;
      List<String> body = new ArrayList<String>();

      Function() throws IOException {
	line = in.readLine();
	String trimmed = line.trim();
	if (!trimmed.startsWith(functionPrefix)) {
	    error("expected '" + functionPrefix + "'");
	}
	int index = trimmed.indexOf('(', functionPrefix.length());
	if (index == -1) {
	    error("missing open paren");
	}
	name = trimmed.substring(functionPrefix.length(), index);
	int index2 = trimmed.indexOf(')', index);
	if (index2 == -1) {
	    error("missing close paren - must be on same line");
	}
	args = trimmed.substring(index+1, index2);
        compareArgs = args.replaceAll("\\s", "");
	String tail = trimmed.substring(index2+1).trim();	
	if (!tail.equals("{")) {
	    error("function declaration first line must end with open bracket '{', instead got '" +
	           tail + "'");
	}
	while(true) {
	    line = in.readLine();
	    if (line == null) {
		line = ""; // so error does not look wierd
		error("unexpected end of file");
	    }
	    if (line.startsWith("}")) {
		break;
	    }
	    body.add(line);
	}
	String expected = "} /* end " + name + " */";
	trimmed = line.replaceAll("\\s","");
	if (!trimmed.equals(expected.replaceAll("\\s",""))) {
	    error("function end is malformed - should be: " + expected);
	}
	// copy over the comment prefix
	comment = before;
	before = new ArrayList<String>();
      }

      void remove() {
	functionMap.remove(name);
      }

      String fileName() {
	return fn;
      }

      void fill(Function filledFunc) {
	if (filledFunc == null) {
	    System.err.println("Warning: function " + name + " missing from filled file");
	    body.add(0, "    /*** warning: function added and not filled in ***/");
	} else {
            int fbsize = filledFunc.body.size();
            int bsize = body.size();
            if (fbsize > bsize  || !body.subList(bsize-fbsize,bsize).equals(filledFunc.body)) {
                // it has actually been filled in
                body = filledFunc.body;
                if (!compareArgs.equals(filledFunc.compareArgs)) {
                    System.err.println("Warning: function " + name + 
                                       ": filled and stub arguments differ");
                    System.err.println("  old (filled): " + filledFunc.args);
                    System.err.println("  new (stub): " + args);
                    body.add(0, "    /*** warning: arguments changed, were: " +
                             filledFunc.args + " ***/");
                }
            }
            filledFunc.remove();  // mark used
	}
      }

      void output(PrintWriter out) {
	    Iterator it = comment.iterator();
 	    while (it.hasNext()) {
	    	out.println(it.next());
	    }
	    out.println("jvmtiError");
	    out.print(functionPrefix);
	    out.print(name);
	    out.print('(');
	    out.print(args);
	    out.println(") {");
	    it = body.iterator();
 	    while (it.hasNext()) {
	    	out.println(it.next());
	    }
	    out.print("} /* end ");
	    out.print(name);
	    out.println(" */");
      }
    }

    SourceFile(String fn) throws IOException {
	this.fn = fn;
	Reader reader = new FileReader(fn);
	in = new LineNumberReader(reader);

	while (readGaps()) {
	    Function func = new Function();
	    functionMap.put(func.name, func);
	    functions.add(func);
	}
	
	in.close();
    }

    void error(String msg) {
	System.err.println("Fatal error parsing file: " + fn);
	System.err.println("Line number: " + in.getLineNumber());
	System.err.println("Error message: " + msg);
	System.err.println("Source line: " + line);
	System.exit(1);
    }

    boolean readGaps() throws IOException {
	while(true) {
	    line = in.readLine();
	    if (line == null) {
		return false; // end of file
	    }
	    if (!inFilePrefix && line.startsWith("}")) {
		error("unexpected close bracket in first column, outside of function.\n");
	    }
	    String trimmed = line.trim();
	    if (line.startsWith("jvmtiError")) {
		if (trimmed.equals("jvmtiError")) {
		    if (inFilePrefix) {
			error("unexpected 'jvmtiError' line in file prefix.\n" +
			      "is '" + endFilePrefix + "'... line missing?");
		    }
		    return true; // beginning of a function
		} else {
		    error("extra characters at end of 'jvmtiError'");
		}
	    }
	    if (inFilePrefix) {
		top.add(line);
	    } else {
		trimmed = line.trim();
		if (!trimmed.equals("") && !trimmed.startsWith("//") && !trimmed.startsWith("#")) {
		    error("only comments and blank lines allowed between functions");
		}
		before.add(line);
	    }
	    if (line.replaceAll("\\s","").toLowerCase().startsWith(endFilePrefix.replaceAll("\\s",""))) {
		if (!inFilePrefix) {
		    error("excess '" + endFilePrefix + "'");
		}
		inFilePrefix = false;
	    }
	}
    }

    void fill(SourceFile filledSF) {
	// copy beginning of file straight from filled file
	top = filledSF.top;

	// file in functions
	Iterator it = functions.iterator();
	while (it.hasNext()) {
	    Function stubFunc = (Function)(it.next());
	    Function filledFunc = (Function)filledSF.functionMap.get(stubFunc.name);
	    stubFunc.fill(filledFunc);
	}
	if (filledSF.functionMap.size() > 0) {
	    System.err.println("Warning: the following functions were present in the " +
				"filled file but missing in the stub file and thus not copied:");
	    it  = filledSF.functionMap.values().iterator();
	    while (it.hasNext()) {
	    	System.err.println("        " + ((Function)(it.next())).name);
	    }
	}
    }	

    void output(PrintWriter out) {
	Iterator it = top.iterator();
	while (it.hasNext()) {
	    out.println(it.next());
	}
	it = functions.iterator();
	while (it.hasNext()) {
	    Function stubFunc = (Function)(it.next());
	    stubFunc.output(out);
	}
    }
}
