mirror of
https://github.com/stjet/bns.git
synced 2025-12-29 11:19:25 +00:00
workinggit status
This commit is contained in:
stjet
221
resolver.ts
Normal file
221
resolver.ts
Normal file
@@ -0,0 +1,221 @@
|
||||
import type { RPC } from "banani";
|
||||
import { get_address_from_public_key, get_public_key_from_address, whole_to_raw } from "banani";
|
||||
|
||||
import type { Address, AccountHistoryRawRPC } from "banani";
|
||||
|
||||
import type { Domain, DomainTransfer } from "./types";
|
||||
import { decode_domain_name, encode_domain_name, LOG } from "./util";
|
||||
|
||||
class Account {
|
||||
readonly rpc: RPC;
|
||||
readonly address: Address;
|
||||
|
||||
constructor(rpc: RPC, address: Address) {
|
||||
this.rpc = rpc;
|
||||
this.address = address;
|
||||
}
|
||||
|
||||
//errors if unopened
|
||||
async get_open_and_frontier(): Promise<[string, string]> {
|
||||
const account_info = await this.rpc.get_account_info(this.address, true);
|
||||
return [account_info.open_block, account_info.confirmation_height_frontier ?? account_info.frontier];
|
||||
}
|
||||
|
||||
//cannot use raw because kalium does not support it...?
|
||||
async get_history_from_open(head: string, count: number): Promise<AccountHistoryRawRPC> {
|
||||
return await this.rpc.get_account_history(this.address, count, true, head, undefined, true) as AccountHistoryRawRPC;
|
||||
}
|
||||
}
|
||||
|
||||
export class TLDAccount extends Account {
|
||||
all_issued: Domain[];
|
||||
|
||||
constructor(rpc: RPC, address: Address) {
|
||||
super(rpc, address);
|
||||
this.all_issued = [];
|
||||
}
|
||||
|
||||
async get_specific(name: string): Promise<Domain | undefined> {
|
||||
const [open_hash, frontier_hash] = await this.get_open_and_frontier();
|
||||
let head_hash = open_hash;
|
||||
while (true) {
|
||||
const { history } = await this.get_history_from_open(head_hash, 100);
|
||||
for (const block of history) {
|
||||
const amount = BigInt(block.amount ?? 0); //no amount if is change rep only
|
||||
const min = whole_to_raw("0.0012070301");
|
||||
const max = whole_to_raw("0.00120703011");
|
||||
if (amount >= min && amount <= max) {
|
||||
const found_name = decode_domain_name(get_public_key_from_address(block.representative));
|
||||
//if already in issued, this one is invalid
|
||||
if (name === found_name) {
|
||||
return {
|
||||
tld: this.address,
|
||||
name,
|
||||
history: [
|
||||
{
|
||||
type: "transfer",
|
||||
block: block,
|
||||
to: get_address_from_public_key(block.link),
|
||||
},
|
||||
],
|
||||
};
|
||||
}
|
||||
}
|
||||
if (block.hash === frontier_hash) {
|
||||
return;
|
||||
}
|
||||
}
|
||||
head_hash = history[history.length - 1].hash;
|
||||
}
|
||||
}
|
||||
|
||||
async get_all_issued(): Promise<Domain[]> {
|
||||
const [open_hash, frontier_hash] = await this.get_open_and_frontier();
|
||||
let head_hash = open_hash;
|
||||
let issued: Record<string, Domain> = {};
|
||||
while (true) {
|
||||
const { history } = await this.get_history_from_open(head_hash, 100);
|
||||
for (const block of history) {
|
||||
const amount = BigInt(block.amount ?? 0); //no amount if change rep only
|
||||
const min = whole_to_raw("0.0012070301");
|
||||
const max = whole_to_raw("0.00120703011");
|
||||
if (block.subtype === "send" && amount >= min && amount <= max) {
|
||||
const name = decode_domain_name(get_public_key_from_address(block.representative));
|
||||
//if already in issued, this one is invalid
|
||||
if (!issued[name]) {
|
||||
issued[name] = {
|
||||
tld: this.address,
|
||||
name,
|
||||
history: [
|
||||
{
|
||||
type: "transfer",
|
||||
block,
|
||||
to: get_address_from_public_key(block.link),
|
||||
},
|
||||
],
|
||||
};
|
||||
} else if (LOG) {
|
||||
console.log(`"${name}" already issued but TLD tried to issue again. Invalid.`);
|
||||
}
|
||||
if (block.hash === frontier_hash) {
|
||||
this.all_issued = Object.values(issued);
|
||||
return this.all_issued;
|
||||
}
|
||||
}
|
||||
}
|
||||
head_hash = history[history.length - 1].hash;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export class DomainAccount extends Account {
|
||||
domain?: Domain;
|
||||
|
||||
constructor(rpc: RPC, address: Address, domain?: Domain) {
|
||||
super(rpc, address);
|
||||
this.domain = domain;
|
||||
}
|
||||
|
||||
async crawl(): Promise<Domain> {
|
||||
let open_hash, frontier_hash;
|
||||
try {
|
||||
[open_hash, frontier_hash] = await this.get_open_and_frontier();
|
||||
} catch {
|
||||
if (LOG) {
|
||||
console.log("Not yet received");
|
||||
}
|
||||
return this.domain;
|
||||
}
|
||||
let head_hash = open_hash;
|
||||
while (true) {
|
||||
const { history } = await this.get_history_from_open(head_hash, 1000) as AccountHistoryRawRPC;
|
||||
for (const block of history) {
|
||||
const amount = BigInt(block.amount ?? 0); //amount is 0 if change rep only, apparently
|
||||
const min = whole_to_raw("0.0012070301");
|
||||
const max = whole_to_raw("0.00120703011");
|
||||
if (block.height === "1") {
|
||||
//domain burned due to not being received as the opening block
|
||||
//? is in case fake domain
|
||||
if (block.link !== this.domain.history[this.domain.history.length - 1]?.block.hash) {
|
||||
if (LOG) console.log("Domain burned");
|
||||
this.domain.burned = true;
|
||||
//do not bother to add to history
|
||||
return this.domain;
|
||||
}
|
||||
this.domain.history.push({
|
||||
type: "receive",
|
||||
block,
|
||||
});
|
||||
} else if (block.subtype === "send" && amount >= min && amount <= max) {
|
||||
const name = decode_domain_name(get_public_key_from_address(block.representative));
|
||||
if (this.domain.name === name) {
|
||||
//domain is transferred away, this domain account no longer owns it
|
||||
//the old resolved address and metadata hash are no longer in effect
|
||||
this.domain.resolved_address = undefined;
|
||||
this.domain.metadata_hash = undefined;
|
||||
this.domain.history.push({
|
||||
type: "transfer",
|
||||
block,
|
||||
to: get_address_from_public_key(block.link),
|
||||
});
|
||||
return this.domain;
|
||||
}
|
||||
} else if (block.subtype === "change") {
|
||||
this.domain.metadata_hash = block.representative;
|
||||
this.domain.history.push({
|
||||
type: "metadata",
|
||||
block,
|
||||
metadata_hash: this.domain.metadata_hash,
|
||||
});
|
||||
//
|
||||
} else if (block.subtype === "send" && amount === 4224n) {
|
||||
this.domain.resolved_address = get_address_from_public_key(block.link);
|
||||
this.domain.history.push({
|
||||
type: "resolver",
|
||||
block,
|
||||
resolved_address: this.domain.resolved_address,
|
||||
});
|
||||
}
|
||||
if (block.hash === frontier_hash) return this.domain;
|
||||
}
|
||||
head_hash = history[history.length - 1].hash;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export class Resolver {
|
||||
readonly rpc: RPC;
|
||||
tld_mapping: Record<string, Address>;
|
||||
|
||||
constructor(rpc: RPC, tld_mapping: Record<string, Address>) {
|
||||
this.rpc = rpc;
|
||||
this.tld_mapping = tld_mapping;
|
||||
}
|
||||
|
||||
async resolve(domain_name: string, tld: string): Promise<Domain | undefined> {
|
||||
if (!this.tld_mapping[tld]) throw new Error("No TLD Account found for that TLD");
|
||||
const tld_account = new TLDAccount(this.rpc, this.tld_mapping[tld]);
|
||||
let domain = await tld_account.get_specific(domain_name);
|
||||
if (!domain) return domain;
|
||||
while (true) {
|
||||
const current_domain_account = (domain.history[domain.history.length - 1] as DomainTransfer).to;
|
||||
const domain_account = new DomainAccount(this.rpc, current_domain_account, domain);
|
||||
const old_l = domain.history.length;
|
||||
domain = await domain_account.crawl();
|
||||
if (domain.history[domain.history.length - 1].type !== "transfer" || domain.burned || old_l === domain.history.length) break; //if length unchanged, means transfer unreceived
|
||||
}
|
||||
return domain;
|
||||
}
|
||||
|
||||
//see what domain a domain account (currently) has
|
||||
async resolve_backwards_ish(domain_account_address: Address, tld: string): Promise<Domain | undefined> {
|
||||
const open_hash = (await this.rpc.get_account_info(domain_account_address, true)).open_block;
|
||||
const transfer_hash = (await this.rpc.get_block_info(open_hash)).contents.link;
|
||||
const transfer_block = await this.rpc.get_block_info(transfer_hash);
|
||||
const domain_name = decode_domain_name(get_public_key_from_address(transfer_block.contents.representative));
|
||||
let domain = await this.resolve(domain_name, tld);
|
||||
const last_transfer = domain.history.reverse().find((b): b is DomainTransfer => b.type === "transfer");
|
||||
if (last_transfer.to === domain_account_address) return domain;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user