From 061390104f70574a847003e6a66282ccccbc28ee Mon Sep 17 00:00:00 2001
From: Huakun Shen <huakun.shen@huakunshen.com>
Date: Tue, 21 Jan 2025 22:12:42 -0500
Subject: [PATCH] feat: support ubuntu

---
 README.md    |  3 +++
 jsr.json     |  2 +-
 package.json | 12 +++++++++++-
 src/index.ts | 48 ++++++++++++++++++++++++++++++++++++++++++++++--
 4 files changed, 61 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index f2abf22..a499980 100644
--- a/README.md
+++ b/README.md
@@ -21,6 +21,9 @@ JSR Package `@kunkun/kunkun-ext-wifi-password`: https://jsr.io/@kunkun/kunkun-ex
   password=$(nmcli device wifi show-password | grep Password | cut -d : -f 2 | xargs)
   ```
 
+  Ubuntu currently only supports showing passwords for connected wifi, I didn't find a way to find password for all wifis.
+  There is command for listing all wifi ssids, but not passwords.
+
 ## Features
 
 - Reveal Wifi Password from current machine
diff --git a/jsr.json b/jsr.json
index 98059fc..9931e6c 100644
--- a/jsr.json
+++ b/jsr.json
@@ -1,6 +1,6 @@
 {
   "name": "@kunkun/kunkun-ext-wifi-password",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "license": "MIT",
   "exports": "./mod.ts",
   "publish": {
diff --git a/package.json b/package.json
index e7288fa..2a6fb93 100644
--- a/package.json
+++ b/package.json
@@ -2,7 +2,7 @@
   "$schema": "https://schema.kunkun.sh",
   "name": "kunkun-ext-wifi-password",
   "license": "MIT",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "type": "module",
   "kunkun": {
     "name": "Wifi Password",
@@ -67,6 +67,16 @@
                 "key=clear"
               ]
             }
+          },
+          {
+            "cmd": {
+              "program": "nmcli",
+              "args": [
+                "device",
+                "wifi",
+                "show-password"
+              ]
+            }
           }
         ]
       }
diff --git a/src/index.ts b/src/index.ts
index 0dbad23..d2195ce 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -59,9 +59,44 @@ async function windowsGetWifiSsids() {
     .map((line) => line.split(":")[1].trim());
 }
 
+async function getCurrentWifiInfo() {
+  // nmcli device wifi show-password
+  const cmd = shell.createCommand("nmcli", [
+    "device",
+    "wifi",
+    "show-password"
+  ])
+  const result = await cmd.execute();
+  if (result.code !== 0) {
+    if (result.stderr.includes("No Wi-Fi device found")) {
+      toast.warning('Not connected to wifi')
+    }
+    return undefined;
+  }
+  const lines = result.stdout.split("\n")
+  const ssidLines = lines.filter(line => line.includes("SSID"))
+  if (ssidLines.length === 0) {
+    toast.error("No wifi ssid found")
+    return undefined
+  }
+  const ssid = ssidLines[0].split(":")[1].trim()
+  const passwordLines = lines.filter(line => line.includes("Password:"))
+  if (passwordLines.length === 0) {
+    toast.error("No wifi password found")
+    return undefined
+  }
+  const password = passwordLines[0].split(":")[1].trim()
+  return {
+    ssid,
+    password
+  }
+  
+}
+
 class ListWifiPasswords extends WorkerExtension {
   networks: string[] = [];
   currentWifiPassword: string | undefined;
+  currentWifiSsid: string | undefined;
 
   get listItems() {
     return this.networks.map(
@@ -72,7 +107,7 @@ class ListWifiPasswords extends WorkerExtension {
           icon: new Icon({
             type: IconEnum.Iconify,
             value: "mdi:wifi",
-            hexColor: this.currentWifiPassword === x ? "#ff0" : undefined,
+            hexColor: this.currentWifiSsid === x ? "#ff0" : undefined,
           }),
         })
     );
@@ -99,9 +134,16 @@ class ListWifiPasswords extends WorkerExtension {
         .slice(1)
         .map((x) => x.trim());
     } else if (platform === "windows") {
-      this.currentWifiPassword = await windowsGetCurrentWifiSsid();
+      this.currentWifiSsid = await windowsGetCurrentWifiSsid();
       this.networks = (await windowsGetWifiSsids()) ?? [];
     } else if (platform === "linux") {
+      const ubuntuWifiInfo = await getCurrentWifiInfo()
+      console.log(ubuntuWifiInfo);
+      if (ubuntuWifiInfo) {
+        this.currentWifiPassword = ubuntuWifiInfo.password
+        this.currentWifiSsid = ubuntuWifiInfo.ssid
+        this.networks = [ubuntuWifiInfo.ssid]
+      }
     }
     return ui.render(
       new List.List({
@@ -137,6 +179,8 @@ class ListWifiPasswords extends WorkerExtension {
       wifiPassword = result.stdout.trim();
     } else if (platform === "windows") {
       wifiPassword = await windowsGetWifiPassword(ssid);
+    } else if (platform === "linux") {
+      wifiPassword = this.currentWifiPassword
     }
 
     if (!wifiPassword) {