Add type to schemas

Signed-off-by: Quoc-Hao Tran <quoc-hao.tran@polymtl.ca>

Author: haoadoreorange

tsp-typescript-client/package.json

@@ -23,7 +23,7 @@
   "dependencies": {
     "node-fetch": "^2.5.0",
     "rimraf": "latest",
-    "when-json-met-bigint": "^0.21.0"
+    "when-json-met-bigint": "^0.26.0"
   },
   "scripts": {
     "prepare": "yarn run clean && yarn run build",

tsp-typescript-client/src/models/annotation.ts

@@ -11,7 +11,7 @@ export interface AnnotationCategoriesModel {
     annotationCategories: string[];
 }
 
-export const AnnotationSchema: Schema = {
+export const AnnotationSchema: Schema<AnnotationModel> = {
     annotations: {
         [Symbol.for(`any`)]: [{
             duration: bigint,

tsp-typescript-client/src/models/bookmark.ts

@@ -1,7 +1,7 @@
 import { Schema } from 'when-json-met-bigint';
 import { bigint } from '../protocol/serialization';
 
-export const BookmarkSchema: Schema = {
+export const BookmarkSchema: Schema<Bookmark> = {
     endTime: bigint,
     startTime: bigint,
 };

tsp-typescript-client/src/models/entry.ts

@@ -2,7 +2,7 @@ import { Schema } from 'when-json-met-bigint';
 import { assertNumber } from '../protocol/serialization';
 import { OutputElementStyle } from './styles';
 
-export const EntrySchema: Schema = {
+export const EntrySchema: Schema<Entry> = {
     id: assertNumber,
     parentId: assertNumber,
 };
@@ -52,7 +52,7 @@ export interface EntryHeader {
     tooltip: string
 }
 
-export const EntryModelSchema = (schema: Schema): Schema => ({ entries: [schema] });
+export const EntryModelSchema = <T extends Entry>(schema: Schema<T>): Schema<EntryModel<T>> => ({ entries: [schema] });
 
 /**
  * Entry model that will be returned by the server

tsp-typescript-client/src/models/experiment.ts

@@ -2,7 +2,7 @@ import { Schema } from 'when-json-met-bigint';
 import { assertNumber, bigint } from '../protocol/serialization';
 import { TraceSchema, Trace } from './trace';
 
-export const ExperimentSchema: Schema = {
+export const ExperimentSchema: Schema<Experiment> = {
     end: bigint,
     nbEvents: assertNumber,
     start: bigint,

tsp-typescript-client/src/models/output-descriptor.ts

@@ -1,7 +1,7 @@
 import { Schema } from 'when-json-met-bigint';
 import { bigint } from '../protocol/serialization';
 
-export const OutputDescriptorSchema: Schema = {
+export const OutputDescriptorSchema: Schema<OutputDescriptor> = {
     end: bigint,
     start: bigint,
 };

tsp-typescript-client/src/models/response/responses.ts

@@ -24,7 +24,7 @@ export enum ResponseStatus {
     CANCELLED = 'CANCELLED'
 }
 
-export const GenericResponseSchema = (schema: Schema): Schema => ({ model: schema });
+export const GenericResponseSchema = <T>(schema: Schema<T>): Schema<GenericResponse<T>> => ({ model: schema });
 
 /**
  * Generic response that contains a model

tsp-typescript-client/src/models/table.ts

@@ -1,7 +1,7 @@
 import { Schema } from 'when-json-met-bigint';
 import { assertNumber } from '../protocol/serialization';
 
-export const ColumnHeaderEntrySchema: Schema = {
+export const ColumnHeaderEntrySchema: Schema<ColumnHeaderEntry> = {
     id: assertNumber,
 };
 
@@ -30,7 +30,7 @@ export interface ColumnHeaderEntry {
     type: string;
 }
 
-export const CellSchema: Schema = {
+export const CellSchema: Schema<Cell> = {
     tags: assertNumber,
 };
 
@@ -49,7 +49,7 @@ export interface Cell {
     tags?: number;
 }
 
-export const LineSchema = {
+export const LineSchema: Schema<Line> = {
     cells: [CellSchema],
     index: assertNumber,
     tags: assertNumber,
@@ -75,7 +75,7 @@ export interface Line {
     tags?: number;
 }
 
-export const TableModelSchema = {
+export const TableModelSchema: Schema<TableModel> = {
     columnIds: [assertNumber],
     lines: [LineSchema],
     lowIndex: assertNumber,

tsp-typescript-client/src/models/timegraph.ts

@@ -3,7 +3,7 @@ import { assertNumber, bigint } from '../protocol/serialization';
 import { Entry } from './entry';
 import { OutputElementStyle } from './styles';
 
-export const TimeGraphEntrySchema: Schema = {
+export const TimeGraphEntrySchema: Schema<TimeGraphEntry> = {
     end: bigint,
     id: assertNumber,
     parentId: assertNumber,
@@ -25,7 +25,7 @@ export interface TimeGraphEntry extends Entry {
     end: bigint;
 }
 
-const TimeGraphStateSchema: Schema = {
+const TimeGraphStateSchema: Schema<TimeGraphState> = {
     end: bigint,
     start: bigint,
     tags: assertNumber,
@@ -61,7 +61,7 @@ export interface TimeGraphState {
     style?: OutputElementStyle;
 }
 
-export const TimeGraphRowSchema: Schema = {
+export const TimeGraphRowSchema: Schema<TimeGraphRow> = {
     entryId: assertNumber,
     states: [TimeGraphStateSchema],
 };
@@ -81,7 +81,7 @@ export interface TimeGraphRow {
     states: TimeGraphState[];
 }
 
-export const TimeGraphModelSchema: Schema = {
+export const TimeGraphModelSchema: Schema<TimeGraphModel> = {
     rows: [TimeGraphRowSchema],
 };
 
@@ -92,7 +92,7 @@ export interface TimeGraphModel {
     rows: TimeGraphRow[];
 }
 
-export const TimeGraphArrowSchema: Schema = {
+export const TimeGraphArrowSchema: Schema<TimeGraphArrow> = {
     end: bigint,
     sourceId: assertNumber,
     start: bigint,

tsp-typescript-client/src/models/trace.ts

@@ -1,7 +1,7 @@
 import { Schema } from 'when-json-met-bigint';
 import { assertNumber, bigint } from '../protocol/serialization';
 
-export const TraceSchema: Schema = {
+export const TraceSchema: Schema<Trace> = {
     end: bigint,
     nbEvents: assertNumber,
     start: bigint,