diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/CLI.ts | 31 | ||||
| -rw-r--r-- | src/Main.ts | 84 | ||||
| -rw-r--r-- | src/clients/Multer.ts | 68 | ||||
| -rw-r--r-- | src/clients/Prisma.ts | 19 | ||||
| -rw-r--r-- | src/routers/Auth.ts | 100 | ||||
| -rw-r--r-- | src/routers/Image.ts | 107 | ||||
| -rw-r--r-- | src/routers/Profile.ts | 58 |
7 files changed, 467 insertions, 0 deletions
diff --git a/src/CLI.ts b/src/CLI.ts new file mode 100644 index 0000000..f1e4faf --- /dev/null +++ b/src/CLI.ts @@ -0,0 +1,31 @@ +// Copyright 2022 NotDankEnough (ilotterytea) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +import { Command } from "commander"; + +/** + * Silly Command-Line Interface. + * @returns options. + */ +function CLI(): Record<string, string | boolean> { + const Program = new Command(); + + Program.option("--init", "Generate the neccessary files.", false); + Program.option("--no-ssl", "Run application in debug mode. Port for HTTP connection is \"8080\" instead of standard \"80\".", false); + + Program.parse(process.argv); + return Program.opts(); +} + +export default CLI;
\ No newline at end of file diff --git a/src/Main.ts b/src/Main.ts new file mode 100644 index 0000000..bf75f2a --- /dev/null +++ b/src/Main.ts @@ -0,0 +1,84 @@ +// Copyright 2022 NotDankEnough (ilotterytea) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +import express from "express"; + +import cookieParser from "cookie-parser"; +import bodyParser from "body-parser"; + +import http from "http"; +import https from "https"; + +import { Logger } from "tslog"; + +import ProfileRouter from "./routers/Profile"; +import AuthRouter from "./routers/Auth"; +import ImageRouter from "./routers/Image"; +import { PrismaClient } from "@prisma/client"; + +const log: Logger = new Logger({name: "main"}); + +function Main(dirPath: string, cfg: {[key: string]: any}, cli_options?: {[key: string]: any}): void { + const App: express.Express = express(); + const prisma: PrismaClient = new PrismaClient(); + + var httpc: http.Server | null = null; + var httpsc: https.Server | null = null; + + App.set("view engine", "ejs"); + App.set("views", `${dirPath}/static/ejs`); + + App.use(cookieParser()); + App.use(bodyParser.urlencoded({extended: false})); + + App.use("/me", ProfileRouter(dirPath, prisma)); + App.use("/auth", AuthRouter(dirPath, cfg, prisma)); + App.use("/", ImageRouter(dirPath, cfg, prisma)); + + App.use(express.static(`${dirPath}/static`)); + + if (cli_options) { + if (!cli_options.noSsl) { + httpc = http.createServer(App); + httpc.listen(parseInt(cfg.Ports.HTTP), () => { + log.info("Image hoster is running on port", cfg.Ports.HTTP); + }); + } else { + if ( + !cfg.Certificate.Key || + !cfg.Certificate.Cert || + !cfg.Certificate.Ca + ) { + log.error("No paths for certificate provided."); + process.exit(1); + } + + const credentials = { + key: cfg.Certificate.Key, + cert: cfg.Certificate.Cert, + ca: cfg.Certificate.Ca + } + + httpsc = https.createServer(credentials, App); + httpsc.listen(parseInt(cfg.Ports.HTTPS), () => { + log.info("Image hoster is running on port", cfg.Ports.HTTPS, "(SSL)"); + }); + } + } else { + log.error("NO CLI OPTIONS PROVIDED!!!") + process.exit(1); + } +} + +export default Main;
\ No newline at end of file diff --git a/src/clients/Multer.ts b/src/clients/Multer.ts new file mode 100644 index 0000000..81f2b0f --- /dev/null +++ b/src/clients/Multer.ts @@ -0,0 +1,68 @@ +// Copyright 2022 NotDankEnough (ilotterytea) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +import multer from "multer"; +import mime from "mime"; +import crypto from "crypto"; +import { readdirSync } from "fs"; + +const fileFilter = (req: Express.Request, res: Express.Response, cb: multer.FileFilterCallback) => { + cb(null, true); +} + +const storage = multer.diskStorage({ + destination: "static/images", + filename(req, file, callback) { + if (!req.headers["pattern"]) req.headers["pattern"] = "random"; + + switch (req.headers["pattern"]) { + case "numeration": { + const img_id: number = readdirSync("images").length + 1; + const max_zero: number = 5; + var blank_string: string = ""; + + for (var i = 0; i < max_zero; i++) { blank_string = blank_string + "0"; } + + blank_string = blank_string.slice(img_id.toString().length, blank_string.length); + + callback(null, `${blank_string}${img_id.toString()}.${mime.getExtension(file.mimetype)}`); + break; + } + case "random": { + crypto.randomBytes(5, (err, raw) => { + if (err) return callback(err, ""); + callback(null, `${raw.toString("hex")}.${mime.getExtension(file.mimetype)}`) + }); + break; + } + default: { + crypto.randomBytes(5, (err, raw) => { + if (err) return callback(err, ""); + callback(null, `${raw.toString("hex")}.${mime.getExtension(file.mimetype)}`) + }); + break; + } + } + } +}); + +const Multer = multer({ + fileFilter: fileFilter, + storage: storage, + limits: { + fieldSize: 1.6e+7 + } +}); + +export default Multer;
\ No newline at end of file diff --git a/src/clients/Prisma.ts b/src/clients/Prisma.ts new file mode 100644 index 0000000..1e17fc9 --- /dev/null +++ b/src/clients/Prisma.ts @@ -0,0 +1,19 @@ +// Copyright 2022 NotDankEnough (ilotterytea) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +import { PrismaClient } from "@prisma/client"; + +const prisma = new PrismaClient(); + +export default prisma;
\ No newline at end of file diff --git a/src/routers/Auth.ts b/src/routers/Auth.ts new file mode 100644 index 0000000..146bafe --- /dev/null +++ b/src/routers/Auth.ts @@ -0,0 +1,100 @@ +// Copyright 2022 NotDankEnough (ilotterytea) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +import { Router, CookieOptions } from "express"; +import { PrismaClient, User } from "@prisma/client"; +import { randomBytes } from "crypto"; +import axios from "axios"; +import { Logger } from "tslog"; + +const log: Logger = new Logger({name: "authlog"}); + +function AuthRouter(dirPath: string, cfg: {[key: string]: any}, prisma: PrismaClient): Router { + const router: Router = Router(); + + router.get("/twitch", async (req, res) => { + if (!("code" in req.query)) { + return res.json({ + status: 400, + reason: "\"code\" query not found." + }).status(400); + } + + try { + const req_token = await axios.post("https://id.twitch.tv/oauth2/token", `client_id=${cfg.Auth.ClientID}&client_secret=${cfg.Auth.ClientSecret}&code=${req.query.code}&grant_type=authorization_code&redirect_uri=${cfg.Auth.RedirectURI}`, {headers: {"Content-Type": "application/x-www-form-urlencoded"}}); + + const ttv_user = await axios.get("https://api.twitch.tv/helix/users", { + responseType: "json", + headers: { + "Authorization": `Bearer ${req_token.data.access_token}`, + "Client-Id": cfg.Auth.ClientID + } + }); + + const user_data = ttv_user.data.data[0]; + + const user: User | null = await prisma.user.findFirst({ + where: { + alias_id: parseInt(user_data.id) + } + }); + + const key: string = randomBytes(16).toString("hex"); + const cookie_opts: CookieOptions = { + httpOnly: false, + secure: true, + sameSite: "lax" + }; + + if (!user) { + await prisma.user.create({ + data: { + alias_id: parseInt(user_data.id), + name: user_data.login, + desc: user_data.description, + pic: user_data.profile_image_url, + key: key + } + }); + + res.cookie("key", key, cookie_opts); + res.cookie("id", user_data.id, cookie_opts); + } else { + await prisma.user.update({ + where: {id: user.id}, + data: { + name: user_data.login, + desc: user_data.description, + pic: user_data.profile_image_url, + key: key + } + }); + + res.cookie("key", key, cookie_opts); + res.cookie("id", user_data.id, cookie_opts); + } + + res.redirect("/me"); + } catch (err: any) { + res.json({ + status: (err.response.data.status) ? err.response.data.status : 400, + message: (err.response.data.message) ? err.response.data.message : "Bad request." + }).status((err.response.data.status) ? err.response.data.status : 400); + } + }); + + return router; +} + +export default AuthRouter;
\ No newline at end of file diff --git a/src/routers/Image.ts b/src/routers/Image.ts new file mode 100644 index 0000000..dfd6a16 --- /dev/null +++ b/src/routers/Image.ts @@ -0,0 +1,107 @@ +// Copyright 2022 NotDankEnough (ilotterytea) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +import { PrismaClient, User, Image } from "@prisma/client"; +import express from "express"; +import mime from "mime"; +import multer from "../clients/Multer"; + +function ImageRouter(dirPath: string, cfg: {[key: string]: any}, prisma: PrismaClient): express.Router { + const router: express.Router = express.Router(); + + router.get("/", async (req, res) => { + const user: User | null = await prisma.user.findFirst({ + where: { + alias_id: (req.cookies.id) ? parseInt(req.cookies.id) : null, + key: (req.cookies.key) ? req.cookies.key : "" + } + }); + + return res.render("pages/home", { + user: user, + cid: cfg.Auth.ClientID, + uri: cfg.Auth.RedirectURI + }); + }); + + router.post("/upload", multer.single("file"), async (req, res) => { + const auth: string[] | undefined = (req.headers["authorization"]) ? new Buffer(req.headers["authorization"], "base64").toString("utf-8").split(':') : undefined; + + var user: User | null = await prisma.user.findFirst({ + where: { + alias_id: (auth) ? parseInt(auth[1]) : (req.cookies.id) ? parseInt(req.cookies["id"]) : null, + key: (auth) ? auth[2] : (req.cookies.key) ? req.cookies["key"] : "" + } + }); + + const image: Image = await prisma.image.create({ + data: { + storage_id: req.file!.filename!, + ext: mime.getExtension(req.file!.mimetype)!, + userId: (user) ? user.id : null + } + }); + + return res.send(`https://i.hmmtodayiwill.ru/${image.id}`).status(200); + }); + + + router.get("/:imageId", async (req, res) => { + const auth: string[] | undefined = (req.headers["authorization"]) ? new Buffer(req.headers["authorization"], "base64").toString("utf-8").split(':') : undefined; + + var user: User | null = await prisma.user.findFirst({ + where: { + alias_id: (auth) ? parseInt(auth[0]) : (req.cookies.id) ? parseInt(req.cookies["id"]) : null, + key: (auth) ? auth[1] : (req.cookies.key) ? req.cookies["key"] : "" + } + }); + + const image: Image | null = await prisma.image.findFirst({ + where: { + id: req.params.imageId + } + }); + + if (!image) { + return res.json({ + status: 404, + reason: "Image ID " + req.params.imageId + " not found in database!" + }).status(404); + } + + if (image.is_hidden) { + if (user) { + if (image.userId !== user.id) { + return res.json({ + status: 401, + reason: "Image ID " + req.params.imageId + " have a hide flag." + }).status(401); + } else { + return res.sendFile(`${__dirname}/static/images/${image.storage_id}`); + } + } else { + return res.json({ + status: 401, + reason: "Image ID " + req.params.imageId + " have a hide flag." + }).status(401); + } + } + + return res.sendFile(`${dirPath}/static/images/${image.storage_id}`); + }); + + return router; +} + +export default ImageRouter;
\ No newline at end of file diff --git a/src/routers/Profile.ts b/src/routers/Profile.ts new file mode 100644 index 0000000..afc5e5d --- /dev/null +++ b/src/routers/Profile.ts @@ -0,0 +1,58 @@ +// Copyright 2022 NotDankEnough (ilotterytea) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +import { PrismaClient, User, Image } from "@prisma/client"; +import { Router } from "express"; + +function ProfileRouter(dirPath: string, prisma: PrismaClient): Router { + const router: Router = Router(); + + router.get("/", async (req, res) => { + if (!req.cookies.id || !req.cookies.key) { + return res.redirect("/"); + } + + const user: User | null = await prisma.user.findFirst({ + where: { + alias_id: parseInt(req.cookies.id), + key: req.cookies.key + } + }); + + if (!user) { + return res.redirect("/"); + } + + var images: Image[] = await prisma.image.findMany({ + where: { + userId: user.id + }, + orderBy: { + timestamp: "desc" + } + }); + + const keyNotEncoded: string = `${user.id}:${user.alias_id}:${user.key}`; + + return res.render("pages/me", { + user: user, + images: images, + authKey: Buffer.alloc(keyNotEncoded.length, keyNotEncoded).toString("base64") + }); + }); + + return router; +} + +export default ProfileRouter;
\ No newline at end of file |