package com.nailabs.abac.test;

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

import com.nailabs.abac.trust.*;
import com.nailabs.abac.process.*;

public class TextGraphObserver implements Observer {

    HashMap nodeMap = new HashMap(10);
    PrintStream out;
    int count = 0;

    public TextGraphObserver(String name) {
	try {
	    out = new PrintStream(new FileOutputStream(name + ".xml"));
	}
	catch(Exception ex) {
	    ex.printStackTrace(out);
	}
	out.println("<?xml version=\"1.0\"?>\n<Negotiation>");
    }


    public void update(TTGNode node) {
	if(!nodeMap.containsKey(node.getGoal().toString())) {
	    nodeMap.put(node.getGoal().toString(), node);
	    node.addObserver(this);
	    //out.print("<!--Now observing node for ");
	    //out.print(node.getGoal().toString());
	    //out.println("-->");
	}
    }

    public void update(Observable o, Object arg) {
	if(o instanceof TTG) {
	    if(arg instanceof TTGNode) {
		update((TTGNode)arg);
		out.println("<!--updating from TTG-->");
		return;
	    }
	    out.print("<Observation count=\"");
	    out.print(++count);
	    out.println("\">");
	    
	} else {	
	    if(o instanceof TTGNode && !(arg instanceof SatisfactionState)) {
		return;		// skip over all but satisfaction states
	    }
	    out.print("<Observation count=\"");
	    out.print(++count);
	    out.println("\">");
	    if(o instanceof XMLizable) {
		out.print(((XMLizable)o).toXML());
	    } else {
		out.print(o);
	    }
	}
	if(arg instanceof XMLizable) {
	    out.print(((XMLizable)arg).toXML());
	} else {
	    out.print(arg);
	}
	out.println("</Observation>\n");

	out.flush();
    }

    public void close() {
	out.print("\n<!--");
	out.print(count);
	out.println(" total observable events-->");
	out.println("</Negotiation>");
	out.flush();
	out.close();
    }

}