From 18a93fe0792bdc14570faf6b50247aca6b4458f7 Mon Sep 17 00:00:00 2001
From: Joakim Hulthe <joakim@hulthe.net>
Date: Thu, 28 Aug 2025 13:59:00 +0200
Subject: [PATCH] nu: Add --server argument to dns command

---
 tree/.local/nu/mullvad.nu | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tree/.local/nu/mullvad.nu b/tree/.local/nu/mullvad.nu
index 37a1f08..03cf82f 100644
--- a/tree/.local/nu/mullvad.nu
+++ b/tree/.local/nu/mullvad.nu
@@ -47,9 +47,15 @@ def "m daemon is-running" [
 }
 
 def dns [
+    --server (-s): string,
     name: string,
 ] {
-    nslookup $name |
+    # HACK: turn the optional argument into a list. not sure how to pass it to nslookup otherwise.
+    let server = ([$server] | compact)
+
+    nslookup $name ...$server |
+      lines |
+      skip until { |line| $line =~ answer } |
       parse -r "Address:\\s(?<ipv4>\\d+\\.\\d+\\.\\d+\\.\\d+)|Address:\\s(?<ipv6>[\\d:a-z]+)" |
       each { |row|
         if $row.ipv4 != "" {