locklift/src/internal/timeMovement/index.ts

Node JS framework for working with Ever contracts. Inspired by Truffle and Hardhat. Helps you to build, test, run and maintain your smart contracts.

import { ConfigState, NetworkValue } from "../config";
import { logger } from "../logger";
import { Clock } from "everscale-standalone-client/nodejs";
import { SeService } from "./seService";

export class TimeMovement {
  constructor(
    private readonly seService: SeService | undefined,
    private readonly clock: Clock,
    readonly isEnabled: boolean,
  ) {}

  /*
   * Returns current offset in seconds
   */
  public getTimeOffset = (): number => {
    if (!this.isEnabled) {
      logger.printWarn("TimeMovement is disabled");
      return 0;
    }
    return toSeconds(this.clock.offset);
  };

  /*
   * Returns current time
   */
  public getCurrentTime = () => this.clock.time;
  /*
   * Set node and provider offset in seconds
   @param offsetInSeconds offset in seconds
   */
  public increaseTime = async (seconds: number): Promise<void> => {
    if (!this.isEnabled) {
      return logger.printWarn("TimeMovement is disabled");
    }
    if (seconds < 0) {
      return logger.printWarn("TimeMovement is not allowed to go back in time");
    }
    if (this.seService) {
      this.clock.offset = await this.seService.setTimeOffset(seconds).then(toMs);
    } else {
      this.clock.offset += toMs(seconds);
    }
  };
  //TODO make it public when it will be resolved
  private resetTimeOffset = async (): Promise<void> => {
    if (!this.isEnabled) {
      return logger.printWarn("TimeMovement is disabled");
    }
    if (this.seService) {
      this.clock.offset = await this.seService.resetTimeOffset().then(toMs);
    } else {
      this.clock.offset = 0;
    }
    console.log(`TimeMovement reset to ${this.getTimeOffset()}`);
  };
}

export const createTimeMovement = async (
  clock: Clock,
  connectionConfig: NetworkValue<ConfigState.INTERNAL>,
): Promise<TimeMovement> => {
  switch (connectionConfig.connection.type) {
    case "proxy":
      return new TimeMovement(undefined, clock, true);
    case "graphql": {
      const rpcUrl = connectionConfig.connection.data.endpoints[0];
      const rpcOrigin = new URL(rpcUrl).origin;
      const seService = new SeService(rpcOrigin);
      const { currentOffsetInSeconds, isEnabled } = await seService
        .getCurrentOffsetTime()
        .then(currentOffsetInSeconds => ({ currentOffsetInSeconds, isEnabled: true }))
        .catch(() => ({ isEnabled: false, currentOffsetInSeconds: 0 }));
      if (currentOffsetInSeconds > 0) {
        logger.printWarn(
          `Current SE time delta is ${currentOffsetInSeconds} seconds. Provider will sync with this offset`,
        );
      }
      clock.offset = toMs(currentOffsetInSeconds);
      return new TimeMovement(seService, clock, isEnabled);
    }
    default:
      return new TimeMovement(undefined, clock, false);
  }
};

const toMs = (seconds: number) => seconds * 1000;
const toSeconds = (ms: number) => ms / 1000;