Merge pull request #14 from munirahmed007/dnsapp/Issue#10

Dnsapp/issue#10
ivanstegic · Jan 23, 2021 · 019eb4d · 019eb4d
2 parents 256bd23 + 71aa8b8
commit 019eb4d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 7 deletions.
diff --git a/Menu Bar DNS/Menu Bar DNS/AppDelegate.swift b/Menu Bar DNS/Menu Bar DNS/AppDelegate.swift
@@ -77,6 +77,7 @@ class AppDelegate: NSObject, NSApplicationDelegate {
         let wifiAddresses = dnsAddresses.1
 
         rebuildMenuWithAddress(ethernetAddresses, wifiAddresses)
+        menu?.autoenablesItems = false
         statusItem?.menu = menu
         statusItem?.button?.performClick(self)
         statusItem?.menu = nil

diff --git a/Menu Bar DNS/Menu Bar DNS/DNSConfiguration.swift b/Menu Bar DNS/Menu Bar DNS/DNSConfiguration.swift
@@ -46,18 +46,23 @@ class DNSConfiguration: NSObject {
         let serviceDNSKey = "State:/Network/Service/\(serviceID)/DNS" as CFString
         let serviceSetupDNSKey = "Setup:/Network/Service/\(serviceID)/DNS" as CFString
         let dynmaicStore =  SCDynamicStoreCreate(kCFAllocatorSystemDefault, "DNSSETTING" as CFString, nil, nil)
+        var allDNSIPAddresses : Array<String> = []
 
         let dynamicPlist = SCDynamicStoreCopyValue(dynmaicStore, serviceDNSKey)
         let manualAddressPlist = SCDynamicStoreCopyValue(dynmaicStore, serviceSetupDNSKey)
 
         if let dnsValues = manualAddressPlist?[kSCPropNetDNSServerAddresses] as? [String] {
-            return dnsValues
+            allDNSIPAddresses += dnsValues
         }
 
-        if let manualDNS = dynamicPlist?[kSCPropNetDNSServerAddresses] as? [String] {
-            return manualDNS
+        if let dhcpValues = dynamicPlist?[kSCPropNetDNSServerAddresses] as? [String] {
+            let uniqueValues = Array(Set(dhcpValues))
+            for dhcpValue in uniqueValues {
+                let newvalue = dhcpValue.appending(" (via DHCP)")
+                allDNSIPAddresses.append(newvalue)
+            }
         }
-        return []
+        return allDNSIPAddresses
     }
 
     static func getAddresses()  -> (Array<String>, Array<String>) {

diff --git a/Menu Bar DNS/Menu Bar DNS/NSMenu+DNSItems.swift b/Menu Bar DNS/Menu Bar DNS/NSMenu+DNSItems.swift
@@ -24,9 +24,11 @@ extension NSMenu {
     func addItemFromList(list items: [String],  action selector: Selector?, keyEquivalent charCode: String, prefix string:String) {
 
         for itemTitle in items {
-            self.addItem(NSMenuItem.init(title: "\(string)\(itemTitle)",
-                                         action: selector,
-                                         keyEquivalent: charCode))
+            let menuItem = NSMenuItem.init(title: "\(string)\(itemTitle)",
+                                           action: selector,
+                                           keyEquivalent: charCode)
+            menuItem.state = .on
+            self.addItem(menuItem)
         }
     }
 }