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

#ifndef COMPONENTS_SAVED_TAB_GROUPS_INTERNAL_ANDROID_TAB_GROUP_SYNC_SERVICE_ANDROID_H_
#define COMPONENTS_SAVED_TAB_GROUPS_INTERNAL_ANDROID_TAB_GROUP_SYNC_SERVICE_ANDROID_H_

#include "base/android/jni_android.h"
#include "base/android/scoped_java_ref.h"
#include "base/memory/raw_ptr.h"
#include "base/supports_user_data.h"
#include "components/saved_tab_groups/internal/android/versioning_message_controller_android.h"
#include "components/saved_tab_groups/public/tab_group_sync_service.h"

using base::android::JavaRef;
using base::android::ScopedJavaGlobalRef;
using base::android::ScopedJavaLocalRef;

namespace tab_groups {

// Helper class responsible for bridging the TabGroupSyncService between
// C++ and Java.
class TabGroupSyncServiceAndroid : public base::SupportsUserData::Data,
                                   public TabGroupSyncService::Observer {
 public:
  explicit TabGroupSyncServiceAndroid(TabGroupSyncService* service);
  ~TabGroupSyncServiceAndroid() override;

  ScopedJavaLocalRef<jobject> GetJavaObject();

  // TabGroupSyncService::Observer overrides.
  void OnInitialized() override;
  void OnTabGroupAdded(const SavedTabGroup& group,
                       TriggerSource source) override;
  void OnTabGroupUpdated(const SavedTabGroup& group,
                         TriggerSource source) override;
  void OnTabGroupRemoved(const LocalTabGroupID& local_id,
                         TriggerSource source) override;
  void OnTabGroupRemoved(const base::Uuid& sync_id,
                         TriggerSource source) override;
  void OnTabGroupLocalIdChanged(
      const base::Uuid& sync_id,
      const std::optional<LocalTabGroupID>& local_id) override;

  // Mutation methods (Java -> native).
  // Mutator methods that result in group metadata mutation.
  void AddGroup(JNIEnv* env, const JavaRef<jobject>& j_saved_tab_group);

  void RemoveGroupByLocalId(JNIEnv* env,
                            const JavaRef<jobject>& j_local_group_id);

  void RemoveGroupBySyncId(JNIEnv* env,
                           const JavaRef<jstring>& j_sync_group_id);

  void UpdateVisualData(JNIEnv* env,
                        const JavaRef<jobject>& j_group_id,
                        const JavaRef<jstring>& j_title,
                        int32_t j_color);

  void MakeTabGroupShared(JNIEnv* env,
                          const JavaRef<jobject>& j_group_id,
                          const JavaRef<jstring>& j_collaboration_id,
                          const JavaRef<jobject>& j_boolean_callback);

  void AboutToUnShareTabGroup(JNIEnv* env,
                              const JavaRef<jobject>& j_group_id,
                              const JavaRef<jobject>& j_callback);

  void OnTabGroupUnShareComplete(JNIEnv* env,
                                 const JavaRef<jobject>& j_group_id,
                                 const bool j_success);

  // Mutator methods that result in tab metadata mutation.
  void AddTab(JNIEnv* env,
              const JavaRef<jobject>& j_group_id,
              int32_t j_tab_id,
              const JavaRef<jstring>& j_title,
              const JavaRef<jobject>& j_url,
              int32_t j_position);

  void UpdateTab(JNIEnv* env,
                 const JavaRef<jobject>& j_group_id,
                 int32_t j_tab_id,
                 const JavaRef<jstring>& j_title,
                 const JavaRef<jobject>& j_url,
                 int32_t j_position);

  void RemoveTab(JNIEnv* env,
                 const JavaRef<jobject>& j_group_id,
                 int32_t j_tab_id);

  void MoveTab(JNIEnv* env,
               const JavaRef<jobject>& j_group_id,
               int32_t j_tab_id,
               int j_new_index_in_group);

  void SetTabSelected(JNIEnv* env,
                      const JavaRef<jobject>& j_group_id,
                      int32_t j_tab_id,
                      const JavaRef<jstring>& j_tab_title);

  // Accessor methods.
  ScopedJavaLocalRef<jobjectArray> GetAllGroupIds(JNIEnv* env);

  ScopedJavaLocalRef<jobject> GetGroupBySyncGroupId(
      JNIEnv* env,
      const JavaRef<jstring>& j_sync_group_id);

  ScopedJavaLocalRef<jobject> GetGroupByLocalGroupId(
      JNIEnv* env,
      const JavaRef<jobject>& j_group_id);

  int32_t GetArchivedGroupCount(JNIEnv* env);

  std::vector<ScopedJavaLocalRef<jobject>> GetDeletedGroupIds(JNIEnv* env);

  // Book-keeping methods to maintain in-memory mapping of sync and local IDs.
  void UpdateLocalTabGroupMapping(JNIEnv* env,
                                  const JavaRef<jstring>& j_sync_id,
                                  const JavaRef<jobject>& j_local_id,
                                  int32_t j_opening_source);
  void RemoveLocalTabGroupMapping(JNIEnv* env,
                                  const JavaRef<jobject>& j_local_id,
                                  int32_t j_closing_source);
  void UpdateLocalTabId(JNIEnv* env,
                        const JavaRef<jobject>& j_group_id,
                        const JavaRef<jstring>& j_sync_tab_id,
                        int32_t j_local_tab_id);
  bool WasTabGroupClosedLocally(JNIEnv* env,
                                const JavaRef<jstring>& j_sync_tab_group_id);

  // Helper methods for attributions.
  bool IsRemoteDevice(JNIEnv* env, const JavaRef<jstring>& j_sync_cache_guid);
  void RecordTabGroupEvent(JNIEnv* env,
                           int32_t j_event_type,
                           const JavaRef<jobject>& j_local_group_id,
                           int32_t j_local_tab_id,
                           int32_t j_opening_source,
                           int32_t j_closing_source);

  void UpdateArchivalStatus(JNIEnv* env,
                            const JavaRef<jstring>& j_sync_group_id,
                            const bool j_archival_status);

  ScopedJavaLocalRef<jobject> GetVersioningMessageController(
      JNIEnv* env,
      const JavaRef<jobject>& j_caller);

  void SetCollaborationAvailableInFinderForTesting(
      JNIEnv* env,
      const JavaRef<jstring>& j_collaboration_id);

 private:
  // A reference to the Java counterpart of this class.  See
  // TabGroupSyncServiceImpl.java.
  ScopedJavaGlobalRef<jobject> java_obj_;

  // Owns the JNI bridge for VersioningMessageController.
  std::unique_ptr<VersioningMessageControllerAndroid>
      versioning_messaging_controller_android_;

  // Not owned. This is safe because the JNI bridge is destroyed and the
  // native pointer in Java is cleared whenever TabGroupSyncService is
  // destroyed. Hence there will be no subsequent unsafe calls to native.
  raw_ptr<TabGroupSyncService> tab_group_sync_service_;
};

}  // namespace tab_groups

#endif  // COMPONENTS_SAVED_TAB_GROUPS_INTERNAL_ANDROID_TAB_GROUP_SYNC_SERVICE_ANDROID_H_