import javax.xml.parsers.*;
import javax.xml.transform.*;
import javax.xml.transform.dom.*;
import javax.xml.transform.stream.*;
import org.w3c.dom.*;
import java.lang.*;
import java.util.*;
import java.io.*;
import junit.framework.*;

interface Vili{
  public void Lisa(String nimi, String kogus);
  public boolean Eemalda(String nimi, String mass);
  public boolean Mass(String nimi);
}

class Hoidla implements Vili{
  int i;

  public void Lisa(String nimi, String kogus){
    try {
      Document d=DocumentBuilderFactory.newInstance().newDocumentBuilder().parse("hoidla.xml");
      NodeList esimeneNimi=d.getElementsByTagName("nimi");
      NodeList esimeneMass=d.getElementsByTagName("mass");
      NodeList esimeneNumber=d.getElementsByTagName("loendur");
      NodeList in=d.getElementsByTagName("sisestus");
      for(i=0; i<esimeneNimi.getLength(); i++){
        String vaartus=esimeneNimi.item(i).getFirstChild().getNodeValue();
        String kaal=esimeneMass.item(i).getFirstChild().getNodeValue();
        String number=esimeneNumber.item(i).getFirstChild().getNodeValue();
        String inside=esimeneNumber.item(i).getFirstChild().getNodeValue();
        if(vaartus.equals(nimi)){
          String vastus=Integer.toString(Integer.parseInt(kogus)+Integer.parseInt(kaal));
          String vastus2=Integer.toString(Integer.parseInt(number)+1);
          String sisse=new Date().toString();
          esimeneMass.item(i).getFirstChild().setNodeValue(vastus);
          esimeneNumber.item(i).getFirstChild().setNodeValue(vastus2);
          in.item(i).getFirstChild().setNodeValue(sisse);
          i=esimeneNimi.getLength();
          Transformer t=TransformerFactory.newInstance().newTransformer(); 
          t.transform(new DOMSource(d), new StreamResult(new FileOutputStream("hoidla.xml")));
        }else{
          if(i==esimeneNimi.getLength()-1){
            Element juurikas=d.createElement("vili");         
            d.getFirstChild().appendChild(juurikas);
            Element e=d.createElement("nimi");
            e.appendChild(d.createTextNode(nimi));
            juurikas.appendChild(e);
            e=d.createElement("mass");
            e.appendChild(d.createTextNode(kogus));
            juurikas.appendChild(e);
            e=d.createElement("loendur");
            e.appendChild(d.createTextNode("0"));
            juurikas.appendChild(e);

            Transformer t=TransformerFactory.newInstance().newTransformer(); 
            t.transform(new DOMSource(d), new StreamResult(new FileOutputStream("hoidla.xml")));

          }
        }
      }
    }
      catch (Exception viga){ viga.printStackTrace();
    }
  }
  
  public boolean Eemalda(String nimi, String mass){
    boolean viga2=true;
    try {
      Document d=DocumentBuilderFactory.newInstance().newDocumentBuilder().parse("hoidla.xml");
      NodeList esimeneNimi=d.getElementsByTagName("nimi");
      NodeList esimeneMass=d.getElementsByTagName("mass");
      NodeList esimeneNumber=d.getElementsByTagName("loendur");
      
      NodeList out=d.getElementsByTagName("eemaldus");
      for(i=0; i<esimeneNimi.getLength(); i++){
        String vaartus=esimeneNimi.item(i).getFirstChild().getNodeValue();
        String kaal=esimeneMass.item(i).getFirstChild().getNodeValue();
        String number=esimeneNumber.item(i).getFirstChild().getNodeValue();
        String outside=esimeneNumber.item(i).getFirstChild().getNodeValue();
        if(vaartus.equals(nimi)){
          if(Integer.parseInt(kaal)>Integer.parseInt(mass)){
            String vastus=Integer.toString(Integer.parseInt(kaal)-Integer.parseInt(mass));
            String vastus2=Integer.toString(Integer.parseInt(number)+1);
            String valja=new Date().toString();
            esimeneMass.item(i).getFirstChild().setNodeValue(vastus);
            esimeneNumber.item(i).getFirstChild().setNodeValue(vastus2);
            out.item(i).getFirstChild().setNodeValue(valja);
          }else{
            viga2=false;
          }
          i=esimeneNimi.getLength();
        }else{
          if(i==esimeneNimi.getLength()-1){
            viga2=false;
          }
        }
      }
      Transformer t=TransformerFactory.newInstance().newTransformer(); 
      t.transform(new DOMSource(d), new StreamResult(new FileOutputStream("hoidla.xml")));
    }
      catch (Exception e){ e.printStackTrace();
    }
    
    return viga2;
  }
  
  public boolean Mass(String nimi){
    boolean vastus=false;
    try {
      Document d=DocumentBuilderFactory.newInstance().newDocumentBuilder().parse("hoidla.xml");
      NodeList esimeneNimi=d.getElementsByTagName("nimi");
      NodeList esimeneMass=d.getElementsByTagName("mass");
      for(i=0; i<esimeneNimi.getLength(); i++){
        String vaartus=esimeneNimi.item(i).getFirstChild().getNodeValue();
        String kaal=esimeneMass.item(i).getFirstChild().getNodeValue();
        if(vaartus.equals(nimi)){
          System.out.println("Teravilja nimega "+vaartus+" on laos "+kaal+" tonni");
          vastus=true;
          i=esimeneNimi.getLength();
        }else{
          vastus=false;
        }
      }           
    }
      catch (Exception viga){ viga.printStackTrace(); 
    }
    return vastus;
  }
}

public class eksam extends TestCase{

  public void testFail() throws Exception{

    Hoidla kola=new Hoidla();
    kola.Eemalda("oder", "10");
    kola.Mass("Mais");


    assertFalse(kola.Mass("Lehm"));
    assertTrue(kola.Mass("mais"));
    assertTrue(kola.Eemalda("oder", "1200"));


  }

  public static void main(String[] argumendid){
    junit.textui.TestRunner.run(new TestSuite(eksam.class));
  }
}