Feature: windows support (#2)

* feat: support windows password fetching

* feat: highlight current wifi ssid on windows. refactor windows code

.github/workflows/jsr-publish.yml

@@ -22,4 +22,5 @@ jobs:
         run: |
           bun install
           bun run build
+          bunx kksh verify --publish
           bunx jsr publish --allow-slow-types

README.md

@@ -11,8 +11,9 @@
   ```
 - Windows
   ```powershell
-  $_, $ssid = ((netsh wlan show interface | findstr "Profile" | findstr /v "mode") -split ":",2).trim();
+  (((netsh wlan show profiles | findstr "Profile") -split ":",2) | findstr /v "Profile").trim(); # find all wifi ssid
-  $_, $pass = ((netsh wlan show profile name=$ssid key=clear | findstr Key) -split ":").trim();
+  $_, $ssid = ((netsh wlan show interface | findstr "Profile" | findstr /v "mode") -split ":",2).trim(); # find current wifi ssid
+  $_, $pass = ((netsh wlan show profile name=$ssid key=clear | findstr Key) -split ":").trim(); # find current wifi password
   ```
 - Linux
   ```bash

jsr.json

@@ -1,6 +1,6 @@
 {
   "name": "@kunkun/kunkun-ext-wifi-password",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "license": "MIT",
   "exports": "./mod.ts",
   "publish": {

package.json

@@ -2,7 +2,7 @@
   "$schema": "https://schema.kunkun.sh",
   "name": "kunkun-ext-wifi-password",
   "license": "MIT",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "type": "module",
   "kunkun": {
     "name": "Wifi Password",
@@ -35,6 +35,38 @@
                 "-w"
               ]
             }
+          },
+          {
+            "cmd": {
+              "program": "netsh",
+              "args": [
+                "wlan",
+                "show",
+                "profiles"
+              ]
+            }
+          },
+          {
+            "cmd": {
+              "program": "netsh",
+              "args": [
+                "wlan",
+                "show",
+                "interfaces"
+              ]
+            }
+          },
+          {
+            "cmd": {
+              "program": "netsh",
+              "args": [
+                "wlan",
+                "show",
+                "profile",
+                "name=.+",
+                "key=clear"
+              ]
+            }
           }
         ]
       }

src/index.ts

@@ -13,12 +13,55 @@ import {
   shell,
   toast,
   ui,
-	WorkerExtension
+  WorkerExtension,
-} from "@kksh/api/ui/worker"
+} from "@kksh/api/ui/worker";
-import qrcode from "qrcode"
+import qrcode from "qrcode";
 
