xevion/rdap
1
0
Fork 0
mirror of https://github.com/Xevion/rdap.git synced 2025-12-18 06:12:55 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add await registry-informed getTypes method with smart entity tag validation

Browse Source 
fuck type coloring madness
This commit is contained in:
Xevion
2024-05-12 02:10:15 -05:00
parent 15666716a6
commit b3048086c3
4 changed files with 84 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/hooks/useLookup.tsx
View File

@@ -1,4 +1,4 @@
import { useEffect, useMemo, useRef, useState } from "react";
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
import { domainMatchPredicate, getBestURL, getType } from "@/rdap";
import type {
AutonomousNumber,
@@ -43,19 +43,19 @@ const useLookup = (warningHandler?: WarningHandler) => {
const [error, setError] = useState<string | null>(null);
const [target, setTarget] = useState<string>("");
const [uriType, setUriType] = useState<Maybe<TargetType>>(Maybe.nothing());
// Used by a callback on LookupInput to forcibly set the type of the lookup.
const [currentType, setTargetType] = useState<TargetType | null>(null);
// Used to allow repeatable lookups when weird errors happen.
const repeatableRef = useRef<string>("");
const uriType = useMemo<Maybe<TargetType>>(
function () {
if (currentType != null) return Maybe.just(currentType);
return getType(target).mapOr(Maybe.nothing(), (type) => Maybe.just(type));
},
[target, currentType]
);
useCallback(async () => {
if (currentType != null) return Maybe.just(currentType);
const uri: Maybe<TargetType> = (await getTypeEasy(target)).mapOr(Maybe.nothing(), (type) => Maybe.just(type));
setUriType(uri);
}, [target, currentType])
// Fetch & load a specific registry's data into memory.
async function loadBootstrap(type: RootRegistryType, force = false) {
@@ -81,6 +81,20 @@ const useLookup = (warningHandler?: WarningHandler) => {
};
}
async function getRegistry(type: RootRegistryType): Promise<Register> {
if (registryDataRef.current[type] == null) await loadBootstrap(type);
if (registryDataRef.current[type] == null)
throw new Error(
`Could not load bootstrap data for ${type} registry.`
);
return registryDataRef.current[type];
}
async function getTypeEasy(target: string): Promise<Result<TargetType, Error>> {
return getType(target, getRegistry);
}
function getRegistryURL(
type: RootRegistryType,
lookupTarget: string
@@ -232,7 +246,7 @@ const useLookup = (warningHandler?: WarningHandler) => {
new Error("A target must be given in order to execute a lookup.")
);
const targetType = getType(target);
const targetType = await getTypeEasy(target);
if (targetType.isErr) {
return Result.err(
@@ -365,7 +379,7 @@ const useLookup = (warningHandler?: WarningHandler) => {
}
}
return { error, setTarget, setTargetType, submit, currentType: uriType };
return { error, setTarget, setTargetType, submit, currentType: uriType, getType: getTypeEasy };
};
export default useLookup;