/*
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF 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 { Effect, Context, Layer, pipe } from 'effect';
import { generateGraphSchema, DEFAULT_SCHEMA_CONFIG } from './schema-generator.js';
import {
  createSchemaCache,
  getCachedSchema,
  invalidateSchemaCache,
  peekCachedSchema,
  refreshSchemaCache,
} from './schema-cache.js';
import type { GraphSchema } from './models/index.js';
import { GremlinClient } from './client.js';
import { AppConfig } from '../config.js';
import type { GremlinConnectionError, GremlinQueryError } from '../errors.js';

// Define the service interface and create a tag
export class SchemaService extends Context.Tag('SchemaService')<
  SchemaService,
  {
    readonly getSchema: Effect.Effect<GraphSchema, GremlinConnectionError | GremlinQueryError>;
    readonly peekSchema: Effect.Effect<GraphSchema | null, never>;
    readonly invalidateSchema: Effect.Effect<void, never>;
    readonly refreshSchema: Effect.Effect<void, GremlinConnectionError | GremlinQueryError>;
  }
>() {}

// Implement the live layer
export const SchemaServiceLive = Layer.effect(
  SchemaService,
  Effect.gen(function* () {
    const gremlinClient = yield* GremlinClient;
    const config = yield* AppConfig;
    const cacheRef = yield* createSchemaCache();

    const generateSchemaEffect = pipe(
      gremlinClient.getConnection,
      Effect.flatMap(conn =>
        generateGraphSchema(conn, {
          ...DEFAULT_SCHEMA_CONFIG,
          includeCounts: config.schema.includeCounts,
          includeSampleValues: config.schema.includeSampleValues,
        })
      )
    );

    const getSchema = getCachedSchema(cacheRef, generateSchemaEffect);
    const peekSchema = peekCachedSchema(cacheRef);
    const invalidateSchema = invalidateSchemaCache(cacheRef);
    const refreshSchema = Effect.asVoid(refreshSchemaCache(cacheRef, generateSchemaEffect));

    return SchemaService.of({
      getSchema,
      peekSchema,
      invalidateSchema,
      refreshSchema,
    });
  })
);
