/*************************************************************************
 * (c) Copyright 2017 Hewlett Packard Enterprise Development Company LP
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 3 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see http://www.gnu.org/licenses/.
 ************************************************************************/
package com.eucalyptus.cloud;

import java.util.Map;
import javax.annotation.Nullable;
import org.apache.log4j.Logger;
import com.eucalyptus.auth.principal.OwnerFullName;
import com.eucalyptus.cloud.run.Allocations;
import com.eucalyptus.cluster.Clusters;
import com.eucalyptus.cluster.common.internal.ResourceToken;
import com.eucalyptus.cluster.common.ClusterController;
import com.eucalyptus.component.Topology;
import com.eucalyptus.compute.common.CloudMetadata;
import com.eucalyptus.compute.common.CloudMetadatas;
import com.eucalyptus.compute.common.internal.blockstorage.Volume;
import com.eucalyptus.compute.common.internal.vm.VmInstance;
import com.eucalyptus.compute.common.network.Networking;
import com.eucalyptus.compute.common.network.ReleaseNetworkResourcesType;
import com.eucalyptus.vm.VmInstances;
import com.eucalyptus.vmtypes.VmTypes;
import com.google.common.collect.Maps;

/**
 *
 */
public class VmInstanceToken extends ResourceToken implements CloudMetadata.VmInstanceMetadata {
  private static Logger LOG    = Logger.getLogger( ResourceToken.class );
  private final Allocations.Allocation allocation;
  private final Integer       launchIndex;
  private final String        instanceId;
  private final String        instanceUuid;
  @Nullable
  private Volume rootVolume;
  @Nullable
  private Map<String, Volume> ebsVolumes;
  @Nullable
  private Map<String, String> ephemeralDisks;
  @Nullable
  private VmInstance vmInst;
  private boolean             aborted;
  private boolean             zombie;

  public VmInstanceToken( final Allocations.Allocation allocInfo, final int launchIndex ) {
    super(
        Clusters.lookupAny( Topology.lookup( ClusterController.class, allocInfo.getPartition( ) ) ),
        allocInfo.getVmType( ),
        VmTypes.isUnorderedType( allocInfo.getVmType( ) ),
        allocInfo.getInstanceId( launchIndex )
    );
    this.allocation = allocInfo;
    this.launchIndex = launchIndex;
    this.instanceId = allocInfo.getInstanceId( launchIndex );
    this.instanceUuid = allocInfo.getInstanceUuid( launchIndex );
    if ( this.instanceId == null || this.instanceUuid == null ) {
      throw new IllegalArgumentException( "Cannot create resource token with null instance id or uuid: " + allocInfo );
    }
  }

  public Allocations.Allocation getAllocationInfo( ) {
    return this.allocation;
  }

  public String getInstanceId( ) {
    return this.instanceId;
  }

  public Integer getAmount( ) {
    return 1;
  }

  public String getInstanceUuid( ) {
    return this.instanceUuid;
  }

  public Integer getLaunchIndex( ) {
    return this.launchIndex;
  }

  public void abort( ) {
    if ( aborted ) return;
    aborted = true;

    LOG.debug( this );
    if ( isPending( ) ) { // release unused resources
      try {
        this.release( );
      } catch ( final Exception ex ) {
        LOG.error( ex, ex );
      }

      try {
        final ReleaseNetworkResourcesType releaseNetworkResourcesType = new ReleaseNetworkResourcesType( );
        releaseNetworkResourcesType.setVpc( allocation.getSubnet( ) == null ?
            null :
            CloudMetadatas.toDisplayName( ).apply( allocation.getSubnet( ).getVpc( ) ) );
        releaseNetworkResourcesType.getResources( ).addAll( getAttribute( VmInstanceLifecycleHelpers.NetworkResourceVmInstanceLifecycleHelper.NetworkResourcesKey ) );
        Networking.getInstance( ).release( releaseNetworkResourcesType );
      } catch ( final Exception ex ) {
        LOG.error( ex, ex );
      }

      if ( this.vmInst != null ) {
        try {
          VmInstances.terminated( this.vmInst );
        } catch ( Exception ex ) {
          LOG.error( ex, ex );
        }
      }
    } else { // redeem and release later (if not used)
      try {
        this.redeem( );
      } catch ( final Exception ex ) {
        LOG.error( ex, ex );
      }
    }
  }

  static Logger getLOG( ) {
    return LOG;
  }

  Allocations.Allocation getAllocation( ) {
    return this.allocation;
  }

  public void setVmInstance( VmInstance vmInst ) {
    this.vmInst = vmInst;
  }

  public VmInstance getVmInstance( ) {
    return this.vmInst;
  }

  @Override
  public String getDisplayName( ) {
    return this.getInstanceId( );
  }

  public boolean isCommitted( ) {
    return this.allocation.isCommitted( );
  }

  @Override
  public OwnerFullName getOwner( ) {
    return this.allocation.getOwnerFullName( );
  }

  public Volume getRootVolume( ) {
    return this.rootVolume;
  }

  public void setRootVolume( Volume rootVolume ) {
    this.rootVolume = rootVolume;
  }

  public Map<String, Volume> getEbsVolumes() {
    if (this.ebsVolumes == null) {
      this.ebsVolumes = Maps.newHashMap();
    }
    return ebsVolumes;
  }

  public void setEbsVolumes(Map<String, Volume> ebsVolumes) {
    this.ebsVolumes = ebsVolumes;
  }

  public Map<String, String> getEphemeralDisks() {
    if (this.ephemeralDisks == null){
      this.ephemeralDisks = Maps.newHashMap();
    }
    return ephemeralDisks;
  }

  public void setEphemeralDisks(Map<String, String> ephemeralDisks) {
    this.ephemeralDisks = ephemeralDisks;
  }

  public boolean isZombie() {
    return zombie;
  }

  public void setZombie( final boolean zombie ) {
    this.zombie = zombie;
  }
}