feat: Add mDNS support for network discovery and dynamic hostname updates

.github/workflows/unittest.yml

@@ -18,7 +18,7 @@ jobs:
         with:
           esp_idf_version: v5.5
           target: esp32s3
-          command: GITHUB_ACTIONS="true" idf.py build
+          command: idf.py add-dependency "espressif/mdns^1.8.0" ; GITHUB_ACTIONS="true" idf.py build
           path: 'test-ci'
 
       - name: Run tests

components/connect/CMakeLists.txt

@@ -13,4 +13,5 @@ REQUIRES
     "esp_wifi"
     "esp_event"
     "stratum"
+    "espressif__mdns"
 )

components/connect/connect.c

@@ -1,6 +1,7 @@
 #include <string.h>
 #include "esp_event.h"
 #include "esp_log.h"
+#include "mdns.h"
 #include "esp_system.h"
 #include "esp_wifi.h"
 #include "freertos/FreeRTOS.h"
@@ -222,6 +223,52 @@ static void event_handler(void * arg, esp_event_base_t event_base, int32_t event
         ESP_LOGI(TAG, "Connected to SSID: %s", GLOBAL_STATE->SYSTEM_MODULE.ssid);
 
         wifi_softap_off();
+
+        /* Initialize mDNS after connection */
+        char * hostname = nvs_config_get_string(NVS_CONFIG_HOSTNAME, CONFIG_LWIP_LOCAL_HOSTNAME);
+        esp_err_t err = mdns_init();
+        if (err != ESP_OK) {
+            ESP_LOGW(TAG, "mDNS/Avahi initialization failed: %s", esp_err_to_name(err));
+            ESP_LOGW(TAG, "Device will not be discoverable via mDNS/Bonjour/Avahi");
+        } else {
+            ESP_LOGI(TAG, "mDNS/Avahi initialized successfully - device discoverable on network");
+
+            /* Set mDNS hostname */
+            err = mdns_hostname_set(hostname);
+            if (err != ESP_OK) {
+                ESP_LOGW(TAG, "mDNS hostname setup failed: %s", esp_err_to_name(err));
+                ESP_LOGW(TAG, "Device hostname not set for mDNS discovery");
+            } else {
+                ESP_LOGI(TAG, "mDNS hostname set to: %s.local", hostname);
+                ESP_LOGI(TAG, "Access device at: http://%s.local", hostname);
+            }
+
+            /* Set mDNS instance name */
+            uint8_t mac[6];
+            esp_wifi_get_mac(WIFI_IF_STA, mac);
+            char mac_suffix[6];
+            snprintf(mac_suffix, sizeof(mac_suffix), "%02X%02X", mac[4], mac[5]);
+
+            char instance_name[64];
+            snprintf(instance_name, sizeof(instance_name), "Bitaxe %s %s (%s)",
+                     GLOBAL_STATE->DEVICE_CONFIG.family.name,
+                     GLOBAL_STATE->DEVICE_CONFIG.board_version,
+                     mac_suffix);
+
+            /* Add HTTP service */
+            err = mdns_service_add(instance_name, "_http", "_tcp", 80, NULL, 0);
+            if (err != ESP_OK) {
+                ESP_LOGW(TAG, "mDNS HTTP service registration failed: %s", esp_err_to_name(err));
+                ESP_LOGW(TAG, "HTTP service not advertised via mDNS");
+            } else {
+                ESP_LOGI(TAG, "mDNS HTTP service registered: _http._tcp port 80");
+                ESP_LOGI(TAG, "Discover with: avahi-browse _http._tcp");
+                ESP_LOGI(TAG, "mDNS instance: %s", instance_name);
+            }
+
+            ESP_LOGI(TAG, "mDNS/Avahi setup complete - device ready for network discovery");
+        }
+        free(hostname);
     }
 }
 
@@ -389,6 +436,7 @@ void wifi_init(void * pvParameters)
             ESP_LOGI(TAG, "ESP_WIFI setting hostname to: %s", hostname);
         }
 
