1 files changed, 121 insertions, 0 deletions
diff --git a/docker/auto-backup/AutoBackup/Program.cs b/docker/auto-backup/AutoBackup/Program.cs
new file mode 100644
index 0000000..c2e7a0d
--- /dev/null
+++ b/docker/auto-backup/AutoBackup/Program.cs
@@ -0,0 +1,121 @@
+// Read args to determine what file to upload
+
+const string DefaultUploadFilePath = "/tmp/data.tar.xz";
+string uploadFilePath = DefaultUploadFilePath;
+string? uploadDestinationPath = null;
+if (args.Length == 0)
+{
+    Console.WriteLine("You don't specify the file to upload, will upload /tmp/data.tar.xz by default.");
+    Console.WriteLine("You don't specify the destination to upload, will use timestamp with proper file extension.");
+}
+else if (args.Length == 1)
+{
+    if (args[0].Length == 0)
+    {
+        Console.Error.WriteLine("File to upload can't be empty string.");
+        Environment.Exit(2);
+    }
+    uploadFilePath = args[0];
+    Console.WriteLine("You don't specify the destination to upload, will use timestamp with proper file extension.");
+}
+else if (args.Length == 2)
+{
+    if (args[0].Length == 0)
+    {
+        Console.Error.WriteLine("File to upload can't be empty string.");
+        Environment.Exit(2);
+    }
+
+    if (args[1].Length == 0)
+    {
+        Console.Error.WriteLine("Destination to upload can't be empty string.");
+        Environment.Exit(2);
+    }
+
+    uploadFilePath = args[0];
+    uploadDestinationPath = args[1];
+}
+else
+{
+    // Write to stderr
+    Console.Error.WriteLine("You can only specify one optional file and one optional destination to upload.");
+    Environment.Exit(2);
+}
+
+// Check the upload exists
+if (!File.Exists(uploadFilePath))
+{
+    Console.Error.WriteLine($"The file {uploadFilePath} doesn't exist.");
+    Environment.Exit(3);
+}
+
+// Check the upload file is not a directory
+if (File.GetAttributes(uploadFilePath).HasFlag(FileAttributes.Directory))
+{
+    Console.Error.WriteLine($"The file {uploadFilePath} is a directory.");
+    Environment.Exit(4);
+}
+
+// Check the upload file is not bigger than 5G
+if (new FileInfo(uploadFilePath).Length > 5L * 1024L * 1024L * 1024L)
+{
+    Console.Error.WriteLine($"The file {uploadFilePath} is bigger than 5G, which is not support now.");
+    Environment.Exit(5);
+}
+
+// Get config from environment variables
+var configNameList = new List<string>{
+    "CRUPEST_AUTO_BACKUP_COS_SECRET_ID",
+    "CRUPEST_AUTO_BACKUP_COS_SECRET_KEY",
+    "CRUPEST_AUTO_BACKUP_COS_REGION",
+    "CRUPEST_AUTO_BACKUP_BUCKET_NAME"
+};
+
+var config = new Dictionary<string, string>();
+foreach (var configName in configNameList)
+{
+    var configValue = Environment.GetEnvironmentVariable(configName);
+    if (configValue is null)
+    {
+        Console.Error.WriteLine($"Environment variable {configName} is required.");
+        Environment.Exit(5);
+    }
+    config.Add(configName, configValue);
+}
+
+var region = config["CRUPEST_AUTO_BACKUP_COS_REGION"];
+var secretId = config["CRUPEST_AUTO_BACKUP_COS_SECRET_ID"];
+var secretKey = config["CRUPEST_AUTO_BACKUP_COS_SECRET_KEY"];
+var bucketName = config["CRUPEST_AUTO_BACKUP_BUCKET_NAME"];
+
+var credentials = new TencentCloudCOSHelper.Credentials(secretId, secretKey);
+
+if (uploadDestinationPath is null)
+{
+    var uploadFileName = Path.GetFileName(uploadFilePath);
+    var firstDotPosition = uploadFileName.IndexOf('.');
+    uploadDestinationPath = DateTime.Now.ToString("s");
+    if (firstDotPosition != -1)
+    {
+        uploadDestinationPath += uploadFileName.Substring(firstDotPosition + 1);
+    }
+}
+
+Console.WriteLine($"Upload file source: {uploadFilePath}");
+Console.WriteLine($"Upload COS region: {config["CRUPEST_AUTO_BACKUP_COS_REGION"]}");
+Console.WriteLine($"Upload bucket name: {config["CRUPEST_AUTO_BACKUP_BUCKET_NAME"]}");
+Console.WriteLine($"Upload file destination: {uploadDestinationPath}");
+
+await using var fileStream = new FileStream(uploadFilePath, FileMode.Open, FileAccess.Read);
+
+// 上传对象
+try
+{
+    await TencentCloudCOSHelper.PutObject(credentials, region, bucketName, uploadDestinationPath, fileStream);
+    Console.WriteLine("Upload completed!");
+}
+catch (Exception e)
+{
+    Console.Error.WriteLine("Exception: " + e);
+    Environment.Exit(6);
+}