auto-archiver-extension/source/js/background.js

// eslint-disable-next-line import/no-unassigned-import
// import './options-storage.js';
import optionsStorage from './options-storage.js';

const API_ENDPOINT = 'http://localhost:8000/tasks'

chrome.runtime.onMessage.addListener(((r, s, sR) => {
	processMessages(r, s, sR)
	return true; // needed for sendResponse to be async
}));

async function processMessages(request, sender, sendResponse) {
	console.info(`action {${request.action}} from ${sender.tab ? 'content-script (' + sender.tab.url + ')' : 'the extension'}`)
	if (request.action === "archive") {
		archiveUrl(sendResponse);
	} else if (request.action === "status") {
		const task_db = await getTaskById(request.task.task_id);
		if (task_db?.status == "SUCCESS" || task_db?.status == 'FAILURE') {
			console.log("ALREADY FINSIHED, NO REQS")
			sendResponse(task_db)
		}
		const task_fresh = await checkTaskStatus(request.task)
		sendResponse(task_fresh)
	} else if (request.action === "getTasks") {
		sendResponse(await getAllTasks());
	}
}

function archiveUrl(sendResponse) {
	chrome.tabs.query({
		active: true,
		lastFocusedWindow: true
	}, async (tabs) => {
		let url = tabs[0].url;
		console.log(`url=${url}`);
		const response = await submitUrlTask(url)
		const new_archive = { url, task_id: response.task_id, status: 'PENDING', result: {} };
		await upsertTask(new_archive);
		sendResponse(new_archive);
	});
}

function submitUrlTask(url) {
	console.log(`API: SUBMIT`)
	return new Promise((resolve, reject) => {
		fetch(API_ENDPOINT, {
			method: 'POST',
			headers: { 'Content-Type': 'application/json' },//, 'X-API-KEY': 'TODO' },
			body: JSON.stringify({ url }),
		}).then(
			response => response.json(),
		).then(response => resolve(response)
		).catch(err => {
			console.log(`There was an error: ${err}`)
			reject(err)
		});
	})
}

function checkTaskStatus(task) {
	console.log(`API: STATUS`)
	return new Promise((resolve, reject) => {
		fetch(`${API_ENDPOINT}/${task.task_id}`, {
			method: 'GET',
			headers: { 'Content-Type': 'application/json' },
		}).then(
			response => response.json(),
		).then(response => {
			const new_task = {
				url: task.url,
				task_id: response.task_id,
				status: response.task_status,
				result: JSON.parse(response.task_result),
			}
			console.log(new_task);
			upsertTask(new_task);
			resolve(new_task)
		}
		).catch(err => reject(err));
	})
}

async function getAllTasks() {
	const storage = await optionsStorage.getAll();
	return storage.archived_urls;
}

//TODO: improve with less reads from storage
async function upsertTask(task) {
	const storage = await optionsStorage.getAll();
	storage.archived_urls[task.task_id] = task;
	await optionsStorage.set(storage);
}

async function getTaskById(task) {
	const storage = await optionsStorage.getAll();
	return storage.archived_urls[task.task_id];
}