import java.io.*;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;


public class testing{
	String command;
	BufferedReader reader;
	String url;
	String source;
	boolean writtensrc;
	boolean writtenpage;
	PrintWriter writer;
	PrintWriter printer;
	String str;
	String page;
	
	public void searchCore()throws IOException{	
		Java help=new Java();
		reader=new BufferedReader(new InputStreamReader(System.in));
	    writer=new PrintWriter("C:\\Users\\Jakub\\Desktop\\Source.txt");
	    printer=new PrintWriter("C:\\Users\\Jakub\\Desktop\\Text.txt");
		
		System.out.print("Choose name of website to be tested: ");
		url=reader.readLine();
		Document doc=Jsoup.connect(url).get();
		System.out.println("Connected.");

	do{
		System.out.print(">");
		command=reader.readLine();		    	    	   
//SIMPLE//		
		if(command.equals("?getnode"))
			System.out.println(doc.nodeName());
		
		if(command.equals("?getname"))
			System.out.println(doc.title());
		
		if(command.equals("?gettag"))
			System.out.println(doc.tag());
		
	    if(command.equals("?commands"))
	    	help.commands();
	     
	    if(writtensrc==true&&command.equals("?printsource"))
	    	System.out.println(source);
	    
	    if(writtenpage==true&&command.equals("?printpage"))
	    	System.out.println(page);
//MAIN//		
		if(command.equals("?getsource")){
			source=doc.html();
			writer.write(source);
		    writer.close();
		    writtensrc=true;
		    System.out.println("Done.");
		}	
	    
	    if(command.equals("?getpage")){
	    	page=doc.text();
	    	printer.write(page);
	    	printer.close();
	    	writtenpage=true;
	    	System.out.println("Done.");
	    }
	    
	    if(command.equals("?isvulnerable")){
	    	if(url.contains("id")||url.contains("ID")){
	    		
	    	Document vuln=Jsoup.connect(url+"'").get();
	    	String message=vuln.text();
	    	  	
	    	if(message.contains("sql")||message.contains("Sql")||message.contains("SQL"))
	    		System.out.println("Website is SQLi vulnerable.");
	    	else
	    		System.out.println("No vulnerable place found.");
	    	}
	    	else
	    		System.out.println("No parameter provided.");
	    }
	    
	    if(command.equals("?getall")){
	    	System.out.println("Node: "+doc.nodeName());
	    	System.out.println("Name: "+doc.title());
	    	System.out.println("Tag: "+doc.tag());
	    }
//STR FILTER//	    
	    if(writtensrc==true&&command.equals("?getsourcestr")){
	    	str=reader.readLine();
	    	if(source.contains(str)){
	    		System.out.println("Source code contains *"+str+"*.");
	    	}
	    	else
	    		System.out.println("Source code does not contain *"+str+"*.");
	    	}	
		else if(writtensrc!=true&&(command.equals("?getsourcestr")||command.equals("?printsource")))
	    	System.out.println("Source not found.");
	    
	    
	    
	    if(writtenpage==true&&command.equals("?getpagestr")){
	    	str=reader.readLine();
	    	if(page.contains(str)){
	    		System.out.println("Page text contains *"+str+"*.");
	    	}
	    	else
	    		System.out.println("Page text does not contain *"+str+"*.");
	    	}
		else if(writtenpage!=true&&(command.equals("?getpagestr")||command.equals("?printpage")))
	    	System.out.println("Text not found.");
	    
	}while(command!="quit");
}
		public static void main(String args[])throws IOException{
			testing x=new testing();
			x.searchCore();
		}
}