From 14881c51f607a99ad0717d2020e3392c8dc91c5e Mon Sep 17 00:00:00 2001
From: Stephen Paul Weber <singpolyma@singpolyma.net>
Date: Sat, 6 Jul 2024 09:21:45 -0500
Subject: [PATCH] Fix direct targets with explicit backend set

---
 lib/trivial_backend_sgx_repo.rb | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/lib/trivial_backend_sgx_repo.rb b/lib/trivial_backend_sgx_repo.rb
index 6e5980b7324d320958df0e02e88f499a9d2c514a..7747aafe0af78a2874333256721a8d329d27dd2c 100644
--- a/lib/trivial_backend_sgx_repo.rb
+++ b/lib/trivial_backend_sgx_repo.rb
@@ -23,7 +23,7 @@ class TrivialBackendSgxRepo
 	end
 
 	def get(customer_id, tel: nil)
-		@redis.get("jmp_customer_backend_sgx-#{customer_id}").then do |sgx_jid|
+		use_jid(customer_id).then do |sgx_jid|
 			creds = sgx_jid ? CONFIG[:sgx_creds][sgx_jid.to_sym] : @creds
 			BackendSgx.new(
 				jid: sgx_jid ? Blather::JID.new(sgx_jid) : @jid, creds: creds,
@@ -37,6 +37,14 @@ class TrivialBackendSgxRepo
 
 protected
 
+	def use_jid(customer_id)
+		if @jid.node.nil?
+			@redis.get("jmp_customer_backend_sgx-#{customer_id}")
+		else
+			EMPromise.resolve(@jid)
+		end
+	end
+
 	def ibr_for(tel)
 		ibr = Blather::Stanza::Iq::IBR.new
 		ibr.registered = true