andreas/saferm
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

fixed retard commit

Browse Source
This commit is contained in:
Hion-V 2022-02-20 17:34:54 +01:00
parent 01f27d6476
commit b1412b4eb2
1 changed files with 112 additions and 126 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
src/Main.hx
View File

@ -8,36 +8,32 @@ package;
import sys.FileSystem; import sys.FileSystem;
import Sys; import Sys;
enum abstract AnsiColors(String) {
var C_BLACK = '\033[0;30m';
var C_RED = '\033[31m';
var C_GREEN = '\033[32m';
var C_YELLOW = '\033[33m';
var C_BLUE = '\033[1;34m';
var C_MAGENTA = '\033[1;35m';
var C_CYAN = '\033[0;36m';
var C_GREY = '\033[0;37m';
var C_WHITE = '\033[1;37m';
var C_DEFAULT = '\033[0m';
}
class Main{ class Main{
static var switches:Map<String, String> = [
"-v" => "Prints the version number",
static var switches:Map<String,String> = [ "-t" => "Turbo mode: Skips recursively listing all content in directories"
"-v"=>"Prints the version number",
"-t"=>"Turbo mode: Skips recursively listing all content in directories"
]; ];
static var ansiColors:Map<String,String> = [];
static function setupColors():Void
{
ansiColors['black'] = '\033[0;30m';
ansiColors['red'] = '\033[31m';
ansiColors['green'] = '\033[32m';
ansiColors['yellow'] = '\033[33m';
ansiColors['blue'] = '\033[1;34m';
ansiColors['magenta'] = '\033[1;35m';
ansiColors['cyan'] = '\033[0;36m';
ansiColors['grey'] = '\033[0;37m';
ansiColors['white'] = '\033[1;37m';
}
static var SWITCHMODE_TURBO:Bool = false; static var SWITCHMODE_TURBO:Bool = false;
static function main(){ static function main() {
// Set up ansiColors
setupColors();
// Store args // Store args
var args = Sys.args(); var args = Sys.args();
// Check if we need to throw usage strings back at the user // Check if we need to throw usage strings back at the user
if(args.length == 0){ if (args.length == 0) {
usage(); usage();
return; return;
} }
@ -45,53 +41,48 @@ class Main{
var _switches:Array<String> = []; var _switches:Array<String> = [];
var files:Array<String> = []; var files:Array<String> = [];
// Loop through args to sort them into switches and files // Loop through args to sort them into switches and files
for(arg in args){ for (arg in args) {
if(arg.charAt(0) == "-" && switches.exists(arg)){ if (arg.charAt(0) == "-" && switches.exists(arg)) {
_switches.push(arg); _switches.push(arg);
//Sys.println("switch found"); // Sys.println("switch found");
} } else {
else{
files.push(arg); files.push(arg);
} }
} }
// Look at switch contents // Look at switch contents
if(_switches.contains("-v")){ if (_switches.contains("-v")) {
Sys.println('version: ${haxe.macro.Compiler.getDefine("version")}'); Sys.println('version: ${haxe.macro.Compiler.getDefine("version")}');
return; return;
} }
if(_switches.contains("-t")){ if (_switches.contains("-t")) {
SWITCHMODE_TURBO = true; SWITCHMODE_TURBO = true;
} }
// Good to go! // Good to go!
run(files, _switches); run(files, _switches);
// Reset color just in case // Reset color just in case
Sys.print(ansiColors['grey']); Sys.print(C_DEFAULT);
Sys.exit(0); Sys.exit(0);
} }
static function run(files:Array<String>, _switches:Array<String>):Void static function run(files:Array<String>, _switches:Array<String>):Void {
{
// Warn user about the operation they're about to perform // Warn user about the operation they're about to perform
Sys.println(ansiColors['red'] + 'Are you ABSOLUTELY sure you want to delete the following files/ directories:' + ansiColors['grey']); Sys.println(C_RED + 'Are you ABSOLUTELY sure you want to delete the following files/ directories:' + C_DEFAULT);
var _files:Array<String> = []; var _files:Array<String> = [];
for(file in files){ for (file in files) {
if(FileSystem.exists(file)){ if (FileSystem.exists(file)) {
var containingDir = FileSystem.absolutePath('$file/..'); var containingDir = FileSystem.absolutePath('$file/..');
var absPath = FileSystem.absolutePath('$file'); var absPath = FileSystem.absolutePath('$file');
var filename = absPath.substr(containingDir.length); var filename = absPath.substr(containingDir.length);
Sys.println(ansiColors['yellow'] + ' - $containingDir$filename' + ansiColors['grey']); Sys.println(C_YELLOW + ' - $containingDir$filename' + C_DEFAULT);
_files.push(file); _files.push(file);
} } else {
else{ Sys.println(C_RED + '$file does not exist' + C_DEFAULT);
Sys.println(ansiColors['red'] + '$file does not exist' + ansiColors['grey']);
} }
} }
Sys.println(ansiColors['white'] + 'To confirm deletion type: "YES"' + ansiColors['grey']); Sys.println(C_WHITE + 'To confirm deletion type: "YES"' + C_DEFAULT);
// Read user input // Read user input
var stdin = Sys.stdin(); var stdin = Sys.stdin();
@ -99,39 +90,34 @@ class Main{
stdin.close(); stdin.close();
// Check user input // Check user input
if(strIn == "YES"){ if (strIn == "YES") {
// Delete those fucking files // Delete those fucking files
deleteFiles(files); deleteFiles(files);
} } else {
else{
// Abort // Abort
Sys.println('Aborting operation.'); Sys.println('Aborting operation.');
return; return;
} }
} }
static function usage():Void {
static function usage():Void
{
Sys.println('Usage: ${haxe.macro.Compiler.getDefine("HAXE_OUTPUT_FILE")} [OPTION] [FILE]...'); Sys.println('Usage: ${haxe.macro.Compiler.getDefine("HAXE_OUTPUT_FILE")} [OPTION] [FILE]...');
Sys.println("Safely runs rm -rf with a confirm prompt so you don´t make an oopsie."); Sys.println("Safely runs rm -rf with a confirm prompt so you don't make an oopsie.");
Sys.println(""); Sys.println("");
Sys.println("Available OPTIONS:"); Sys.println("Available OPTIONS:");
for(key in switches.keys()){ for (key in switches.keys()) {
Sys.println(' $key: ${switches[key]}'); Sys.println(' $key: ${switches[key]}');
} }
//Sys.println(""); // Sys.println("");
//Sys.println("Safely run rm -rf with a confirm prompt so you don´t make an oopsie"); // Sys.println("Safely run rm -rf with a confirm prompt so you don´t make an oopsie");
} }
static function deleteFiles(files:Array<String>):Void
{ static function deleteFiles(files:Array<String>):Void {
for(file in files){ for (file in files) {
Sys.print('Removing ${FileSystem.absolutePath('$file')} ...'); Sys.print('Removing ${FileSystem.absolutePath('$file')} ...');
Sys.command('rm -rf $file'); Sys.command('rm -rf $file');
Sys.println('Done!'); Sys.println('Done!');
} }
Sys.println('Removed ${files.length} files'); Sys.println('Removed ${files.length} files');
} }
} }