/*
    *    Copyright 2016-2024 the original author or authors.
    *
    *    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
    *
    *       https://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.mybatis.dynamic.sql;
  
  import java.sql.JDBCType;
  import java.util.Objects;
  import java.util.Optional;
  import java.util.function.Supplier;
  
  import org.jetbrains.annotations.NotNull;
  
  public class SqlTable implements TableExpression {
  
      protected Supplier<String> nameSupplier;
  
      protected SqlTable(String tableName) {
          Objects.requireNonNull(tableName);
  
          this.nameSupplier = () -> tableName;
      }
  
      /**
       * Creates an SqlTable whose name can be changed at runtime.
       *
       * @param tableNameSupplier table name supplier
       * @deprecated please use {@link AliasableSqlTable} if you need to change the table name at runtime
       */
      @Deprecated
      protected SqlTable(Supplier<String> tableNameSupplier) {
          Objects.requireNonNull(tableNameSupplier);
  
          this.nameSupplier = tableNameSupplier;
      }
  
      /**
       * Creates an SqlTable whose name can be changed at runtime.
       *
       * @param schemaSupplier schema supplier
       * @param tableName table name
       * @deprecated please use {@link AliasableSqlTable} if you need to change the table name at runtime
       */
      @Deprecated
      protected SqlTable(Supplier<Optional<String>> schemaSupplier, String tableName) {
          this(Optional::empty, schemaSupplier, tableName);
      }
  
      /**
       * Creates an SqlTable whose name can be changed at runtime.
       *
       * @param catalogSupplier catalog supplier
       * @param schemaSupplier schema supplier
       * @param tableName table name
       * @deprecated please use {@link AliasableSqlTable} if you need to change the table name at runtime
       */
      @Deprecated
      protected SqlTable(Supplier<Optional<String>> catalogSupplier, Supplier<Optional<String>> schemaSupplier,
              String tableName) {
          Objects.requireNonNull(catalogSupplier);
          Objects.requireNonNull(schemaSupplier);
          Objects.requireNonNull(tableName);
  
          this.nameSupplier = () -> compose(catalogSupplier, schemaSupplier, tableName);
      }
  
      private String compose(Supplier<Optional<String>> catalogSupplier, Supplier<Optional<String>> schemaSupplier,
              String tableName) {
          return catalogSupplier.get().map(c -> compose(c, schemaSupplier, tableName))
                  .orElseGet(() -> compose(schemaSupplier, tableName));
      }
  
      private String compose(String catalog, Supplier<Optional<String>> schemaSupplier, String tableName) {
          return schemaSupplier.get().map(s -> composeCatalogSchemaAndTable(catalog, s, tableName))
                  .orElseGet(() -> composeCatalogAndTable(catalog, tableName));
      }
  
      private String compose(Supplier<Optional<String>> schemaSupplier, String tableName) {
          return schemaSupplier.get().map(s -> composeSchemaAndTable(s, tableName))
                  .orElse(tableName);
      }
  
      private String composeCatalogAndTable(String catalog, String tableName) {
          return catalog + ".." + tableName; //$NON-NLS-1$
      }
  
      private String composeSchemaAndTable(String schema, String tableName) {
          return schema + "." + tableName; //$NON-NLS-1$
     }
 
     private String composeCatalogSchemaAndTable(String catalog, String schema, String tableName) {
         return catalog + "." + schema + "." + tableName; //$NON-NLS-1$ //$NON-NLS-2$
     }
 
     public String tableNameAtRuntime() {
         return nameSupplier.get();
     }
 
     public BasicColumn allColumns() {
         return SqlColumn.of("*", this); //$NON-NLS-1$
     }
 
     @NotNull
     public <T> SqlColumn<T> column(String name) {
         return SqlColumn.of(name, this);
     }
 
     @NotNull
     public <T> SqlColumn<T> column(String name, JDBCType jdbcType) {
         return SqlColumn.of(name, this, jdbcType);
     }
 
     @NotNull
     public <T> SqlColumn<T> column(String name, JDBCType jdbcType, String typeHandler) {
         SqlColumn<T> column = SqlColumn.of(name, this, jdbcType);
         return column.withTypeHandler(typeHandler);
     }
 
     @Override
     public <R> R accept(TableExpressionVisitor<R> visitor) {
         return visitor.visit(this);
     }
 
     public Optional<String> tableAlias() {
         return Optional.empty();
     }
 
     public static SqlTable of(String name) {
         return new SqlTable(name);
     }
 }