mirror of
https://github.com/ReVanced/revanced-bots.git
synced 2026-01-18 16:53:57 +00:00
b3b7723b4f
BREAKING CHANGES: - Heartbeating removed - `config.consoleLogLevel` -> `config.logLevel` NEW FEATURES: - Training messages - Sequence number system - WebSocket close codes used instead of disconnect packets FIXES: - Improved error handling - Some performance improvements - Made code more clean - Updated dependencies
93 lines
2.5 KiB
TypeScript
Executable File
93 lines
2.5 KiB
TypeScript
Executable File
import {
|
|
url,
|
|
type AnySchema,
|
|
type NullSchema,
|
|
type ObjectSchema,
|
|
type Output,
|
|
array,
|
|
enum_,
|
|
null_,
|
|
object,
|
|
parse,
|
|
special,
|
|
string,
|
|
// merge
|
|
} from 'valibot'
|
|
import DisconnectReason from '../constants/DisconnectReason'
|
|
import { ClientOperation, Operation, ServerOperation } from '../constants/Operation'
|
|
|
|
/**
|
|
* Schema to validate packets
|
|
*/
|
|
export const PacketSchema = special<Packet>(input => {
|
|
if (
|
|
typeof input === 'object' &&
|
|
input &&
|
|
'op' in input &&
|
|
typeof input.op === 'number' &&
|
|
input.op in Operation &&
|
|
'd' in input &&
|
|
typeof input.d === 'object'
|
|
) {
|
|
if (input.op in ServerOperation && !('s' in input && typeof input.s === 'number')) return false
|
|
|
|
try {
|
|
parse(PacketDataSchemas[input.op as Operation], input.d)
|
|
return true
|
|
} catch {
|
|
return false
|
|
}
|
|
}
|
|
return false
|
|
}, 'Invalid packet data')
|
|
|
|
/**
|
|
* Schema to validate packet data for each possible operations
|
|
*/
|
|
export const PacketDataSchemas = {
|
|
[ServerOperation.Hello]: null_(),
|
|
[ServerOperation.ParsedText]: object({
|
|
labels: array(
|
|
object({
|
|
name: string(),
|
|
confidence: special<number>(input => typeof input === 'number' && input >= 0 && input <= 1),
|
|
}),
|
|
),
|
|
}),
|
|
[ServerOperation.ParsedImage]: object({
|
|
text: string(),
|
|
}),
|
|
[ServerOperation.ParseTextFailed]: null_(),
|
|
[ServerOperation.ParseImageFailed]: null_(),
|
|
[ServerOperation.Disconnect]: object({
|
|
reason: enum_(DisconnectReason),
|
|
}),
|
|
[ServerOperation.TrainedMessage]: null_(),
|
|
[ServerOperation.TrainMessageFailed]: null_(),
|
|
|
|
[ClientOperation.ParseText]: object({
|
|
text: string(),
|
|
}),
|
|
[ClientOperation.ParseImage]: object({
|
|
image_url: string([url()]),
|
|
}),
|
|
[ClientOperation.TrainMessage]: object({
|
|
text: string(),
|
|
label: string(),
|
|
}),
|
|
} as const satisfies Record<
|
|
Operation,
|
|
// biome-ignore lint/suspicious/noExplicitAny: This is a schema, it's not possible to type it
|
|
ObjectSchema<any> | AnySchema | NullSchema
|
|
>
|
|
|
|
export type Packet<TOp extends Operation = Operation> = TOp extends ServerOperation
|
|
? PacketWithSequenceNumber<TOp>
|
|
: Omit<PacketWithSequenceNumber<TOp>, 's'>
|
|
|
|
type PacketWithSequenceNumber<TOp extends Operation> = {
|
|
op: TOp
|
|
d: Output<(typeof PacketDataSchemas)[TOp]>
|
|
s: number
|
|
}
|