1 files changed, 65 insertions, 0 deletions
diff --git a/main/mdns_service.c b/main/mdns_service.c
new file mode 100644
index 0000000..761ade5
--- /dev/null
+++ b/main/mdns_service.c
@@ -0,0 +1,65 @@
+#include "mdns_service.h"
+
+#include <stdio.h>
+#include <string.h>
+
+#include "esp_check.h"
+#include "esp_log.h"
+#include "mdns.h"
+
+#include "viewport_state.h"
+
+static const char *TAG       = "mdns";
+static const char *SERVICE   = "_scrypted-viewport";
+static const char *PROTO     = "_tcp";
+static const uint16_t PORT   = 80;
+
+static esp_err_t apply_records(void)
+{
+    viewport_state_lock();
+    viewport_state_t *st = viewport_state_get();
+
+    char hostname[80];
+    if (st->viewport_name[0]) {
+        snprintf(hostname, sizeof(hostname), "viewport-%s", st->viewport_name);
+    } else {
+        snprintf(hostname, sizeof(hostname), "viewport");
+    }
+
+    const char *resolution  = viewport_state_resolution_str();
+    const char *orientation = (st->orientation == VIEWPORT_ORIENTATION_PORTRAIT)
+                                  ? "portrait" : "landscape";
+    const char *name_value  = st->viewport_name[0] ? st->viewport_name : "";
+
+    viewport_state_unlock();
+
+    mdns_txt_item_t txt[] = {
+        { "version",     VIEWPORT_VERSION },
+        { "resolution",  resolution },
+        { "orientation", orientation },
+        { "name",        name_value },
+    };
+    const size_t txt_count = sizeof(txt) / sizeof(txt[0]);
+
+    ESP_RETURN_ON_ERROR(mdns_hostname_set(hostname), TAG, "hostname_set");
+    ESP_RETURN_ON_ERROR(mdns_service_txt_set(SERVICE, PROTO, txt, txt_count),
+                       TAG, "txt_set");
+
+    ESP_LOGI(TAG, "advertising %s.local on %s.%s.local:%u",
+             hostname, SERVICE, PROTO, PORT);
+    return ESP_OK;
+}
+
+esp_err_t mdns_service_start(void)
+{
+    ESP_RETURN_ON_ERROR(mdns_init(), TAG, "mdns_init");
+    ESP_RETURN_ON_ERROR(
+        mdns_service_add(NULL, SERVICE, PROTO, PORT, NULL, 0),
+        TAG, "service_add");
+    return apply_records();
+}
+
+esp_err_t mdns_service_refresh(void)
+{
+    return apply_records();
+}