-class ExtensionTemplate extends WorkerExtension {
+async function windowsGetCurrentWifiSsid() {
-	networks: string[] = []
+  const cmd = shell.createCommand("netsh", ["wlan", "show", "interfaces"]);
+  const result = await cmd.execute();
+  if (result.code !== 0) {
+    return undefined;
+  }
+  const lines = result.stdout.split("\n");
+  const ssid = lines.find((line) => line.includes("SSID"));
+  return ssid ? ssid.split(":")[1].trim() : undefined;
+}
+
+async function windowsGetWifiPassword(ssid: string) {
+  const cmd = shell.createCommand("netsh", [
+    "wlan",
+    "show",
+    "profile",
+    "name=" + ssid,
+    "key=clear",
+  ]);
+  const result = await cmd.execute();
+  if (result.code !== 0) {
+    return undefined;
+  }
+  return result.stdout
+    .split("\n")
+    .find((line) => line.includes("Key Content"))
+    ?.split(":")[1]
+    .trim();
+}
+
+async function windowsGetWifiSsids() {
+  const cmd = shell.createCommand("netsh", ["wlan", "show", "profiles"]);
+  const result = await cmd.execute();
+  if (result.code !== 0) {
+    return undefined;
+  }
+  return result.stdout
+    .split("\n")
+    .filter((line) => line.includes("User Profile"))
+    .map((line) => line.split(":")[1].trim());
+}
+
+class ListWifiPasswords extends WorkerExtension {
+  networks: string[] = [];
+  currentWifiPassword: string | undefined;
 
   get listItems() {
     return this.networks.map(
@@ -28,70 +71,92 @@ class ExtensionTemplate extends WorkerExtension {
           value: x,
           icon: new Icon({
             type: IconEnum.Iconify,
-						value: "mdi:wifi"
+            value: "mdi:wifi",
+            hexColor: this.currentWifiPassword === x ? "#ff0" : undefined,
+          }),
         })
-				})
+    );
-		)
   }
 
   async load() {
-		ui.setSearchBarPlaceholder("Search for wifi and press enter to get password")
+    ui.setSearchBarPlaceholder(
-		const platform = await os.platform()
+      "Search for wifi and press enter to get password"
+    );
+    const platform = await os.platform();
     if (platform === "macos") {
-			const cmd = shell.createCommand("networksetup", ["-listpreferredwirelessnetworks", "en0"])
+      const cmd = shell.createCommand("networksetup", [
-			const result = await cmd.execute()
+        "-listpreferredwirelessnetworks",
+        "en0",
+      ]);
+      const result = await cmd.execute();
       if (result.code !== 0) {
-				toast.error("Failed to get wifi password")
+        toast.error("Failed to get wifi password");
-				return
+        return;
       }
       this.networks = result.stdout
         .trim()
         .split("\n")
         .slice(1)
-				.map((x) => x.trim())
+        .map((x) => x.trim());
+    } else if (platform === "windows") {
+      this.currentWifiPassword = await windowsGetCurrentWifiSsid();
+      this.networks = (await windowsGetWifiSsids()) ?? [];
+    } else if (platform === "linux") {
     }
     return ui.render(
       new List.List({
-				items: this.listItems
+        items: this.listItems,
       })
-		)
+    );
   }
 
   async onListItemSelected(ssid: string): Promise<void> {
+    const platform = await os.platform();
     ui.render(
       new List.List({
         inherits: ["items"],
-				detail: undefined
+        detail: undefined,
       })
-		)
+    );
+    let wifiPassword: string | undefined;
+    if (platform === "macos") {
       const cmd = shell.createCommand("security", [
         "find-generic-password",
         "-D",
         `AirPort network password`,
         "-a",
         ssid,
-			"-w"
+        "-w",
-		])
+      ]);
-		const result = await cmd.execute()
+      const result = await cmd.execute();
       if (result.code !== 0) {
-			console.log(result.stderr)
+        console.log(result.stderr);
-			toast.error("Failed to get wifi password")
+        toast.error("Failed to get wifi password");
-			return
+        return;
       }
-		const wifiPassword = result.stdout.trim()
+      wifiPassword = result.stdout.trim();
-		toast.success(`Wifi password: ${wifiPassword}, written to clipboard`)
+    } else if (platform === "windows") {
-		clipboard.writeText(wifiPassword)
+      wifiPassword = await windowsGetWifiPassword(ssid);
-		const wifiConnectUrl = `WIFI:S:${ssid};T:WPA;P:${wifiPassword};;`
+    }
-		let qrcodeSvg: string | undefined
+
+    if (!wifiPassword) {
+      toast.error("Failed to get wifi password");
+      return;
+    }
+
+    toast.success(`Wifi password: ${wifiPassword}, written to clipboard`);
+    clipboard.writeText(wifiPassword);
+    const wifiConnectUrl = `WIFI:S:${ssid};T:WPA;P:${wifiPassword};;`;
+    let qrcodeSvg: string | undefined;
     try {
-			qrcodeSvg = await qrcode.toString(wifiConnectUrl)
+      qrcodeSvg = await qrcode.toString(wifiConnectUrl);
     } catch (error) {
-			toast.error(`Failed to generate QR code: ${error}`)
+      toast.error(`Failed to generate QR code: ${error}`);
-			console.error(error)
+      console.error(error);
     }
     // <img src="${qrcodeImageBase64}" style="display: block; margin: 0 auto;" />
     if (!qrcodeSvg) {
-			return toast.error("Failed to generate QR code")
+      return toast.error("Failed to generate QR code");
     }
     const markdown = `
 <div style="display: flex; flex-direction: column; justify-content: center; align-items: center; padding: 2em;">
@@ -101,17 +166,17 @@ class ExtensionTemplate extends WorkerExtension {
 <p style="text-align: center;">Wifi Password: <strong>${wifiPassword}</strong></p>
 <p style="text-align: center;">Connect URL: <strong>${wifiConnectUrl}</strong></p>
 </div>
-`
+`;
     return ui.render(
       new List.List({
         inherits: ["items"],
         detail: new List.ItemDetail({
           width: 60,
-					children: [new Markdown(markdown)]
+          children: [new Markdown(markdown)],
+        }),
       })
-			})
+    );
-		)
   }
 }
 
-expose(new ExtensionTemplate())
+expose(new ListWifiPasswords());