nip42 support

This commit is contained in:
Liran Cohen 2023-01-22 20:45:53 -05:00 committed by Kieran
parent 1f62afacb1
commit 25f8dd5ef7
Signed by: Kieran
GPG Key ID: DE71CEB3925BE941
3 changed files with 50 additions and 39 deletions

View File

@ -90,22 +90,8 @@ export default function useEventPublisher() {
return Promise.reject(); return Promise.reject();
} }
useEffect(() =>{
const nip42AuthEvent = async (event: NIP42AuthChallenge) => {
if(event.challenge && event.relay) {
const signedEvent = await nip42Auth(event.challenge, event.relay);
const response = new NIP42AuthResponse(event.challenge, signedEvent);
window.dispatchEvent(response);
}
}
window.addEventListener("nip42auth", nip42AuthEvent)
return () => {
window.removeEventListener("nip42auth", nip42AuthEvent)
}
}, [])
return { return {
nip42Auth: nip42Auth,
broadcast: (ev: NEvent | undefined) => { broadcast: (ev: NEvent | undefined) => {
if (ev) { if (ev) {
console.debug("Sending event: ", ev); console.debug("Sending event: ", ev);

View File

@ -50,9 +50,9 @@ export default class Connection {
LastState: Readonly<StateSnapshot>; LastState: Readonly<StateSnapshot>;
IsClosed: boolean; IsClosed: boolean;
ReconnectTimer: ReturnType<typeof setTimeout> | null; ReconnectTimer: ReturnType<typeof setTimeout> | null;
EventsCallback: Map<u256, () => void>; EventsCallback: Map<u256, (msg?:any) => void>;
AwaitingAuth: Map<string, boolean>;
Authed: boolean; Authed: boolean;
AwaitingAuth: boolean;
constructor(addr: string, options: RelaySettings) { constructor(addr: string, options: RelaySettings) {
this.Id = uuid(); this.Id = uuid();
@ -78,8 +78,8 @@ export default class Connection {
this.IsClosed = false; this.IsClosed = false;
this.ReconnectTimer = null; this.ReconnectTimer = null;
this.EventsCallback = new Map(); this.EventsCallback = new Map();
this.AwaitingAuth = new Map();
this.Authed = false; this.Authed = false;
this.AwaitingAuth = false;
this.Connect(); this.Connect();
} }
@ -128,11 +128,10 @@ export default class Connection {
this.ConnectTimeout = DefaultConnectTimeout; this.ConnectTimeout = DefaultConnectTimeout;
console.log(`[${this.Address}] Open!`); console.log(`[${this.Address}] Open!`);
setTimeout(() => { setTimeout(() => {
if(this.AwaitingAuth) { if(this.Authed || this.AwaitingAuth.size === 0) {
return
}
this._InitSubscriptions(); this._InitSubscriptions();
}, 500) }
}, 150)
} }
OnClose(e: CloseEvent) { OnClose(e: CloseEvent) {
@ -156,7 +155,7 @@ export default class Connection {
let tag = msg[0]; let tag = msg[0];
switch (tag) { switch (tag) {
case "AUTH": { case "AUTH": {
this._OnAuth(msg[1]) this._OnAuthAsync(msg[1])
this.Stats.EventsReceived++; this.Stats.EventsReceived++;
this._UpdateState(); this._UpdateState();
break; break;
@ -178,7 +177,7 @@ export default class Connection {
if (this.EventsCallback.has(id)) { if (this.EventsCallback.has(id)) {
let cb = this.EventsCallback.get(id)!; let cb = this.EventsCallback.get(id)!;
this.EventsCallback.delete(id); this.EventsCallback.delete(id);
cb(); cb(msg);
} }
break; break;
} }
@ -333,6 +332,11 @@ export default class Connection {
} }
_SendSubscription(sub: Subscriptions) { _SendSubscription(sub: Subscriptions) {
if(!this.Authed && this.AwaitingAuth.size > 0) {
this.Pending.push(sub);
return;
}
let req = ["REQ", sub.Id, sub.ToObject()]; let req = ["REQ", sub.Id, sub.ToObject()];
if (sub.OrSubs.length > 0) { if (sub.OrSubs.length > 0) {
req = [ req = [
@ -367,25 +371,33 @@ export default class Connection {
} }
} }
async _OnAuth(challenge: string, timeout: number = 5000):Promise<void> { async _OnAuthAsync(challenge: string) {
const challengeEvent = new NIP42AuthChallenge(challenge, this.Address) const challengeEvent = new NIP42AuthChallenge(challenge, this.Address)
const authCleanup = () => {
this.AwaitingAuth.delete(challenge)
}
const authCallback = (e:NIP42AuthResponse):Promise<void> => { const authCallback = (e:NIP42AuthResponse):Promise<void> => {
return new Promise((resolve,reject) => { window.removeEventListener(`nip42response:${challenge}`, authCallback)
return new Promise((resolve,_) => {
if(!e.event) { if(!e.event) {
authCleanup();
return Promise.reject('no event'); return Promise.reject('no event');
} }
let t = setTimeout(() => { let t = setTimeout(() => {
window.removeEventListener(`nip42response:${challenge}`, authCallback); authCleanup();
this.AwaitingAuth = false; resolve();
reject('timeout'); }, 10_000);
}, timeout);
this.EventsCallback.set(e.event.Id, () => { this.EventsCallback.set(e.event.Id, (msg:any[]) => {
clearTimeout(t); clearTimeout(t);
this.AwaitingAuth = false; authCleanup();
window.removeEventListener(`nip42response:${challenge}`, authCallback) if(msg.length > 3 && msg[2] === true) {
this.Authed = true;
this._InitSubscriptions();
}
resolve(); resolve();
}); });
@ -396,7 +408,7 @@ export default class Connection {
}) })
} }
this.AwaitingAuth = true; this.AwaitingAuth.set(challenge, true)
window.addEventListener(`nip42response:${challenge}`, authCallback) window.addEventListener(`nip42response:${challenge}`, authCallback)
window.dispatchEvent(challengeEvent) window.dispatchEvent(challengeEvent)
} }

View File

@ -1,12 +1,12 @@
import "./Layout.css"; import "./Layout.css";
import { useEffect, useState } from "react" import { useEffect } from "react"
import { useDispatch, useSelector } from "react-redux"; import { useDispatch, useSelector } from "react-redux";
import { Outlet, useNavigate } from "react-router-dom"; import { Outlet, useNavigate } from "react-router-dom";
import { faBell, faMessage, faSearch } from "@fortawesome/free-solid-svg-icons"; import { faBell, faMessage, faSearch } from "@fortawesome/free-solid-svg-icons";
import { FontAwesomeIcon } from "@fortawesome/react-fontawesome"; import { FontAwesomeIcon } from "@fortawesome/react-fontawesome";
import { RootState } from "State/Store"; import { RootState } from "State/Store";
import { init, setPreferences, UserPreferences } from "State/Login"; import { init, UserPreferences } from "State/Login";
import { HexKey, RawEvent, TaggedRawEvent } from "Nostr"; import { HexKey, RawEvent, TaggedRawEvent } from "Nostr";
import { RelaySettings } from "Nostr/Connection"; import { RelaySettings } from "Nostr/Connection";
import { System } from "Nostr/System" import { System } from "Nostr/System"
@ -14,6 +14,8 @@ import ProfileImage from "Element/ProfileImage";
import useLoginFeed from "Feed/LoginFeed"; import useLoginFeed from "Feed/LoginFeed";
import { totalUnread } from "Pages/MessagesPage"; import { totalUnread } from "Pages/MessagesPage";
import { SearchRelays } from 'Const'; import { SearchRelays } from 'Const';
import useEventPublisher from "Feed/EventPublisher";
import { NIP42AuthChallenge, NIP42AuthResponse } from "Nostr/Auth";
export default function Layout() { export default function Layout() {
const dispatch = useDispatch(); const dispatch = useDispatch();
@ -25,13 +27,20 @@ export default function Layout() {
const readNotifications = useSelector<RootState, number>(s => s.login.readNotifications); const readNotifications = useSelector<RootState, number>(s => s.login.readNotifications);
const dms = useSelector<RootState, RawEvent[]>(s => s.login.dms); const dms = useSelector<RootState, RawEvent[]>(s => s.login.dms);
const prefs = useSelector<RootState, UserPreferences>(s => s.login.preferences); const prefs = useSelector<RootState, UserPreferences>(s => s.login.preferences);
const pub = useEventPublisher();
const [keyword, setKeyword] = useState<string>('');
useLoginFeed(); useLoginFeed();
useEffect(() => { useEffect(() => {
if (relays) { if (relays) {
const nip42AuthEvent = async (event: NIP42AuthChallenge) => {
if(event.challenge && event.relay) {
const signedEvent = await pub.nip42Auth(event.challenge, event.relay);
const response = new NIP42AuthResponse(event.challenge, signedEvent);
window.dispatchEvent(response);
}
}
window.addEventListener("nip42auth", nip42AuthEvent)
for (let [k, v] of Object.entries(relays)) { for (let [k, v] of Object.entries(relays)) {
System.ConnectToRelay(k, v); System.ConnectToRelay(k, v);
} }
@ -40,6 +49,10 @@ export default function Layout() {
System.DisconnectRelay(k); System.DisconnectRelay(k);
} }
} }
return () => {
window.removeEventListener("nip42auth", nip42AuthEvent)
}
} }
}, [relays]); }, [relays]);