PatchSet.java

package com.chrome.codereview.model;

import android.os.Parcel;
import android.os.Parcelable;

import org.json.JSONException;
import org.json.JSONObject;

import java.text.ParseException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

/**
 * Created by sergeyv on 21/4/14.
 */
public class PatchSet implements Parcelable {

    public static final Parcelable.Creator<PatchSet> CREATOR = new Creator<PatchSet>() {

        public PatchSet createFromParcel(Parcel source) {
            String message = source.readString();
            List<PatchSetFile> files = new ArrayList<PatchSetFile>();
            source.readTypedList(files, PatchSetFile.CREATOR);
            int numComments = source.readInt();
            int patchSetId = source.readInt();
            return new PatchSet(message, files, numComments, patchSetId, new ArrayList<TryBotResult>());
        }

        @Override
        public PatchSet[] newArray(int size) {
            return new PatchSet[size];
        }

    };

    private final String message;
    private final List<PatchSetFile> files;
    private final int numComments;
    private final int patchSetId;
    private final List<TryBotResult> tryBotResults;
    private final Map<String, TryBotResult.Result> botToState = new HashMap<String, TryBotResult.Result>();

    public PatchSet(String message, List<PatchSetFile> files, int numComments, int patchSetId, List<TryBotResult> tryBotResults) {
        this.numComments = numComments;
        this.patchSetId = patchSetId;
        this.tryBotResults = Collections.unmodifiableList(tryBotResults);
        this.message = message != null ? message : "";
        this.files = files;
        prepareBotToState();
    }

    public static PatchSet from(int patchSetId, JSONObject jsonObject) throws JSONException, ParseException {
        JSONObject filesJsonObject = jsonObject.getJSONObject("files");
        List<PatchSetFile> files = new ArrayList<PatchSetFile>();
        for (Iterator iterator = filesJsonObject.keys(); iterator.hasNext(); ) {
            String file = (String) iterator.next();
            JSONObject meta = filesJsonObject.getJSONObject(file);
            files.add(PatchSetFile.from(file, meta));
        }
        String message = !jsonObject.isNull("message") ? jsonObject.getString("message") : null;
        int numComments = jsonObject.getInt("num_comments");
        List<TryBotResult> tryJobResults = TryBotResult.from(jsonObject.getJSONArray("try_job_results"));
        return new PatchSet(message, files, numComments, patchSetId, tryJobResults);
    }

    public String message() {
        return message;
    }

    public List<PatchSetFile> files() {
        return files;
    }

    public int linesAdded() {
        int result = 0;
        for (PatchSetFile file : files()) {
            result += file.numAdded();
        }
        return result;
    }

    public int linesRemoved() {
        int result = 0;
        for (PatchSetFile file : files()) {
            result += file.numRemoved();
        }
        return result;
    }

    public int numComments() {
        return numComments;
    }

    public int numDrafts() {
        int drafts = 0;
        for (PatchSetFile file : files()) {
            drafts += file.numberOfDrafts();
        }
        return drafts;
    }

    public int id() {
        return patchSetId;
    }

    public List<TryBotResult> tryBotResults() {
        return tryBotResults;
    }

    public Map<String, TryBotResult.Result> botToState() {
        return botToState;
    }

    public boolean hasTryBotsResults() {
        return !tryBotResults.isEmpty();
    }

    private void prepareBotToState() {
        ArrayList<TryBotResult> results = new ArrayList<TryBotResult>(tryBotResults);
        Collections.sort(results, new Comparator<TryBotResult>() {
            @Override
            public int compare(TryBotResult lhs, TryBotResult rhs) {
                return lhs.result().order() - rhs.result().order();
            }
        });

        for (TryBotResult result : results) {
            botToState.put(result.builder(), result.result());
        }
    }

    @Override
    public int describeContents() {
        return 0;
    }

    @Override
    public void writeToParcel(Parcel dest, int flags) {
        dest.writeString(message);
        dest.writeTypedList(files);
        dest.writeInt(numComments);
        dest.writeInt(patchSetId);
    }
}