| // Copyright (C) 2016 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.account; |
| |
| import com.google.gerrit.extensions.api.changes.StarsInput; |
| import com.google.gerrit.extensions.common.ChangeInfo; |
| import com.google.gerrit.extensions.registration.DynamicMap; |
| import com.google.gerrit.extensions.restapi.AuthException; |
| import com.google.gerrit.extensions.restapi.BadRequestException; |
| import com.google.gerrit.extensions.restapi.ChildCollection; |
| import com.google.gerrit.extensions.restapi.IdString; |
| import com.google.gerrit.extensions.restapi.ResourceNotFoundException; |
| import com.google.gerrit.extensions.restapi.RestModifyView; |
| import com.google.gerrit.extensions.restapi.RestReadView; |
| import com.google.gerrit.extensions.restapi.RestView; |
| import com.google.gerrit.extensions.restapi.TopLevelResource; |
| import com.google.gerrit.server.CurrentUser; |
| import com.google.gerrit.server.IdentifiedUser; |
| import com.google.gerrit.server.StarredChangesUtil; |
| import com.google.gerrit.server.StarredChangesUtil.IllegalLabelException; |
| import com.google.gerrit.server.account.AccountResource.Star; |
| import com.google.gerrit.server.change.ChangeResource; |
| import com.google.gerrit.server.change.ChangesCollection; |
| import com.google.gerrit.server.query.change.QueryChanges; |
| import com.google.gwtorm.server.OrmException; |
| import com.google.inject.Inject; |
| import com.google.inject.Provider; |
| import com.google.inject.Singleton; |
| import java.util.Collection; |
| import java.util.List; |
| import java.util.Set; |
| import java.util.SortedSet; |
| |
| @Singleton |
| public class Stars implements ChildCollection<AccountResource, AccountResource.Star> { |
| |
| private final ChangesCollection changes; |
| private final ListStarredChanges listStarredChanges; |
| private final StarredChangesUtil starredChangesUtil; |
| private final DynamicMap<RestView<AccountResource.Star>> views; |
| |
| @Inject |
| Stars( |
| ChangesCollection changes, |
| ListStarredChanges listStarredChanges, |
| StarredChangesUtil starredChangesUtil, |
| DynamicMap<RestView<AccountResource.Star>> views) { |
| this.changes = changes; |
| this.listStarredChanges = listStarredChanges; |
| this.starredChangesUtil = starredChangesUtil; |
| this.views = views; |
| } |
| |
| @Override |
| public Star parse(AccountResource parent, IdString id) |
| throws ResourceNotFoundException, OrmException { |
| IdentifiedUser user = parent.getUser(); |
| ChangeResource change = changes.parse(TopLevelResource.INSTANCE, id); |
| Set<String> labels = starredChangesUtil.getLabels(user.getAccountId(), change.getId()); |
| return new AccountResource.Star(user, change, labels); |
| } |
| |
| @Override |
| public DynamicMap<RestView<Star>> views() { |
| return views; |
| } |
| |
| @Override |
| public ListStarredChanges list() { |
| return listStarredChanges; |
| } |
| |
| @Singleton |
| public static class ListStarredChanges implements RestReadView<AccountResource> { |
| private final Provider<CurrentUser> self; |
| private final ChangesCollection changes; |
| |
| @Inject |
| ListStarredChanges(Provider<CurrentUser> self, ChangesCollection changes) { |
| this.self = self; |
| this.changes = changes; |
| } |
| |
| @Override |
| @SuppressWarnings("unchecked") |
| public List<ChangeInfo> apply(AccountResource rsrc) |
| throws BadRequestException, AuthException, OrmException { |
| if (self.get() != rsrc.getUser()) { |
| throw new AuthException("not allowed to list stars of another account"); |
| } |
| QueryChanges query = changes.list(); |
| query.addQuery("has:stars"); |
| return (List<ChangeInfo>) query.apply(TopLevelResource.INSTANCE); |
| } |
| } |
| |
| @Singleton |
| public static class Get implements RestReadView<AccountResource.Star> { |
| private final Provider<CurrentUser> self; |
| private final StarredChangesUtil starredChangesUtil; |
| |
| @Inject |
| Get(Provider<CurrentUser> self, StarredChangesUtil starredChangesUtil) { |
| this.self = self; |
| this.starredChangesUtil = starredChangesUtil; |
| } |
| |
| @Override |
| public SortedSet<String> apply(AccountResource.Star rsrc) throws AuthException, OrmException { |
| if (self.get() != rsrc.getUser()) { |
| throw new AuthException("not allowed to get stars of another account"); |
| } |
| return starredChangesUtil.getLabels(self.get().getAccountId(), rsrc.getChange().getId()); |
| } |
| } |
| |
| @Singleton |
| public static class Post implements RestModifyView<AccountResource.Star, StarsInput> { |
| private final Provider<CurrentUser> self; |
| private final StarredChangesUtil starredChangesUtil; |
| |
| @Inject |
| Post(Provider<CurrentUser> self, StarredChangesUtil starredChangesUtil) { |
| this.self = self; |
| this.starredChangesUtil = starredChangesUtil; |
| } |
| |
| @Override |
| public Collection<String> apply(AccountResource.Star rsrc, StarsInput in) |
| throws AuthException, BadRequestException, OrmException { |
| if (self.get() != rsrc.getUser()) { |
| throw new AuthException("not allowed to update stars of another account"); |
| } |
| try { |
| return starredChangesUtil.star( |
| self.get().getAccountId(), |
| rsrc.getChange().getProject(), |
| rsrc.getChange().getId(), |
| in.add, |
| in.remove); |
| } catch (IllegalLabelException e) { |
| throw new BadRequestException(e.getMessage()); |
| } |
| } |
| } |
| } |