advent-of-code/2021/common.ts

114 lines
2.9 KiB
TypeScript

import { ensureDir, path, readAll, readLines } from "./deps.ts";
export const HOME = Deno.env.get("HOME") || "/tmp";
export const XDG_CACHE_HOME = Deno.env.get("XDG_CACHE_HOME") ||
path.join(HOME, ".config");
const INPUT_CACHE_DIR = path.join(XDG_CACHE_HOME, "aoc2021");
async function fileExists(filePath: string) {
try {
await Deno.lstat(filePath);
return true;
} catch (e) {
if (e instanceof Deno.errors.NotFound) {
return false;
}
throw e;
}
}
function inputFilePath(day: string) {
return path.join(INPUT_CACHE_DIR, `${day}.input`);
}
async function ensureInput(day: string) {
const filePath = inputFilePath(day);
await ensureDir(INPUT_CACHE_DIR);
if (await fileExists(filePath)) return;
const cookieFile = path.join(HOME, ".advent-of-code-2021-auth-cookie");
let cookie = "";
try {
cookie = Deno.env.get("ADVENT_OF_CODE_2021_AUTH_COOKIE") ||
new TextDecoder().decode(
await readAll(await Deno.open(cookieFile)),
);
} catch (e) {
console.error(
`Failed to load adventofcode.com cookie from ADVENT_OF_CODE_2021_AUTH_COOKIE environment variable or from file ${cookieFile}`,
);
throw e;
}
const url = `https://adventofcode.com/2021/day/${day}/input`;
const response = await fetch(url, { headers: { cookie } });
await Deno.writeTextFile(filePath, await response.text());
}
async function fileLines(
filePath: string,
): Promise<AsyncIterableIterator<string>> {
return readLines(await Deno.open(filePath));
}
/** Returns an generator that will yield lines of input for the given Advent of
* Code 2021 day.
*/
export async function inputLines(
day: string,
): Promise<AsyncIterableIterator<string>> {
await ensureInput(day);
return fileLines(inputFilePath(day));
}
async function* readStringsAsNumbers(
input: AsyncIterableIterator<string>,
): AsyncIterableIterator<number> {
for await (const s of input) {
yield parseInt(s);
}
}
/** Returns an generator that will yield lines of input parsed as integers for
* the given Advent of Code 2021 day.
*/
export async function inputNumbers(
day: string,
): Promise<AsyncIterableIterator<number>> {
return readStringsAsNumbers(await inputLines(day));
}
export async function* asyncIterator<T>(
array: T[],
): AsyncIterableIterator<T> {
for (const v of array) {
yield v;
}
}
export async function preloadedAsyncIterator<T>(
iterator: AsyncIterableIterator<T>,
): Promise<AsyncIterableIterator<T>> {
const arr: T[] = [];
for await (const v of iterator) {
arr.push(v);
}
return asyncIterator(arr);
}
export async function measureDuration(cb: () => Promise<void> | void) {
const p1t = window.performance.now();
await cb();
const p1t2 = window.performance.now();
console.log(p1t2 - p1t, "ms");
}
export async function collectArray<T>(
iterator: AsyncIterableIterator<T>,
): Promise<T[]> {
const arr: T[] = [];
for await (const v of iterator) {
arr.push(v);
}
return arr;
}