/*
 * Copyright (C) 2012 University of Washington
 * 
 * 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.odk.collect.android.widgets;

import android.app.Activity;
import android.content.ActivityNotFoundException;
import android.content.ContentValues;
import android.content.Context;
import android.content.Intent;
import android.graphics.Bitmap;
import android.net.Uri;
import android.provider.MediaStore.Images;
import android.util.Log;
import android.view.Display;
import android.view.View;
import android.view.WindowManager;
import android.view.inputmethod.InputMethodManager;
import android.widget.Button;
import android.widget.ImageView;
import android.widget.LinearLayout;
import android.widget.TableLayout;
import android.widget.TextView;
import android.widget.Toast;

import org.javarosa.core.model.data.IAnswerData;
import org.javarosa.core.model.data.StringData;
import org.javarosa.form.api.FormEntryPrompt;
import org.odk.collect.android.R;
import org.odk.collect.android.activities.DrawActivity;
import org.odk.collect.android.activities.FormEntryActivity;
import org.odk.collect.android.application.Collect;
import org.odk.collect.android.utilities.FileUtils;
import org.odk.collect.android.utilities.MediaUtils;
import org.odk.collect.android.views.ButtonBuilder;

import java.io.File;

/**
 * Image widget that supports annotations on the image.
 *
 * @author BehrAtherton@gmail.com
 * @author Carl Hartung (carlhartung@gmail.com)
 * @author Yaw Anokwa (yanokwa@gmail.com)
 */
public class AnnotateWidget extends QuestionWidget implements IBinaryWidget {
  private final static String t = "AnnotateWidget";

  private Button mCaptureButton;
  private Button mChooseButton;
  private Button mAnnotateButton;
  private ImageView mImageView;

  private String mBinaryName;

  private String mInstanceFolder;

  private TextView mErrorTextView;

