/*
    * 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.
   */
  
  package org.apache.shardingsphere.sharding.rewrite.token.pojo;
  
  import org.apache.shardingsphere.infra.datanode.DataNode;
  import org.apache.shardingsphere.infra.rewrite.sql.token.pojo.RouteUnitAware;
  import org.apache.shardingsphere.infra.rewrite.sql.token.pojo.generic.InsertValue;
  import org.apache.shardingsphere.infra.rewrite.sql.token.pojo.generic.InsertValuesToken;
  import org.apache.shardingsphere.infra.route.context.RouteUnit;
  
  /**
   * Insert values token for sharding.
   */
  public final class ShardingInsertValuesToken extends InsertValuesToken implements RouteUnitAware {
      
      public ShardingInsertValuesToken(final int startIndex, final int stopIndex) {
          super(startIndex, stopIndex);
      }
      
      @Override
      public String toString(final RouteUnit routeUnit) {
          StringBuilder result = new StringBuilder();
          appendInsertValue(routeUnit, result);
          result.delete(result.length() - 2, result.length());
          return result.toString();
      }
      
      @Override
      public String toString() {
          return toString(null);
      }
      
      private void appendInsertValue(final RouteUnit routeUnit, final StringBuilder stringBuilder) {
          for (InsertValue each : getInsertValues()) {
              if (isAppend(routeUnit, (ShardingInsertValue) each)) {
                  stringBuilder.append(each).append(", ");
              }
          }
      }
      
      private boolean isAppend(final RouteUnit routeUnit, final ShardingInsertValue insertValueToken) {
          if (insertValueToken.getDataNodes().isEmpty() || null == routeUnit) {
              return true;
          }
          for (DataNode each : insertValueToken.getDataNodes()) {
              if (routeUnit.findTableMapper(each.getDataSourceName(), each.getTableName()).isPresent()) {
                  return true;
              }
          }
          return false;
      }
  }