From 57804f3ce8fc5f81e7c9d292fbbf5796b08012c8 Mon Sep 17 00:00:00 2001
From: Yaya4 <jemefaitchierdu65@gmail.com>
Date: Thu, 6 May 2021 01:02:06 +0200
Subject: [PATCH] Some fixes.

---
 .gitignore        |  1 +
 index.js          | 59 +++++++++++++++++++++++++++++------------------
 package-lock.json |  6 ++++-
 package.json      |  2 +-
 4 files changed, 44 insertions(+), 24 deletions(-)
 create mode 100644 .gitignore

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..b512c09
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+node_modules
\ No newline at end of file
diff --git a/index.js b/index.js
index c93bc33..e105f32 100644
--- a/index.js
+++ b/index.js
@@ -3,34 +3,49 @@ const args = process.argv.slice(2).join(' ');
 
 const pongoOSDevice = usb.findByIds(0x05ac, 0x4141);
 
-function checkRoot() {
-    return process.getuid && process.getuid() === 0
+function log(message, type) {
+  switch (type) {
+    case 'success':
+      console.log("[+] " + message)
+      break;
+    case 'error':
+      console.error("[-] " + message)
+      break;
+  }
 }
 
+function checkRoot() {
+  return process.getuid && process.getuid() === 0
+};
+
 if (!checkRoot()) {
-    console.error("Please run as root!")
-    process.exit()
-}
+  log("Please run ass root", "error");
+  process.exit(1);
+};
 
 if (!pongoOSDevice) {
-    console.error('pongoOS Device not found !');
-    process.exit();
-}
+  log("No device with pongoOS Booted has been found.", "error");
+  process.exit(1);
+};
 if (!args) {
-    console.error('No command provided.');
-    process.exit();
-}
+  log("No argumment has been provided.", "error");
+  process.exit(1);
+};
 try {
-	pongoOSDevice.open(true)
-} catch(e) {
-	console.error('An error as occured while i was opening the device.' + e)
-	process.exit()
-}
+  pongoOSDevice.open(true)
+} catch (e) {
+  log("An error as occured while i was opening the device.", "error");
+  process.exit(1)
+};
+if (args.length > 510) {
+  log("You reached the length of the argumment.", "error");
+  process.exit(1)
+};
 pongoOSDevice.controlTransfer(0x21, 3, 0, 0, new Buffer.from(`${args} \n`), (error) => {
-    if (error) {
-        console.error("An error as occured while sending cmd error :" + error);
-        process.exit();
-    }
-    console.log('Command sent: ' + args);
-    process.exit();
+  if (error) {
+    log(`An error as occured while sending cmd error: ${error}`, "error")
+    process.exit(1);
+  }
+  log(`Success, command "${args}" has been sent.`, "success")
+  process.exit(0);
 })
\ No newline at end of file
diff --git a/package-lock.json b/package-lock.json
index 1ad8083..fc207c7 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,9 +1,13 @@
 {
-  "name": "cum",
+  "name": "pongoos-cmd-sender",
+  "version": "1.0.0",
   "lockfileVersion": 2,
   "requires": true,
   "packages": {
     "": {
+      "name": "pongoos-cmd-sender",
+      "version": "1.0.0",
+      "license": "unlicensed",
       "dependencies": {
         "usb": "^1.7.0"
       }
diff --git a/package.json b/package.json
index 53bda87..5b960dc 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "pongoos-cmd-sender",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "Send command to a device connected and running pongOS.",
   "main": "index.js",
   "repository": {