content: /*
   * 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.camel.component.digitalocean.producer;
  
  import com.myjeeva.digitalocean.exception.DigitalOceanException;
  import com.myjeeva.digitalocean.exception.RequestUnsuccessfulException;
  import com.myjeeva.digitalocean.pojo.Delete;
  import com.myjeeva.digitalocean.pojo.Snapshot;
  import com.myjeeva.digitalocean.pojo.Snapshots;
  import org.apache.camel.Exchange;
  import org.apache.camel.component.digitalocean.DigitalOceanConfiguration;
  import org.apache.camel.component.digitalocean.DigitalOceanEndpoint;
  import org.apache.camel.component.digitalocean.constants.DigitalOceanHeaders;
  import org.apache.camel.component.digitalocean.constants.DigitalOceanSnapshotTypes;
  import org.apache.camel.util.ObjectHelper;
  
  /**
   * The DigitalOcean producer for Snapshots API.
   */
  public class DigitalOceanSnapshotsProducer extends DigitalOceanProducer {
  
      public DigitalOceanSnapshotsProducer(DigitalOceanEndpoint endpoint, DigitalOceanConfiguration configuration) {
          super(endpoint, configuration);
      }
  
      @Override
      public void process(Exchange exchange) throws Exception {
  
          switch (determineOperation(exchange)) {
  
              case list:
                  getSnapshots(exchange);
                  break;
              case get:
                  getSnapshot(exchange);
                  break;
              case delete:
                  deleteSnapshot(exchange);
                  break;
  
              default:
                  throw new IllegalArgumentException("Unsupported operation");
          }
  
      }
  
      private void getSnapshots(Exchange exchange) throws RequestUnsuccessfulException, DigitalOceanException {
          DigitalOceanSnapshotTypes type = exchange.getIn().getHeader(DigitalOceanHeaders.TYPE, DigitalOceanSnapshotTypes.class);
          Snapshots snapshots;
  
          if (ObjectHelper.isNotEmpty(type)) {
              if (type == DigitalOceanSnapshotTypes.droplet) {
                  snapshots = getEndpoint().getDigitalOceanClient().getAllDropletSnapshots(configuration.getPage(),
                          configuration.getPerPage());
              } else {
                  snapshots = getEndpoint().getDigitalOceanClient().getAllVolumeSnapshots(configuration.getPage(),
                          configuration.getPerPage());
              }
          } else {
              snapshots = getEndpoint().getDigitalOceanClient().getAvailableSnapshots(configuration.getPage(),
                      configuration.getPerPage());
          }
  
          LOG.trace("All Snapshots : page {} / {} per page [{}] ", configuration.getPage(), configuration.getPerPage(),
                  snapshots.getSnapshots());
          exchange.getMessage().setBody(snapshots.getSnapshots());
      }
  
      private void getSnapshot(Exchange exchange) throws RequestUnsuccessfulException, DigitalOceanException {
          String snapshotId = exchange.getIn().getHeader(DigitalOceanHeaders.ID, String.class);
  
          if (ObjectHelper.isEmpty(snapshotId)) {
              throw new IllegalArgumentException(DigitalOceanHeaders.ID + " must be specified");
          }
  
          Snapshot snapshot = getEndpoint().getDigitalOceanClient().getSnaphotInfo(snapshotId);
          LOG.trace("Snapshot [{}] ", snapshot);
          exchange.getMessage().setBody(snapshot);
  
      }
  
      private void deleteSnapshot(Exchange exchange) throws RequestUnsuccessfulException, DigitalOceanException {
          String snapshotId = exchange.getIn().getHeader(DigitalOceanHeaders.ID, String.class);
  
          if (ObjectHelper.isEmpty(snapshotId)) {
              throw new IllegalArgumentException(DigitalOceanHeaders.ID + " must be specified");
          }
  
          Delete delete = getEndpoint().getDigitalOceanClient().deleteSnapshot(snapshotId);
          LOG.trace("Delete Snapshot [{}] ", delete);
          exchange.getMessage().setBody(delete);
  
      }
  
  }
  