source.dussan.org Git - archiva.git/blob

   1 package org.apache.maven.archiva.indexer.filecontent;
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements.  See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership.  The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License.  You may obtain a copy of the License at
  11  *
  12  *  http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied.  See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import org.apache.commons.lang.StringUtils;
  23 import org.apache.lucene.document.Document;
  24 import org.apache.maven.archiva.indexer.ArtifactKeys;
  25 import org.apache.maven.archiva.indexer.lucene.LuceneDocumentMaker;
  26 import org.apache.maven.archiva.indexer.lucene.LuceneEntryConverter;
  27 import org.apache.maven.archiva.indexer.lucene.LuceneRepositoryContentRecord;
  28 import org.apache.maven.archiva.model.ArchivaArtifact;
  29
  30 import java.text.ParseException;
  31
  32 /**
  33  * FileContentConverter
  34  *
  35  * @version $Id$
  36  */
  37 public class FileContentConverter
  38     implements LuceneEntryConverter
  39 {
  40     public Document convert( LuceneRepositoryContentRecord record )
  41     {
  42         if ( !( record instanceof FileContentRecord ) )
  43         {
  44             throw new ClassCastException( "Unable to convert type " + record.getClass().getName() + " to "
  45                 + FileContentRecord.class.getName() + "." );
  46         }
  47
  48         FileContentRecord filecontent = (FileContentRecord) record;
  49
  50         LuceneDocumentMaker doc = new LuceneDocumentMaker( filecontent );
  51
  52         if( filecontent.getArtifact() != null )
  53         {
  54             // Artifact Reference
  55             doc.addFieldTokenized( ArtifactKeys.GROUPID, filecontent.getArtifact().getGroupId() );
  56             doc.addFieldExact( ArtifactKeys.GROUPID_EXACT, filecontent.getArtifact().getGroupId() );
  57             doc.addFieldTokenized( ArtifactKeys.ARTIFACTID, filecontent.getArtifact().getArtifactId() );
  58             doc.addFieldExact( ArtifactKeys.ARTIFACTID_EXACT, filecontent.getArtifact().getArtifactId() );
  59             doc.addFieldTokenized( ArtifactKeys.VERSION, filecontent.getArtifact().getVersion() );
  60             doc.addFieldExact( ArtifactKeys.VERSION_EXACT, filecontent.getArtifact().getVersion() );
  61             doc.addFieldTokenized( ArtifactKeys.TYPE, filecontent.getArtifact().getType() );
  62             doc.addFieldUntokenized( ArtifactKeys.CLASSIFIER, filecontent.getArtifact().getClassifier() );
  63         }
  64
  65         doc.addFieldTokenized( FileContentKeys.FILENAME, filecontent.getFilename() );
  66
  67         return doc.getDocument();
  68     }
  69
  70     public LuceneRepositoryContentRecord convert( Document document )
  71         throws ParseException
  72     {
  73         FileContentRecord record = new FileContentRecord();
  74
  75         record.setRepositoryId( document.get( LuceneDocumentMaker.REPOSITORY_ID ) );
  76
  77         // Artifact Reference
  78         String groupId = document.get( ArtifactKeys.GROUPID );
  79         String artifactId = document.get( ArtifactKeys.ARTIFACTID );
  80         String version = document.get( ArtifactKeys.VERSION );
  81         String classifier = document.get( ArtifactKeys.CLASSIFIER );
  82         String type = document.get( ArtifactKeys.TYPE );
  83
  84         if( StringUtils.isNotBlank( groupId ) && StringUtils.isNotBlank( artifactId ) )
  85         {
  86             ArchivaArtifact artifact = new ArchivaArtifact( groupId, artifactId, version, classifier, type );
  87             record.setArtifact( artifact );
  88         }
  89
  90         // Filecontent Specifics
  91         record.setFilename( document.get( FileContentKeys.FILENAME ) );
  92
  93         return record;
  94     }
  95
  96 }