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

#import "ios/chrome/browser/intelligence/bwg/utils/gemini_prefs.h"

#import "base/time/time.h"
#import "components/pref_registry/pref_registry_syncable.h"
#import "components/prefs/pref_registry_simple.h"
#import "components/prefs/pref_service.h"
#import "ios/chrome/browser/intelligence/bwg/utils/gemini_constants.h"
#import "ios/chrome/browser/intelligence/features/features.h"
#import "ios/chrome/browser/shared/model/prefs/pref_names.h"

namespace gemini {
bool GeminiAllowedByPolicy(PrefService* prefs) {
  return GenAiAllowedByEnterprise(prefs) && GeminiAllowedByEnterprise(prefs);
}

bool GenAiAllowedByEnterprise(PrefService* prefs) {
  return prefs->GetInteger(::prefs::kGenAiEnabledByPolicy) !=
         static_cast<int>(gemini::GenAiDefaultSettingsPolicy::kNotAllowed);
}

bool GeminiAllowedByEnterprise(PrefService* prefs) {
  return prefs->GetInteger(::prefs::kGeminiEnabledByPolicy) !=
         static_cast<int>(gemini::SettingsPolicy::kNotAllowed);
}

void ResetGeminiConsent(PrefService* prefs) {
  UpdateUserConsentPrefs(false, prefs);
}

FirstRunState CurrentFirstRunState(PrefService* prefs) {
  if (DidUserSeeGeminiPromo(prefs)) {
    return DidUserConsentToGemini(prefs) ? FirstRunState::kCompleted
                                         : FirstRunState::kStarted;
  }

  return FirstRunState::kPending;
}

bool DidUserConsentToGemini(PrefService* prefs) {
  return prefs->GetBoolean(::prefs::kIOSBwgConsent);
}

bool DidUserConsentToGeminiLive(PrefService* prefs) {
  return prefs->GetBoolean(::prefs::kIOSGeminiLiveConsent);
}

bool DidUserSeeGeminiPromo(PrefService* prefs) {
  return prefs->GetInteger(prefs::kIOSBWGPromoImpressionCount) > 0;
}

bool DidGeminiLiveIntroPlay(PrefService* prefs) {
  return prefs->GetBoolean(::prefs::kIOSGeminiLiveIntroPlayed);
}

void SetGeminiLiveIntroPlayed(PrefService* prefs) {
  prefs->SetBoolean(::prefs::kIOSGeminiLiveIntroPlayed, true);
}

void UpdateUserConsentPrefs(bool value, PrefService* prefs) {
  prefs->SetBoolean(::prefs::kIOSBwgConsent, value);
}

void UpdateUserConsentToLivePrefs(bool value, PrefService* prefs) {
  prefs->SetBoolean(::prefs::kIOSGeminiLiveConsent, value);
}

void CreateOrUpdateConversationIdPrefs(const std::string& conversation_id,
                                       const std::string& url_spec,
                                       PrefService* prefs) {
  if (conversation_id.empty()) {
    return;
  }
  prefs->SetTime(prefs::kLastGeminiInteractionTimestamp, base::Time::Now());
  prefs->SetString(prefs::kLastGeminiInteractionURL, url_spec);
  prefs->SetString(prefs::kGeminiConversationId, conversation_id);
}

std::optional<std::string> GetConversationId(PrefService* prefs) {
  base::Time last_interaction_timestamp =
      prefs->GetTime(prefs::kLastGeminiInteractionTimestamp);
  const std::string conversation_id =
      prefs->GetString(prefs::kGeminiConversationId);
  if (base::Time::Now() - last_interaction_timestamp <
      GetGeminiSessionValidityDuration()) {
    if (!conversation_id.empty()) {
      return conversation_id;
    }
  }
  return std::nullopt;
}

void DeleteGeminiSessionInStorage(PrefService* prefs) {
  prefs->ClearPref(prefs::kGeminiConversationId);
}

}  // namespace gemini