import { DataSanitizerInput } from '../internal/logging.generated.js';
import { BaseUrls } from './baseUrls.generated.js';
import { Interceptor } from './interceptors.generated.js';
import { Agent } from '../internal/utils.js';
import { AgentOptions } from '../internal/utils.js';
import { createAgent } from '../internal/utils.js';
import { ProxyConfig } from './proxyConfig.generated.js';
import { BoxNetworkClient } from './boxNetworkClient.js';
import { NetworkClient } from './networkClient.generated.js';
import { RetryStrategy } from './retries.generated.js';
import { BoxRetryStrategy } from './retries.generated.js';
import { DataSanitizer } from '../internal/logging.generated.js';
export class NetworkSession {
  readonly additionalHeaders: {
    readonly [key: string]: string;
  } = {};
  readonly baseUrls: BaseUrls = new BaseUrls({});
  readonly interceptors: readonly Interceptor[] = [];
  readonly agent: Agent = createAgent(void 0, void 0);
  readonly agentOptions?: AgentOptions;
  readonly proxyConfig?: ProxyConfig;
  readonly networkClient: NetworkClient = new BoxNetworkClient({});
  readonly retryStrategy: RetryStrategy = new BoxRetryStrategy({});
  readonly dataSanitizer: DataSanitizer = new DataSanitizer({});
  constructor(
    fields: Omit<
      NetworkSession,
      | 'additionalHeaders'
      | 'baseUrls'
      | 'interceptors'
      | 'agent'
      | 'networkClient'
      | 'retryStrategy'
      | 'dataSanitizer'
      | 'withAdditionalHeaders'
      | 'withCustomBaseUrls'
      | 'withCustomAgentOptions'
      | 'withInterceptors'
      | 'withProxy'
      | 'withNetworkClient'
      | 'withRetryStrategy'
      | 'withDataSanitizer'
    > &
      Partial<
        Pick<
          NetworkSession,
          | 'additionalHeaders'
          | 'baseUrls'
          | 'interceptors'
          | 'agent'
          | 'networkClient'
          | 'retryStrategy'
          | 'dataSanitizer'
        >
      >,
  ) {
    if (fields.additionalHeaders !== undefined) {
      this.additionalHeaders = fields.additionalHeaders;
    }
    if (fields.baseUrls !== undefined) {
      this.baseUrls = fields.baseUrls;
    }
    if (fields.interceptors !== undefined) {
      this.interceptors = fields.interceptors;
    }
    if (fields.agent !== undefined) {
      this.agent = fields.agent;
    }
    if (fields.agentOptions !== undefined) {
      this.agentOptions = fields.agentOptions;
    }
    if (fields.proxyConfig !== undefined) {
      this.proxyConfig = fields.proxyConfig;
    }
    if (fields.networkClient !== undefined) {
      this.networkClient = fields.networkClient;
    }
    if (fields.retryStrategy !== undefined) {
      this.retryStrategy = fields.retryStrategy;
    }
    if (fields.dataSanitizer !== undefined) {
      this.dataSanitizer = fields.dataSanitizer;
    }
  }
  /**
     * Generate a fresh network session by duplicating the existing configuration and network parameters, while also including additional headers to be attached to every API call.
     * @param {{
        readonly [key: string]: string;
    }} additionalHeaders Headers, which are appended to each API request
     * @returns {NetworkSession}
     */
  withAdditionalHeaders(
    additionalHeaders: {
      readonly [key: string]: string;
    } = {},
  ): NetworkSession {
    return new NetworkSession({
      additionalHeaders: { ...this.additionalHeaders, ...additionalHeaders },
      baseUrls: this.baseUrls,
      interceptors: this.interceptors,
      agent: this.agent,
      agentOptions: this.agentOptions,
      proxyConfig: this.proxyConfig,
      networkClient: this.networkClient,
      retryStrategy: this.retryStrategy,
      dataSanitizer: this.dataSanitizer,
    });
  }
  /**
   * Generate a fresh network session by duplicating the existing configuration and network parameters, while also including custom base urls to be used for every API call.
   * @param {BaseUrls} baseUrls Custom base urls
   * @returns {NetworkSession}
   */
  withCustomBaseUrls(baseUrls: BaseUrls): NetworkSession {
    return new NetworkSession({
      additionalHeaders: this.additionalHeaders,
      baseUrls: baseUrls,
      interceptors: this.interceptors,
      agent: this.agent,
      agentOptions: this.agentOptions,
      proxyConfig: this.proxyConfig,
      networkClient: this.networkClient,
      retryStrategy: this.retryStrategy,
      dataSanitizer: this.dataSanitizer,
    });
  }
  /**
   * Generate a fresh network session by duplicating the existing configuration and network parameters, while also including custom agent options to be used for every API call.
   * @param {AgentOptions} agentOptions Custom agent options
   * @returns {NetworkSession}
   */
  withCustomAgentOptions(agentOptions: AgentOptions): NetworkSession {
    return new NetworkSession({
      additionalHeaders: this.additionalHeaders,
      baseUrls: this.baseUrls,
      interceptors: this.interceptors,
      agent: createAgent(agentOptions, this.proxyConfig),
      agentOptions: this.agentOptions,
      proxyConfig: this.proxyConfig,
      networkClient: this.networkClient,
      retryStrategy: this.retryStrategy,
      dataSanitizer: this.dataSanitizer,
    });
  }
  /**
   * Generate a fresh network session by duplicating the existing configuration and network parameters, while also additional including custom interceptors.
   * @param {readonly Interceptor[]} interceptors Custom base urls
   * @returns {NetworkSession}
   */
  withInterceptors(interceptors: readonly Interceptor[]): NetworkSession {
    return new NetworkSession({
      additionalHeaders: this.additionalHeaders,
      baseUrls: this.baseUrls,
      interceptors: this.interceptors.concat(interceptors),
      agent: this.agent,
      agentOptions: this.agentOptions,
      proxyConfig: this.proxyConfig,
      networkClient: this.networkClient,
      retryStrategy: this.retryStrategy,
      dataSanitizer: this.dataSanitizer,
    });
  }
  /**
   * Generate a fresh network session by duplicating the existing configuration and network parameters, while also including a custom proxy configuration.
   * @param {ProxyConfig} proxyConfig
   * @returns {NetworkSession}
   */
  withProxy(proxyConfig: ProxyConfig): NetworkSession {
    return new NetworkSession({
      additionalHeaders: this.additionalHeaders,
      baseUrls: this.baseUrls,
      interceptors: this.interceptors,
      agent: createAgent(this.agentOptions, proxyConfig),
      agentOptions: this.agentOptions,
      proxyConfig: proxyConfig,
      networkClient: this.networkClient,
      retryStrategy: this.retryStrategy,
      dataSanitizer: this.dataSanitizer,
    });
  }
  /**
   * Generate a fresh network session by duplicating the existing configuration and network parameters, while also including a custom network client.
   * @param {NetworkClient} networkClient
   * @returns {NetworkSession}
   */
  withNetworkClient(networkClient: NetworkClient): NetworkSession {
    return new NetworkSession({
      additionalHeaders: this.additionalHeaders,
      baseUrls: this.baseUrls,
      interceptors: this.interceptors,
      agent: this.agent,
      agentOptions: this.agentOptions,
      proxyConfig: this.proxyConfig,
      networkClient: networkClient,
      retryStrategy: this.retryStrategy,
      dataSanitizer: this.dataSanitizer,
    });
  }
  /**
   * Generate a fresh network session by duplicating the existing configuration and network parameters, while also applying retry strategy
   * @param {RetryStrategy} retryStrategy
   * @returns {NetworkSession}
   */
  withRetryStrategy(retryStrategy: RetryStrategy): NetworkSession {
    return new NetworkSession({
      additionalHeaders: this.additionalHeaders,
      baseUrls: this.baseUrls,
      interceptors: this.interceptors,
      agent: this.agent,
      agentOptions: this.agentOptions,
      proxyConfig: this.proxyConfig,
      networkClient: this.networkClient,
      retryStrategy: retryStrategy,
      dataSanitizer: this.dataSanitizer,
    });
  }
  /**
   * Generate a fresh network session by duplicating the existing configuration and network parameters, while also applying data sanitizer
   * @param {DataSanitizerInput} dataSanitizerInput
   * @returns {NetworkSession}
   */
  withDataSanitizer(dataSanitizerInput: DataSanitizerInput): NetworkSession {
    const dataSanitizer: DataSanitizer = new DataSanitizer({});
    return new NetworkSession({
      additionalHeaders: this.additionalHeaders,
      baseUrls: this.baseUrls,
      interceptors: this.interceptors,
      agent: this.agent,
      agentOptions: this.agentOptions,
      proxyConfig: this.proxyConfig,
      networkClient: this.networkClient,
      retryStrategy: this.retryStrategy,
      dataSanitizer: dataSanitizer,
    });
  }
}
export interface NetworkSessionInput {
  readonly additionalHeaders?: {
    readonly [key: string]: string;
  };
  readonly baseUrls?: BaseUrls;
  readonly interceptors?: readonly Interceptor[];
  readonly agent?: Agent;
  readonly agentOptions?: AgentOptions;
  readonly proxyConfig?: ProxyConfig;
  readonly networkClient?: NetworkClient;
  readonly retryStrategy?: RetryStrategy;
  readonly dataSanitizer?: DataSanitizer;
}