001/*
002    Licensed to the Apache Software Foundation (ASF) under one
003    or more contributor license agreements.  See the NOTICE file
004    distributed with this work for additional information
005    regarding copyright ownership.  The ASF licenses this file
006    to you under the Apache License, Version 2.0 (the
007    "License"); you may not use this file except in compliance
008    with the License.  You may obtain a copy of the License at
009
010       http://www.apache.org/licenses/LICENSE-2.0
011
012    Unless required by applicable law or agreed to in writing,
013    software distributed under the License is distributed on an
014    "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
015    KIND, either express or implied.  See the License for the
016    specific language governing permissions and limitations
017    under the License.
018 */
019package org.apache.wiki.markdown.extensions.jspwikilinks.attributeprovider;
020
021import org.apache.wiki.WikiContext;
022import org.apache.wiki.markdown.nodes.JSPWikiLink;
023import org.apache.wiki.parser.LinkParsingOperations;
024import org.apache.wiki.parser.MarkupParser;
025
026import com.vladsch.flexmark.util.html.Attributes;
027import com.vladsch.flexmark.util.sequence.CharSubSequence;
028
029
030/**
031 * {@link NodeAttributeProviderState} which sets the attributes for local links.
032 */
033public class LocalLinkAttributeProviderState implements NodeAttributeProviderState< JSPWikiLink > {
034
035    private final boolean hasRef;
036    private final WikiContext wikiContext;
037    private final LinkParsingOperations linkOperations;
038
039    public LocalLinkAttributeProviderState( final WikiContext wikiContext, final boolean hasRef ) {
040        this.hasRef = hasRef;
041        this.wikiContext = wikiContext;
042        this.linkOperations = new LinkParsingOperations( wikiContext );
043    }
044
045    /**
046     * {@inheritDoc}
047     *
048     * @see NodeAttributeProviderState#setAttributes(Attributes, JSPWikiLink)
049     */
050    @Override
051    public void setAttributes( final Attributes attributes, final JSPWikiLink link ) {
052        final int hashMark = link.getUrl().toString().indexOf( '#' );
053        final String attachment = wikiContext.getEngine().getAttachmentManager().getAttachmentInfoName( wikiContext, link.getWikiLink() );
054        if( attachment != null ) {
055            if( !linkOperations.isImageLink( link.getUrl().toString() ) ) {
056                attributes.replaceValue( "class", MarkupParser.CLASS_ATTACHMENT );
057                final String attlink = wikiContext.getURL( WikiContext.ATTACH, link.getWikiLink() );
058                attributes.replaceValue( "href", attlink );
059            } else {
060                new ImageLinkAttributeProviderState( wikiContext, attachment, hasRef ).setAttributes( attributes, link );
061            }
062        } else if( hashMark != -1 ) { // It's an internal Wiki link, but to a named section
063            final String namedSection = link.getUrl().toString().substring( hashMark + 1 );
064            final String matchedLink = linkOperations.linkIfExists( link.getUrl().toString() );
065            if( matchedLink != null ) {
066                String sectref = "#section-" + wikiContext.getEngine().encodeName( matchedLink + "-" + MarkupParser.wikifyLink( namedSection ) );
067                sectref = sectref.replace('%', '_');
068                link.setUrl( CharSubSequence.of( link.getUrl().toString() + sectref ) );
069                new LocalReadLinkAttributeProviderState( wikiContext ).setAttributes( attributes, link );
070            } else {
071                new LocalEditLinkAttributeProviderState( wikiContext, link.getWikiLink() ).setAttributes( attributes, link );
072            }
073        } else {
074            if( linkOperations.linkExists( link.getWikiLink() ) ) {
075                new LocalReadLinkAttributeProviderState( wikiContext ).setAttributes( attributes, link );
076            } else {
077                new LocalEditLinkAttributeProviderState( wikiContext, link.getWikiLink() ).setAttributes( attributes, link );
078            }
079        }
080    }
081
082}