content/browser/resources/quota/quota_internals_browser_proxy.ts

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import {Time} from 'chrome://resources/mojo/mojo/public/mojom/base/time.mojom-webui.js';
import {Origin} from 'chrome://resources/mojo/url/mojom/origin.mojom-webui.js';

import {QuotaInternalsHandler} from './quota_internals.mojom-webui.js';

enum StorageType {
  TEMPORARY,
  PERSISTENT,
  SYNCABLE,
}

interface BucketTableEntry {
  'bucketId': bigint;
  'storageKey': string;
  'type': StorageType;
  'name': string;
  'usage': bigint;
  'useCount': bigint;
  'lastAccessed': Time;
  'lastModified': Time;
}

interface GetDiskAvailabilityAndTempPoolSizeResult {
  totalSpace: bigint;
  availableSpace: bigint;
  tempPoolSize: bigint;
}

interface GetGlobalUsageResult {
  usage: bigint;
  unlimitedUsage: bigint;
}

interface GetStatisticsResult {
  evictionStatistics: {
    'errors-on-getting-usage-and-quota': string,
    'evicted-buckets': string,
    'eviction-rounds': string,
    'skipped-eviction-rounds': string,
  };
}

interface RetrieveBucketsTableResult {
  entries: BucketTableEntry[];
}

function urlPort(url: URL): number {
  if (url.port) {
    return Number.parseInt(url.port, 10);
  }
  if (url.protocol === 'https:') {
    return 443;
  } else if (url.protocol === 'http:') {
    return 80;
  } else {
    return 0;
  }
}

function enumerateStorageType(storageType: string): number {
  switch (storageType) {
    case 'temporary':
      return 0;
    case 'persistent':
      return 1;
    case 'syncable':
      return 2;
    default:
      return 0;
  }
}

export class QuotaInternalsBrowserProxy {
  private handler = QuotaInternalsHandler.getRemote();

  getDiskAvailabilityAndTempPoolSize():
      Promise<GetDiskAvailabilityAndTempPoolSizeResult> {
    return this.handler.getDiskAvailabilityAndTempPoolSize();
  }

  getGlobalUsage(storageType: string): Promise<GetGlobalUsageResult> {
    return this.handler.getGlobalUsageForInternals(
        enumerateStorageType(storageType));
  }

  getStatistics(): Promise<GetStatisticsResult> {
    return this.handler.getStatistics();
  }

  simulateStoragePressure() {
    const originToTest = (document.body.querySelector<HTMLInputElement>(
        '#origin-to-test'))!.value;
    const originUrl = new URL(originToTest);
    const newOrigin = new Origin();
    newOrigin.scheme = originUrl.protocol.replace(/:$/, '');
    newOrigin.host = originUrl.host;
    newOrigin.port = urlPort(originUrl);

    this.handler.simulateStoragePressure(newOrigin);
  }

  retrieveBucketsTable(): Promise<RetrieveBucketsTableResult> {
    return this.handler.retrieveBucketsTable();
  }

  static getInstance(): QuotaInternalsBrowserProxy {
    return instance || (instance = new QuotaInternalsBrowserProxy());
  }
}

let instance: QuotaInternalsBrowserProxy|null = null;