145 lines
4.9 KiB
JavaScript
145 lines
4.9 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;
|
|
}
|
|
|
|
addNote(note) {
|
|
return this.retrieve().then(info => {
|
|
const model = info.models.find(m => m.name === note.modelName);
|
|
if (!model) {
|
|
return Promise.reject('invalid model');
|
|
}
|
|
|
|
const fields = [];
|
|
for (const field of model.fields) {
|
|
fields.push(note.fields[field]);
|
|
}
|
|
|
|
const form = new FormData();
|
|
form.append('data', [fields, note.tags.join(' ')]);
|
|
form.append('mid', model.id);
|
|
form.append('deck', note.deckname);
|
|
|
|
return AnkiWeb.loadAccountPage('https://ankiweb.net/edit/save', 'POST', form, this.username, this.password);
|
|
}).then(response => {
|
|
return true;
|
|
});
|
|
}
|
|
|
|
canAddNotes(notes) {
|
|
return Promise.resolve(new Array(notes.length).fill(true));
|
|
}
|
|
|
|
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 AnkiWeb.scrape(this.username, this.password).then(({deckNames, models}) => {
|
|
this.noteInfo = {deckNames, models};
|
|
return this.noteInfo;
|
|
});
|
|
}
|
|
|
|
static scrape(username, password) {
|
|
return AnkiWeb.loadAccountPage('https://ankiweb.net/edit/', 'GET', null, username, password).then(response => {
|
|
const modelsMatch = /editor\.models = (.*}]);/.exec(response);
|
|
if (modelsMatch === null) {
|
|
return Promise.reject('failed to scrape model data');
|
|
}
|
|
|
|
const decksMatch = /editor\.decks = (.*}});/.exec(response);
|
|
if (decksMatch === null) {
|
|
return Promise.reject('failed to scrape deck data');
|
|
}
|
|
|
|
const modelsJson = JSON.parse(modelsMatch[1]);
|
|
const decksJson = JSON.parse(decksMatch[1]);
|
|
|
|
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)
|
|
});
|
|
}
|
|
|
|
return {deckNames, models};
|
|
});
|
|
}
|
|
|
|
static login(username, password) {
|
|
if (username.length === 0 || password.length === 0) {
|
|
return Promise.reject('unspecified login credentials');
|
|
}
|
|
|
|
const form = new FormData();
|
|
form.append('username', username);
|
|
form.append('password', password);
|
|
form.append('submitted', 1);
|
|
|
|
return AnkiWeb.loadPage('https://ankiweb.net/account/login', 'POST', form).then(response => {
|
|
if (!response.includes('class="mitem"')) {
|
|
return Promise.reject('failed to authenticate');
|
|
}
|
|
});
|
|
}
|
|
|
|
static loadAccountPage(url, method, form, username, password) {
|
|
return AnkiWeb.loadPage(url, method, form).then(response => {
|
|
if (response.includes('name="password"')) {
|
|
return AnkiWeb.login(username, password).then(() => AnkiWeb.loadPage(url, method, form));
|
|
} else {
|
|
return response;
|
|
}
|
|
});
|
|
}
|
|
|
|
static loadPage(url, method, form) {
|
|
return new Promise((resolve, reject) => {
|
|
const xhr = new XMLHttpRequest();
|
|
xhr.addEventListener('error', () => reject('failed to execute request'));
|
|
xhr.addEventListener('load', () => resolve(xhr.responseText));
|
|
xhr.open(method, url);
|
|
xhr.send(form);
|
|
});
|
|
}
|
|
}
|