mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-11 13:36:13 +03:00
85 lines
2.1 KiB
C++
85 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2023, Cameron Youell <cameronyouell@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "StringUtils.h"
|
|
#include <AK/Forward.h>
|
|
#include <AK/String.h>
|
|
#include <QCompleter>
|
|
#include <QNetworkReply>
|
|
#include <QTreeView>
|
|
|
|
namespace Ladybird {
|
|
|
|
class AutoCompleteModel final : public QAbstractListModel {
|
|
Q_OBJECT
|
|
public:
|
|
explicit AutoCompleteModel(QObject* parent)
|
|
: QAbstractListModel(parent)
|
|
{
|
|
}
|
|
|
|
virtual int rowCount(QModelIndex const& parent = QModelIndex()) const override { return parent.isValid() ? 0 : m_suggestions.size(); }
|
|
virtual QVariant data(QModelIndex const& index, int role = Qt::DisplayRole) const override
|
|
{
|
|
if (role == Qt::DisplayRole || role == Qt::EditRole)
|
|
return qstring_from_ak_string(m_suggestions[index.row()]);
|
|
return {};
|
|
}
|
|
|
|
void add(String const& result)
|
|
{
|
|
beginInsertRows({}, m_suggestions.size(), m_suggestions.size());
|
|
m_suggestions.append(result);
|
|
endInsertRows();
|
|
}
|
|
|
|
void clear()
|
|
{
|
|
beginResetModel();
|
|
m_suggestions.clear();
|
|
endResetModel();
|
|
}
|
|
|
|
private:
|
|
AK::Vector<String> m_suggestions;
|
|
};
|
|
class AutoComplete final : public QCompleter {
|
|
Q_OBJECT
|
|
|
|
public:
|
|
AutoComplete(QWidget* parent);
|
|
|
|
virtual QString pathFromIndex(QModelIndex const& index) const override
|
|
{
|
|
return index.data(Qt::DisplayRole).toString();
|
|
}
|
|
|
|
ErrorOr<void> get_search_suggestions(StringView);
|
|
void clear_suggestions();
|
|
static ErrorOr<String> auto_complete_url_from_query(StringView query);
|
|
|
|
signals:
|
|
void activated(QModelIndex const&);
|
|
|
|
private:
|
|
ErrorOr<void> got_network_response(QNetworkReply* reply);
|
|
|
|
ErrorOr<void> parse_google_autocomplete(Vector<JsonValue> const&);
|
|
ErrorOr<void> parse_duckduckgo_autocomplete(Vector<JsonValue> const&);
|
|
ErrorOr<void> parse_yahoo_autocomplete(JsonObject const&);
|
|
|
|
QNetworkAccessManager* m_manager;
|
|
AutoCompleteModel* m_auto_complete_model;
|
|
QTreeView* m_tree_view;
|
|
QNetworkReply* m_reply { nullptr };
|
|
|
|
String m_query;
|
|
};
|
|
|
|
}
|