package org.apache.maven.archiva.database.jdo;
   
   /*
    * 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.
   */
  
  import org.apache.maven.archiva.database.ArchivaDatabaseException;
  import org.apache.maven.archiva.database.ArtifactDAO;
  import org.apache.maven.archiva.database.Constraint;
  import org.apache.maven.archiva.database.DeclarativeConstraint;
  import org.apache.maven.archiva.database.ObjectNotFoundException;
  import org.apache.maven.archiva.model.ArchivaArtifact;
  import org.apache.maven.archiva.model.ArchivaArtifactModel;
  import org.apache.maven.archiva.model.jpox.ArchivaArtifactModelKey;
  
  import java.util.ArrayList;
  import java.util.List;
  
  /**
   * JdoArtifactDAO 
   *
   * @version $Id: JdoArtifactDAO.java 1043850 2010-12-09 07:58:00Z brett $
   * 
   * @plexus.component role-hint="jdo"
   */
  public class JdoArtifactDAO
      implements ArtifactDAO
  {
      /**
       * @plexus.requirement role-hint="archiva"
       */
      private JdoAccess jdo;
  
      /* .\ Archiva Artifact \. _____________________________________________________________ */
  
      public ArchivaArtifact createArtifact( String groupId, String artifactId, String version, String classifier,
                                             String type, String repositoryId )
      {
          ArchivaArtifact artifact;
  
          try
          {
              artifact = getArtifact( groupId, artifactId, version, classifier, type, repositoryId );
          }
          catch ( ArchivaDatabaseException e )
          {
              artifact = new ArchivaArtifact( groupId, artifactId, version, classifier, type, repositoryId );
          }
  
          return artifact;
      }
  
      public ArchivaArtifact getArtifact( String groupId, String artifactId, String version, String classifier,
                                          String type, String repositoryId )
          throws ObjectNotFoundException, ArchivaDatabaseException
      {
          ArchivaArtifactModelKey key = new ArchivaArtifactModelKey();
          key.setGroupId( groupId );
          key.setArtifactId( artifactId );
          key.setVersion( version );
          key.setClassifier( classifier );
          key.setType( type );
          key.setRepositoryId( repositoryId );
  
          ArchivaArtifactModel model = (ArchivaArtifactModel) jdo.getObjectById( ArchivaArtifactModel.class, key, null );
  
          return new ArchivaArtifact( model );
      }
  
      @SuppressWarnings("unchecked")
      public List<ArchivaArtifact> queryArtifacts( Constraint constraint )
          throws ObjectNotFoundException, ArchivaDatabaseException
      {
          List<ArchivaArtifactModel> results = (List<ArchivaArtifactModel>) jdo.queryObjects( ArchivaArtifactModel.class, constraint );
  
          List<ArchivaArtifact> ret = null;
          if ( results != null )
          {
              ret = new ArrayList<ArchivaArtifact>();
              for ( ArchivaArtifactModel model : results )
              {
                  ret.add( new ArchivaArtifact( model ) );
              }
          }
 
         return ret;
     }
 
     public ArchivaArtifact saveArtifact( ArchivaArtifact artifact )
         throws ArchivaDatabaseException
     {
         ArchivaArtifactModel model = (ArchivaArtifactModel) jdo.saveObject( artifact.getModel() );
         if ( model == null )
         {
             return null;
         }
 
         return new ArchivaArtifact( model );
     }
 
     public void deleteArtifact( ArchivaArtifact artifact )
         throws ArchivaDatabaseException
     {
         jdo.removeObject( artifact.getModel() );
     }
 
     public long countArtifacts( DeclarativeConstraint constraint )
     {
         return jdo.countObjects( ArchivaArtifactModel.class, constraint );
     }
 }