Hopeless-cloud/update-csharp
3
0
Fork 0
Code Issues Pull Requests Actions Releases Activity

Major rewrites && Added args && Using tmp

Browse Source 
Complete rewrite of pacman database logic && Check if db.lck exists && Create db.lck when the program backups the database and delete it afterwords
This commit is contained in:
ProfessionalUwU 2023-02-01 03:17:45 +01:00
parent e07cdcfeb0
commit 58ae26b4f7
3 changed files with 153 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -1,3 +1,4 @@
.vscode/ .vscode/
obj/ obj/
bin/ bin/
justdoit.sh

64
Program.cs
View File

@ -1,27 +1,57 @@
public class Program { public class Program {
static void Main(string[] args) { static void Main(string[] args) {
string result = Update.copyEverthingBeforeUpdateToBackupLocation();
Console.WriteLine(result);
bool pacmanDatabaseResult = Update.zipPacmanDatabase(); string beforeUpdate() {
Console.WriteLine(pacmanDatabaseResult);
bool result2 = Update.zipAllContentInBackupLocation("pre-backup.zip"); Update.copyEverthingBeforeUpdateToBackupLocation();
Console.WriteLine(result2); Update.zipAllContentInBackupLocation("pre-backup.zip");
Update.zipPacmanDatabase();
string result3 = Update.copyEverthingAfterUpdateToBackupLocation();
Console.WriteLine(result3);
bool result4 = Update.zipAllContentInBackupLocation("post-backup.zip");
if (result4) {
Console.WriteLine(result4);
} else {
Console.ForegroundColor = ConsoleColor.Green;
return "pre-backup complete";
} }
string result5 = Update.copyEverthingFromBackupLocationToFinalDestination(args[0]); // "/artemis/test/" string postUpdate() {
Console.WriteLine(result5);
//Console.WriteLine(args[0]); Update.copyEverthingAfterUpdateToBackupLocation();
Update.zipAllContentInBackupLocation("post-backup.zip");
Update.copyEverthingFromBackupLocationToFinalDestination(args[0]);
Console.ForegroundColor = ConsoleColor.Green;
return "post-backup complete";
}
string testPacmanDatabase() {
Update.zipPacmanDatabase();
Console.ForegroundColor = ConsoleColor.Yellow;
return "Test complete";
}
// Use https://learn.microsoft.com/en-us/dotnet/api/system.io.filesystemwatcher?redirectedfrom=MSDN&view=net-7.0
try {
switch(args[1]) {
case "pre-backup":
beforeUpdate();
Console.ResetColor();
break;
case "post-backup":
postUpdate();
Console.ResetColor();
break;
case "testPacmanDatabase":
testPacmanDatabase();
Console.ResetColor();
break;
default:
Console.WriteLine("Wait! How did you do that?");
break;
}
} catch (Exception e) {
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine(e.Message);
}
} }
} }

143
Update.cs
View File

