Fix types

mermaid-js · Jan 20, 2022 · aee4460 · aee4460
1 parent b52f2c4
commit aee4460
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/src/lib/util/serde.ts b/src/lib/util/serde.ts
@@ -32,15 +32,15 @@ export const pakoSerde: Serde = {
  return inflate(data, { to: 'string' });
  }
 };
-const serdes: Record<SerdeType, Serde> = {
- [SerdeType.Base64]: base64Serde,
- [SerdeType.Pako]: pakoSerde
-};
+const serdes: Map<string, Serde> = new Map([
+ [SerdeType.Base64, base64Serde],
+ [SerdeType.Pako, pakoSerde]
+]);
 
 export const serializeState = (state: State): string => {
  const json = JSON.stringify(state);
  const defaultSerde = SerdeType.Pako;
- const serialized = serdes[defaultSerde].serialize(json);
+ const serialized = serdes.get(defaultSerde).serialize(json);
  return `${defaultSerde}:${serialized}`;
 };
 
@@ -52,6 +52,6 @@ export const deserializeState = (state: string): State => {
  type = SerdeType.Base64;
  serialized = state;
  }
- const json = serdes[type].deserialize(serialized);
- return JSON.parse(json);
+ const json = serdes.get(type).deserialize(serialized);
+ return JSON.parse(json) as State;
 };
diff --git a/src/lib/util/state.ts b/src/lib/util/state.ts
@@ -46,7 +46,7 @@ export const loadState = (data: string): void => {
  console.log('Loading', data);
  try {
  state = deserializeState(data);
- const mermaidConfig =
+ const mermaidConfig: { [key: string]: string } =
  typeof state.mermaid === 'string' ? JSON.parse(state.mermaid) : state.mermaid;
  if (
  mermaidConfig.securityLevel &&