path: root/app/src/main/java/net/lacolaco/smileessence/view/page/SearchFragment.java

/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2012-2014 lacolaco.net
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package net.lacolaco.smileessence.view.page;

import android.os.Bundle;
import android.support.annotation.NonNull;
import android.text.Spannable;
import android.text.TextUtils;
import android.text.method.ArrowKeyMovementMethod;
import android.view.*;
import android.view.inputmethod.EditorInfo;
import android.widget.EditText;
import android.widget.ImageButton;
import android.widget.ListView;
import android.widget.TextView;
import com.handmark.pulltorefresh.library.PullToRefreshBase;
import com.handmark.pulltorefresh.library.PullToRefreshListView;
import net.lacolaco.smileessence.Application;
import net.lacolaco.smileessence.R;
import net.lacolaco.smileessence.command.Command;
import net.lacolaco.smileessence.command.CommandOpenSearch;
import net.lacolaco.smileessence.data.Account;
import net.lacolaco.smileessence.entity.SavedSearch;
import net.lacolaco.smileessence.entity.Tweet;
import net.lacolaco.smileessence.notification.Notificator;
import net.lacolaco.smileessence.preference.InternalPreferenceHelper;
import net.lacolaco.smileessence.twitter.task.SavedSearches;
import net.lacolaco.smileessence.twitter.task.SearchTask;
import net.lacolaco.smileessence.util.SystemServiceHelper;
import net.lacolaco.smileessence.util.UIHandler;
import net.lacolaco.smileessence.view.DialogHelper;
import net.lacolaco.smileessence.view.adapter.SearchListAdapter;
import net.lacolaco.smileessence.view.dialog.SelectSearchQueryDialogFragment;
import net.lacolaco.smileessence.viewmodel.StatusViewModel;
import twitter4j.Query;

import java.util.List;

