dataService.tsx

import { CommonServiceIds, IExtensionDataManager, IExtensionDataService } from 'azure-devops-extension-api';
import { appInsights } from '../utilities/telemetryClient';
import { SDKContext } from './azureDevOpsContextProvider';
import React from 'react';

let extensionDataManager: IExtensionDataManager;

async function getDataService(): Promise<IExtensionDataManager> {
  if (!extensionDataManager) {
    const { SDK } = React.useContext(SDKContext);
    const accessToken = await SDK.getAccessToken();
    const extensionDataService = await SDK.getService<IExtensionDataService>(CommonServiceIds.ExtensionDataService);
    extensionDataManager = await extensionDataService.getExtensionDataManager(SDK.getExtensionContext().id, accessToken);
  }

  return extensionDataManager;
}

/**
 * Read user/account scoped documents.
 */
export async function readDocuments<T>(
  collectionName: string, isPrivate?: boolean, throwCollectionDoesNotExistException?: boolean): Promise<T[]> {
  const dataService: IExtensionDataManager = await getDataService();
  let data: T[];

  try {
    data = await dataService.getDocuments(collectionName, isPrivate ? { scopeType: 'User' } : undefined);
    // eslint-disable-next-line @typescript-eslint/no-explicit-any
  } catch (e: any) {
    console.error(e);
    appInsights.trackException(e);
    if (e.serverError?.typeKey === 'DocumentCollectionDoesNotExistException') {
      if (throwCollectionDoesNotExistException) {
        throw e;
      }
    }

    appInsights.trackException(e);

    data = [];
  }

  return data;
}

/**
 * Read a specific user/account scoped document.
 */
export async function readDocument<T>(collectionName: string, id: string, isPrivate?: boolean): Promise<T> {
  if (id === "emptyFeedbackItem") {
    return undefined;
  }
  const dataService: IExtensionDataManager = await getDataService();
  let data: T;
  try {
    data = await dataService.getDocument(collectionName, id, isPrivate ? { scopeType: 'User' } : undefined);
  } catch (e) {
    appInsights.trackException(e);
    console.error('An exception occurred while trying to read the document: ', e);
    data = undefined;
  }

  return data;
}

/**
 * Create user/account scoped document.
 */
export async function createDocument<T>(collectionName: string, data: T, isPrivate?: boolean): Promise<T> {
  const dataService: IExtensionDataManager = await getDataService();
  return dataService.createDocument(collectionName, data, isPrivate ? { scopeType: 'User' } : undefined);
}

/**
 * Create or Update user/account scoped document.
 */
export async function createOrUpdateDocument<T>(collectionName: string, data: T, isPrivate?: boolean): Promise<T> {
  const dataService: IExtensionDataManager = await getDataService();
  return dataService.setDocument(collectionName, data, isPrivate ? { scopeType: 'User' } : undefined);
}

/**
 * Update user/account scoped document.
 */
export async function updateDocument<T>(collectionName: string, data: T, isPrivate?: boolean): Promise<T> {
  const dataService: IExtensionDataManager = await getDataService();

  let updatedData: T;
  try {
    updatedData = await dataService.updateDocument(collectionName, data, isPrivate ? { scopeType: 'User' } : undefined);
  } catch (e) {
    appInsights.trackException(e);
    console.error('An exception occurred while trying to update the document: ', e);
    updatedData = undefined;
  }

  return updatedData;
}

/**
 * Delete user/account scoped document.
 */
export async function deleteDocument(collectionName: string, id: string, isPrivate?: boolean): Promise<void> {
  const dataService: IExtensionDataManager = await getDataService();
  return dataService.deleteDocument(collectionName, id, isPrivate ? { scopeType: 'User' } : undefined);
}

/**
 * Set user/account scoped value.
 */
export async function setValue<T>(id: string, data: T, isPrivate?: boolean): Promise<T> {
  const dataService: IExtensionDataManager = await getDataService();

  let updatedData: T;
  try {
    return dataService.setValue(id, data, isPrivate ? { scopeType: 'User' } : undefined);
  } catch (e) {
    appInsights.trackException(e);
    console.error('An exception occurred while trying to read the value: ', e);
    updatedData = undefined;
  }

  return updatedData;
}

/**
 * Get user/account scoped value.
 */
export async function getValue<T>(id: string, isPrivate?: boolean): Promise<T> {
  const dataService: IExtensionDataManager = await getDataService();

  let data: T;
  try {
    data = await dataService.getValue<T>(id, isPrivate ? { scopeType: 'User' } : undefined);
  } catch (e) {
    appInsights.trackException(e);
    console.error('An exception occurred while trying to read the value: ', e);
    data = undefined;
  }

  return data;
}