feat: sar_download_file

This could potentially be unsafe, but it could also automate srconfigs updates, challenge_maplist, fast taunts, containerridesave, vaultsave, tram save, etc. Just be careful with usage, it overwrites!

Smartly reads through game directories for existing file to overwrite (even p2common), falls back to main game directory i.e. portal2 / portalreloaded / whatever if file doesn't exist.

also fixed a bug with createTempPath -- instead of creating tmp_0_1_2 now it creates tmp_2

Author: ThisAMJ

src/Features/ConfigPlus.cpp

@@ -1,7 +1,9 @@
 #include "Event.hpp"
 #include "Features/Session.hpp"
+#include "Features/Updater.hpp"
 #include "Modules/Client.hpp"
 #include "Modules/Engine.hpp"
+#include "Modules/FileSystem.hpp"
 
 #include <cstdlib>
 #include <cstring>
@@ -11,6 +13,8 @@
 #include <vector>
 #include <unordered_set>
 #include <fstream>
+#include <curl/curl.h>
+#include <regex>
 
 // Fuck you Windows
 #ifdef _WIN32
@@ -52,6 +56,64 @@ static void SavePersistentSvars() {
 	}
 }
 
+CON_COMMAND_F(sar_download_file, "sar_download_file <url> <filepath> [directory] - Downloads a file from a URL and saves it to a path relative to game directory (e.g. portal2)\nIf directory isn't specified or invalid, looks for and overwrites existing file or uses base game directory\n", FCVAR_DONTRECORD) {
+	if (args.ArgC() < 3 || args.ArgC() > 4 || !args[1][0] || !args[2][0]) {
+		return console->Print(sar_download_file.ThisPtr()->m_pszHelpString);
+	}
+
+	if (!Utils::StartsWith(args[1], "https://s.portal2.sr/") && !Utils::StartsWith(args[1], "https://raw.githubusercontent.com/p2sr/")) {
+		return console->Print("URL domain is not portal2.sr.\n");
+	}
+
+	std::string filepath = args[2];
+	std::string gamedir = args.ArgC() > 3 ? args[3] : "";
+	
+	auto result = fileSystem->FindFileSomewhere(filepath, gamedir);
+	if (result.has_value()) {
+		filepath = result.value();
+	} else {
+		filepath = std::string(engine->GetGameDirectory()) + "/" + filepath;
+		if (gamedir != "") for (auto path : fileSystem->GetSearchPaths()) {
+			if (path.find(gamedir) != std::string::npos) {
+				filepath = path + filepath;
+				break;
+			}
+		}
+	}
+
+	CURL *curl = curl_easy_init();
+	FILE *fp;
+	CURLcode res;
+	auto temp = createTempPath("sar_download_file.tmp");
+	if (curl) {
+		fp = fopen(temp.c_str(), "wb");
+		curl_easy_setopt(curl, CURLOPT_URL, args[1]);
+		curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, &fwrite);
+		curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp);
+		res = curl_easy_perform(curl);
+		curl_easy_cleanup(curl);
+		fclose(fp);
+	}
+
+	if (res == CURLE_OK) {
+		
+		// Create any intermediate directories
+		std::string dirpath = filepath;
+		std::replace(dirpath.begin(), dirpath.end(), '\\', '/');
+		dirpath = dirpath.substr(0, dirpath.rfind('/'));
+		std::filesystem::create_directories(dirpath);
+
+		std::filesystem::remove(filepath);
+		std::filesystem::copy(temp, filepath); // for some reason moving the file doesn't work
+
+	} else {
+		console->Print("An error occurred\n");
+	}
+
+	std::filesystem::remove(temp);
+
+}
+
 static void SetSvar(std::string name, std::string val) {
 	g_svars[name] = val;
 	if (g_persistentSvars.count(name) != 0) SavePersistentSvars();

src/Features/Updater.cpp

@@ -199,7 +199,7 @@ static bool getLatestVersion(std::string *name, std::string *dlUrl, std::string
 	return true;
 }
 
-static std::string createTempPath(const char *filename) {
+std::string createTempPath(const char *filename) {
 	auto base = std::filesystem::temp_directory_path().append(filename);
 	if (!std::filesystem::exists(base)) {
 		return base.string();
@@ -209,7 +209,7 @@ static std::string createTempPath(const char *filename) {
 	std::filesystem::path p;
 
 	do {
-		p = base.concat(Utils::ssprintf("_%d", n++));
+		p = base.string() + Utils::ssprintf("_%d", n++);
 	} while (std::filesystem::exists(p));
 
 	return p.string();

src/Features/Updater.hpp

@@ -1 +1,3 @@
 #pragma once
+
+extern std::string createTempPath(const char *filename);