UserInteractions.js 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296
  1. import bodyParser from "body-parser";
  2. import express from "express";
  3. import session from "express-session";
  4. import cwh from "./Singletons.js";
  5. import fs from "fs";
  6. import webp from "webp-converter";
  7. export default function UserInteractions(opts) {
  8. async function sha256(message) {
  9. // encode as UTF-8
  10. const msgBuffer = new TextEncoder().encode(message);
  11. // hash the message
  12. const hashBuffer = await crypto.subtle.digest('SHA-256', msgBuffer);
  13. // convert ArrayBuffer to Array
  14. const hashArray = Array.from(new Uint8Array(hashBuffer));
  15. // convert bytes to hex string
  16. const hashHex = hashArray.map(b => b.toString(16).padStart(2, '0')).join('');
  17. return hashHex;
  18. }
  19. let app = opts.app;
  20. app.use(session({
  21. secret: "rtifhg5878fj",
  22. resave: false,
  23. saveUninitialized: false,
  24. cookie: {
  25. sameSite: "lax", secure: "auto"
  26. }
  27. }))
  28. app.use(bodyParser.json({"limit": "200mb"}));
  29. app.use(express.json());
  30. let db = opts.db;
  31. app.options("/*", (req, res) => {
  32. res.set(cwh).end("FUCK YOU CORS")
  33. })
  34. app.get("/delacc", async (req, res) => {
  35. if (!req.session.uuid) {
  36. res.set(cwh).status(500).json({code: 500, R: "IO"})
  37. return
  38. }
  39. let uuid = req.session.uuid;
  40. db.execute("DELETE FROM user WHERE uuid = ?", [uuid], async (err, rows) => {
  41. if (err) {
  42. res.set(cwh).status(500).json({code: 500, R: "ERR"})
  43. req.session.destroy()
  44. } else {
  45. req.session.destroy()
  46. res.set(cwh).status(200).json({code: 200, R: "SUCCESS"})
  47. }
  48. })
  49. })
  50. app.get("/userapi", (req, res) => {
  51. if (req.query.getname) {
  52. db.execute("SELECT username from user where uuid = ?", [req.query.uuid], (err, result) => {
  53. res.set(cwh).end(result[0].username);
  54. })
  55. } else {
  56. if (!req.session.uuid) {
  57. res.set(cwh).status(500).json({code: 500, R: "IO"})
  58. return
  59. }
  60. db.execute("SELECT username, email from user where uuid = ?", [req.session.uuid], (err, result) => {
  61. res.set(cwh).end(JSON.stringify({
  62. uuid: req.session.uuid,
  63. username: result[0].username,
  64. email: result[0].email,
  65. }));
  66. })
  67. }
  68. })
  69. app.get("/logout", (req, res) => {
  70. req.session.destroy()
  71. res.set(cwh).status(200)
  72. })
  73. app.post("/login", async function (req, res) {
  74. if (!req.body.username || !req.body.password) {
  75. res.set(cwh).status(500).json({code: 500, R: "IO"})
  76. } else {
  77. db.execute("SELECT uuid from user where username = ? and password = ?", [req.body.username, await sha256(req.body.password)], function (err, result) {
  78. if (result.length === 0) {
  79. res.set(cwh).status(500).json({code: 500, R: "DNE"})
  80. return;
  81. }
  82. req.session.uuid = result[0].uuid;
  83. res.set(cwh).status(200).json({
  84. code: 200,
  85. R: "SS",
  86. uid: result[0].uuid
  87. });
  88. })
  89. }
  90. })
  91. app.post("/playlist-owned", async function (req, res) {
  92. if (!req.session.uuid) {
  93. res.set(cwh).status(500).json({code: 500, R: "AD"})
  94. return;
  95. }
  96. db.execute("SELECT * from playlist where owner = ?", [req.session.uuid], function (err, result) {
  97. if (err) {
  98. console.log(err);
  99. res.set(cwh).status(500).json({code: 500, R: "AD"})
  100. return;
  101. }
  102. res.set(cwh).json(result);
  103. })
  104. })
  105. app.get("/playlist", async function (req, res) {
  106. db.execute("SELECT owner, uuid, content, private FROM playlist where uuid=?", [req.query.playlistuuid], function (err, result) {
  107. if (err) {
  108. console.log(err);
  109. res.set(cwh).status(500).json({code: 500, R: "AD"})
  110. return;
  111. }
  112. if (result.length === 0) {
  113. res.set(cwh).status(404).json({code: 404, R: "DNF"})
  114. return;
  115. }
  116. res.set(cwh).json(result[0]);
  117. })
  118. })
  119. app.get("/playlist-name", async function (req, res) {
  120. db.execute("SELECT title FROM playlist where uuid=?", [req.query.playlistuuid], function (err, result) {
  121. if (err) {
  122. console.log(err);
  123. res.set(cwh).status(500).json({code: 500, R: "AD"})
  124. return;
  125. }
  126. if (result.length === 0) {
  127. res.set(cwh).status(404).json({code: 404, R: "DNF"})
  128. return;
  129. }
  130. res.set(cwh).end(result[0].title);
  131. })
  132. })
  133. app.get('/playlist-tmb', async function (req, res) {
  134. db.execute("SELECT tmb FROM playlist where uuid=?", [req.query.playlistuuid], function (err, result) {
  135. if (err) {
  136. console.log(err);
  137. res.set(cwh).status(500).json({code: 500, R: "AD"})
  138. return;
  139. }
  140. if (result.length === 0) {
  141. res.set(cwh).status(404).json({code: 404, R: "DNF"})
  142. return;
  143. }
  144. res.set(cwh).json(result[0]);
  145. })
  146. })
  147. app.post("/remove-playlist", async function (req, res) {
  148. if (!req.session.uuid) {
  149. res.set(cwh).status(500).json({code: 500, R: "AD"})
  150. return;
  151. }
  152. if (req.body.playlistuuid) {
  153. db.execute("SELECT owner FROM playlist WHERE uuid = ?", [req.body.playlistuuid], function (err, result) {
  154. if (result.length === 0) {
  155. res.set(cwh).status(200).json({code: 301, R: "DNM"})
  156. }
  157. if (result[0].owner === req.session.uuid) {
  158. db.execute("DELETE FROM playlist WHERE uuid = ?", [req.body.playlistuuid], (err, result) => {
  159. if (err) {
  160. console.log(err);
  161. res.set(cwh).status(500).json({code: 500, R: "Err"})
  162. return
  163. }
  164. res.set(cwh).status(200).json({code: 200, R: "SS"})
  165. })
  166. } else {
  167. res.set(cwh).status(403).json({code: 403, R: "Forbidden"})
  168. }
  169. })
  170. }
  171. })
  172. app.post("/upload-playlist", async function (req, res) {
  173. if (!req.session.uuid) {
  174. res.set(cwh).status(500).json({code: 500, R: "AD"})
  175. return;
  176. }
  177. if (req.body.playlistuuid) {
  178. db.execute("SELECT owner FROM playlist WHERE uuid = ?", [req.body.playlistuuid], async function (err, result) {
  179. if (result[0].owner === req.session.uuid) {
  180. let uuid = crypto.randomUUID();
  181. fs.writeFileSync("tmp/"+uuid, Buffer.from(Object.values(req.body.tmb.data)));
  182. try{
  183. await webp.cwebp("tmp/"+uuid, "tmp/"+uuid+".webp", "-q 80 -size 100000 -mt -metadata all", "-v")
  184. }catch(e){
  185. console.log(e)
  186. res.set(cwh).status(500).json({code: 500, R: "Err"})
  187. fs.rm("tmp/"+uuid, (err) => {
  188. if (err) {
  189. console.log(err);
  190. }
  191. });
  192. fs.rm("tmp/"+uuid+".webp", (err) => {
  193. if (err) {
  194. console.log(err);
  195. }
  196. });
  197. return;
  198. }
  199. let pl_cover = fs.readFileSync("tmp/"+uuid+".webp");
  200. fs.rm("tmp/"+uuid, (err) => {
  201. if (err) {
  202. console.log(err);
  203. }
  204. });
  205. fs.rm("tmp/"+uuid+".webp", (err) => {
  206. if (err) {
  207. console.log(err);
  208. }
  209. });
  210. res.set(cwh).status(200).json({code: 200, R: "SS"})
  211. db.execute("UPDATE playlist SET content = ?, private = ?, tmb = ?, title = ? WHERE uuid = ?", [req.body.content, req.body.private, pl_cover, req.body.title, req.body.playlistuuid], (err, result) => {
  212. if (err) {
  213. console.log(err);
  214. res.set(cwh).status(500).json({code: 500, R: "Err"})
  215. }
  216. })
  217. } else {
  218. res.set(cwh).status(403).json({code: 403, R: "Forbidden"})
  219. }
  220. })
  221. } else {
  222. let key = crypto.randomUUID();
  223. db.execute("INSERT INTO playlist(owner, uuid, content, private, tmb, title) values (?,?,?,?,?,?)", [req.session.uuid, key, req.body.content, req.body.private, req.body.tmb, req.body.title], function (err, result) {
  224. if (err) {
  225. console.log(err);
  226. res.set(cwh).status(500).json({code: 500, R: "AD"})
  227. return;
  228. }
  229. res.set(cwh).json({
  230. code: 200, R: "SS", UUID: key
  231. });
  232. })
  233. }
  234. })
  235. app.post("/register", function (req, res) {
  236. db.execute("SELECT uuid FROM user WHERE username = ?", [req.body.username], async function (err, rows) {
  237. if (!req.body.username || !req.body.password) {
  238. res.set(cwh).status(500).json({code: 500, R: "PE"})
  239. return;
  240. }
  241. if (err) {
  242. console.log(err);
  243. res.set(cwh).status(500).json({code: 500, R: "UNE"});
  244. return;
  245. }
  246. if (rows.length === 0) {
  247. db.execute("INSERT INTO user (uuid, username, email, password, avatar, time) values (?,?,?,?,?,?)", [crypto.randomUUID(), req.body.username, !req.body.email ? null : req.body.email, await sha256(req.body.password), null, Date.now()]);
  248. res.status(200).set(cwh).json({code: 200, R: "SS"});
  249. return;
  250. }
  251. res.set(cwh).status(500).json({code: 500, R: "UE"});
  252. return;
  253. })
  254. })
  255. }