renders.js 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. var logging = require("../logging");
  2. var helpers = require("../helpers");
  3. var renders = require("../renders");
  4. var config = require("../../config");
  5. var cache = require("../cache");
  6. var skins = require("../skins");
  7. var path = require("path");
  8. var url = require("url");
  9. var fs = require("fs");
  10. // valid types: head, body
  11. // helmet is query param
  12. function handle_default(rid, scale, helm, body, img_status, userId, size, def, req, err, callback) {
  13. def = def || skins.default_skin(userId);
  14. if (def !== "steve" && def !== "mhf_steve" && def !== "alex" && def !== "mhf_alex") {
  15. if (helpers.id_valid(def)) {
  16. // clean up the old URL to match new image
  17. var parsed = req.url;
  18. delete parsed.query.default;
  19. delete parsed.search;
  20. parsed.pathname = parsed.pathname.replace(userId, def);
  21. var newUrl = url.format(parsed);
  22. callback({
  23. status: img_status,
  24. redirect: newUrl,
  25. err: err
  26. });
  27. } else {
  28. callback({
  29. status: img_status,
  30. redirect: def,
  31. err: err
  32. });
  33. }
  34. } else {
  35. // handle steve and alex
  36. def = def.toLowerCase();
  37. if (def.substr(0, 4) !== "mhf_") {
  38. def = "mhf_" + def;
  39. }
  40. fs.readFile(path.join(__dirname, "..", "public", "images", def + "_skin.png"), function (fs_err, buf) {
  41. // we render the default skins, but not custom images
  42. renders.draw_model(rid, buf, scale, helm, body, function(render_err, def_img) {
  43. callback({
  44. status: img_status,
  45. body: def_img,
  46. type: "image/png",
  47. hash: def,
  48. err: render_err || fs_err || err
  49. });
  50. });
  51. });
  52. }
  53. }
  54. // GET render request
  55. module.exports = function(req, callback) {
  56. var raw_type = (req.url.path_list[1] || "");
  57. var rid = req.id;
  58. var body = raw_type === "body";
  59. var userId = (req.url.path_list[2] || "").split(".")[0];
  60. var def = req.url.query.default;
  61. var scale = parseInt(req.url.query.scale) || config.renders.default_scale;
  62. var helm = req.url.query.hasOwnProperty("helm");
  63. // check for extra paths
  64. if (req.url.path_list.length > 3) {
  65. callback({
  66. status: -2,
  67. body: "Invalid Path",
  68. code: 404
  69. });
  70. return;
  71. }
  72. // validate type
  73. if (raw_type !== "body" && raw_type !== "head") {
  74. callback({
  75. status: -2,
  76. body: "Invalid Render Type"
  77. });
  78. return;
  79. }
  80. if (scale < config.renders.min_scale || scale > config.renders.max_scale) {
  81. callback({
  82. status: -2,
  83. body: "Invalid Scale"
  84. });
  85. return;
  86. } else if (!helpers.id_valid(userId)) {
  87. callback({
  88. status: -2,
  89. body: "Invalid UserID"
  90. });
  91. return;
  92. }
  93. // strip dashes
  94. userId = userId.replace(/-/g, "");
  95. try {
  96. helpers.get_render(rid, userId, scale, helm, body, function(err, status, hash, image) {
  97. if (err) {
  98. if (err.code === "ENOENT") {
  99. // no such file
  100. cache.remove_hash(rid, userId);
  101. }
  102. }
  103. if (image) {
  104. callback({
  105. status: status,
  106. body: image,
  107. type: "image/png",
  108. hash: hash,
  109. err: err
  110. });
  111. } else {
  112. logging.debug(rid, "image not found, using default.");
  113. handle_default(rid, scale, helm, body, status, userId, scale, def, req, err, callback);
  114. }
  115. });
  116. } catch(e) {
  117. handle_default(rid, scale, helm, body, -1, userId, scale, def, req, e, callback);
  118. }
  119. };