import {
  ChatService,
  ContactService,
  AssetService,
  MetricsService,
  TransactionService,
  FeedbackService,
} from '@open-gsio/services';
import { types, type Instance, getMembers } from 'mobx-state-tree';

const RequestContext = types
  .model('RequestContext', {
    chatService: ChatService,
    contactService: types.optional(ContactService, {}),
    assetService: types.optional(AssetService, {}),
    metricsService: types.optional(MetricsService, {}),
    transactionService: types.optional(TransactionService, {}),
    feedbackService: types.optional(FeedbackService, {}),
  })
  .actions(self => {
    const services = Object.keys(getMembers(self).properties);

    return {
      setEnv(env: Env) {
        services.forEach(service => {
          // @ts-expect-error - override indexing type error
          if (typeof self[service]?.setEnv === 'function') {
            // @ts-expect-error - override indexing type error
            self[service].setEnv(env);
          }
        });
      },
      setCtx(ctx: ExecutionContext) {
        services.forEach(service => {
          // @ts-expect-error - override indexing type error
          if (typeof self[service]?.setCtx === 'function') {
            // @ts-expect-error - override indexing type error
            self[service].setCtx(ctx);
          }
        });
      },
    };
  });

export type RequestContextInstance = Instance<typeof RequestContext>;

const createRequestContext = (env: Env, ctx: any) => {
  const instance = RequestContext.create({
    contactService: ContactService.create({}),
    assetService: AssetService.create({}),
    transactionService: TransactionService.create({}),
    feedbackService: FeedbackService.create({}),
    metricsService: MetricsService.create({
      isCollectingMetrics: true,
    }),
    chatService: ChatService.create({
      openAIApiKey: env.OPENAI_API_KEY,
      openAIBaseURL: env.OPENAI_API_ENDPOINT,
      activeStreams: undefined,
      maxTokens: 16384,
      systemPrompt:
        'You are an assistant designed to provide accurate, concise, and context-aware responses while demonstrating your advanced reasoning capabilities.',
    }),
  });
  instance.setEnv(env);
  instance.setCtx(ctx);
  return instance;
};

export { createRequestContext };