public class SearchFragment extends CustomListFragment<SearchListAdapter> implements View.OnClickListener, View.OnFocusChangeListener,
        SearchListAdapter.OnQueryChangeListener {

    // ------------------------------ FIELDS ------------------------------

    private EditText editText;

    // --------------------- GETTER / SETTER METHODS ---------------------

    @Override
    protected PullToRefreshBase.Mode getRefreshMode() {
        return PullToRefreshBase.Mode.BOTH;
    }

    // ------------------------ INTERFACE METHODS ------------------------

    @Override
    public void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setHasOptionsMenu(true);
        SearchListAdapter adapter = new SearchListAdapter(getActivity());
        setAdapter(adapter);

        if (Application.getInstance().getCurrentAccount() != null) {
            refresh();
        }
    }

    @Override
    public void refresh() { //TODO
        String lastUsedSearchQuery = InternalPreferenceHelper.getInstance().get(R.string.key_last_used_search_query, "");
        if (!TextUtils.isEmpty(lastUsedSearchQuery)) {
            startSearch(lastUsedSearchQuery);
        }
    }

    // --------------------- Interface OnClickListener ---------------------

    @Override
    public void onClick(View v) {
        switch (v.getId()) {
            case R.id.button_search_queries: {
                openSearchQueryDialog();
                break;
            }
            case R.id.button_search_execute: {
                search();
                break;
            }
            case R.id.button_search_save: {
                saveQuery();
            }
        }
    }

    // --------------------- Interface OnFocusChangeListener ---------------------

    @Override
    public void onFocusChange(View v, boolean hasFocus) {
        if (!hasFocus) {
            SystemServiceHelper.hideIM(getActivity(), editText);
        }
    }

    // --------------------- Interface OnQueryChangeListener ---------------------

    @Override
    public void onQueryChange(String newQuery) {
        if (editText != null) {
            editText.setText(newQuery);
        }
    }

    // --------------------- Interface OnRefreshListener2 ---------------------

    @Override
    public void onPullDownToRefresh(final PullToRefreshBase<ListView> refreshView) {
        final SearchListAdapter adapter = getAdapter();
        String queryString = adapter.getQuery();
        if (TextUtils.isEmpty(queryString)) {
            new UIHandler().post(() -> {
                notifyTextEmpty();
                refreshView.onRefreshComplete();
            });
            return;
        }
        final Query query = new Query();
        query.setQuery(queryString);
        query.setCount(200);
        query.setResultType(Query.RECENT);
        if (adapter.getCount() > 0) {
            query.setSinceId(adapter.getTopID());
        }
        runRefreshTask(
                new SearchTask(Application.getInstance().getCurrentAccount(), query),
                () -> {
                    updateListViewWithNotice(refreshView.getRefreshableView(), true);
                    refreshView.onRefreshComplete();
                });
    }

    @Override
    public void onPullUpToRefresh(final PullToRefreshBase<ListView> refreshView) {
        final SearchListAdapter adapter = getAdapter();
        String queryString = adapter.getQuery();
        if (TextUtils.isEmpty(queryString)) {
            new UIHandler().post(() -> {
                notifyTextEmpty();
                refreshView.onRefreshComplete();
            });
            return;
        }
        final Query query = new Query();
        query.setQuery(queryString);
        query.setCount(200);
        query.setResultType(Query.RECENT);
        if (adapter.getCount() > 0) {
            query.setMaxId(adapter.getLastID() - 1);
        }
        runRefreshTask(
                new SearchTask(Application.getInstance().getCurrentAccount(), query),
                () -> {
                    updateListViewWithNotice(refreshView.getRefreshableView(), false);
                    refreshView.onRefreshComplete();
                });
    }

    // ------------------------ OVERRIDE METHODS ------------------------

    @Override
    protected PullToRefreshListView getListView(View page) {
        return (PullToRefreshListView) page.findViewById(R.id.listview_search);
    }

    @Override
    public View onCreateView(LayoutInflater inflater, ViewGroup container, Bundle savedInstanceState) {
        View page = inflater.inflate(R.layout.fragment_search, container, false);
        PullToRefreshListView listView = getListView(page);
        SearchListAdapter adapter = getAdapter();
        listView.setAdapter(adapter);
        listView.setOnScrollListener(this);
        listView.setOnRefreshListener(this);
        listView.setMode(getRefreshMode());
        ImageButton buttonQueries = getQueriesButton(page);
        buttonQueries.setOnClickListener(this);
        ImageButton buttonExecute = getExecuteButton(page);
        buttonExecute.setOnClickListener(this);
        ImageButton buttonSave = getSaveButton(page);
        buttonSave.setOnClickListener(this);
        editText = getEditText(page);
        editText.setOnFocusChangeListener(this);
        editText.setText(adapter.getQuery());
        editText.setOnEditorActionListener((textView, i, keyEvent) -> {
            if (i == EditorInfo.IME_ACTION_SEARCH ||
                    keyEvent != null &&
                    keyEvent.getAction() == KeyEvent.ACTION_DOWN &&
                    keyEvent.getKeyCode() == KeyEvent.KEYCODE_ENTER) {
                search();
            }
            return true;
        });
        editText.setMovementMethod(new ArrowKeyMovementMethod() {
            @Override
            protected boolean right(@NonNull TextView widget, Spannable buffer) {
                //Don't move page
                return widget.getSelectionEnd() == widget.length() || super.right(widget, buffer);
            }

            @Override
            protected boolean left(@NonNull TextView widget, Spannable buffer) {
                //Don't move page
                return widget.getSelectionStart() == 0 || super.left(widget, buffer);
            }
        });
        adapter.setOnQueryChangeListener(this);
        return page;
    }

    private EditText getEditText(View page) {
        return (EditText) page.findViewById(R.id.edittext_search);
    }

    private ImageButton getExecuteButton(View page) {
        return (ImageButton) page.findViewById(R.id.button_search_execute);
    }

    private ImageButton getQueriesButton(View page) {
        return (ImageButton) page.findViewById(R.id.button_search_queries);
    }

    private ImageButton getSaveButton(View page) {
        return (ImageButton) page.findViewById(R.id.button_search_save);
    }

    private void notifyTextEmpty() {
        Notificator.getInstance().alert(R.string.notice_search_text_empty);
    }

    private void openSearchQueryDialog() {
        DialogHelper.showDialog(getActivity(), new SelectSearchQueryDialogFragment() {
            @Override
            protected void executeCommand(Command command) {
                super.executeCommand(command);
                SavedSearch ss = ((CommandOpenSearch) command).getQuery();
                editText.setText(ss.getQuery());
                SystemServiceHelper.hideIM(SearchFragment.this.getActivity(), editText);
            }
        });
    }

    private void saveQuery() {
        String text = editText.getText().toString();
        if (TextUtils.isEmpty(text)) {
            Notificator.getInstance().alert(R.string.notice_query_is_empty);
        } else {
            new SavedSearches.Create(Application.getInstance().getCurrentAccount(), text).onDoneUI(cb -> {
                Notificator.getInstance().publish(R.string.notice_query_saved);
                Application.getInstance().getCurrentAccount().refreshSavedSearches();
            }).onFailUI(ex -> {
                Notificator.getInstance().publish("Query is not saved");
            }).execute();
        }
    }

    private void search() {
        if (editText != null) {
            String text = editText.getText().toString();
            if (TextUtils.isEmpty(text)) {
                Notificator.getInstance().alert(R.string.notice_query_is_empty);
            } else {
                startSearch(text);
                SystemServiceHelper.hideIM(getActivity(), editText);
            }
        }
    }

    public void startSearch(final String queryString) {
        InternalPreferenceHelper.getInstance().set(R.string.key_last_used_search_query, queryString);
        if (!TextUtils.isEmpty(queryString)) {
            final SearchListAdapter adapter = getAdapter();
            adapter.initSearch(queryString);
            adapter.updateForce();
            final Query query = new Query();
            query.setQuery(queryString);
            query.setCount(200);
            query.setResultType(Query.RECENT);
            runRefreshTask(
                    new SearchTask(Application.getInstance().getCurrentAccount(), query),
                    adapter::updateForce);
        }
    }

    private void runRefreshTask(SearchTask task, Runnable onFinish) {
        final SearchListAdapter adapter = getAdapter();
        final Account account = Application.getInstance().getCurrentAccount();
        task
                .onFail(x -> Notificator.getInstance().alert(R.string.notice_error_search))
                .onDoneUI(queryResult -> {
                    if (queryResult != null) {
                        List<Tweet> tweets = Tweet.fromTwitter(queryResult.getTweets(), account.getUserId());
                        for (Tweet tweet : tweets) {
                            if (!tweet.isRetweet()) {
                                adapter.addItem(new StatusViewModel(tweet));
                            }
                        }
                    }
                })
                .onFinishUI(onFinish)
                .execute();
    }
}