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.postprocessor;
020
021import com.vladsch.flexmark.util.ast.Node;
022import com.vladsch.flexmark.util.ast.NodeTracker;
023import com.vladsch.flexmark.util.sequence.CharSubSequence;
024import org.apache.oro.text.regex.Pattern;
025import org.apache.wiki.api.core.Context;
026import org.apache.wiki.api.core.ContextEnum;
027import org.apache.wiki.attachment.AttachmentManager;
028import org.apache.wiki.markdown.nodes.JSPWikiLink;
029import org.apache.wiki.parser.LinkParsingOperations;
030import org.apache.wiki.parser.MarkupParser;
031
032import java.util.List;
033
034
035/**
036 * {@link NodePostProcessorState} which further post processes local links.
037 */
038public class LocalLinkNodePostProcessorState implements NodePostProcessorState< JSPWikiLink > {
039
040    private final Context wikiContext;
041    private final LinkParsingOperations linkOperations;
042    private final boolean isImageInlining;
043    private final List< Pattern > inlineImagePatterns;
044
045    public LocalLinkNodePostProcessorState( final Context wikiContext,
046                                            final boolean isImageInlining,
047                                            final List< Pattern > inlineImagePatterns ) {
048        this.wikiContext = wikiContext;
049        this.linkOperations = new LinkParsingOperations( wikiContext );
050        this.isImageInlining = isImageInlining;
051        this.inlineImagePatterns = inlineImagePatterns;
052    }
053
054    /**
055     * {@inheritDoc}
056     *
057     * @see NodePostProcessorState#process(NodeTracker, Node) 
058     */
059    @Override
060    public void process( final NodeTracker state, final JSPWikiLink link ) {
061        final int hashMark = link.getUrl().toString().indexOf( '#' );
062        final String attachment = wikiContext.getEngine().getManager( AttachmentManager.class ).getAttachmentInfoName( wikiContext, link.getUrl().toString() );
063        if( attachment != null  ) {
064            if( !linkOperations.isImageLink( link.getUrl().toString(), isImageInlining, inlineImagePatterns ) ) {
065                final String attlink = wikiContext.getURL( ContextEnum.PAGE_ATTACH.getRequestContext(), link.getUrl().toString() );
066                link.setUrl( CharSubSequence.of( attlink ) );
067                link.removeChildren();
068                final WikiHtmlInline content = WikiHtmlInline.of( link.getText().toString(), wikiContext );
069                link.appendChild( content );
070                state.nodeAddedWithChildren( content );
071                addAttachmentLink( state, link );
072            } else {
073                new ImageLinkNodePostProcessorState( wikiContext, attachment, link.hasRef() ).process( state, link );
074            }
075        } else if( hashMark != -1 ) { // It's an internal Wiki link, but to a named section
076            final String namedSection = link.getUrl().toString().substring( hashMark + 1 );
077            link.setUrl( CharSubSequence.of( link.getUrl().toString().substring( 0, hashMark ) ) );
078            final String matchedLink = linkOperations.linkIfExists( link.getUrl().toString() );
079            if( matchedLink != null ) {
080                String sectref = "#section-" + wikiContext.getEngine().encodeName( matchedLink + "-" + MarkupParser.wikifyLink( namedSection ) );
081                sectref = sectref.replace( '%', '_' );
082                link.setUrl( CharSubSequence.of( wikiContext.getURL( ContextEnum.PAGE_VIEW.getRequestContext(), link.getUrl().toString() + sectref ) ) );
083            } else {
084                link.setUrl( CharSubSequence.of( wikiContext.getURL( ContextEnum.PAGE_EDIT.getRequestContext(), link.getUrl().toString() ) ) );
085            }
086        } else {
087            if( linkOperations.linkExists( link.getUrl().toString() ) ) {
088                link.setUrl( CharSubSequence.of( wikiContext.getURL( ContextEnum.PAGE_VIEW.getRequestContext(), link.getUrl().toString() ) ) );
089            } else {
090                link.setUrl( CharSubSequence.of( wikiContext.getURL( ContextEnum.PAGE_EDIT.getRequestContext(), link.getUrl().toString() ) ) );
091            }
092        }
093    }
094
095    void addAttachmentLink( final NodeTracker state, final JSPWikiLink link ) {
096        final String infolink = wikiContext.getURL( ContextEnum.PAGE_INFO.getRequestContext(), link.getWikiLink() );
097        final String imglink = wikiContext.getURL( ContextEnum.PAGE_NONE.getRequestContext(), "images/attachment_small.png" );
098        final WikiHtmlInline aimg = WikiHtmlInline.of( "<a href=\""+ infolink + "\" class=\"infolink\">" +
099                                                              "<img src=\""+ imglink + "\" border=\"0\" alt=\"(info)\" />" +
100                                                           "</a>" ) ;
101        link.insertAfter( aimg );
102        state.nodeAdded( aimg );
103    }
104
105}