1 files changed, 141 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..0b5845a
--- /dev/null
+++ b/docker/auto-backup/AutoBackup/Program.cs
@@ -0,0 +1,141 @@
+using COSXML;
+using COSXML.Auth;
+using COSXML.Transfer;
+
+// Check I'm root
+if (Environment.UserName != "root")
+{
+    Console.WriteLine("You must run this program as root");
+    Environment.Exit(1);
+}
+
+// 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);
+}
+
+// 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);
+    }
+}
+
+var cosConfig = new CosXmlConfig.Builder()
+    .IsHttps(true)
+    .SetRegion(config["CRUPEST_AUTO_BACKUP_COS_REGION"])
+    .Build();
+
+QCloudCredentialProvider cosCredentialProvider =
+    new DefaultQCloudCredentialProvider(
+        config["CRUPEST_AUTO_BACKUP_COS_SECRET_ID"],
+        config["CRUPEST_AUTO_BACKUP_COS_SECRET_KEY"],
+        60
+    );
+
+CosXml cosXml = new CosXmlServer(cosConfig, cosCredentialProvider);
+
+TransferConfig transferConfig = new TransferConfig();
+
+TransferManager transferManager = new TransferManager(cosXml, transferConfig);
+
+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}");
+
+// 上传对象
+COSXMLUploadTask uploadTask = new COSXMLUploadTask(config["CRUPEST_AUTO_BACKUP_BUCKET_NAME"], uploadDestinationPath);
+uploadTask.SetSrcPath(uploadFilePath);
+
+uploadTask.progressCallback = delegate (long completed, long total)
+{
+    Console.WriteLine(String.Format("progress = {0:##.##}%", completed * 100.0 / total));
+};
+
+try
+{
+    COSXMLUploadTask.UploadTaskResult result = await transferManager.UploadAsync(uploadTask);
+    Console.WriteLine("Upload completed!");
+}
+catch (Exception e)
+{
+    Console.Error.WriteLine("CosException: " + e);
+    Environment.Exit(6);
+}