/*
 * Autopsy Forensic Browser
 * 
 * Copyright 2011 Basis Technology Corp.
 * Contact: carrier <at> sleuthkit <dot> org
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.sleuthkit.autopsy.hashdatabase;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CancellationException;
import java.util.logging.Level;

import org.openide.util.NbBundle;
import org.sleuthkit.autopsy.coreutils.Logger;
import javax.swing.JOptionPane;
import javax.swing.SwingWorker;
import org.netbeans.api.progress.ProgressHandle;
import org.netbeans.api.progress.ProgressHandleFactory;
import org.openide.util.Cancellable;
import org.sleuthkit.datamodel.AbstractFile;
import org.sleuthkit.datamodel.FsContent;

class HashDbSearchThread extends SwingWorker<Object,Void> {
    private Logger logger = Logger.getLogger(HashDbSearchThread.class.getName());
    private ProgressHandle progress;
    private Map<String, List<AbstractFile>> map;
    private ArrayList<String> hashes = new ArrayList<String>();
    private AbstractFile file;
    
    HashDbSearchThread(AbstractFile file) {
        this.file = file;
        this.hashes.add(this.file.getMd5Hash());
    }
    HashDbSearchThread(ArrayList<String> hashes) {
        this.hashes = hashes;
    }

    @Override
    protected Object doInBackground() throws Exception {
        logger.log(Level.INFO, "Starting background processing for file search by MD5 hash.");
            
        // Setup progress bar
        final String displayName = NbBundle.getMessage(this.getClass(), "HashDbSearchThread.name.searching");
        progress = ProgressHandleFactory.createHandle(displayName, new Cancellable() {
            @Override
            public boolean cancel() {
                if (progress != null)
                    progress.setDisplayName(
                            NbBundle.getMessage(this.getClass(), "HashDbSearchThread.progress.cancellingSearch",
                                                displayName));
                return HashDbSearchThread.this.cancel(true);
            }
        });
        // Start the progress bar as indeterminate
        progress.start();
        progress.switchToIndeterminate();
        
        // Do the querying
        map = HashDbSearcher.findFilesBymd5(hashes, progress, this);
        logger.log(Level.INFO, "Done background processing");
        
        return null;
    }
        
    @Override
    protected void done() {
        try {
            super.get(); //block and get all exceptions thrown while doInBackground()
        } catch (CancellationException ex) {
            logger.log(Level.INFO, "File search by MD5 hash was canceled.");
        } catch (InterruptedException ex) {
            logger.log(Level.INFO, "File search by MD5 hash was interrupted.");
        } catch (Exception ex) {
            logger.log(Level.SEVERE, "Fatal error during file search by MD5 hash.", ex);
        } finally {
            progress.finish();
            if (!this.isCancelled()) {
                logger.log(Level.INFO, "File search by MD5 hash completed without cancellation.");
                // If its a right click action, we are given an FsContent which
                // is the file right clicked, so we can remove that from the search
                if(file!=null) {
                    boolean quit = true;
                    for(List<AbstractFile> files: map.values()) {
                        files.remove(file);
                        if(!files.isEmpty()) {
                            quit = false;
                        }
                    }
                    if(quit) {
                        JOptionPane.showMessageDialog(null,
                                                      NbBundle.getMessage(this.getClass(),
                                                                        "HashDbSearchThread.noMoreFilesWithMD5Msg"));
                        return;
                    }
                }
                HashDbSearchManager man = new HashDbSearchManager(map);
                man.execute();
            } else {
                logger.log(Level.INFO, "File search by MD5 hash was canceled.");
            }
        }
    }
    
}