goldendict-ng/ankiconnector.cc

#include "ankiconnector.hh"
#include <QJsonDocument>
#include <QJsonObject>
#include <QJsonValue>
#include "utils.hh"

QString markTargetWord(QString const& sentence, QString const& word)
{
  // TODO properly handle inflected words.
  QString result = sentence;
  return result.replace(word, "<b>" + word + "</b>", Qt::CaseInsensitive);
}

AnkiConnector::AnkiConnector( QObject * parent, Config::Class const & _cfg ) : QObject{ parent }, cfg( _cfg )
{
  mgr = new QNetworkAccessManager( this );
  connect( mgr, &QNetworkAccessManager::finished, this, &AnkiConnector::finishedSlot );
}

void AnkiConnector::sendToAnki( QString const & word, QString text, QString const & sentence )
{
  if ( word.isEmpty() ) {
    emit this->errorText( tr( "anki: can't create a card without a word" ) );
    return;
  }

  // Anki doesn't understand the newline character, so it should be escaped.
  text = text.replace( "\n", "<br>" );

  QString const postTemplate = R"anki({
      "action": "addNote",
      "version": 6,
      "params": {
          "note": {
              "deckName": "%1",
              "modelName": "%2",
              "fields": %3,
              "options": {
                  "allowDuplicate": true
              },
              "tags": []
          }
      }
  })anki";

  QJsonObject fields;
  fields.insert( cfg.preferences.ankiConnectServer.word, word );
  fields.insert( cfg.preferences.ankiConnectServer.text, text );
  if (!cfg.preferences.ankiConnectServer.sentence.isEmpty()) {
    QString sentence_changed = markTargetWord(sentence, word);
    fields.insert( cfg.preferences.ankiConnectServer.sentence, sentence_changed );
  }

  QString postData = postTemplate.arg( cfg.preferences.ankiConnectServer.deck,
                                       cfg.preferences.ankiConnectServer.model,
                                       Utils::json2String( fields ) );

//  qDebug().noquote() << postData;
  postToAnki( postData );
}

void AnkiConnector::ankiSearch( QString const & word )
{
    if( !cfg.preferences.ankiConnectServer.enabled ) {
      emit this->errorText( tr( "Anki search: AnkiConnect is not enabled." ) );
      return;
    }

    QString postTemplate = R"anki({
        "action": "guiBrowse",
        "version": 6,
        "params": {
            "query": "%1"
        }
    })anki";
    postToAnki( postTemplate.arg( word ) );
}

void AnkiConnector::postToAnki( QString const & postData )
{
  QUrl url;
  url.setScheme( "http" );
  url.setHost( cfg.preferences.ankiConnectServer.host );
  url.setPort( cfg.preferences.ankiConnectServer.port );
  QNetworkRequest request( url );
  request.setTransferTimeout( transfer_timeout );
  //  request.setAttribute( QNetworkRequest::RedirectPolicyAttribute, QNetworkRequest::NoLessSafeRedirectPolicy );
  request.setHeader( QNetworkRequest::ContentTypeHeader, "application/json" );
  auto reply = mgr->post( request, postData.toUtf8() );
  connect( reply,
           &QNetworkReply::errorOccurred,
           this,
           [ this ]( QNetworkReply::NetworkError e )
           {
             qWarning() << e;
             emit this->errorText( tr( "anki: post to anki failed" ) );
           } );
}

void AnkiConnector::finishedSlot( QNetworkReply * reply )
{
  if ( reply->error() == QNetworkReply::NoError ) {
      QByteArray const bytes   = reply->readAll();
      QJsonDocument const json = QJsonDocument::fromJson( bytes );
      auto const obj           = json.object();

      // Normally AnkiConnect always returns result and error,
      // unless Anki is not running.
      if ( obj.size() == 2 && obj.contains( "result" ) && obj.contains( "error" ) && obj[ "error" ].isNull() ) {
        emit errorText( tr( "anki: post to anki success" ) );
      }
      else {
        emit errorText( tr( "anki: post to anki failed" ) );
      }

      qDebug().noquote() << "anki response:" << Utils::json2String( obj );
  }
  else {
      qDebug() << "anki connect error" << reply->errorString();
      emit errorText( "anki:" + reply->errorString() );
  }

  reply->deleteLater();
}