@ -1,10 +1,11 @@
using System.Diagnostics;
using System.IO.Compression; using System.IO.Compression;
public class Update { public class Update {
public static string getHomePath() { public static string getHomePath() {
string homePath = string.Empty; string homePath = string.Empty;
if(Environment.OSVersion.Platform == PlatformID.Unix) { if (Environment.OSVersion.Platform == PlatformID.Unix) {
homePath = Environment.GetEnvironmentVariable("HOME"); homePath = Environment.GetEnvironmentVariable("HOME");
return homePath; return homePath;
} else { } else {
@ -12,19 +13,19 @@ public class Update {
} }
} }
public static string copyEverthingBeforeUpdateToBackupLocation() { public static string copyEverthingBeforeUpdateToBackupLocation() {
string targetPath = getHomePath() + "/backup/uncompressed/"; string targetPath = "/tmp/backup/uncompressed/";
if (File.Exists(targetPath + "/pacman-after.txt") && File.Exists(targetPath + "/flatpak-after.txt")) { if (File.Exists(targetPath + "pacman-after.txt") && File.Exists(targetPath + "flatpak-after.txt")) {
File.Delete(targetPath + "/pacman-after.txt"); File.Delete(targetPath + "pacman-after.txt");
File.Delete(targetPath + "/flatpak-after.txt"); File.Delete(targetPath + "flatpak-after.txt");
} }
string[] systemFilesToCopy = {"/etc/fstab", "/etc/makepkg.conf"}; string[] systemFilesToCopy = {"/etc/fstab", "/etc/makepkg.conf"};
List<string> filesToCopy = new List<string>(systemFilesToCopy); List<string> filesToCopy = new List<string>(systemFilesToCopy);
string pacmanPackageListBeforeUpdate = getHomePath() + "/pacman-pre.txt"; string pacmanPackageListBeforeUpdate = targetPath + "pacman-pre.txt";
filesToCopy.Add(pacmanPackageListBeforeUpdate); filesToCopy.Add(pacmanPackageListBeforeUpdate);
string flatpakListBeforeUpdate = getHomePath() + "/flatpak-pre.txt"; string flatpakListBeforeUpdate = targetPath+ "flatpak-pre.txt";
filesToCopy.Add(flatpakListBeforeUpdate); filesToCopy.Add(flatpakListBeforeUpdate);
if (!Directory.Exists(targetPath)) { if (!Directory.Exists(targetPath)) {
@ -45,19 +46,19 @@ public class Update {
} }
public static string copyEverthingAfterUpdateToBackupLocation() { public static string copyEverthingAfterUpdateToBackupLocation() {
string targetPath = getHomePath() + "/backup/uncompressed/"; // Use /tmp to zip and then move into /backup/compressed/ string targetPath = "/tmp/backup/uncompressed/"; // Use /tmp to zip and then move into /backup/compressed/
if (File.Exists(targetPath + "/pacman-pre.txt") && File.Exists(targetPath + "/flatpak-pre.txt")) { if (File.Exists(targetPath + "pacman-pre.txt") && File.Exists(targetPath + "flatpak-pre.txt")) {
File.Delete(targetPath + "/pacman-pre.txt"); File.Delete(targetPath + "pacman-pre.txt");
File.Delete(targetPath + "/flatpak-pre.txt"); File.Delete(targetPath + "flatpak-pre.txt");
File.Delete(targetPath + "/fstab"); File.Delete(targetPath + "fstab");
File.Delete(targetPath + "/makepkg.conf"); File.Delete(targetPath + "makepkg.conf");
} }
List<string> filesToCopy = new List<string>(); List<string> filesToCopy = new List<string>();
string pacmanPackageListBeforeUpdate = getHomePath() + "/pacman-after.txt"; string pacmanPackageListBeforeUpdate = targetPath + "pacman-after.txt";
filesToCopy.Add(pacmanPackageListBeforeUpdate); filesToCopy.Add(pacmanPackageListBeforeUpdate);
string flatpakListBeforeUpdate = getHomePath() + "/flatpak-after.txt"; string flatpakListBeforeUpdate = targetPath + "flatpak-after.txt";
filesToCopy.Add(flatpakListBeforeUpdate); filesToCopy.Add(flatpakListBeforeUpdate);
if (!Directory.Exists(targetPath)) { if (!Directory.Exists(targetPath)) {
@ -115,7 +116,7 @@ public class Update {
Directory.CreateDirectory(targetPath); Directory.CreateDirectory(targetPath);
} }
string sourcePath = getHomePath() + "/backup/uncompressed/"; // Moved to /tmp string sourcePath = "/tmp/backup/uncompressed/";
string targetZip = getHomePath() + "/backup/compressed/" + finalZipName; string targetZip = getHomePath() + "/backup/compressed/" + finalZipName;
if (!Directory.Exists("/tmp/backup/")) { if (!Directory.Exists("/tmp/backup/")) {
@ -126,7 +127,6 @@ public class Update {
ZipFile.CreateFromDirectory(sourcePath, newFinalZip); ZipFile.CreateFromDirectory(sourcePath, newFinalZip);
if (File.Exists(targetZip)) { if (File.Exists(targetZip)) {
// ToDo verify if current zip matched old zip. If yes don't create the zip and leave the old one. If it doesn't match delete the old one.
if (!checkForIdenticalFile(targetZip, newFinalZip)) { if (!checkForIdenticalFile(targetZip, newFinalZip)) {
Console.ForegroundColor = ConsoleColor.Red; Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine($"{finalZipName} is outdated"); Console.WriteLine($"{finalZipName} is outdated");
@ -154,42 +154,61 @@ public class Update {
public static bool zipPacmanDatabase() { public static bool zipPacmanDatabase() {
string pacmanDatabaseLocation = "/var/lib/pacman/local/"; string pacmanDatabaseLocation = "/var/lib/pacman/local/";
string pacmanDatabaseZip = getHomePath() + "/backup/compressed/pacman-database.zip"; string oldPacmanDatabaseZip = getHomePath() + "/backup/compressed/pacman-database.zip";
string newPacmanDatabaseZip = "/tmp/backup/compressed/pacman-database.zip";
if (!Directory.Exists("/tmp/backup/")) {
Directory.CreateDirectory("/tmp/backup/");
}
string newPacmanDatabaseZip = "/tmp/backup/pacman-database.zip";
File.Delete(newPacmanDatabaseZip); // Delete residual Pacman Database in tmp
ZipFile.CreateFromDirectory(pacmanDatabaseLocation, newPacmanDatabaseZip);
if (File.Exists(pacmanDatabaseZip)) { if (!Directory.Exists("/tmp/backup/compressed/")) {
if (!checkForIdenticalFile(pacmanDatabaseZip, newPacmanDatabaseZip)) { Directory.CreateDirectory("/tmp/backup/compressed/");
Console.ForegroundColor = ConsoleColor.Red; }
Console.WriteLine("Pacman Database is outdated");
File.Delete(pacmanDatabaseZip); try {
if (File.Exists(newPacmanDatabaseZip)) { if(checkForLckFile("/var/lib/pacman/") == false) { // Only creates the zip if the db.lck doesn't exist
makeDatabaseLock();
if (File.Exists(newPacmanDatabaseZip)) { // Delete residual pacman database in tmp
File.Delete(newPacmanDatabaseZip); File.Delete(newPacmanDatabaseZip);
zipPacmanDatabase(); deleteDatabaseLock(); // Delete previous created database lock
} else { } else {
File.Move(newPacmanDatabaseZip, pacmanDatabaseZip); ZipFile.CreateFromDirectory(pacmanDatabaseLocation, newPacmanDatabaseZip); // If no zip exists create one
}
if (File.Exists(oldPacmanDatabaseZip)) {
if (!checkForIdenticalFile(oldPacmanDatabaseZip, newPacmanDatabaseZip)) {
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("Pacman Database is outdated");
File.Delete(oldPacmanDatabaseZip);
File.Move(newPacmanDatabaseZip, oldPacmanDatabaseZip);
} else {
Console.ForegroundColor = ConsoleColor.Green;
Console.WriteLine("Pacman Database is up to date");
File.Delete(newPacmanDatabaseZip);
}
} else {
if (!File.Exists(newPacmanDatabaseZip)) {
ZipFile.CreateFromDirectory(pacmanDatabaseLocation, newPacmanDatabaseZip); // Create the zip in tmp
}
File.Move(newPacmanDatabaseZip, oldPacmanDatabaseZip);
} }
} else { } else {
Console.ForegroundColor = ConsoleColor.Green; throw new ApplicationException("db.lck exists. Please try again later.");
Console.WriteLine("Pacman Database is up to date");
} }
} else { } catch (Exception e) {
ZipFile.CreateFromDirectory(pacmanDatabaseLocation, pacmanDatabaseZip); //deleteDatabaseLock(); // Bad practice. Only for debug purpose!
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine(e.Message);
} }
if (File.Exists(pacmanDatabaseZip)) { if (File.Exists(oldPacmanDatabaseZip)) {
deleteDatabaseLock();
return true; return true;
} else { } else {
return false; return false;
} }
} }
public static bool checkForIdenticalFile(string existingFilePath, string newFilePath) { private static bool checkForIdenticalFile(string existingFilePath, string newFilePath) {
byte[] existingFile = File.ReadAllBytes(existingFilePath); byte[] existingFile = File.ReadAllBytes(existingFilePath);
byte[] newFile = File.ReadAllBytes(newFilePath); byte[] newFile = File.ReadAllBytes(newFilePath);
@ -203,4 +222,50 @@ public class Update {
} }
return false; return false;
} }
private static bool checkForLckFile(string folderToCheck) {
if (Directory.GetFiles(folderToCheck, "*.lck").Length == 1) {
return true; // lck file exists
} else {
return false; // lck file doesn't exists
}
}
private static void makeDatabaseLock() {
var psi = new ProcessStartInfo
{
FileName = "/bin/bash",
UseShellExecute = false,
RedirectStandardOutput = true,
Arguments = string.Format("-c \"cd /var/lib/pacman/ && sudo touch db.lck && sudo chmod 000 db.lck\"")
};
using (var p = Process.Start(psi))
{
if (p != null)
{
var strOutput = p.StandardOutput.ReadToEnd();
p.WaitForExit();
}
}
}
private static void deleteDatabaseLock() {
var psi = new ProcessStartInfo
{
FileName = "/bin/bash",
UseShellExecute = false,
RedirectStandardOutput = true,
Arguments = string.Format("-c \"cd /var/lib/pacman/ && sudo rm -f db.lck\"")
};
using (var p = Process.Start(psi))
{
if (p != null)
{
var strOutput = p.StandardOutput.ReadToEnd();
p.WaitForExit();
}
}
}
} }