yomichan/ext/bg/js/ankiweb.js
2016-10-15 20:33:04 -07:00

144 lines
4.5 KiB
JavaScript

/*
* Copyright (C) 2016 Alex Yatskov <alex@foosoft.net>
* Author: Alex Yatskov <alex@foosoft.net>
*
* 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.
*
* This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
*/
class AnkiWeb {
constructor(username, password) {
this.username = username;
this.password = password;
this.noteInfo = null;
this.logged = false;
}
addNote(note) {
return Promise.resolve(true);
}
canAddNotes(notes) {
return Promise.resolve([]);
}
getDeckNames() {
return this.retrieve().then(info => info.deckNames);
}
getModelNames() {
return this.retrieve().then(info => info.models.map(m => m.name));
}
getModelFieldNames(modelName) {
return this.retrieve().then(info => {
const model = info.models.find(m => m.name === modelName);
return model ? model.fields : [];
});
}
retrieve() {
if (this.noteInfo !== null) {
return Promise.resolve(this.noteInfo);
}
return this.authenticate().then(() => {
return AnkiWeb.scrape();
}).then(({deckNames, models}) => {
this.noteInfo = {deckNames, models};
return this.noteInfo;
});
}
authenticate() {
if (this.logged) {
return Promise.resolve(true);
}
return AnkiWeb.logout().then(() => {
return AnkiWeb.login(this.username, this.password);
}).then(() => {
this.logged = true;
return true;
});
}
static scrape() {
return new Promise((resolve, reject) => {
const xhr = new XMLHttpRequest();
xhr.addEventListener('error', () => reject('failed to execute scrape request'));
xhr.addEventListener('load', () => {
const modelsJson = JSON.parse(/editor\.models = (.*}]);/.exec(xhr.responseText)[1]);
if (!modelsJson) {
reject('failed to scrape model data');
return;
}
const decksJson = JSON.parse(/editor\.decks = (.*}});/.exec(xhr.responseText)[1]);
if (!decksJson) {
reject('failed to scrape deck data');
return;
}
const deckNames = Object.keys(decksJson).map(d => decksJson[d].name);
const models = [];
for (const modelJson of modelsJson) {
models.push({
name: modelJson.name,
id: modelJson.id,
fields: modelJson.flds.map(f => f.name)
});
}
resolve({deckNames, models});
});
xhr.open('GET', 'https://ankiweb.net/edit/');
xhr.send();
});
}
static login(username, password) {
return new Promise((resolve, reject) => {
const xhr = new XMLHttpRequest();
xhr.addEventListener('error', () => reject('failed to execute login request'));
xhr.addEventListener('load', () => {
if (xhr.responseText.includes('class="mitem"')) {
resolve();
} else {
reject('failed to authenticate');
}
});
const form = new FormData();
form.append('username', username);
form.append('password', password);
form.append('submitted', 1);
xhr.open('POST', 'https://ankiweb.net/account/login');
xhr.send(form);
});
}
static logout() {
return new Promise((resolve, reject) => {
const xhr = new XMLHttpRequest();
xhr.addEventListener('error', () => reject('failed to execute logout request'));
xhr.addEventListener('load', () => resolve());
xhr.open('GET', 'https://ankiweb.net/account/logout');
xhr.send();
});
}
}