package org.zaproxy.zap.extension.bruteforce;

import java.net.URL;

import com.sittinglittleduck.DirBuster.BaseCase;
import com.sittinglittleduck.DirBuster.Manager;

public class DirBusterManager extends Manager {
	
	private BruteForceListenner listenner;
	private int done = 0;
	private int total = 100;

	public DirBusterManager (BruteForceListenner listenner) {
		super();
		this.listenner = listenner;
	}
	
    public synchronized void foundDir(URL url, int statusCode, String responce, String baseCase, String rawResponce, BaseCase baseCaseObj) {
    	super.foundDir(url, statusCode, responce, baseCase, rawResponce, baseCaseObj);
    	System.out.println("DirBusterManager.foundDir " + url.toString() + " code:" + statusCode);
    	listenner.foundDir(url, statusCode, responce, baseCase, rawResponce, baseCaseObj);
    }
    
    public synchronized void foundFile(URL url, int statusCode, String responce, String baseCase, String rawResponce, BaseCase baseCaseObj) {
    	super.foundFile(url, statusCode, responce, baseCase, rawResponce, baseCaseObj);
    	listenner.foundDir(url, statusCode, responce, baseCase, rawResponce, baseCaseObj);
    }

    public synchronized void foundError(URL url, String reason) {
    	super.foundError(url, reason);
    	System.out.println("DirBusterManager.foundError " + url.toString() + " reason:" + reason);
    	
    	(new Exception("")).printStackTrace(); 
    }

    
    public int getTotalDone()
    {
    	if (this.areWorkersAlive()) {
    		done = super.getTotalDone();
    	}
        return done;
    }

    public int getTotal() {
    	if (this.areWorkersAlive()) {
    		long bigTotal = this.getTotalDone() + this.getTotalToDo();
    		if (bigTotal > Integer.MAX_VALUE) {
    			total = Integer.MAX_VALUE;
    		} else if (bigTotal > total){
    			// More work - ignore if less than before - this happens if its stopped early
    			total = (int) bigTotal;
    		}
    		if (total == 0) {
    			// Havt started yet
    			total = 100;
    		}
    	}
    	return total;
    }
    
    private long getTotalToDo() {
        int totalDirs = 1;

        if (this.isRecursive() && this.getDoDirs()) {
            totalDirs = 1 + this.getTotalDirsFound();
        }

        int doingFiles = 1;

        //only if we are doing both files and dirs do we need to times by 2
        if (this.getDoFiles() && this.getDoDirs()) {
            doingFiles = 1 + this.getExtToUse().size();
        } else if ((this.getDoFiles() && !this.getDoDirs())) {
            doingFiles = this.getExtToUse().size();
        }
        long totalToDo = ((long) this.getTotalPass()) * totalDirs * doingFiles;

        //add the number of base cases and the frist inital request
        totalToDo = totalToDo + this.getNumberOfBaseCasesProduced();

        //correct to deal with the intial dir we need to test
        if (this.getDoDirs()) {
            totalToDo = totalToDo + 1;
        }

        //add the process parsed links
        totalToDo = totalToDo + this.getParsedLinksProcessed();

        //correct the total to do but removing items we have skipped
        totalToDo = totalToDo - this.getWorkAmountCorrection();
        
        return totalToDo;
    }

}