/*
 * Autopsy Forensic Browser
 *
 * Copyright 2011-2016 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.modules.android;

import java.util.ArrayList;

import org.sleuthkit.autopsy.casemodule.Case;
import org.sleuthkit.autopsy.casemodule.services.FileManager;
import org.sleuthkit.autopsy.ingest.DataSourceIngestModuleProgress;
import org.sleuthkit.autopsy.ingest.IngestModule;
import org.sleuthkit.datamodel.Content;
import org.sleuthkit.autopsy.ingest.DataSourceIngestModule;
import org.sleuthkit.autopsy.ingest.IngestJobContext;
import org.sleuthkit.autopsy.ingest.IngestMessage;

class AndroidIngestModule implements DataSourceIngestModule {

    private IngestJobContext context = null;

    @Override
    public void startUp(IngestJobContext context) throws IngestModuleException {
        this.context = context;
    }

    @Override
    public ProcessResult process(Content dataSource, DataSourceIngestModuleProgress progressBar) {
        ArrayList<String> errors = new ArrayList<>();
        progressBar.switchToDeterminate(9);
        FileManager fileManager = Case.getCurrentCase().getServices().getFileManager();

        try {
            ContactAnalyzer.findContacts(dataSource, fileManager, context);
            progressBar.progress(1);
            if (context.dataSourceIngestIsCancelled()) {
                return IngestModule.ProcessResult.OK;
            }
        } catch (Exception e) {
            errors.add("Error getting Contacts"); //NON-NLS
        }

        try {
            CallLogAnalyzer.findCallLogs(dataSource, fileManager, context);
            progressBar.progress(2);
            if (context.dataSourceIngestIsCancelled()) {
                return IngestModule.ProcessResult.OK;
            }
        } catch (Exception e) {
            errors.add("Error getting Call Logs"); //NON-NLS
        }

        try {
            TextMessageAnalyzer.findTexts(dataSource, fileManager, context);
            progressBar.progress(3);
            if (context.dataSourceIngestIsCancelled()) {
                return IngestModule.ProcessResult.OK;
            }
        } catch (Exception e) {
            errors.add("Error getting Text Messages"); //NON-NLS
        }

        try {
            TangoMessageAnalyzer.findTangoMessages(dataSource, fileManager, context);
            progressBar.progress(4);
            if (context.dataSourceIngestIsCancelled()) {
                return IngestModule.ProcessResult.OK;
            }
        } catch (Exception e) {
            errors.add("Error getting Tango Messages"); //NON-NLS
        }

        try {
            WWFMessageAnalyzer.findWWFMessages(dataSource, fileManager, context);
            progressBar.progress(5);
            if (context.dataSourceIngestIsCancelled()) {
                return IngestModule.ProcessResult.OK;
            }
        } catch (Exception e) {
            errors.add("Error getting Words with Friends Messages"); //NON-NLS
        }

        try {
            GoogleMapLocationAnalyzer.findGeoLocations(dataSource, fileManager, context);
            progressBar.progress(6);
            if (context.dataSourceIngestIsCancelled()) {
                return IngestModule.ProcessResult.OK;
            }
        } catch (Exception e) {
            errors.add("Error getting Google Map Locations"); //NON-NLS
        }

        try {
            BrowserLocationAnalyzer.findGeoLocations(dataSource, fileManager, context);
            progressBar.progress(7);
            if (context.dataSourceIngestIsCancelled()) {
                return IngestModule.ProcessResult.OK;
            }
        } catch (Exception e) {
            errors.add("Error getting Browser Locations"); //NON-NLS
        }

        try {
            CacheLocationAnalyzer.findGeoLocations(dataSource, fileManager, context);
            progressBar.progress(8);
        } catch (Exception e) {
            errors.add("Error getting Cache Locations"); //NON-NLS
        }

        // create the final message for inbox
        StringBuilder errorMessage = new StringBuilder();
        String errorMsgSubject;
        IngestMessage.MessageType msgLevel = IngestMessage.MessageType.INFO;
        if (errors.isEmpty() == false) {
            msgLevel = IngestMessage.MessageType.ERROR;
            errorMessage.append("Errors were encountered"); //NON-NLS
            for (String msg : errors) {
                errorMessage.append("<li>").append(msg).append("</li>\n"); //NON-NLS
            }
            errorMessage.append("</ul>\n"); //NON-NLS

            if (errors.size() == 1) {
                errorMsgSubject = "One error was found"; //NON-NLS
            } else {
                errorMsgSubject = "errors found: " + errors.size(); //NON-NLS
            }
        } else {
            errorMessage.append("No errors"); //NON-NLS
            errorMsgSubject = "No errors"; //NON-NLS
        }

        return IngestModule.ProcessResult.OK;
    }
}