cluster.ts

/*
 * Licensed to Elasticsearch B.V. under one or more contributor
 * license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright
 * ownership. Elasticsearch B.V. licenses this file to you under
 * the Apache License, Version 2.0 (the "License"); you may
 * not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

import { Request } from 'hapi';
import { errors } from 'elasticsearch';
import { CallAPIOptions, ClusterClient, FakeRequest } from 'kibana/server';

export class Cluster {
  public readonly errors = errors;

  constructor(private readonly clusterClient: ClusterClient) {}

  public callWithRequest = async (
    req: Request | FakeRequest,
    endpoint: string,
    clientParams?: Record<string, unknown>,
    options?: CallAPIOptions
  ) => {
    return await this.clusterClient
      .asScoped(req)
      .callAsCurrentUser(endpoint, clientParams, options);
  };

  public callWithInternalUser = async (
    endpoint: string,
    clientParams?: Record<string, unknown>,
    options?: CallAPIOptions
  ) => {
    return await this.clusterClient.callAsInternalUser(endpoint, clientParams, options);
  };

  public close() {
    this.clusterClient.close();
  }
}