/*
 * Copyright 2025 Hirokazu Kobayashi
 *
 * Licensed 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.
 */

package org.idp.server.core.adapters.datasource.multi_tenancy.tenant.query;

import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import org.idp.server.platform.datasource.cache.CacheStore;
import org.idp.server.platform.json.JsonConverter;
import org.idp.server.platform.log.LoggerWrapper;
import org.idp.server.platform.multi_tenancy.tenant.*;

public class TenantQueryDataSource implements TenantQueryRepository {

  private static final LoggerWrapper log = LoggerWrapper.getLogger(TenantQueryDataSource.class);
  TenantQuerySqlExecutor executor;
  JsonConverter jsonConverter;
  CacheStore cacheStore;

  public TenantQueryDataSource(TenantQuerySqlExecutor executor, CacheStore cacheStore) {
    this.executor = executor;
    this.jsonConverter = JsonConverter.snakeCaseInstance();
    this.cacheStore = cacheStore;
  }

  @Override
  public Tenant get(TenantIdentifier tenantIdentifier) {
    String key = key(tenantIdentifier);
    Optional<Tenant> optionalTenant = cacheStore.find(key, Tenant.class);

    if (optionalTenant.isPresent()) {
      Tenant cached = optionalTenant.get();
      log.trace("tenant cache HIT: tenant={}", tenantIdentifier.value());
      return cached;
    }

    log.trace("tenant cache MISS: tenant={}", tenantIdentifier.value());
    Map<String, String> result = executor.selectOne(tenantIdentifier);

    if (Objects.isNull(result) || result.isEmpty()) {
      throw new TenantNotFoundException(
          String.format("Tenant is not found (%s)", tenantIdentifier.value()));
    }

    Tenant convert = ModelConverter.convert(result);

    // Cache only when enabled = true
    if (convert.isEnabled()) {
      cacheStore.put(key, convert);
    }

    return convert;
  }

  @Override
  public Tenant find(TenantIdentifier tenantIdentifier) {
    String key = key(tenantIdentifier);
    Optional<Tenant> optionalTenant = cacheStore.find(key, Tenant.class);

    if (optionalTenant.isPresent()) {
      return optionalTenant.get();
    }

    Map<String, String> result = executor.selectOne(tenantIdentifier);

    if (Objects.isNull(result) || result.isEmpty()) {
      return new Tenant();
    }

    Tenant convert = ModelConverter.convert(result);

    // Cache only when enabled = true
    if (convert.isEnabled()) {
      cacheStore.put(key, convert);
    }

    return convert;
  }

  @Override
  public Tenant findWithDisabled(TenantIdentifier tenantIdentifier, boolean includeDisabled) {
    String key = key(tenantIdentifier);
    Optional<Tenant> optionalTenant = cacheStore.find(key, Tenant.class);

    if (optionalTenant.isPresent()) {
      return optionalTenant.get();
    }

    Map<String, String> result = executor.selectOne(tenantIdentifier, includeDisabled);

    if (Objects.isNull(result) || result.isEmpty()) {
      return new Tenant();
    }

    Tenant convert = ModelConverter.convert(result);

    // Cache only when enabled = true
    if (convert.isEnabled()) {
      cacheStore.put(key, convert);
    }

    return convert;
  }

  @Override
  public Tenant getAdmin() {

    Map<String, String> result = executor.selectAdmin();

    if (Objects.isNull(result) || result.isEmpty()) {
      throw new TenantNotFoundException("Admin Tenant is unregistered.");
    }

    return ModelConverter.convert(result);
  }

  @Override
  public Tenant findAdmin() {

    Map<String, String> result = executor.selectAdmin();

    if (Objects.isNull(result) || result.isEmpty()) {
      return new Tenant();
    }

    return ModelConverter.convert(result);
  }

  @Override
  public List<Tenant> findList(List<TenantIdentifier> tenantIdentifiers) {

    List<Map<String, String>> results = executor.selectList(tenantIdentifiers);

    if (Objects.isNull(results) || results.isEmpty()) {
      return List.of();
    }

    return results.stream().map(ModelConverter::convert).toList();
  }

  private String key(TenantIdentifier tenantIdentifier) {
    return "tenantId:" + tenantIdentifier.value() + ":" + Tenant.class.getSimpleName();
  }
}