learn-request-queue/src/app.ts

152 lines
4.9 KiB
TypeScript

import * as config from "./config";
import * as requests from "./requests";
import * as twitch from "./twitch";
import { URLSearchParams } from "url";
import { QueryResult } from "pg";
import express from "express";
import session from "express-session";
import pgSessionStore from "connect-pg-simple";
import fetch, { Response as FetchResponse } from "node-fetch";
import * as eta from "eta";
import db from "./db";
import errorHandler from "./errors";
const app = express();
app.use(express.static('public'));
app.use(express.urlencoded({extended: false}));
app.use(session({
secret: config.sessionSecret,
saveUninitialized: false,
resave: false,
store: new (pgSessionStore(session))({
pool: db
})
}));
// API
app.get("/api/getRequests", async (request, response) => {
var requestCount = ( request.query.count ? parseInt(request.query.count as string, 10) : 5 );
requests.getRequests(requestCount).then((val: QueryResult) => response.send(val))
.catch((e: any) => errorHandler(request,response,e));
});
app.get("/api/getAllRequests", async (request, response) => {
var requestCount = ( request.query.count ? parseInt(request.query.count as string, 10) : 5 );
requests.getAllRequests(requestCount).then((val: QueryResult) => response.send(val))
.catch((e: any) => errorHandler(request,response,e));
});
app.post("/api/addRequest", async (request, response) => {
response.type('text/plain');
if (!request.session || !request.session.user) {
response.status(401);
response.send("Must be logged in");
return;
}
if (!request.body.url) {
response.status(400);
response.send("Missing url");
return
}
var url = request.body.url as string;
var requester = request.session.user.display_name;
requests.addRequest(url,requester).then((val: [number,string]) => {
response.status(val[0]);
response.send(val[1]);
})
.catch((e: any) => errorHandler(request,response,e));
});
app.post("/api/updateRequestState", async (request, response) => {
response.type('text/plain');
if (!request.body.url) {
response.status(400);
response.send("Missing url");
return
}
if (!request.body.state) {
response.status(400);
response.send("Missing scoreDiff");
return
}
var url = request.body.url as string;
var state = request.body.state as string;
requests.updateRequestState(url,state).then((val: [number,string]) => {
response.status(val[0]);
response.send(val[1]);
})
.catch((e: any) => errorHandler(request,response,e));
});
app.post("/api/updateRequestScore", async (request, response) => {
response.type('text/plain');
if (!request.body.url) {
response.status(400);
response.send("Missing url");
return
}
if (!request.body.scoreDiff) {
response.status(400);
response.send("Missing scoreDiff");
return
}
var url = request.body.url as string;
var scoreDiff = parseInt(request.body.scoreDiff as string, 10);
requests.updateRequestScore(url,scoreDiff).then((val: string) => response.send(val))
.catch((e: any) => errorHandler(request,response,e));
});
app.post("/api/deleteRequest", async (request, response) => {
response.type('text/plain');
if (!request.body.url) {
response.status(400);
response.send("Missing url");
return
}
var url = request.body.url as string;
requests.deleteRequest(url).then((val: string) => response.send(val))
.catch((e: any) => errorHandler(request,response,e));
});
// Twitch callback
app.get("/callback", async (request, response) => {
if (request.query.error) {
response.redirect(307, '/');
return;
}
var authcode = request.query.code as string;
var tokenResponse = await fetch("https://id.twitch.tv/oauth2/token", { method: "POST", body: new URLSearchParams({
client_id: config.twitchClientId,
client_secret: config.twitchSecret,
code: authcode,
grant_type: "authorization_code",
redirect_uri: `${config.urlPrefix}/callback`
})}).then((res: FetchResponse) => res.json() as Promise<twitch.TokenResponse>)
.catch((e: any) => errorHandler(request,response,e));
if (typeof request.session == 'undefined') throw new Error('Session is undefined');
if (typeof tokenResponse == 'undefined') throw new Error('tokenResponse is undefined');
request.session.tokenpair = { access_token: tokenResponse.access_token, refresh_token: tokenResponse.refresh_token };
request.session.user = (await twitch.apiRequest(request.session.tokenpair,"GET","/users")).data[0];
response.redirect(307, '/');
});
// Frontend templates
app.get("/", async (request, response) => {
if (!request.session) {
throw new Error ("Missing request.session")
}
if (!request.session.user) {
response.render('main.eta', { loggedIn: false });
} else {
response.render('main.eta', { loggedIn: true, userName: request.session.user.display_name, userProfilePicture: request.session.user.profile_image_url });
}
});
// Logout
app.get ("/logout", async (request, response) => request.session!.destroy((err) => response.redirect(307, '/')));
const server = app.listen(config.port, () => {
console.log(`Listening on port ${config.port}`);
});