WsURLStreamHandlerImpl.java

/**
 * Copyright 2007-2015, Kaazing Corporation. All rights reserved.
 *
 * 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 org.kaazing.netx.ws.internal.url;

import static org.kaazing.netx.ws.internal.util.Util.changeScheme;

import java.io.IOException;
import java.net.URI;
import java.net.URL;
import java.net.URLStreamHandler;
import java.util.Map;
import java.util.Random;

import org.kaazing.netx.URLConnectionHelper;
import org.kaazing.netx.ws.WsURLConnection;
import org.kaazing.netx.ws.internal.WebSocketExtensionFactory;
import org.kaazing.netx.ws.internal.WebSocketInputStateMachine;
import org.kaazing.netx.ws.internal.WebSocketOutputStateMachine;
import org.kaazing.netx.ws.internal.WsURLConnectionImpl;

final class WsURLStreamHandlerImpl extends URLStreamHandler {

    private final URLConnectionHelper helper;
    private final Map<String, String> supportedProtocols;
    private final WebSocketExtensionFactory extensionFactory;
    private final Random random;
    private final WebSocketInputStateMachine inputStateMachine;
    private final WebSocketOutputStateMachine outputStateMachine;

    public WsURLStreamHandlerImpl(
            URLConnectionHelper helper,
            Map<String, String> supportedProtocols,
            Random random,
            WebSocketExtensionFactory extensionFactory,
            WebSocketInputStateMachine inputStateMachine,
            WebSocketOutputStateMachine outputStateMachine) {
        this.helper = helper;
        this.supportedProtocols = supportedProtocols;
        this.extensionFactory = extensionFactory;
        this.random = random;
        this.inputStateMachine = inputStateMachine;
        this.outputStateMachine = outputStateMachine;
    }

    @Override
    protected int getDefaultPort() {
        return 80;
    }

    @Override
    protected WsURLConnection openConnection(URL location) throws IOException {
        URI locationURI = URI.create(location.toString());
        String scheme = locationURI.getScheme();
        String httpScheme = supportedProtocols.get(scheme);
        assert httpScheme != null;
        URI httpLocation = changeScheme(locationURI, httpScheme);

        return new WsURLConnectionImpl(
                helper,
                location,
                httpLocation,
                random,
                extensionFactory,
                inputStateMachine,
                outputStateMachine);
    }

}