  public AnnotateWidget(Context context, FormEntryPrompt prompt) {
    super(context, prompt);

    mInstanceFolder = Collect.getInstance().getFormController()
        .getInstancePath().getParent();

    setOrientation(LinearLayout.VERTICAL);

    TableLayout.LayoutParams params = new TableLayout.LayoutParams();
    params.setMargins(7, 5, 7, 5);

    mErrorTextView = new TextView(context);
    mErrorTextView.setId(QuestionWidget.newUniqueId());
    mErrorTextView.setText("Selected file is not a valid image");

    // setup capture button
    final ButtonBuilder captureButtonBuilder = new ButtonBuilder(context);
    captureButtonBuilder.readOnly(prompt.isReadOnly());
    captureButtonBuilder.withTextResource(R.string.capture_image);
    captureButtonBuilder.withParent(this);
    mCaptureButton = captureButtonBuilder.build();
    mCaptureButton.setLayoutParams(params);

    // launch capture intent on click
    mCaptureButton.setOnClickListener(new View.OnClickListener() {
      @Override
      public void onClick(View v) {
        Collect.getInstance()
            .getActivityLogger()
            .logInstanceAction(this, "captureButton", "click",
                mPrompt.getIndex());
        mErrorTextView.setVisibility(View.GONE);
        Intent i = new Intent(
            android.provider.MediaStore.ACTION_IMAGE_CAPTURE);
        // We give the camera an absolute filename/path where to put the
        // picture because of bug:
        // http://code.google.com/p/android/issues/detail?id=1480
        // The bug appears to be fixed in Android 2.0+, but as of feb 2,
        // 2010, G1 phones only run 1.6. Without specifying the path the
        // images returned by the camera in 1.6 (and earlier) are ~1/4
        // the size. boo.

        // if this gets modified, the onActivityResult in
        // FormEntyActivity will also need to be updated.
        i.putExtra(android.provider.MediaStore.EXTRA_OUTPUT,
            Uri.fromFile(new File(Collect.TMPFILE_PATH)));
        try {
          Collect.getInstance().getFormController()
              .setIndexWaitingForData(mPrompt.getIndex());
          ((Activity) getContext()).startActivityForResult(i,
              FormEntryActivity.IMAGE_CAPTURE);
        } catch (ActivityNotFoundException e) {
          Toast.makeText(
              getContext(),
              getContext().getString(R.string.activity_not_found,
                  "image capture"), Toast.LENGTH_SHORT)
              .show();
          Collect.getInstance().getFormController()
              .setIndexWaitingForData(null);
        }

      }
    });

    // setup chooser button
    final ButtonBuilder chooseButtonBuilder = new ButtonBuilder(context);
    chooseButtonBuilder.withTextResource(R.string.choose_image);
    chooseButtonBuilder.withParent(this);
    chooseButtonBuilder.readOnly(prompt.isReadOnly());
    mChooseButton = chooseButtonBuilder.build();
    mChooseButton.setLayoutParams(params);

    // launch capture intent on click
    mChooseButton.setOnClickListener(new View.OnClickListener() {
      @Override
      public void onClick(View v) {
        Collect.getInstance()
            .getActivityLogger()
            .logInstanceAction(this, "chooseButton", "click",
                mPrompt.getIndex());
        mErrorTextView.setVisibility(View.GONE);
        Intent i = new Intent(Intent.ACTION_GET_CONTENT);
        i.setType("image/*");

        try {
          Collect.getInstance().getFormController()
              .setIndexWaitingForData(mPrompt.getIndex());
          ((Activity) getContext()).startActivityForResult(i,
              FormEntryActivity.IMAGE_CHOOSER);
        } catch (ActivityNotFoundException e) {
          Toast.makeText(
              getContext(),
              getContext().getString(R.string.activity_not_found,
                  "choose image"), Toast.LENGTH_SHORT).show();
          Collect.getInstance().getFormController()
              .setIndexWaitingForData(null);
        }

      }
    });

    // setup Blank Image Button
    final ButtonBuilder annotateButtonBuilder = new ButtonBuilder(context);
    annotateButtonBuilder.readOnly(prompt.isReadOnly());
    annotateButtonBuilder.withParent(this);
    annotateButtonBuilder.withTextResource(R.string.markup_image);
    mAnnotateButton = annotateButtonBuilder.build();
    mAnnotateButton.setLayoutParams(params);
    // launch capture intent on click
    mAnnotateButton.setOnClickListener(new View.OnClickListener() {
      @Override
      public void onClick(View v) {
        Collect.getInstance()
            .getActivityLogger()
            .logInstanceAction(this, "annotateButton", "click",
                mPrompt.getIndex());
        launchAnnotateActivity();
      }
    });

    // finish complex layout
    addView(mCaptureButton);
    addView(mChooseButton);
    addView(mAnnotateButton);
    addView(mErrorTextView);

    // and hide the capture, choose and annotate button if read-only
    if (prompt.isReadOnly()) {
      mCaptureButton.setVisibility(View.GONE);
      mChooseButton.setVisibility(View.GONE);
      mAnnotateButton.setVisibility(View.GONE);
    }
    mErrorTextView.setVisibility(View.GONE);

    // retrieve answer from data model and update ui
    mBinaryName = prompt.getAnswerText();

    // Only add the imageView if the user has taken a picture
    if (mBinaryName != null) {
      if (!prompt.isReadOnly()) {
        mAnnotateButton.setEnabled(true);
      }
      mImageView = new ImageView(getContext());
      mImageView.setId(QuestionWidget.newUniqueId());
      Display display = ((WindowManager) getContext().getSystemService(
          Context.WINDOW_SERVICE)).getDefaultDisplay();
      int screenWidth = display.getWidth();
      int screenHeight = display.getHeight();

      File f = new File(mInstanceFolder + File.separator + mBinaryName);

      if (f.exists()) {
        Bitmap bmp = FileUtils.getBitmapScaledToDisplay(f,
            screenHeight, screenWidth);
        if (bmp == null) {
          mErrorTextView.setVisibility(View.VISIBLE);
        }
        mImageView.setImageBitmap(bmp);
      } else {
        mImageView.setImageBitmap(null);
      }

      mImageView.setPadding(10, 10, 10, 10);
      mImageView.setAdjustViewBounds(true);
      mImageView.setOnClickListener(new View.OnClickListener() {
        @Override
        public void onClick(View v) {
          Collect.getInstance()
              .getActivityLogger()
              .logInstanceAction(this, "viewImage", "click",
                  mPrompt.getIndex());
          launchAnnotateActivity();
        }
      });

      addView(mImageView);
    }
  }

