001    package org.apache.maven.model.building;
002    
003    /*
004     * Licensed to the Apache Software Foundation (ASF) under one
005     * or more contributor license agreements.  See the NOTICE file
006     * distributed with this work for additional information
007     * regarding copyright ownership.  The ASF licenses this file
008     * to you under the Apache License, Version 2.0 (the
009     * "License"); you may not use this file except in compliance
010     * with the License.  You may obtain a copy of the License at
011     *
012     *   http://www.apache.org/licenses/LICENSE-2.0
013     *
014     * Unless required by applicable law or agreed to in writing,
015     * software distributed under the License is distributed on an
016     * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
017     * KIND, either express or implied.  See the License for the
018     * specific language governing permissions and limitations
019     * under the License.
020     */
021    
022    import java.io.File;
023    import java.io.FileInputStream;
024    import java.io.IOException;
025    import java.io.InputStream;
026    import java.net.URI;
027    
028    /**
029     * Wraps an ordinary {@link File} as a model source.
030     * 
031     * @author Benjamin Bentmann
032     */
033    public class FileModelSource
034        implements ModelSource2
035    {
036        private final File pomFile;
037    
038        /**
039         * Creates a new model source backed by the specified file.
040         * 
041         * @param pomFile The POM file, must not be {@code null}.
042         */
043        public FileModelSource( File pomFile )
044        {
045            if ( pomFile == null )
046            {
047                throw new IllegalArgumentException( "no POM file specified" );
048            }
049            this.pomFile = pomFile.getAbsoluteFile();
050        }
051    
052        public InputStream getInputStream()
053            throws IOException
054        {
055            return new FileInputStream( pomFile );
056        }
057    
058        public String getLocation()
059        {
060            return pomFile.getPath();
061        }
062    
063        /**
064         * Gets the POM file of this model source.
065         * 
066         * @return The underlying POM file, never {@code null}.
067         */
068        public File getPomFile()
069        {
070            return pomFile;
071        }
072    
073        @Override
074        public String toString()
075        {
076            return getLocation();
077        }
078    
079        public ModelSource2 getRelatedSource( String relPath )
080        {
081            relPath = relPath.replace( '\\', File.separatorChar ).replace( '/', File.separatorChar );
082    
083            File relatedPom = new File( pomFile.getParentFile(), relPath );
084    
085            if ( relatedPom.isDirectory() )
086            {
087                // TODO figure out how to reuse ModelLocator.locatePom(File) here
088                relatedPom = new File( relatedPom, "pom.xml" );
089            }
090    
091            if ( relatedPom.isFile() && relatedPom.canRead() )
092            {
093                return new FileModelSource( new File( relatedPom.toURI().normalize() ) );
094            }
095    
096            return null;
097        }
098    
099        public URI getLocationURI()
100        {
101            return pomFile.toURI();
102        }
103    }