+
         free(hostname);
 
         ESP_LOGI(TAG, "wifi_init_sta finished.");
@@ -466,6 +514,22 @@ static const wifi_reason_desc_t wifi_reasons[] = {
     {0,                                               NULL},
 };
 
+esp_err_t update_mdns_hostname(const char *new_hostname) {
+    if (new_hostname == NULL || strlen(new_hostname) == 0) {
+        ESP_LOGW(TAG, "Invalid hostname provided for mDNS update");
+        return ESP_ERR_INVALID_ARG;
+    }
+
+    esp_err_t err = mdns_hostname_set(new_hostname);
+    if (err != ESP_OK) {
+        ESP_LOGW(TAG, "Failed to update mDNS hostname to: %s, error: %s", new_hostname, esp_err_to_name(err));
+        return err;
+    }
+
+    ESP_LOGI(TAG, "mDNS hostname updated to: %s", new_hostname);
+    return ESP_OK;
+}
+
 static const char *get_wifi_reason_string(int reason) {
     for (int i = 0; wifi_reasons[i].reason != 0; i++) {
         if (wifi_reasons[i].reason == reason) {

components/connect/include/connect.h

@@ -18,5 +18,6 @@ void toggle_wifi_softap(void);
 void wifi_init(void * GLOBAL_STATE);
 esp_err_t wifi_scan(wifi_ap_record_simple_t *ap_records, uint16_t *ap_count);
 esp_err_t get_wifi_current_rssi(int8_t *rssi);
+esp_err_t update_mdns_hostname(const char *new_hostname);
 
 #endif /* CONNECT_H_ */

main/http_server/axe-os/src/app/components/swarm/swarm.component.html

@@ -28,8 +28,8 @@
         <tr>
             <th *ngFor="let field of [
                 {
-                    name: 'IP',
-                    label: 'IP'
+                    name: 'address',
+                    label: 'Address'
                 },
                 {
                     name: 'hostname',
@@ -85,11 +85,11 @@
             <tr>
                 <td>
                     <a
-                       [ngClass]="'text-'+axe.swarmColor+'-500'"
-                       [href]="'http://'+axe.IP"
-                       target="_blank"
-                       [pTooltip]="axe.deviceModel || 'Other'"
-                       tooltipPosition="top">{{axe.IP}}</a>
+                        [ngClass]="'text-'+axe.swarmColor+'-500'"
+                        [href]="'http://'+axe.address"
+                        target="_blank"
+                        [pTooltip]="axe.deviceModel || 'Other'"
+                        tooltipPosition="top">{{axe.address}}</a>
                 </td>
                 <td>{{axe.hostname}}</td>
                 <td>{{axe.hashRate * 1000000000 | hashSuffix}}</td>
@@ -155,10 +155,10 @@
 <div class="card p-3 mt-4">
     <form [formGroup]="form">
         <div class="field grid p-fluid mb-0">
-            <label htmlFor="ip" class="col-12 mb-2 md:col-4 md:mb-0">Manual Addition</label>
+            <label htmlFor="manualAddAddress" class="col-12 mb-2 md:col-4 md:mb-0">Manual Addition</label>
             <div class="col-12 md:col-8">
                 <p-inputGroup>
-                    <input pInputText id="manualAddIp" formControlName="manualAddIp" type="text" />
+                    <input pInputText id="manualAddAddress" formControlName="manualAddAddress" type="text" placeholder="192.168.1.100 or bitaxe.local" />
                     <button pButton  [disabled]="form.invalid" (click)="add()">Add</button>
                 </p-inputGroup>
 
@@ -187,6 +187,6 @@
     </div>
 </div>
 
-<app-modal [headline]="selectedAxeOs?.IP">
-    <app-edit *ngIf="selectedAxeOs" [uri]="'http://' + selectedAxeOs.IP"></app-edit>
+<app-modal [headline]="selectedAxeOs?.address">
+    <app-edit *ngIf="selectedAxeOs" [uri]="'http://' + selectedAxeOs.address"></app-edit>
 </app-modal>