Line data    Source code

       1             : // Copyright (C) 2017 The Android Open Source Project
       2             : //
       3             : // Licensed under the Apache License, Version 2.0 (the "License");
       4             : // you may not use this file except in compliance with the License.
       5             : // You may obtain a copy of the License at
       6             : //
       7             : // http://www.apache.org/licenses/LICENSE-2.0
       8             : //
       9             : // Unless required by applicable law or agreed to in writing, software
      10             : // distributed under the License is distributed on an "AS IS" BASIS,
      11             : // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      12             : // See the License for the specific language governing permissions and
      13             : // limitations under the License.
      14             : 
      15             : package com.google.gerrit.server.query.group;
      16             : 
      17             : import com.google.common.flogger.FluentLogger;
      18             : import com.google.gerrit.entities.InternalGroup;
      19             : import com.google.gerrit.exceptions.NoSuchGroupException;
      20             : import com.google.gerrit.index.query.IsVisibleToPredicate;
      21             : import com.google.gerrit.server.CurrentUser;
      22             : import com.google.gerrit.server.account.GroupControl;
      23             : import com.google.gerrit.server.index.IndexUtils;
      24             : import com.google.gerrit.server.query.account.AccountQueryBuilder;
      25             : 
      26             : public class GroupIsVisibleToPredicate extends IsVisibleToPredicate<InternalGroup> {
      27          12 :   private static final FluentLogger logger = FluentLogger.forEnclosingClass();
      28             : 
      29             :   protected final GroupControl.GenericFactory groupControlFactory;
      30             :   protected final CurrentUser user;
      31             : 
      32             :   public GroupIsVisibleToPredicate(
      33             :       GroupControl.GenericFactory groupControlFactory, CurrentUser user) {
      34          12 :     super(AccountQueryBuilder.FIELD_VISIBLETO, IndexUtils.describe(user));
      35          12 :     this.groupControlFactory = groupControlFactory;
      36          12 :     this.user = user;
      37          12 :   }
      38             : 
      39             :   @Override
      40             :   public boolean match(InternalGroup group) {
      41             :     try {
      42          12 :       boolean canSee = groupControlFactory.controlFor(user, group.getGroupUUID()).isVisible();
      43          12 :       if (!canSee) {
      44           2 :         logger.atFine().log("Filter out non-visisble group: %s", group.getGroupUUID());
      45             :       }
      46          12 :       return canSee;
      47           0 :     } catch (NoSuchGroupException e) {
      48             :       // Ignored
      49           0 :       return false;
      50             :     }
      51             :   }
      52             : 
      53             :   @Override
      54             :   public int getCost() {
      55           0 :     return 1;
      56             :   }
      57             : }