ProfessionalUwU
58ae26b4f7
Complete rewrite of pacman database logic && Check if db.lck exists && Create db.lck when the program backups the database and delete it afterwords
271 lines
11 KiB
C#
271 lines
11 KiB
C#
using System.Diagnostics;
|
|
using System.IO.Compression;
|
|
|
|
public class Update {
|
|
public static string getHomePath() {
|
|
string homePath = string.Empty;
|
|
|
|
if (Environment.OSVersion.Platform == PlatformID.Unix) {
|
|
homePath = Environment.GetEnvironmentVariable("HOME");
|
|
return homePath;
|
|
} else {
|
|
throw new ApplicationException("This script doesn't support your operating system.");
|
|
}
|
|
}
|
|
public static string copyEverthingBeforeUpdateToBackupLocation() {
|
|
string targetPath = "/tmp/backup/uncompressed/";
|
|
|
|
if (File.Exists(targetPath + "pacman-after.txt") && File.Exists(targetPath + "flatpak-after.txt")) {
|
|
File.Delete(targetPath + "pacman-after.txt");
|
|
File.Delete(targetPath + "flatpak-after.txt");
|
|
}
|
|
|
|
string[] systemFilesToCopy = {"/etc/fstab", "/etc/makepkg.conf"};
|
|
List<string> filesToCopy = new List<string>(systemFilesToCopy);
|
|
|
|
string pacmanPackageListBeforeUpdate = targetPath + "pacman-pre.txt";
|
|
filesToCopy.Add(pacmanPackageListBeforeUpdate);
|
|
string flatpakListBeforeUpdate = targetPath+ "flatpak-pre.txt";
|
|
filesToCopy.Add(flatpakListBeforeUpdate);
|
|
|
|
if (!Directory.Exists(targetPath)) {
|
|
Directory.CreateDirectory(targetPath);
|
|
copyEverthingBeforeUpdateToBackupLocation();
|
|
} else {
|
|
foreach (string file in filesToCopy) {
|
|
FileInfo info = new FileInfo(file);
|
|
string destFile = Path.Combine(targetPath, info.Name);
|
|
File.Copy(info.FullName, destFile, true);
|
|
}
|
|
}
|
|
|
|
string copiedFiles = string.Join(", ", filesToCopy);
|
|
|
|
Console.ForegroundColor = ConsoleColor.Green;
|
|
return $"Copied {copiedFiles} to {targetPath}";
|
|
}
|
|
|
|
public static string copyEverthingAfterUpdateToBackupLocation() {
|
|
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")) {
|
|
File.Delete(targetPath + "pacman-pre.txt");
|
|
File.Delete(targetPath + "flatpak-pre.txt");
|
|
File.Delete(targetPath + "fstab");
|
|
File.Delete(targetPath + "makepkg.conf");
|
|
}
|
|
|
|
List<string> filesToCopy = new List<string>();
|
|
string pacmanPackageListBeforeUpdate = targetPath + "pacman-after.txt";
|
|
filesToCopy.Add(pacmanPackageListBeforeUpdate);
|
|
string flatpakListBeforeUpdate = targetPath + "flatpak-after.txt";
|
|
filesToCopy.Add(flatpakListBeforeUpdate);
|
|
|
|
if (!Directory.Exists(targetPath)) {
|
|
Directory.CreateDirectory(targetPath);
|
|
copyEverthingAfterUpdateToBackupLocation();
|
|
} else {
|
|
foreach (string file in filesToCopy) {
|
|
FileInfo info = new FileInfo(file);
|
|
string destFile = Path.Combine(targetPath, info.Name);
|
|
File.Copy(info.FullName, destFile, true);
|
|
}
|
|
}
|
|
|
|
string copiedFiles = string.Join(", ", filesToCopy);
|
|
|
|
Console.ForegroundColor = ConsoleColor.Green;
|
|
return $"Copied {copiedFiles} to {targetPath}";
|
|
}
|
|
|
|
/// <summary>
|
|
/// Method <c>copyEverthingFromBackupLocationToFinalDestination</c> copies everything to second Backup Location which should be a external drive or a network share. Offsite/Second Backup.
|
|
/// </summary>
|
|
public static string copyEverthingFromBackupLocationToFinalDestination(string finalBackupLocation) {
|
|
string targetPath = finalBackupLocation;
|
|
|
|
if (!Directory.Exists(targetPath)) {
|
|
Console.ForegroundColor = ConsoleColor.Red;
|
|
return $"Backup location does not exist! Please specify one in the config.";
|
|
}
|
|
|
|
if (targetPath is not null) {
|
|
string sourcePath = getHomePath() + "/backup/compressed/";
|
|
string[] intermediateBackupLocation = Directory.GetFiles(sourcePath);
|
|
|
|
if (!Directory.Exists(targetPath)) {
|
|
throw new DirectoryNotFoundException("Target directory not found");
|
|
} else {
|
|
foreach (string file in intermediateBackupLocation) {
|
|
FileInfo info = new FileInfo(file);
|
|
string destFile = Path.Combine(targetPath, info.Name);
|
|
File.Copy(info.FullName, destFile, true);
|
|
}
|
|
}
|
|
Console.ForegroundColor = ConsoleColor.Green;
|
|
return $"Copied everything successfully to {targetPath}";
|
|
} else {
|
|
Console.ForegroundColor = ConsoleColor.Red;
|
|
return "You have not configured a backup location!";
|
|
}
|
|
}
|
|
|
|
public static bool zipAllContentInBackupLocation(string finalZipName) {
|
|
string targetPath = getHomePath() + "/backup/compressed/";
|
|
if (!Directory.Exists(targetPath)) {
|
|
Directory.CreateDirectory(targetPath);
|
|
}
|
|
|
|
string sourcePath = "/tmp/backup/uncompressed/";
|
|
string targetZip = getHomePath() + "/backup/compressed/" + finalZipName;
|
|
|
|
if (!Directory.Exists("/tmp/backup/")) {
|
|
Directory.CreateDirectory("/tmp/backup/");
|
|
}
|
|
string newFinalZip = "/tmp/backup/" + finalZipName;
|
|
File.Delete(newFinalZip); // Delete residual zip's in tmp
|
|
ZipFile.CreateFromDirectory(sourcePath, newFinalZip);
|
|
|
|
if (File.Exists(targetZip)) {
|
|
if (!checkForIdenticalFile(targetZip, newFinalZip)) {
|
|
Console.ForegroundColor = ConsoleColor.Red;
|
|
Console.WriteLine($"{finalZipName} is outdated");
|
|
File.Delete(targetZip);
|
|
if (File.Exists(newFinalZip)) {
|
|
File.Delete(newFinalZip);
|
|
zipAllContentInBackupLocation(finalZipName);
|
|
} else {
|
|
File.Move(newFinalZip, targetZip);
|
|
}
|
|
} else {
|
|
Console.ForegroundColor = ConsoleColor.Green;
|
|
Console.WriteLine($"{finalZipName} is up to date");
|
|
}
|
|
} else {
|
|
ZipFile.CreateFromDirectory(sourcePath, targetZip);
|
|
}
|
|
|
|
if (File.Exists(targetZip)) {
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public static bool zipPacmanDatabase() {
|
|
string pacmanDatabaseLocation = "/var/lib/pacman/local/";
|
|
string oldPacmanDatabaseZip = getHomePath() + "/backup/compressed/pacman-database.zip";
|
|
string newPacmanDatabaseZip = "/tmp/backup/compressed/pacman-database.zip";
|
|
|
|
if (!Directory.Exists("/tmp/backup/compressed/")) {
|
|
Directory.CreateDirectory("/tmp/backup/compressed/");
|
|
}
|
|
|
|
try {
|
|
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);
|
|
deleteDatabaseLock(); // Delete previous created database lock
|
|
} else {
|
|
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 {
|
|
throw new ApplicationException("db.lck exists. Please try again later.");
|
|
}
|
|
} catch (Exception e) {
|
|
//deleteDatabaseLock(); // Bad practice. Only for debug purpose!
|
|
Console.ForegroundColor = ConsoleColor.Red;
|
|
Console.WriteLine(e.Message);
|
|
}
|
|
|
|
if (File.Exists(oldPacmanDatabaseZip)) {
|
|
deleteDatabaseLock();
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
|
|
}
|
|
|
|
private static bool checkForIdenticalFile(string existingFilePath, string newFilePath) {
|
|
byte[] existingFile = File.ReadAllBytes(existingFilePath);
|
|
byte[] newFile = File.ReadAllBytes(newFilePath);
|
|
|
|
if (existingFile.Length == newFile.Length) {
|
|
for (int i=0; i < existingFile.Length; i++) {
|
|
if (existingFile[i] != newFile[i]) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
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();
|
|
}
|
|
}
|
|
}
|
|
} |