package app.fedilab.android.viewmodel.mastodon; /* Copyright 2021 Thomas Schneider * * This file is a part of Fedilab * * This program is free software; you can redistribute it and/or modify it under the terms of the * GNU General Public License as published by the Free Software Foundation; either version 3 of the * License, or (at your option) any later version. * * Fedilab is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General * Public License for more details. * * You should have received a copy of the GNU General Public License along with Fedilab; if not, * see . */ import android.app.Application; import android.os.Handler; import android.os.Looper; import androidx.annotation.NonNull; import androidx.lifecycle.AndroidViewModel; import androidx.lifecycle.LiveData; import androidx.lifecycle.MutableLiveData; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import java.util.ArrayList; import java.util.List; import java.util.concurrent.TimeUnit; import app.fedilab.android.client.endpoints.MastodonSearchService; import app.fedilab.android.client.entities.api.Results; import app.fedilab.android.client.entities.api.Status; import app.fedilab.android.client.entities.app.StatusCache; import app.fedilab.android.exception.DBException; import app.fedilab.android.helper.Helper; import app.fedilab.android.helper.SpannableHelper; import okhttp3.OkHttpClient; import retrofit2.Call; import retrofit2.Response; import retrofit2.Retrofit; import retrofit2.converter.gson.GsonConverterFactory; public class SearchVM extends AndroidViewModel { final OkHttpClient okHttpClient = new OkHttpClient.Builder() .readTimeout(60, TimeUnit.SECONDS) .connectTimeout(60, TimeUnit.SECONDS) .callTimeout(60, TimeUnit.SECONDS) .proxy(Helper.getProxy(getApplication().getApplicationContext())) .build(); private MutableLiveData resultsMutableLiveData; public SearchVM(@NonNull Application application) { super(application); } private MastodonSearchService init(@NonNull String instance) { Gson gson = new GsonBuilder().setDateFormat("MMM dd, yyyy HH:mm:ss").create(); Retrofit retrofit = new Retrofit.Builder() .baseUrl("https://" + instance + "/api/v2/") .addConverterFactory(GsonConverterFactory.create(gson)) .client(okHttpClient) .build(); return retrofit.create(MastodonSearchService.class); } /** * Search for content in accounts, statuses and hashtags with API v2 * * @param instance Instance domain of the active account * @param token Access token of the active account * @param q String - search words * @param account_id String - If provided, statuses returned will be authored only by this account * @param type String - Enum(accounts, hashtags, statuses) * @param exclude_unreviewed boolean - Filter out unreviewed tags? Defaults to false. Use true when trying to find trending tags. * @param resolve boolean - Attempt WebFinger lookup. Defaults to false. * @param following boolean - Only include accounts that the user is following. Defaults to false. * @param offset int - Offset in search results. Used for pagination. Defaults to 0. * @param max_id String - Return results older than this id * @param min_id String - Return results immediately newer than this id * @param limit int - Maximum number of results to load, per type. Defaults to 20. Max 40. * @return {@link LiveData} containing an {@link Results} */ public LiveData search(@NonNull String instance, String token, @NonNull String q, String account_id, String type, Boolean exclude_unreviewed, Boolean resolve, Boolean following, Integer offset, String max_id, String min_id, Integer limit) { MastodonSearchService mastodonSearchService = init(instance); resultsMutableLiveData = new MutableLiveData<>(); new Thread(() -> { Call resultsCall = mastodonSearchService.search( token, q, account_id, type, exclude_unreviewed, resolve, following, offset, max_id, min_id, limit); Results results = null; if (resultsCall != null) { try { Response resultsResponse = resultsCall.execute(); if (resultsResponse.isSuccessful()) { results = resultsResponse.body(); if (results != null) { if (results.statuses == null) { results.statuses = new ArrayList<>(); } else { results.statuses = SpannableHelper.convertStatus(getApplication(), results.statuses); } if (results.accounts == null) { results.accounts = new ArrayList<>(); } if (results.hashtags == null) { results.hashtags = new ArrayList<>(); } } } } catch (Exception e) { e.printStackTrace(); } } Handler mainHandler = new Handler(Looper.getMainLooper()); Results finalResults = results; Runnable myRunnable = () -> resultsMutableLiveData.setValue(finalResults); mainHandler.post(myRunnable); }).start(); return resultsMutableLiveData; } public LiveData searchCache(@NonNull String instance, String userId, @NonNull String q) { resultsMutableLiveData = new MutableLiveData<>(); new Thread(() -> { Results results = new Results(); try { results.statuses = new ArrayList<>(); List statuses = new StatusCache(getApplication()).searchStatus(StatusCache.CacheEnum.HOME, instance, userId, q); statuses = SpannableHelper.convertStatus(getApplication(), statuses); results.statuses.addAll(statuses); } catch (DBException e) { e.printStackTrace(); } Handler mainHandler = new Handler(Looper.getMainLooper()); Runnable myRunnable = () -> resultsMutableLiveData.setValue(results); mainHandler.post(myRunnable); }).start(); return resultsMutableLiveData; } }