All files / json-crdt/codec/structural/verbose Encoder.ts

98.87% Statements 88/89
100% Branches 26/26
100% Functions 16/16
98.64% Lines 73/74

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 18014x 14x 14x 14x         14x       1917x 1917x 1917x 1917x             1607x 1607x 1607x 1607x 26722x 1607x       244547x       88400x 60167x 52139x 33502x 32331x 4007x 3x         28233x 28233x 76602x   28233x               3x 3x 3x 3x 3x 9x 9x 8x   3x               8028x 8028x   8028x 8028x               6938x 632x       632x   6306x 6306x   8702x   6306x       18637x 18637x   86599x 18637x               86599x 43241x       43241x   43358x       43358x       4004x 4004x   60692x 4004x               60692x 45247x       45247x   15445x       15445x       3088x               28324x       28324x 28324x 1x 1x   28323x   28324x      
import * as nodes from '../../../nodes';
import {toBase64} from '@jsonjoy.com/base64/lib/toBase64';
import {SESSION} from '../../../../json-crdt-patch/constants';
import {type ITimestampStruct, type IClockVector, Timestamp} from '../../../../json-crdt-patch/clock';
import type {Model} from '../../../model';
import type * as types from './types';
import type {Chunk} from '../../../nodes/rga';
 
export class Encoder {
  protected model!: Model;
 
  public encode(model: Model<any>): types.JsonCrdtVerboseDocument {
    this.model = model;
    const clock = model.clock;
    const isServerClock = clock.sid === SESSION.SERVER;
    return {
      time: isServerClock ? clock.time : this.cClock(model.clock),
      root: this.cVal(model.root),
    };
  }
 
  public cClock(clock: IClockVector): types.JsonCrdtVerboseLogicalTimestamp[] {
    const data: types.JsonCrdtVerboseLogicalTimestamp[] = [];
    const sessionId = clock.sid;
    const localTs = clock.peers.get(sessionId);
    if (!localTs) data.push([sessionId, clock.time]);
    for (const c of clock.peers.values()) data.push([c.sid, c.time]);
    return data;
  }
 
  public cTs(ts: ITimestampStruct): types.JsonCrdtVerboseLogicalTimestamp | types.JsonCrdtVerboseServerTimestamp {
    return ts.sid === SESSION.SERVER ? ts.time : [ts.sid, ts.time];
  }
 
  public cNode(node: nodes.JsonNode): types.JsonCrdtNode {
    if (node instanceof nodes.ObjNode) return this.cObj(node);
    else if (node instanceof nodes.ArrNode) return this.cArr(node);
    else if (node instanceof nodes.StrNode) return this.cStr(node);
    else if (node instanceof nodes.ValNode) return this.cVal(node);
    else if (node instanceof nodes.ConNode) return this.cCon(node);
    else if (node instanceof nodes.BinNode) return this.cBin(node);
    else if (node instanceof nodes.VecNode) return this.cVec(node);
    throw new Error('UNKNOWN_NODE');
  }
 
  public cObj(obj: nodes.ObjNode): types.JsonCrdtVerboseObj {
    const map: Record<string, types.JsonCrdtNode> = {};
    obj.nodes((node, key) => {
      map[key] = this.cNode(node);
    });
    return {
      type: 'obj',
      id: this.cTs(obj.id),
      map,
    };
  }
 
  public cVec(obj: nodes.VecNode): types.JsonCrdtVerboseVec {
    const map: types.JsonCrdtVerboseVec['map'] = [];
    const elements = obj.elements;
    const length = elements.length;
    const index = this.model.index;
    for (let i = 0; i < length; i++) {
      const element = elements[i];
      if (element === undefined) map.push(null);
      else map.push(this.cNode(index.get(element)!));
    }
    return {
      type: 'vec',
      id: this.cTs(obj.id),
      map,
    };
  }
 
  public cArr(obj: nodes.ArrNode): types.JsonCrdtVerboseArr {
    const chunks: (types.JsonCrdtVerboseArrChunk | types.JsonCrdtVerboseTombstone)[] = [];
    const iterator = obj.iterator();
    let chunk: Chunk<ITimestampStruct[]> | undefined;
    while ((chunk = iterator())) chunks.push(this.cArrChunk(chunk));
    return {
      type: 'arr',
      id: this.cTs(obj.id),
      chunks,
    };
  }
 
  public cArrChunk(chunk: nodes.ArrChunk): types.JsonCrdtVerboseArrChunk | types.JsonCrdtVerboseTombstone {
    if (chunk.del) {
      const tombstone: types.JsonCrdtVerboseTombstone = {
        id: this.cTs(chunk.id),
        span: chunk.span,
      };
      return tombstone;
    }
    const index = this.model.index;
    const res: types.JsonCrdtVerboseArrChunk = {
      id: this.cTs(chunk.id),
      value: chunk.data!.map((n) => this.cNode(index.get(n)!)),
    };
    return res;
  }
 
  public cStr(obj: nodes.StrNode): types.JsonCrdtVerboseStr {
    const chunks: (types.JsonCrdtVerboseStrChunk | types.JsonCrdtVerboseTombstone)[] = [];
    const iterator = obj.iterator();
    let chunk: ReturnType<typeof iterator>;
    while ((chunk = iterator())) chunks.push(this.cStrChunk(chunk as nodes.StrChunk));
    return {
      type: 'str',
      id: this.cTs(obj.id),
      chunks,
    };
  }
 
  public cStrChunk(chunk: nodes.StrChunk): types.JsonCrdtVerboseStrChunk | types.JsonCrdtVerboseTombstone {
    if (chunk.del) {
      const tombstone: types.JsonCrdtVerboseTombstone = {
        id: this.cTs(chunk.id),
        span: chunk.span,
      };
      return tombstone;
    }
    const res: types.JsonCrdtVerboseStrChunk = {
      id: this.cTs(chunk.id),
      value: chunk.data!,
    };
    return res;
  }
 
  public cBin(obj: nodes.BinNode): types.JsonCrdtVerboseBin {
    const chunks: (types.JsonCrdtVerboseBinChunk | types.JsonCrdtVerboseTombstone)[] = [];
    const iterator = obj.iterator();
    let chunk: ReturnType<typeof iterator>;
    while ((chunk = iterator())) chunks.push(this.cBinChunk(chunk as nodes.BinChunk));
    return {
      type: 'bin',
      id: this.cTs(obj.id),
      chunks,
    };
  }
 
  public cBinChunk(chunk: nodes.BinChunk): types.JsonCrdtVerboseBinChunk | types.JsonCrdtVerboseTombstone {
    if (chunk.del) {
      const tombstone: types.JsonCrdtVerboseTombstone = {
        id: this.cTs(chunk.id),
        span: chunk.span,
      };
      return tombstone;
    }
    const res: types.JsonCrdtVerboseStrChunk = {
      id: this.cTs(chunk.id),
      value: toBase64(chunk.data!),
    };
    return res;
  }
 
  public cVal(obj: nodes.ValNode): types.JsonCrdtVerboseVal {
    return {
      type: 'val',
      id: this.cTs(obj.id),
      value: this.cNode(obj.node()),
    };
  }
 
  public cCon(obj: nodes.ConNode): types.JsonCrdtVerboseCon {
    const node: types.JsonCrdtVerboseCon = {
      type: 'con',
      id: this.cTs(obj.id),
    };
    const val = obj.val;
    if (val instanceof Timestamp) {
      node.timestamp = true;
      node.value = this.cTs(val);
    } else {
      if (val !== undefined) node.value = val;
    }
    return node;
  }
}