UserInteractions.js 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  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("/upload-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[0].owner === req.session.uuid){
  125. db.execute("UPDATE playlist SET content = ?, private = ?, tmb = ?, title = ? WHERE uuid = ?", [req.body.content, req.body.private, req.body.tmb, req.body.title, req.body.playlistuuid], (err, result) => {
  126. if (err) {
  127. console.log(err);
  128. res.set(cwh).status(500).json({code: 500, R: "Err"})
  129. }
  130. })
  131. }else{
  132. res.set(cwh).status(403).json({code: 403, R: "Forbidden"})
  133. }
  134. })
  135. }else{
  136. let key = crypto.randomUUID();
  137. 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) {
  138. if (err) {
  139. console.log(err);
  140. res.set(cwh).status(500).json({code: 500, R: "AD"})
  141. return;
  142. }
  143. res.set(cwh).json({
  144. code: 200, R: "SS", UUID:key
  145. });
  146. })
  147. }
  148. })
  149. app.post("/register", function (req, res) {
  150. db.execute("SELECT uuid FROM user WHERE username = ?", [req.body.username], async function (err, rows) {
  151. if (!req.body.username || !req.body.password) {
  152. res.set(cwh).status(500).json({code: 500, R: "PE"})
  153. return;
  154. }
  155. if (err) {
  156. console.log(err);
  157. res.set(cwh).status(500).json({code: 500, R: "UNE"});
  158. return;
  159. }
  160. if (rows.length === 0) {
  161. 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()]);
  162. res.status(200).set(cwh).json({code: 200, R: "SS"});
  163. return;
  164. }
  165. res.set(cwh).status(500).json({code: 500, R: "UE"});
  166. return;
  167. })
  168. })
  169. }