com.predic8.membrane.core.interceptor.ntlm.HeaderNTLMRetriever Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of service-proxy-core Show documentation
There is a newer version: 5.6.0
/* Copyright 2019 predic8 GmbH, www.predic8.com

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

   http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License. */
package com.predic8.membrane.core.interceptor.ntlm;

import com.predic8.membrane.annot.MCAttribute;
import com.predic8.membrane.annot.MCElement;
import com.predic8.membrane.core.exchange.Exchange;

@MCElement(name = "headerRetriever", topLevel = false)
public class HeaderNTLMRetriever implements NTLMRetriever {

    String userHeaderName;
    String passwordHeaderName;
    String domainHeaderName;
    String workstationHeaderName;

    public HeaderNTLMRetriever() {
    }

    public HeaderNTLMRetriever(String userHeaderName, String passwordHeaderName, String domainHeaderName, String workstationHeaderName) {
        if(userHeaderName == null || userHeaderName.isEmpty())
            throw new RuntimeException("userHeaderName attribute cannot be null or empty");
        if(passwordHeaderName == null || passwordHeaderName.isEmpty())
            throw new RuntimeException("passwordHeaderName attribute cannot be null or empty");
        /*if(domainHeaderName == null || domainHeaderName.isEmpty())
            throw new RuntimeException("domainHeaderName attribute cannot be null or empty");
        if(workstationHeaderName == null || workstationHeaderName.isEmpty())
            throw new RuntimeException("workstationHeaderName attribute cannot be null or empty");*/

        this.userHeaderName = userHeaderName;
        this.passwordHeaderName = passwordHeaderName;
        this.domainHeaderName = domainHeaderName;
        this.workstationHeaderName = workstationHeaderName;
    }

    @Override
    public String fetchUsername(Exchange exc) {
        return exc.getRequest().getHeader().getFirstValue(userHeaderName);
    }

    @Override
    public String fetchPassword(Exchange exc) {
        return exc.getRequest().getHeader().getFirstValue(passwordHeaderName);
    }

    @Override
    public String fetchDomain(Exchange exc) {
        return exc.getRequest().getHeader().getFirstValue(domainHeaderName);
    }

    @Override
    public String fetchWorkstation(Exchange exc) {
        return exc.getRequest().getHeader().getFirstValue(workstationHeaderName);
    }

    public String getUserHeaderName() {
        return userHeaderName;
    }

    @MCAttribute(attributeName = "user")
    public void setUserHeaderName(String userHeaderName) {
        this.userHeaderName = userHeaderName;
    }

    public String getPasswordHeaderName() {
        return passwordHeaderName;
    }

    @MCAttribute(attributeName = "pass")
    public void setPasswordHeaderName(String passwordHeaderName) {
        this.passwordHeaderName = passwordHeaderName;
    }

    public String getDomainHeaderName() {
        return domainHeaderName;
    }

    @MCAttribute(attributeName = "domain")
    public void setDomainHeaderName(String domainHeaderName) {
        this.domainHeaderName = domainHeaderName;
    }

    public String getWorkstationHeaderName() {
        return workstationHeaderName;
    }

    @MCAttribute(attributeName = "workstation")
    public void setWorkstationHeaderName(String workstationHeaderName) {
        this.workstationHeaderName = workstationHeaderName;
    }
}