| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/android/ntp/ntp_snippets_bridge.h" |
| |
| #include <jni.h> |
| #include <utility> |
| #include <vector> |
| |
| #include "base/android/callback_android.h" |
| #include "base/android/jni_android.h" |
| #include "base/android/jni_array.h" |
| #include "base/android/jni_string.h" |
| #include "base/callback.h" |
| #include "base/time/time.h" |
| #include "chrome/browser/history/history_service_factory.h" |
| #include "chrome/browser/ntp_snippets/content_suggestions_service_factory.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/browser/profiles/profile_android.h" |
| #include "chrome/browser/profiles/profile_manager.h" |
| #include "components/history/core/browser/history_service.h" |
| #include "components/ntp_snippets/content_suggestion.h" |
| #include "components/ntp_snippets/content_suggestions_metrics.h" |
| #include "components/ntp_snippets/remote/remote_suggestions_provider.h" |
| #include "jni/SnippetsBridge_jni.h" |
| #include "ui/base/window_open_disposition.h" |
| #include "ui/gfx/android/java_bitmap.h" |
| #include "ui/gfx/image/image.h" |
| |
| using base::android::AttachCurrentThread; |
| using base::android::ConvertJavaStringToUTF8; |
| using base::android::ConvertUTF8ToJavaString; |
| using base::android::ConvertUTF16ToJavaString; |
| using base::android::JavaIntArrayToIntVector; |
| using base::android::AppendJavaStringArrayToStringVector; |
| using base::android::JavaParamRef; |
| using base::android::ScopedJavaGlobalRef; |
| using base::android::ScopedJavaLocalRef; |
| using base::android::ToJavaIntArray; |
| using ntp_snippets::Category; |
| using ntp_snippets::CategoryInfo; |
| using ntp_snippets::CategoryStatus; |
| using ntp_snippets::KnownCategories; |
| using ntp_snippets::ContentSuggestion; |
| |
| namespace { |
| |
| // TODO(treib): Move this into the Time class itself. |
| base::Time TimeFromJavaTime(jlong timestamp_ms) { |
| return base::Time::UnixEpoch() + |
| base::TimeDelta::FromMilliseconds(timestamp_ms); |
| } |
| |
| // Converts a vector of ContentSuggestions to its Java equivalent. |
| ScopedJavaLocalRef<jobject> ToJavaSuggestionList( |
| JNIEnv* env, |
| const Category& category, |
| const std::vector<ContentSuggestion>& suggestions) { |
| Profile* profile = ProfileManager::GetLastUsedProfile(); |
| ntp_snippets::ContentSuggestionsService* content_suggestions_service = |
| ContentSuggestionsServiceFactory::GetForProfile(profile); |
| |
| // Get layout for the category. |
| base::Optional<CategoryInfo> info = |
| content_suggestions_service->GetCategoryInfo(category); |
| DCHECK(info); |
| |
| ScopedJavaLocalRef<jobject> result = |
| Java_SnippetsBridge_createSuggestionList(env); |
| for (const ContentSuggestion& suggestion : suggestions) { |
| Java_SnippetsBridge_addSuggestion( |
| env, result, category.id(), |
| ConvertUTF8ToJavaString(env, suggestion.id().id_within_category()), |
| ConvertUTF16ToJavaString(env, suggestion.title()), |
| ConvertUTF16ToJavaString(env, suggestion.publisher_name()), |
| ConvertUTF16ToJavaString(env, suggestion.snippet_text()), |
| ConvertUTF8ToJavaString(env, suggestion.url().spec()), |
| ConvertUTF8ToJavaString(env, suggestion.amp_url().spec()), |
| suggestion.publish_date().ToJavaTime(), suggestion.score(), |
| static_cast<int>(info->card_layout())); |
| if (suggestion.id().category().IsKnownCategory( |
| KnownCategories::DOWNLOADS) && |
| suggestion.download_suggestion_extra() != nullptr && |
| suggestion.download_suggestion_extra()->is_download_asset) { |
| Java_SnippetsBridge_setDownloadAssetDataForLastSuggestion( |
| env, result, |
| ConvertUTF8ToJavaString( |
| env, |
| suggestion.download_suggestion_extra()->target_file_path.value()), |
| ConvertUTF8ToJavaString( |
| env, suggestion.download_suggestion_extra()->mime_type)); |
| } |
| if (suggestion.id().category().IsKnownCategory( |
| KnownCategories::RECENT_TABS) && |
| suggestion.recent_tab_suggestion_extra() != nullptr) { |
| Java_SnippetsBridge_setRecentTabDataForLastSuggestion( |
| env, result, |
| ConvertUTF8ToJavaString( |
| env, suggestion.recent_tab_suggestion_extra()->tab_id), |
| ConvertUTF8ToJavaString( |
| env, suggestion.recent_tab_suggestion_extra()->offline_page_id)); |
| } |
| } |
| |
| return result; |
| } |
| |
| } // namespace |
| |
| static jlong Init(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| const JavaParamRef<jobject>& j_profile) { |
| NTPSnippetsBridge* snippets_bridge = new NTPSnippetsBridge(env, j_profile); |
| return reinterpret_cast<intptr_t>(snippets_bridge); |
| } |
| |
| static void FetchSnippets(JNIEnv* env, |
| const JavaParamRef<jclass>& caller, |
| jboolean j_force_request) { |
| Profile* profile = ProfileManager::GetLastUsedProfile(); |
| // Temporary check while investigating crbug.com/647920. |
| CHECK(profile); |
| |
| ntp_snippets::ContentSuggestionsService* content_suggestions_service = |
| ContentSuggestionsServiceFactory::GetForProfile(profile); |
| |
| // Can maybe be null in some cases? (Incognito profile?) crbug.com/647920 |
| if (!content_suggestions_service) |
| return; |
| |
| ntp_snippets::RemoteSuggestionsProvider* service = |
| content_suggestions_service->ntp_snippets_service(); |
| |
| // Can be null if the feature has been disabled but the scheduler has not been |
| // unregistered yet. The next start should unregister it. |
| if (!service) |
| return; |
| |
| service->FetchSnippets(j_force_request); |
| } |
| |
| // Reschedules the fetching of snippets. If tasks are already scheduled, they |
| // will be rescheduled anyway, so all running intervals will be reset. |
| static void RescheduleFetching(JNIEnv* env, |
| const JavaParamRef<jclass>& caller) { |
| Profile* profile = ProfileManager::GetLastUsedProfile(); |
| // Temporary check while investigating crbug.com/647920. |
| CHECK(profile); |
| |
| ntp_snippets::ContentSuggestionsService* content_suggestions_service = |
| ContentSuggestionsServiceFactory::GetForProfile(profile); |
| |
| // Can maybe be null in some cases? (Incognito profile?) crbug.com/647920 |
| if (!content_suggestions_service) |
| return; |
| |
| ntp_snippets::RemoteSuggestionsProvider* service = |
| content_suggestions_service->ntp_snippets_service(); |
| |
| // Can be null if the feature has been disabled but the scheduler has not been |
| // unregistered yet. The next start should unregister it. |
| if (!service) |
| return; |
| |
| service->RescheduleFetching(/*force=*/true); |
| } |
| |
| static void OnSuggestionTargetVisited(JNIEnv* env, |
| const JavaParamRef<jclass>& caller, |
| jint category, |
| jlong visit_time_ms) { |
| Profile* profile = ProfileManager::GetLastUsedProfile(); |
| ntp_snippets::ContentSuggestionsService* content_suggestions_service = |
| ContentSuggestionsServiceFactory::GetForProfile(profile); |
| ntp_snippets::metrics::OnSuggestionTargetVisited( |
| content_suggestions_service->category_factory()->FromIDValue(category), |
| base::TimeDelta::FromMilliseconds(visit_time_ms)); |
| } |
| |
| NTPSnippetsBridge::NTPSnippetsBridge(JNIEnv* env, |
| const JavaParamRef<jobject>& j_profile) |
| : content_suggestions_service_observer_(this), weak_ptr_factory_(this) { |
| Profile* profile = ProfileAndroid::FromProfileAndroid(j_profile); |
| content_suggestions_service_ = |
| ContentSuggestionsServiceFactory::GetForProfile(profile); |
| history_service_ = |
| HistoryServiceFactory::GetForProfile(profile, |
| ServiceAccessType::EXPLICIT_ACCESS); |
| content_suggestions_service_observer_.Add(content_suggestions_service_); |
| } |
| |
| void NTPSnippetsBridge::Destroy(JNIEnv* env, const JavaParamRef<jobject>& obj) { |
| delete this; |
| } |
| |
| void NTPSnippetsBridge::SetObserver(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| const JavaParamRef<jobject>& j_observer) { |
| observer_.Reset(env, j_observer); |
| } |
| |
| ScopedJavaLocalRef<jintArray> NTPSnippetsBridge::GetCategories( |
| JNIEnv* env, |
| const base::android::JavaParamRef<jobject>& obj) { |
| std::vector<int> category_ids; |
| for (Category category : content_suggestions_service_->GetCategories()) { |
| category_ids.push_back(category.id()); |
| } |
| return ToJavaIntArray(env, category_ids); |
| } |
| |
| int NTPSnippetsBridge::GetCategoryStatus(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint category) { |
| return static_cast<int>(content_suggestions_service_->GetCategoryStatus( |
| CategoryFromIDValue(category))); |
| } |
| |
| base::android::ScopedJavaLocalRef<jobject> NTPSnippetsBridge::GetCategoryInfo( |
| JNIEnv* env, |
| const base::android::JavaParamRef<jobject>& obj, |
| jint category) { |
| base::Optional<CategoryInfo> info = |
| content_suggestions_service_->GetCategoryInfo( |
| CategoryFromIDValue(category)); |
| if (!info) |
| return base::android::ScopedJavaLocalRef<jobject>(env, nullptr); |
| return Java_SnippetsBridge_createSuggestionsCategoryInfo( |
| env, category, ConvertUTF16ToJavaString(env, info->title()), |
| static_cast<int>(info->card_layout()), info->has_more_action(), |
| info->has_reload_action(), info->has_view_all_action(), |
| info->show_if_empty(), |
| ConvertUTF16ToJavaString(env, info->no_suggestions_message())); |
| } |
| |
| ScopedJavaLocalRef<jobject> NTPSnippetsBridge::GetSuggestionsForCategory( |
| JNIEnv* env, |
| const base::android::JavaParamRef<jobject>& obj, |
| jint j_category_id) { |
| Category category = CategoryFromIDValue(j_category_id); |
| return ToJavaSuggestionList( |
| env, category, |
| content_suggestions_service_->GetSuggestionsForCategory(category)); |
| } |
| |
| void NTPSnippetsBridge::FetchSuggestionImage( |
| JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint category, |
| const JavaParamRef<jstring>& id_within_category, |
| const JavaParamRef<jobject>& j_callback) { |
| base::android::ScopedJavaGlobalRef<jobject> callback(j_callback); |
| content_suggestions_service_->FetchSuggestionImage( |
| ContentSuggestion::ID(CategoryFromIDValue(category), |
| ConvertJavaStringToUTF8(env, id_within_category)), |
| base::Bind(&NTPSnippetsBridge::OnImageFetched, |
| weak_ptr_factory_.GetWeakPtr(), callback)); |
| } |
| |
| void NTPSnippetsBridge::Fetch( |
| JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint j_category_id, |
| const JavaParamRef<jobjectArray>& j_displayed_suggestions) { |
| std::vector<std::string> known_suggestion_ids; |
| AppendJavaStringArrayToStringVector(env, j_displayed_suggestions, |
| &known_suggestion_ids); |
| |
| Category category = CategoryFromIDValue(j_category_id); |
| content_suggestions_service_->Fetch( |
| category, std::set<std::string>(known_suggestion_ids.begin(), |
| known_suggestion_ids.end()), |
| base::Bind(&NTPSnippetsBridge::OnSuggestionsFetched, |
| weak_ptr_factory_.GetWeakPtr(), category)); |
| } |
| |
| void NTPSnippetsBridge::DismissSuggestion( |
| JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| const JavaParamRef<jstring>& jurl, |
| jint global_position, |
| jint category, |
| jint category_position, |
| const JavaParamRef<jstring>& id_within_category) { |
| content_suggestions_service_->DismissSuggestion( |
| ContentSuggestion::ID(CategoryFromIDValue(category), |
| ConvertJavaStringToUTF8(env, id_within_category))); |
| |
| history_service_->QueryURL( |
| GURL(ConvertJavaStringToUTF8(env, jurl)), /*want_visits=*/false, |
| base::Bind( |
| [](int global_position, Category category, int category_position, |
| bool success, const history::URLRow& row, |
| const history::VisitVector& visit_vector) { |
| bool visited = success && row.visit_count() != 0; |
| ntp_snippets::metrics::OnSuggestionDismissed( |
| global_position, category, category_position, visited); |
| }, |
| global_position, CategoryFromIDValue(category), category_position), |
| &tracker_); |
| } |
| |
| void NTPSnippetsBridge::DismissCategory(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint category) { |
| content_suggestions_service_->DismissCategory(CategoryFromIDValue(category)); |
| } |
| |
| void NTPSnippetsBridge::RestoreDismissedCategories( |
| JNIEnv* env, |
| const JavaParamRef<jobject>& obj) { |
| content_suggestions_service_->RestoreDismissedCategories(); |
| } |
| |
| void NTPSnippetsBridge::OnPageShown( |
| JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| const JavaParamRef<jintArray>& jcategories, |
| const JavaParamRef<jintArray>& jsuggestions_per_category) { |
| std::vector<int> categories_int; |
| JavaIntArrayToIntVector(env, jcategories, &categories_int); |
| std::vector<int> suggestions_per_category_int; |
| JavaIntArrayToIntVector(env, jsuggestions_per_category, |
| &suggestions_per_category_int); |
| DCHECK_EQ(categories_int.size(), suggestions_per_category_int.size()); |
| std::vector<std::pair<Category, int>> suggestions_per_category; |
| for (size_t i = 0; i < categories_int.size(); i++) { |
| suggestions_per_category.push_back( |
| std::make_pair(CategoryFromIDValue(categories_int[i]), |
| suggestions_per_category_int[i])); |
| } |
| ntp_snippets::metrics::OnPageShown(suggestions_per_category); |
| content_suggestions_service_->user_classifier()->OnEvent( |
| ntp_snippets::UserClassifier::Metric::NTP_OPENED); |
| } |
| |
| void NTPSnippetsBridge::OnSuggestionShown(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint global_position, |
| jint category, |
| jint category_position, |
| jlong publish_timestamp_ms, |
| jfloat score) { |
| ntp_snippets::metrics::OnSuggestionShown( |
| global_position, CategoryFromIDValue(category), category_position, |
| TimeFromJavaTime(publish_timestamp_ms), score); |
| if (global_position == 0) { |
| content_suggestions_service_->user_classifier()->OnEvent( |
| ntp_snippets::UserClassifier::Metric::SUGGESTIONS_SHOWN); |
| } |
| } |
| |
| void NTPSnippetsBridge::OnSuggestionOpened(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint global_position, |
| jint category, |
| jint category_position, |
| jlong publish_timestamp_ms, |
| jfloat score, |
| int windowOpenDisposition) { |
| ntp_snippets::metrics::OnSuggestionOpened( |
| global_position, CategoryFromIDValue(category), category_position, |
| TimeFromJavaTime(publish_timestamp_ms), score, |
| static_cast<WindowOpenDisposition>(windowOpenDisposition)); |
| content_suggestions_service_->user_classifier()->OnEvent( |
| ntp_snippets::UserClassifier::Metric::SUGGESTIONS_USED); |
| } |
| |
| void NTPSnippetsBridge::OnSuggestionMenuOpened(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint global_position, |
| jint category, |
| jint category_position, |
| jlong publish_timestamp_ms, |
| jfloat score) { |
| ntp_snippets::metrics::OnSuggestionMenuOpened( |
| global_position, CategoryFromIDValue(category), category_position, |
| TimeFromJavaTime(publish_timestamp_ms), score); |
| } |
| |
| void NTPSnippetsBridge::OnMoreButtonShown(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint category, |
| jint position) { |
| ntp_snippets::metrics::OnMoreButtonShown(CategoryFromIDValue(category), |
| position); |
| } |
| |
| void NTPSnippetsBridge::OnMoreButtonClicked(JNIEnv* env, |
| const JavaParamRef<jobject>& obj, |
| jint category, |
| jint position) { |
| ntp_snippets::metrics::OnMoreButtonClicked(CategoryFromIDValue(category), |
| position); |
| content_suggestions_service_->user_classifier()->OnEvent( |
| ntp_snippets::UserClassifier::Metric::SUGGESTIONS_USED); |
| } |
| |
| NTPSnippetsBridge::~NTPSnippetsBridge() {} |
| |
| void NTPSnippetsBridge::OnNewSuggestions(Category category) { |
| if (observer_.is_null()) |
| return; |
| |
| JNIEnv* env = base::android::AttachCurrentThread(); |
| Java_SnippetsBridge_onNewSuggestions(env, observer_, |
| static_cast<int>(category.id())); |
| } |
| |
| void NTPSnippetsBridge::OnCategoryStatusChanged(Category category, |
| CategoryStatus new_status) { |
| if (observer_.is_null()) |
| return; |
| |
| JNIEnv* env = base::android::AttachCurrentThread(); |
| Java_SnippetsBridge_onCategoryStatusChanged(env, observer_, |
| static_cast<int>(category.id()), |
| static_cast<int>(new_status)); |
| } |
| |
| void NTPSnippetsBridge::OnSuggestionInvalidated( |
| const ContentSuggestion::ID& suggestion_id) { |
| if (observer_.is_null()) |
| return; |
| |
| JNIEnv* env = base::android::AttachCurrentThread(); |
| Java_SnippetsBridge_onSuggestionInvalidated( |
| env, observer_.obj(), static_cast<int>(suggestion_id.category().id()), |
| ConvertUTF8ToJavaString(env, suggestion_id.id_within_category()).obj()); |
| } |
| |
| void NTPSnippetsBridge::ContentSuggestionsServiceShutdown() { |
| observer_.Reset(); |
| content_suggestions_service_observer_.Remove(content_suggestions_service_); |
| } |
| |
| void NTPSnippetsBridge::OnImageFetched(ScopedJavaGlobalRef<jobject> callback, |
| const gfx::Image& image) { |
| ScopedJavaLocalRef<jobject> j_bitmap; |
| if (!image.IsEmpty()) |
| j_bitmap = gfx::ConvertToJavaBitmap(image.ToSkBitmap()); |
| |
| base::android::RunCallbackAndroid(callback, j_bitmap); |
| } |
| |
| void NTPSnippetsBridge::OnSuggestionsFetched( |
| Category category, |
| ntp_snippets::Status status, |
| std::vector<ContentSuggestion> suggestions) { |
| JNIEnv* env = AttachCurrentThread(); |
| // TODO(fhorschig, dgn): Allow refetch or show notification acc. to status. |
| Java_SnippetsBridge_onMoreSuggestions( |
| env, observer_, category.id(), |
| ToJavaSuggestionList(env, category, suggestions)); |
| } |
| |
| Category NTPSnippetsBridge::CategoryFromIDValue(jint id) { |
| return content_suggestions_service_->category_factory()->FromIDValue(id); |
| } |
| |
| // static |
| bool NTPSnippetsBridge::Register(JNIEnv* env) { |
| return RegisterNativesImpl(env); |
| } |