applist image attribute

sunshine/nvhttp.cpp

@@ -761,12 +761,39 @@ void cancel(resp_https_t response, req_https_t request) {
   }
 }
 
+#define CHECK_EXPECTED_EXTENTIONS(extention) (extention == "png" || extention == "jpg"  || extention == "jpeg")
+
 void appasset(resp_https_t response, req_https_t request) {
   print_req<SimpleWeb::HTTPS>(request);
 
-  std::ifstream in(SUNSHINE_ASSETS_DIR "/box.png");
-  response->write(SimpleWeb::StatusCode::success_ok, in);
+  auto args = request->parse_query_string();
+  auto appid = util::from_view(args.at("appid")) - 1;
+  auto app_image = proc::proc.get_app_image(appid);
+  
+  BOOST_LOG(debug) << "/appasset: ["sv << app_image << "] -- image"sv;
+
+  if (app_image.empty()) {
+    app_image = "steam.png";
+  }
+  auto file_path = SUNSHINE_ASSETS_DIR "/" + app_image;
+  
+  BOOST_LOG(debug) << "/appasset: ["sv << file_path << "] -- image path"sv;
+    // check if files exists
+  std::error_code code;
+  auto image_extention = std::filesystem::path(file_path).extension().string();
+  image_extention = image_extention.substr(1, image_extention.length() - 1);
+  if (!std::filesystem::exists(file_path, code) || !CHECK_EXPECTED_EXTENTIONS(image_extention)) {
+    BOOST_LOG(debug) << "/appasset: ["sv << file_path << "] [extention="sv << image_extention << "] -- file not found"sv;
+    response->write(SimpleWeb::StatusCode::client_error_not_found);
+    return;
+  }
+
+  SimpleWeb::CaseInsensitiveMultimap header;
+  header.emplace("Content-Type", "image/" + image_extention);
+  std::ifstream in(file_path);
+  response->write(SimpleWeb::StatusCode::success_ok, in, header);
   response->close_connection_after_response = true;
+  BOOST_LOG(debug) << "/appasset: ["sv << file_path << "] [extention="sv << image_extention << "] -- sent"sv;
 }
 
 void start() {