Skip to content

Instantly share code, notes, and snippets.

@YBogomolov
Last active November 26, 2022 16:45
Show Gist options
  • Star 10 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save YBogomolov/c5675b788a7046b6edf9e76ef7337af7 to your computer and use it in GitHub Desktop.
Save YBogomolov/c5675b788a7046b6edf9e76ef7337af7 to your computer and use it in GitHub Desktop.
Session types for TypeScript
// Inspired by https://github.com/Munksgaard/session-types
interface Chan<Env, Protocol> {
env_and_protocol: [Env, Protocol];
}
class Eps implements HasDual {
readonly tag: 'Eps' = 'Eps';
readonly dual!: Eps;
}
abstract class Send<A, Protocol> implements HasDual {
readonly dual!: Recv<A, Dual<Protocol>>;
readonly tag: 'Send' = 'Send';
}
abstract class Recv<A, Protocol> implements HasDual {
readonly dual!: Send<A, Dual<Protocol>>;
readonly tag: 'Recv' = 'Recv';
}
interface HasDual {
dual: HasDual;
}
type ExtractProtocol<C> = C extends Chan<infer _, infer P> ? P : never;
type ExtractEnv<C> = C extends Chan<infer E, infer _> ? E : never;
type Dual<X> =
X extends Send<infer A, infer P> ? Recv<A, P extends HasDual ? P['dual'] : never> :
X extends Recv<infer A, infer P> ? Send<A, P extends HasDual ? P['dual'] : never> :
X extends Eps ? Eps :
never;
declare const srv: Chan<void, Send<number, Recv<boolean, Recv<Float32Array, Eps>>>>;
declare const cli: Chan<void, Dual<ExtractProtocol<typeof srv>>>;
declare const cli2: Chan<void, Recv<string, Send<boolean, Send<Float32Array, Eps>>>>;
const connect = <A, B extends Dual<A>>(_server: Chan<void, A>, _client: Chan<void, B>) => { };
connect(srv, cli); // Ok
connect(cli, srv); // Ok
connect(srv, cli2); // Type 'string' is not assignable to type 'number'
connect(srv, srv); // Type 'Send<boolean, Eps>' is not assignable to type 'Recv<boolean, Eps>'
connect(cli, cli); // Type 'Send<boolean, Eps>' is not assignable to type 'Recv<boolean, Eps>'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment