From 48ab96dda936b3f1b7055dfd51f3ca25c0b5c01f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A2u=20Cao?= <raucao@kosmos.org>
Date: Wed, 19 Jun 2024 20:57:22 +0200
Subject: [PATCH] Support "_" placeholder username for domain's own NIP-05

---
 app/controllers/well_known_controller.rb | 29 ++++++++++++++++--------
 spec/requests/well_known_spec.rb         | 27 ++++++++++++++++++----
 2 files changed, 43 insertions(+), 13 deletions(-)

diff --git a/app/controllers/well_known_controller.rb b/app/controllers/well_known_controller.rb
index 8e4ac07..bc342db 100644
--- a/app/controllers/well_known_controller.rb
+++ b/app/controllers/well_known_controller.rb
@@ -1,18 +1,23 @@
 class WellKnownController < ApplicationController
+  before_action :require_nostr_enabled, only: [ :nostr ]
+
   def nostr
     http_status :unprocessable_entity and return if params[:name].blank?
     domain = request.headers["X-Forwarded-Host"].presence || Setting.primary_domain
-    @user = User.where(cn: params[:name], ou: domain).first
-    http_status :not_found and return if @user.nil? || @user.nostr_pubkey.blank?
+    relay_url = Setting.nostr_relay_url
 
-    res = {
-      names: { @user.cn => @user.nostr_pubkey }
-    }
+    if params[:name] == "_"
+      # pubkey for the primary domain without a username (e.g. kosmos.org)
+      res = { names: { "_": Setting.nostr_public_key } }
+    else
+      @user = User.where(cn: params[:name], ou: domain).first
+      http_status :not_found and return if @user.nil? || @user.nostr_pubkey.blank?
 
-    if Setting.nostr_relay_url
-      res[:relays] = {
-        @user.nostr_pubkey => [ Setting.nostr_relay_url ]
-      }
+      res = { names: { @user.cn => @user.nostr_pubkey } }
+    end
+
+    if relay_url
+      res[:relays] = { @user.nostr_pubkey => [ relay_url ] }
     end
 
     respond_to do |format|
@@ -21,4 +26,10 @@ class WellKnownController < ApplicationController
       end
     end
   end
+
+  private
+
+    def require_nostr_enabled
+      http_status :not_found unless Setting.nostr_enabled?
+    end
 end
diff --git a/spec/requests/well_known_spec.rb b/spec/requests/well_known_spec.rb
index eba66e8..0300edf 100644
--- a/spec/requests/well_known_spec.rb
+++ b/spec/requests/well_known_spec.rb
@@ -2,21 +2,21 @@ require 'rails_helper'
 
 RSpec.describe "Well-known URLs", type: :request do
   describe "GET /nostr" do
-    context "without username param" do
+    describe "without username param" do
       it "returns a 422 status" do
         get "/.well-known/nostr.json"
         expect(response).to have_http_status(:unprocessable_entity)
       end
     end
 
-    context "non-existent user" do
+    describe "non-existent user" do
       it "returns a 404 status" do
         get "/.well-known/nostr.json?name=bob"
         expect(response).to have_http_status(:not_found)
       end
     end
 
-    context "user does not have a nostr pubkey configured" do
+    describe "user does not have a nostr pubkey configured" do
       let(:user) { create :user, cn: 'spongebob', ou: 'kosmos.org' }
 
       before do
@@ -30,7 +30,7 @@ RSpec.describe "Well-known URLs", type: :request do
       end
     end
 
-    context "user with nostr pubkey" do
+    describe "user with nostr pubkey" do
       let(:user) { create :user, cn: 'bobdylan', ou: 'kosmos.org' }
       before do
         user.save!
@@ -67,5 +67,24 @@ RSpec.describe "Well-known URLs", type: :request do
         end
       end
     end
+
+    describe "placeholder username for domain's own pubkey" do
+      it "returns the configured nostr pubkey" do
+        get "/.well-known/nostr.json?name=_"
+        res = JSON.parse(response.body)
+        expect(res["names"]["_"]).to eq(Setting.nostr_public_key)
+      end
+
+      context "nostr service integration not enabled" do
+        before do
+          Setting.nostr_enabled = false
+        end
+
+        it "returns a 404 status" do
+          get "/.well-known/nostr.json?name=_"
+          expect(response).to have_http_status(:not_found)
+        end
+      end
+    end
   end
 end