gerrit-server/src/main/java/com/google/gerrit/server/query/change/LabelPredicate.java

// Copyright (C) 2010 The Android Open Source Project
//
// 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 com.google.gerrit.server.query.change;

import com.google.common.collect.Lists;
import com.google.gerrit.reviewdb.client.Account;
import com.google.gerrit.reviewdb.client.AccountGroup;
import com.google.gerrit.reviewdb.server.ReviewDb;
import com.google.gerrit.server.IdentifiedUser;
import com.google.gerrit.server.project.ChangeControl;
import com.google.gerrit.server.project.ProjectCache;
import com.google.gerrit.server.query.OrPredicate;
import com.google.gerrit.server.query.Predicate;
import com.google.gerrit.server.util.LabelVote;
import com.google.inject.Provider;

import java.util.List;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class LabelPredicate extends OrPredicate<ChangeData> {
  private static final int MAX_LABEL_VALUE = 4;

  private static enum Test {
    EQ, GT_EQ, LT_EQ;

    boolean isEq() {
      return EQ.equals(this);
    }

    boolean isGtEq() {
      return GT_EQ.equals(this);
    }

    static Test op(String op) {
      if ("=".equals(op)) {
        return EQ;

      } else if (">=".equals(op)) {
        return GT_EQ;

      } else if ("<=".equals(op)) {
        return LT_EQ;

      } else {
        throw new IllegalArgumentException("Unsupported operation " + op);
      }
    }
  }

  private final String value;

  LabelPredicate(ProjectCache projectCache,
      ChangeControl.GenericFactory ccFactory,
      IdentifiedUser.GenericFactory userFactory, Provider<ReviewDb> dbProvider,
      String value, Set<Account.Id> accounts, AccountGroup.UUID group) {
    super(predicates(projectCache, ccFactory, userFactory, dbProvider, value,
        accounts, group));
    this.value = value;
  }

  private static List<Predicate<ChangeData>> predicates(
      ProjectCache projectCache, ChangeControl.GenericFactory ccFactory,
      IdentifiedUser.GenericFactory userFactory, Provider<ReviewDb> dbProvider,
      String value, Set<Account.Id> accounts, AccountGroup.UUID group) {
    String label = null;
    Test test = null;
    int expVal = 0;

    try {
      LabelVote v = LabelVote.parse(value);
      test = Test.EQ;
      label = v.getLabel();
      expVal = v.getValue();
    } catch (IllegalArgumentException e) {
      // Try next format.
    }

    try {
      LabelVote v = LabelVote.parseWithEquals(value);
      test = Test.EQ;
      label = v.getLabel();
      expVal = v.getValue();
    } catch (IllegalArgumentException e) {
      // Try next format.
    }

    if (label == null) {
      Matcher m = Pattern.compile("(>=|<=)([+-]?\\d+)$").matcher(value);
      if (m.find()) {
        label = value.substring(0, m.start());
        test = Test.op(m.group(1));
        expVal = value(m.group(2));
      } else {
        label = value;
        test = Test.EQ;
        expVal = 1;
      }
    }

    List<Predicate<ChangeData>> r = Lists.newArrayListWithCapacity(2 * MAX_LABEL_VALUE);
    if (test.isEq()) {
      if (expVal != 0) {
        r.add(equalsLabelPredicate(projectCache, ccFactory, userFactory,
            dbProvider, label, expVal, accounts, group));
      } else {
        r.add(noLabelQuery(projectCache, ccFactory, userFactory,
            dbProvider, label, accounts, group));
      }
    } else {
      for (int i = test.isGtEq() ? expVal : neg(expVal); i <= MAX_LABEL_VALUE; i++) {
        if (i != 0) {
          r.add(equalsLabelPredicate(projectCache, ccFactory, userFactory,
              dbProvider, label, test.isGtEq() ? i : neg(i), accounts, group));
        } else {
          r.add(noLabelQuery(projectCache, ccFactory, userFactory,
              dbProvider, label, accounts, group));
        }
      }
    }
    return r;
  }

  private static int value(String value) {
    if (value.startsWith("+")) {
      value = value.substring(1);
    }
    return Integer.parseInt(value);
  }

  private static int neg(int value) {
    return -1 * value;
  }

  private static Predicate<ChangeData> noLabelQuery(ProjectCache projectCache,
      ChangeControl.GenericFactory ccFactory,
      IdentifiedUser.GenericFactory userFactory, Provider<ReviewDb> dbProvider,
      String label, Set<Account.Id> accounts, AccountGroup.UUID group) {
    List<Predicate<ChangeData>> r =
        Lists.newArrayListWithCapacity(2 * MAX_LABEL_VALUE);
    for (int i = 1; i <= MAX_LABEL_VALUE; i++) {
      r.add(not(equalsLabelPredicate(projectCache, ccFactory, userFactory,
          dbProvider, label, i, accounts, group)));
      r.add(not(equalsLabelPredicate(projectCache, ccFactory, userFactory,
          dbProvider, label, neg(i), accounts, group)));
    }
    return and(r);
  }

  private static Predicate<ChangeData> equalsLabelPredicate(
      ProjectCache projectCache, ChangeControl.GenericFactory ccFactory,
      IdentifiedUser.GenericFactory userFactory, Provider<ReviewDb> dbProvider,
      String label, int expVal, Set<Account.Id> accounts,
      AccountGroup.UUID group) {
    if (accounts == null || accounts.isEmpty()) {
      return new EqualsLabelPredicate(projectCache, ccFactory, userFactory,
          dbProvider, label, expVal, null, group);
    } else {
      List<Predicate<ChangeData>> r = Lists.newArrayList();
      for (Account.Id a : accounts) {
        r.add(new EqualsLabelPredicate(projectCache, ccFactory, userFactory,
            dbProvider, label, expVal, a, group));
      }
      return or(r);
    }
  }

  @Override
  public String toString() {
    return ChangeQueryBuilder.FIELD_LABEL + ":" + value;
  }
}