// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_THUMBNAIL_GENERATOR_ANDROID_THUMBNAIL_GENERATOR_H_
#define CHROME_BROWSER_THUMBNAIL_GENERATOR_ANDROID_THUMBNAIL_GENERATOR_H_

#include <memory>

#include "base/android/jni_android.h"
#include "base/memory/weak_ptr.h"
#include "chrome/browser/thumbnail/generator/image_thumbnail_request.h"
#include "components/media_gallery_util/public/mojom/media_parser.mojom.h"

class ThumbnailMediaParser;

// Kicks off asynchronous pipelines for creating thumbnails for local files.
// The native-side ThumbnailGenerator is owned by the Java-side and can be
// safely destroyed while a request is being processed.
class ThumbnailGenerator {
 public:
  explicit ThumbnailGenerator(const base::android::JavaRef<jobject>& jobj);

  // Destroys the ThumbnailGenerator.  Any currently running ImageRequest will
  // delete itself when it has completed.
  void Destroy(JNIEnv* env);

  ThumbnailGenerator(const ThumbnailGenerator&) = delete;
  ThumbnailGenerator& operator=(const ThumbnailGenerator&) = delete;

  // Kicks off an asynchronous process to retrieve the thumbnail for the file
  // located at |file_path| with a max size of |icon_size| in each dimension.
  // Invokes the Java #onthumbnailRetrieved(String, int, Bitmap, boolean) method
  // when finished.
  void RetrieveThumbnail(JNIEnv* env,
                         const base::android::JavaRef<jstring>& jcontent_id,
                         const base::android::JavaRef<jstring>& jfile_path,
                         const base::android::JavaRef<jstring>& jmime_type,
                         int32_t icon_size,
                         const base::android::JavaRef<jobject>& callback);

 private:
  ~ThumbnailGenerator();

  // Called when the image thumbnail is ready.  |thumbnail| will be empty on
  // failure.
  void OnImageThumbnailRetrieved(
      base::OnceCallback<void(SkBitmap)> java_callback,
      const SkBitmap& thumbnail);

  // Called when the video thumbnail is ready.  |thumbnail| will be empty on
  // failure.
  void OnVideoThumbnailRetrieved(
      base::OnceCallback<void(SkBitmap)> java_callback,
      int icon_size,
      std::unique_ptr<ThumbnailMediaParser> parser,
      bool success,
      chrome::mojom::MediaMetadataPtr media_metadata,
      SkBitmap thumbnail);

  // This is a {@link ThumbnailGenerator} Java object.
  base::android::ScopedJavaGlobalRef<jobject> java_delegate_;
  base::WeakPtrFactory<ThumbnailGenerator> weak_factory_{this};
};

#endif  // CHROME_BROWSER_THUMBNAIL_GENERATOR_ANDROID_THUMBNAIL_GENERATOR_H_