/* Copyright (c) 2012-2014 Boundless and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Distribution License v1.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/org/documents/edl-v10.html
 *
 * Contributors:
 * Gabriel Roldan (Boundless) - initial implementation
 */
package org.locationtech.geogig.api.porcelain;

import java.util.Collections;
import java.util.List;

import org.locationtech.geogig.api.AbstractGeoGigOp;
import org.locationtech.geogig.api.Ref;
import org.locationtech.geogig.api.plumbing.ForEachRef;

import com.google.common.base.Predicate;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Lists;

/**
 * Creates a new head ref (branch) pointing to the specified tree-ish or the current HEAD if no
 * tree-ish was specified.
 * <p>
 * 
 * @TODO: support branch descriptions
 * @TODO: support setting up the branch to track a remote branch
 */
public class BranchListOp extends AbstractGeoGigOp<ImmutableList<Ref>> {

    private boolean remotes;

    private boolean locals;

    public BranchListOp() {
        locals = true;
        remotes = false;
    }

    public BranchListOp setRemotes(boolean remotes) {
        this.remotes = remotes;
        return this;
    }

    public BranchListOp setLocal(boolean locals) {
        this.locals = locals;
        return this;
    }

    protected  ImmutableList<Ref> _call() {

        final Predicate<Ref> filter = new Predicate<Ref>() {
            @Override
            public boolean apply(Ref input) {
                if (locals && input.getName().startsWith(Ref.HEADS_PREFIX)) {
                    return true;
                }
                if (remotes && input.getName().startsWith(Ref.REMOTES_PREFIX)) {
                    return true;
                }
                return false;
            }
        };

        List<Ref> refs = Lists.newArrayList(command(ForEachRef.class).setFilter(filter).call());
        Collections.sort(refs);
        return ImmutableList.copyOf(refs);
    }

}