UserInteractions.js 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229
  1. import bodyParser from "body-parser";
  2. import express from "express";
  3. import session from "express-session";
  4. import cwh from "./Singletons.js";
  5. export default function UserInteractions(opts) {
  6. async function sha256(message) {
  7. // encode as UTF-8
  8. const msgBuffer = new TextEncoder().encode(message);
  9. // hash the message
  10. const hashBuffer = await crypto.subtle.digest('SHA-256', msgBuffer);
  11. // convert ArrayBuffer to Array
  12. const hashArray = Array.from(new Uint8Array(hashBuffer));
  13. // convert bytes to hex string
  14. const hashHex = hashArray.map(b => b.toString(16).padStart(2, '0')).join('');
  15. return hashHex;
  16. }
  17. let app = opts.app;
  18. app.use(session({
  19. secret: "rtifhg5878fj",
  20. resave: false,
  21. saveUninitialized: false,
  22. cookie: {
  23. sameSite: "lax", secure: "auto"
  24. }
  25. }))
  26. app.use(bodyParser.json({"limit": "200mb"}));
  27. app.use(express.json());
  28. let db = opts.db;
  29. app.options("/*", (req, res) => {
  30. res.set(cwh).end("FUCK YOU CORS")
  31. })
  32. app.get("/delacc", async (req, res) => {
  33. if (!req.session.uuid) {
  34. res.set(cwh).status(500).json({code: 500, R: "IO"})
  35. return
  36. }
  37. let uuid = req.session.uuid;
  38. db.execute("DELETE FROM user WHERE uuid = ?", [uuid], async (err, rows) => {
  39. if (err) {
  40. res.set(cwh).status(500).json({code: 500, R: "ERR"})
  41. req.session.destroy()
  42. } else {
  43. req.session.destroy()
  44. res.set(cwh).status(200).json({code: 200, R: "SUCCESS"})
  45. }
  46. })
  47. })
  48. app.get("/userapi", (req, res) => {
  49. if (req.query.getname) {
  50. db.execute("SELECT username from user where uuid = ?", [req.query.uuid], (err, result) => {
  51. res.set(cwh).end(result[0].username);
  52. })
  53. } else {
  54. if (!req.session.uuid) {
  55. res.set(cwh).status(500).json({code: 500, R: "IO"})
  56. return
  57. }
  58. db.execute("SELECT username, email from user where uuid = ?", [req.session.uuid], (err, result) => {
  59. res.set(cwh).end(JSON.stringify({
  60. uuid: req.session.uuid,
  61. username: result[0].username,
  62. email: result[0].email,
  63. }));
  64. })
  65. }
  66. })
  67. app.get("/logout", (req, res) => {
  68. req.session.destroy()
  69. res.set(cwh).status(200)
  70. })
  71. app.post("/login", async function (req, res) {
  72. if (!req.body.username || !req.body.password) {
  73. res.set(cwh).status(500).json({code: 500, R: "IO"})
  74. } else {
  75. db.execute("SELECT uuid from user where username = ? and password = ?", [req.body.username, await sha256(req.body.password)], function (err, result) {
  76. if (result.length === 0) {
  77. res.set(cwh).status(500).json({code: 500, R: "DNE"})
  78. return;
  79. }
  80. req.session.uuid = result[0].uuid;
  81. res.set(cwh).status(200).json({
  82. code: 200,
  83. R: "SS",
  84. uid: result[0].uuid
  85. });
  86. })
  87. }
  88. })
  89. app.post("/playlist-owned", async function (req, res) {
  90. if (!req.session.uuid) {
  91. res.set(cwh).status(500).json({code: 500, R: "AD"})
  92. return;
  93. }
  94. db.execute("SELECT * from playlist where owner = ?", [req.session.uuid], function (err, result) {
  95. if (err) {
  96. console.log(err);
  97. res.set(cwh).status(500).json({code: 500, R: "AD"})
  98. return;
  99. }
  100. res.set(cwh).json(result);
  101. })
  102. })
  103. app.get("/playlist", async function (req, res) {
  104. db.execute("SELECT * FROM playlist where uuid=?", [req.query.playlistuuid], function (err, result) {
  105. if (err) {
  106. console.log(err);
  107. res.set(cwh).status(500).json({code: 500, R: "AD"})
  108. return;
  109. }
  110. if (result.length === 0) {
  111. res.set(cwh).status(404).json({code: 404, R: "DNF"})
  112. return;
  113. }
  114. res.set(cwh).json(result[0]);
  115. })
  116. })
  117. app.post("/remove-playlist", async function (req, res) {
  118. if (!req.session.uuid) {
  119. res.set(cwh).status(500).json({code: 500, R: "AD"})
  120. return;
  121. }
  122. if (req.body.playlistuuid) {
  123. db.execute("SELECT owner FROM playlist WHERE uuid = ?", [req.body.playlistuuid], function (err, result) {
  124. if(result.length === 0) {
  125. res.set(cwh).status(200).json({code: 301, R: "DNM"})
  126. }
  127. if (result[0].owner === req.session.uuid) {
  128. db.execute("DELETE FROM playlist WHERE uuid = ?", [req.body.playlistuuid], (err, result) => {
  129. if (err) {
  130. console.log(err);
  131. res.set(cwh).status(500).json({code: 500, R: "Err"})
  132. return
  133. }
  134. res.set(cwh).status(200).json({code: 200, R: "SS"})
  135. })
  136. } else {
  137. res.set(cwh).status(403).json({code: 403, R: "Forbidden"})
  138. }
  139. })
  140. }
  141. })
  142. app.post("/upload-playlist", async function (req, res) {
  143. if (!req.session.uuid) {
  144. res.set(cwh).status(500).json({code: 500, R: "AD"})
  145. return;
  146. }
  147. if (req.body.playlistuuid) {
  148. db.execute("SELECT owner FROM playlist WHERE uuid = ?", [req.body.playlistuuid], function (err, result) {
  149. if (result[0].owner === req.session.uuid) {
  150. db.execute("UPDATE playlist SET content = ?, private = ?, tmb = ?, title = ? WHERE uuid = ?", [req.body.content, req.body.private, Buffer.from( Object.values(req.body.tmb.data)), req.body.title, req.body.playlistuuid], (err, result) => {
  151. if (err) {
  152. console.log(err);
  153. res.set(cwh).status(500).json({code: 500, R: "Err"})
  154. }
  155. })
  156. } else {
  157. res.set(cwh).status(403).json({code: 403, R: "Forbidden"})
  158. }
  159. })
  160. } else {
  161. let key = crypto.randomUUID();
  162. 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) {
  163. if (err) {
  164. console.log(err);
  165. res.set(cwh).status(500).json({code: 500, R: "AD"})
  166. return;
  167. }
  168. res.set(cwh).json({
  169. code: 200, R: "SS", UUID: key
  170. });
  171. })
  172. }
  173. })
  174. app.post("/register", function (req, res) {
  175. db.execute("SELECT uuid FROM user WHERE username = ?", [req.body.username], async function (err, rows) {
  176. if (!req.body.username || !req.body.password) {
  177. res.set(cwh).status(500).json({code: 500, R: "PE"})
  178. return;
  179. }
  180. if (err) {
  181. console.log(err);
  182. res.set(cwh).status(500).json({code: 500, R: "UNE"});
  183. return;
  184. }
  185. if (rows.length === 0) {
  186. 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()]);
  187. res.status(200).set(cwh).json({code: 200, R: "SS"});
  188. return;
  189. }
  190. res.set(cwh).status(500).json({code: 500, R: "UE"});
  191. return;
  192. })
  193. })
  194. }