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}