stack/rest/src/test/java/org/apache/usergrid/rest/test/resource2point0/RestClient.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.usergrid.rest.test.resource2point0;

import org.apache.usergrid.rest.test.resource2point0.endpoints.*;
import org.apache.usergrid.rest.test.resource2point0.endpoints.mgmt.ManagementResource;
import org.apache.usergrid.rest.test.resource2point0.state.ClientContext;

import com.fasterxml.jackson.databind.JsonNode;
import com.sun.jersey.api.client.Client;
import com.sun.jersey.api.client.WebResource;
import com.sun.jersey.api.client.config.ClientConfig;
import com.sun.jersey.api.client.config.DefaultClientConfig;
import com.sun.jersey.api.client.filter.HTTPBasicAuthFilter;

import javax.ws.rs.core.MediaType;
import java.util.UUID;


/**
 * This REST client was made to be able to send calls to any backend system that accepts calls. To do this It needs to
 * work independently of the existing REST test framework.
 */
public class RestClient implements UrlResource {

    private final String serverUrl;
    private ClientContext context;

    public WebResource resource;
    ClientConfig config = new DefaultClientConfig();
    Client client = Client.create( config );

    /**
     *
     * @param serverUrl
     */
    public RestClient( final String serverUrl ) {
        this.serverUrl = serverUrl;
        this.context = new ClientContext();
        resource = client.resource( serverUrl );
        resource.path( serverUrl );
    }


    /**
     * TODO: should this method return the base path or the total path we have built?
     */
    @Override
    public String getPath() {
        return resource.getURI().toString();
    }

    @Override
    public WebResource getResource() {
        return client.resource( serverUrl );
    }

    public ClientContext getContext() {
        return context;
    }

    public SystemResource system() {
        return new SystemResource(context, this);
    }

    public TestPropertiesResource testPropertiesResource() {
        return new TestPropertiesResource( context, this );
    }
    /**
     * Get the management resource
     */
    public ManagementResource management() {
        return new ManagementResource( context, this );
    }


    /**
     * Get hte organization resource
     */
    public OrganizationResource org( final String orgName ) {
        return new OrganizationResource( orgName, context, this );
    }

    public TokenResource token(){
        return new TokenResource(context,this);
    }
    public void refreshIndex(String orgname, String appName, String appid) {
        //TODO: add error checking and logging
        this.getResource().path( "/refreshindex" )
                .queryParam( "org_name", orgname )
                .queryParam( "app_name",appName )
                .queryParam("app_id", appid)
            .accept( MediaType.APPLICATION_JSON ).post();
    }

    public NamedResource pathResource(String path){
        return new NamedResource(path,context,this);
    }

    public void superuserSetup() {
        //TODO: change this when we upgrade to new version of jersey
        HTTPBasicAuthFilter httpBasicAuthFilter = new HTTPBasicAuthFilter( "superuser","superpassword" );
        client.addFilter( httpBasicAuthFilter );

        this.getResource().path( "system/superuser/setup" )
            .accept( MediaType.APPLICATION_JSON ).type( MediaType.APPLICATION_JSON ).get( JsonNode.class );
        client.removeFilter( httpBasicAuthFilter );
    }

    //todo:fix this method for the client.
//    public void loginAdminUser( final String username, final String password ) {
//        //Post isn't implemented yet, but using the method below we should be able to get a superuser password as well.
//        //final String token = management().token().post(username, password);
//
//        //context.setToken( token );
//    }

//
//    public void createOrgandOwner


}