package ru.sape; import com.juick.www.ad.models.Site; import org.apache.commons.lang3.StringUtils; import java.net.URI; import java.util.Collections; import java.util.List; import java.util.stream.Collectors; public class SapePageLinks { private final boolean showCode; private final String linkDelimiter; private final List pageLinks; public SapePageLinks(Site site, String sapeUser, URI request, String sapeCookie, boolean showPageCode) { String req = StringUtils.isNotEmpty(request.getQuery()) ? request.getPath() + "?" + request.getQuery() : request.getPath(); linkDelimiter = site.getDelimiter(); this.showCode = showPageCode || sapeUser.equals(sapeCookie); if (showCode) { pageLinks = Collections.singletonList(site.getCode()); } else { pageLinks = site.getPages().stream().filter(page -> page.getUri().equals(req)) .flatMap(page -> page.getLinks().stream()).collect(Collectors.toList()); } } public String render() { StringBuilder s = new StringBuilder(); s.append(String.join(linkDelimiter, pageLinks)); if (showCode) { s.insert(0, ""); s.append(""); } return s.toString(); } }