org.addhen.birudo.data.net.JenkinsClient.java Source code

Introduction

Here is the source code for org.addhen.birudo.data.net.JenkinsClient.java

Source

/*
 * Copyright 2015 Henry Addo
 *
 * 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.addhen.birudo.data.net;

import com.google.gson.Gson;

import com.squareup.okhttp.Credentials;
import com.squareup.okhttp.FormEncodingBuilder;
import com.squareup.okhttp.Response;

import org.addhen.birudo.data.entity.CrumbInfoEntity;
import org.addhen.birudo.data.entity.JenkinsBuildInfoJsonEntity;
import org.addhen.birudo.data.entity.JenkinsUserEntity;
import org.addhen.birudo.data.entity.mapper.JenkinsBuildInfoEntityJsonMapper;
import org.addhen.birudo.core.entity.GcmRegistrationStatus;

import java.io.IOException;

public class JenkinsClient extends BaseHttpClient {

    JenkinsUserEntity mJenkinsUserEntity;

    public JenkinsClient(String url) {
        this(url, null);
    }

    public JenkinsClient(JenkinsUserEntity jenkinsUserEntity) {
        this(jenkinsUserEntity.getUrl(), jenkinsUserEntity);
    }

    public JenkinsClient(String url, JenkinsUserEntity jenkinsUserEntity) {
        super(url);
        mJenkinsUserEntity = jenkinsUserEntity;
    }

    private CrumbInfoEntity retrieveCrumbInfo() {
        if (mJenkinsUserEntity == null) {
            return null;
        }
        super.url = String.format("%scrumbIssuer/api/json", mJenkinsUserEntity.getUrl());
        try {
            setMethod(HttpMethod.POST);
            execute();
        } catch (Exception e) {
            log("failed to set request method.", e);
        }

        final int statuCode = getResponse().code();
        if (statuCode == 404) {
            return null;
        }

        final Response response = getResponse();
        final Gson gson = new Gson();
        String resp = "";
        try {
            resp = response.body().string();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return gson.fromJson(resp, CrumbInfoEntity.class);
    }

    public GcmRegistrationStatus sendGcmTokenToServer() throws Exception {
        if (mJenkinsUserEntity == null) {
            return null;
        }

        // Retrieve crumbInfo
        CrumbInfoEntity crumbInfoEntity = retrieveCrumbInfo();
        if (crumbInfoEntity != null && crumbInfoEntity.isCsrfEnabled()) {
            setHeader(crumbInfoEntity.getCrumbRequestField(), crumbInfoEntity.getCrumb());
        }

        FormEncodingBuilder formEncodingBuilder = new FormEncodingBuilder();

        formEncodingBuilder.add("token", mJenkinsUserEntity.getSenderId());

        String credential = Credentials.basic(mJenkinsUserEntity.getUsername(), mJenkinsUserEntity.getToken());

        setHeader("Authorization", credential);

        setRequestBody(formEncodingBuilder.build());

        // Send to gcm/regsiter
        super.url = String.format("%sgcm/register", mJenkinsUserEntity.getUrl());
        setMethod(HttpMethod.POST);
        execute();

        final int statusCode = getResponse().code();
        GcmRegistrationStatus status = new GcmRegistrationStatus();
        status.setStatusCode(statusCode);
        if (statusCode != 200) {
            log("Sending registering token failed with satus code %s", statusCode);
            status.setStatus(false);
        } else {
            status.setStatus(true);
        }

        return status;
    }

    public JenkinsBuildInfoJsonEntity fetchBuildInfo() throws Exception {
        // Send info request
        final String tmpUrl = super.url;
        url = String.format("%s/api/json", tmpUrl);
        setMethod(HttpMethod.GET);
        execute();

        JenkinsBuildInfoEntityJsonMapper jsonMapper = new JenkinsBuildInfoEntityJsonMapper();

        final String jsonReponse = getResponse().body().string();
        return jsonMapper.map(jsonReponse);

    }
}