selfBrokerSchema

This commit is contained in:
Aleksandr Kraiz
2023-05-17 00:02:42 +04:00
parent 1382243a01
commit 0eec75fb80

View File

@@ -63,6 +63,12 @@ const baseOrderSchema = z.object({
const selfBrokers = exchanges.map((exchange) => `SELF_BROKER_${exchange}` as const);
type SelfBroker = typeof selfBrokers[number];
const isSelfBroker = (value: string): value is SelfBroker => selfBrokers.some((broker) => broker === value);
const selfBrokerSchema = z.custom<SelfBroker>((value) => {
if (typeof value === 'string' && isSelfBroker(value)) {
return true;
}
return false;
});
const subOrderSchema = baseOrderSchema.extend({
price: z.number(),
id: z.number(),
@@ -72,12 +78,7 @@ const subOrderSchema = baseOrderSchema.extend({
exchange: z.enum(exchanges),
brokerAddress:
z.enum(['ORION_BROKER', 'SELF_BROKER'])
.or(z.custom<SelfBroker>((value) => {
if (typeof value === 'string' && isSelfBroker(value)) {
return true;
}
return false;
}))
.or(selfBrokerSchema)
.or(z.string().refine(ethers.utils.isAddress, (value) => ({
message: `subOrder.subOrders.[n].brokerAddress must be an address, got ${value}`,
}))),