auth/auth-batch/src/main/java/org/onap/aaf/auth/actions/URModify.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.auth.actions;
  23
  24 import java.io.IOException;
  25 import java.util.List;
  26
  27 import org.onap.aaf.auth.dao.cass.Status;
  28 import org.onap.aaf.auth.dao.cass.UserRoleDAO;
  29 import org.onap.aaf.auth.dao.cass.UserRoleDAO.Data;
  30 import org.onap.aaf.auth.env.AuthzTrans;
  31 import org.onap.aaf.auth.helpers.UserRole;
  32 import org.onap.aaf.auth.layer.Result;
  33 import org.onap.aaf.misc.env.APIException;
  34
  35 import com.datastax.driver.core.Cluster;
  36
  37 public class URModify extends ActionDAO<UserRole,Void,URModify.Modify> {
  38         public URModify(AuthzTrans trans, Cluster cluster, boolean dryRun) throws APIException, IOException {
  39                 super(trans, cluster,dryRun);
  40         }
  41
  42         public URModify(AuthzTrans trans, ActionDAO<?,?,?> adao) {
  43                 super(trans, adao);
  44         }
  45
  46         @Override
  47         public Result<Void> exec(AuthzTrans trans, UserRole ur,Modify modify) {
  48                 if(dryRun) {
  49                         trans.info().printf("Would Update %s %s", ur.user(), ur.role());
  50                         return Result.ok();
  51                 } else {
  52                         Result<List<Data>> rr = q.userRoleDAO.read(trans, ur.user(),ur.role());
  53                         if(rr.notOKorIsEmpty()) {
  54                                 return Result.err(rr);
  55                         }
  56                         for(Data d : rr.value) {
  57                                 modify.change(d);
  58                                 if(!(ur.expires().equals(d.expires))) {
  59                                         ur.expires(d.expires);
  60                                 }
  61                                 if(ur.user().equals(d.user) && ur.role().equals(d.role)){
  62                                         Result<Void> rv = q.userRoleDAO.update(trans, d);
  63                                         if(rv.isOK()) {
  64                                                 trans.info().printf("Updated %s %s to %s", ur.user(), ur.role(), d.toString());
  65                                         } else {
  66                                                 trans.info().log(rv.errorString());
  67                                         }
  68                                 } else {
  69                                         return Result.err(Status.ERR_Denied, "You cannot change the key of this Data");
  70                                 }
  71                         }
  72                         return Result.err(Status.ERR_UserRoleNotFound,"No User Role with %s %s",ur.user(),ur.role());
  73                 }
  74         }
  75
  76         public static interface Modify {
  77                 void change(UserRoleDAO.Data ur);
  78         }
  79
  80 }