  private void launchAnnotateActivity() {
    mErrorTextView.setVisibility(View.GONE);
    final DrawActivity.DrawActivityDelegate drawActivityDelegate = Collect.getInstance().getDrawActivityDelegate();
    final Context context = getContext();
    final boolean hasDelegate = drawActivityDelegate == null;
    final Intent defaultDrawActivityIntent = new Intent(context, DrawActivity.class);
    Intent i = hasDelegate ? defaultDrawActivityIntent : drawActivityDelegate.getDrawActivityIntent(context);
    i.putExtra(DrawActivity.OPTION, DrawActivity.OPTION_ANNOTATE);
    // copy...
    if (mBinaryName != null) {
      File f = new File(mInstanceFolder + File.separator + mBinaryName);
      i.putExtra(DrawActivity.REF_IMAGE, Uri.fromFile(f));
    }
    i.putExtra(DrawActivity.EXTRA_OUTPUT,
        Uri.fromFile(new File(Collect.TMPFILE_PATH)));

    try {
      Collect.getInstance().getFormController()
          .setIndexWaitingForData(mPrompt.getIndex());
      ((Activity) getContext()).startActivityForResult(i,
          FormEntryActivity.ANNOTATE_IMAGE);
    } catch (ActivityNotFoundException e) {
      Toast.makeText(
          getContext(),
          getContext().getString(R.string.activity_not_found,
              "annotate image"), Toast.LENGTH_SHORT).show();
      Collect.getInstance().getFormController()
          .setIndexWaitingForData(null);
    }
  }

  private void deleteMedia() {
    // get the file path and delete the file
    String name = mBinaryName;
    // clean up variables
    mBinaryName = null;
    // delete from media provider
    int del = MediaUtils.deleteImageFileFromMediaProvider(mInstanceFolder
        + File.separator + name);
    Log.i(t, "Deleted " + del + " rows from media content provider");
  }

  @Override
  public void clearAnswer() {
    // remove the file
    deleteMedia();
    mImageView.setImageBitmap(null);
    mErrorTextView.setVisibility(View.GONE);
    if (!mPrompt.isReadOnly()) {
      mAnnotateButton.setEnabled(false);
    }

    // reset buttons
    mCaptureButton.setText(getContext().getString(R.string.capture_image));
  }

  @Override
  public IAnswerData getAnswer() {
    if (mBinaryName != null) {
      return new StringData(mBinaryName.toString());
    } else {
      return null;
    }
  }

  @Override
  public void setBinaryData(Object newImageObj) {
    // you are replacing an answer. delete the previous image using the
    // content provider.
    if (mBinaryName != null) {
      deleteMedia();
    }

    File newImage = (File) newImageObj;
    if (newImage.exists()) {
      // Add the new image to the Media content provider so that the
      // viewing is fast in Android 2.0+
      ContentValues values = new ContentValues(6);
      values.put(Images.Media.TITLE, newImage.getName());
      values.put(Images.Media.DISPLAY_NAME, newImage.getName());
      values.put(Images.Media.DATE_TAKEN, System.currentTimeMillis());
      values.put(Images.Media.MIME_TYPE, "image/jpeg");
      values.put(Images.Media.DATA, newImage.getAbsolutePath());

      Uri imageURI = getContext().getContentResolver().insert(
          Images.Media.EXTERNAL_CONTENT_URI, values);
      Log.i(t, "Inserting image returned uri = " + imageURI.toString());

      mBinaryName = newImage.getName();
      Log.i(t, "Setting current answer to " + newImage.getName());
    } else {
      Log.e(t, "NO IMAGE EXISTS at: " + newImage.getAbsolutePath());
    }

    Collect.getInstance().getFormController().setIndexWaitingForData(null);
  }

  @Override
  public void setFocus(Context context) {
    // Hide the soft keyboard if it's showing.
    InputMethodManager inputManager = (InputMethodManager) context
        .getSystemService(Context.INPUT_METHOD_SERVICE);
    inputManager.hideSoftInputFromWindow(this.getWindowToken(), 0);
  }

  @Override
  public boolean isWaitingForBinaryData() {
    return mPrompt.getIndex().equals(
        Collect.getInstance().getFormController()
            .getIndexWaitingForData());
  }

  @Override
  public void cancelWaitingForBinaryData() {
    Collect.getInstance().getFormController().setIndexWaitingForData(null);
  }

  @Override
  public void setOnLongClickListener(OnLongClickListener l) {
    mCaptureButton.setOnLongClickListener(l);
    mChooseButton.setOnLongClickListener(l);
    mAnnotateButton.setOnLongClickListener(l);
    if (mImageView != null) {
      mImageView.setOnLongClickListener(l);
    }
  }

  @Override
  public void cancelLongPress() {
    super.cancelLongPress();
    mCaptureButton.cancelLongPress();
    mChooseButton.cancelLongPress();
    mAnnotateButton.cancelLongPress();
    if (mImageView != null) {
      mImageView.cancelLongPress();
    }
  }

}
