53 lines
1.8 KiB
TypeScript
53 lines
1.8 KiB
TypeScript
import { SQL } from "bun";
|
|
import type { Token } from "./api/infos/token";
|
|
|
|
export const db = new SQL({url: process.env["URL_POSTGRESQL"]});
|
|
|
|
export const createTables = async (database: SQL): Promise<void> => {
|
|
return await database.begin(sql => sql`
|
|
CREATE TABLE IF NOT EXISTS tokens (
|
|
access_token text NOT NULL,
|
|
service text NOT NULL,
|
|
expires bigserial NOT NULL
|
|
)
|
|
`);
|
|
};
|
|
|
|
export const removeExpiredTokens = async (database: SQL): Promise<number> => {
|
|
const now = new Date();
|
|
const deleted_tokens: Token[] = await database.begin(sql => sql`
|
|
DELETE FROM tokens
|
|
WHERE expires <= ${Number(now)}
|
|
RETURNING *
|
|
`);
|
|
|
|
deleted_tokens.forEach(token => console.log("(DATABASE)", token.service, "token had expired on", new Date(Number(token.expires)), "and has been removed!"));
|
|
return deleted_tokens.length;
|
|
};
|
|
|
|
export const addToken = async (database: SQL, token: {access_token: string, service: string, expires?: Date}): Promise<Token> => {
|
|
if (!token.expires) {
|
|
// Assume it expires in one day
|
|
token.expires = new Date();
|
|
token.expires.setHours(token.expires.getHours() + 24);
|
|
}
|
|
|
|
const returned: Token[] = await database.begin(sql => sql`
|
|
INSERT INTO tokens (access_token, service, expires)
|
|
VALUES (${token.access_token}, ${token.service}, ${Number(token.expires)})
|
|
RETURNING *
|
|
`);
|
|
|
|
returned.forEach(token => console.log("(DATABASE)", token.service, "token has been added"));
|
|
return returned[0];
|
|
};
|
|
|
|
export const getToken = async (database: SQL, service: string): Promise<Token> => {
|
|
const now = new Date();
|
|
const tokens: Token[] = await database.begin(sql => sql`
|
|
SELECT * FROM tokens
|
|
WHERE service = ${service}
|
|
AND expires > ${Number(now)}
|
|
`);
|
|
return tokens[Math.floor(Math.random() * tokens.length)];
|
|
};
|