main/src/main/java/org/onap/cli/main/OnapCli.java

   1 /*
   2  * Copyright 2017 Huawei Technologies Co., Ltd.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.onap.cli.main;
  18
  19 import java.io.IOException;
  20 import java.util.ArrayList;
  21 import java.util.Arrays;
  22 import java.util.HashMap;
  23 import java.util.List;
  24 import java.util.Map;
  25 import java.util.Map.Entry;
  26 import java.util.Optional;
  27
  28 import org.apache.commons.io.IOUtils;
  29 import org.onap.cli.fw.cmd.OnapCommand;
  30 import org.onap.cli.fw.conf.OnapCommandConfig;
  31 import org.onap.cli.fw.conf.OnapCommandConstants;
  32 import org.onap.cli.fw.error.OnapCommandException;
  33 import org.onap.cli.fw.error.OnapCommandHelpFailed;
  34 import org.onap.cli.fw.error.OnapCommandInvalidSample;
  35 import org.onap.cli.fw.error.OnapCommandWarning;
  36 import org.onap.cli.fw.input.OnapCommandParameter;
  37 import org.onap.cli.fw.output.OnapCommandPrintDirection;
  38 import org.onap.cli.fw.output.OnapCommandResult;
  39 import org.onap.cli.fw.output.OnapCommandResultAttribute;
  40 import org.onap.cli.fw.output.OnapCommandResultAttributeScope;
  41 import org.onap.cli.fw.output.OnapCommandResultType;
  42 import org.onap.cli.fw.registrar.OnapCommandRegistrar;
  43 import org.onap.cli.fw.store.OnapCommandExecutionStore;
  44 import org.onap.cli.fw.store.OnapCommandExecutionStore.ExecutionStoreContext;
  45 import org.onap.cli.fw.utils.OnapCommandDiscoveryUtils;
  46 import org.onap.cli.main.conf.OnapCliConstants;
  47 import org.onap.cli.main.interactive.StringCompleter;
  48 import org.onap.cli.main.utils.OnapCliArgsParser;
  49 import org.onap.cli.sample.yaml.SampleYamlGenerator;
  50 import org.open.infc.grpc.Result;
  51 import org.slf4j.Logger;
  52 import org.slf4j.LoggerFactory;
  53
  54 import jline.TerminalFactory;
  55 import jline.console.ConsoleReader;
  56
  57 /**
  58  * OCLIP Command Line Interface (CLI).
  59  *
  60  */
  61 public class OnapCli {
  62
  63     private static Logger log = LoggerFactory.getLogger(OnapCli.class);
  64
  65     private List<String> args = new ArrayList<>();
  66
  67     private List<String> argsParamFile = new ArrayList<>();
  68
  69     private String product = null;
  70
  71     private String profile = null;
  72
  73     private String paramFile = null;
  74
  75     private String rpcHost = null;
  76
  77     private String rpcPort = null;
  78
  79     private boolean printHelp = false;
  80
  81     private boolean printVersion = false;
  82
  83     private String requestId = null;
  84
  85     private String cmdName = null;
  86
  87     private int exitCode = -1;
  88
  89     public OnapCli(String[] args) {
  90         this.setArgs(args);
  91     }
  92
  93     public OnapCli() {
  94     }
  95
  96     public void resetExitCode() {
  97         this.exitCode = -1;
  98     }
  99
 100     public void setArgs(String [] args) {
 101         //--help --version --requestId --rpc-host xxx --rpc-port xxx --product xxx --profile xxx --param-file xxx CMD blah blah
 102
 103         int cmdIdx = 0; //index of CMD
 104         while(args.length > cmdIdx) {
 105             //no options given, directly command name invoked
 106             if (!args[cmdIdx].startsWith("-")) {
 107                 break;
 108             }
 109
 110             if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCommandConstants.RPC_PRODUCT))) {
 111                 this.product = args[++cmdIdx];
 112                 cmdIdx++; //move to next option
 113             } else if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCommandConstants.RPC_PROFILE))) {
 114                 this.profile = args[++cmdIdx];
 115                 cmdIdx++; //move to next option
 116             } else if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCommandConstants.RPC_HOST))) {
 117                 this.rpcHost = args[++cmdIdx];
 118                 cmdIdx++; //move to next option
 119             } else if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCommandConstants.RPC_PORT))) {
 120                 this.rpcPort = args[++cmdIdx];
 121                 cmdIdx++; //move to next option
 122             } else if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCliConstants.PARAM_PARAM_FILE_LONG))) {
 123                 this.paramFile = args[++cmdIdx];
 124                 cmdIdx++; //move to next option
 125             } else if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCommandConstants.RPC_REQID))) {
 126                 this.requestId = args[++cmdIdx];
 127                 cmdIdx++; //move to next option
 128             } else if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCliConstants.PARAM_HELP_LOGN)) ||
 129                     args[cmdIdx].equals(OnapCommandParameter.printShortOption(OnapCliConstants.PARAM_HELP_SHORT))) {
 130                 this.printHelp = true;
 131                 cmdIdx++; //move to next option
 132             } else if (args[cmdIdx].equals(OnapCommandParameter.printLongOption(OnapCliConstants.PARAM_VERSION_LONG)) ||
 133                     args[cmdIdx].equals(OnapCommandParameter.printShortOption(OnapCliConstants.PARAM_VERSION_SHORT))) {
 134                 this.printVersion = true;
 135                 cmdIdx++; //move to next option
 136             }
 137         }
 138
 139         if (args.length > cmdIdx) {
 140             this.cmdName = args[cmdIdx];
 141             cmdIdx ++;
 142         }
 143
 144         this.args.clear();
 145
 146         //add all args starting from the command name
 147         for (int i=cmdIdx; i<args.length; i++) {
 148             this.args.add(args[i]);
 149         }
 150     }
 151
 152     private void exitSuccessfully() {
 153         this.exitCode = OnapCliConstants.EXIT_SUCCESS;
 154     }
 155
 156     private void exitFailure() {
 157         this.exitCode = OnapCliConstants.EXIT_FAILURE;
 158     }
 159
 160     protected void print(String msg) {
 161         System.out.println(msg); //NOSONAR
 162     }
 163
 164     protected void printerr(String msg) {
 165         System.err.println(msg); //NOSONAR
 166     }
 167
 168     private void print(Throwable throwable) {
 169         String error = throwable.getMessage() != null ? throwable.getMessage() : "";
 170         this.print(error);
 171         log.error(error, throwable);
 172     }
 173
 174     private String getShortOption(String opt) {
 175         return OnapCommandParameter.printShortOption(opt);
 176     }
 177
 178     private String getLongOption(String opt) {
 179         return OnapCommandParameter.printLongOption(opt);
 180     }
 181
 182     public int getExitCode() {
 183         return this.exitCode;
 184     }
 185
 186     /**
 187      * Handles help. --help or -h
 188      */
 189     public void handleHelp() {
 190         try {
 191             if (this.printHelp) {
 192                 this.print(IOUtils.toString(this.getClass().getClassLoader().getResourceAsStream("oclip-readme.txt")));
 193                 String help = OnapCommandRegistrar.getRegistrar().getHelp();
 194                 this.print(help);
 195                 this.exitSuccessfully();
 196             }
 197         } catch (Exception e) {
 198             this.print(e);
 199             this.exitFailure();
 200         }
 201     }
 202
 203     /**
 204      * Handles version. --version or -v
 205      */
 206     public void handleVersion() {
 207         try {
 208             if (this.printVersion) {
 209                 String version = OnapCommandRegistrar.getRegistrar().getVersion();
 210                 this.print(version);
 211                 this.exitSuccessfully();
 212             }
 213         } catch (Exception e) {
 214             this.print(e);
 215             this.exitFailure();
 216         }
 217     }
 218
 219
 220     /**
 221      * Handles profile. --profile
 222      */
 223     public void handleProfile() {
 224         try {
 225             if (this.profile != null) {
 226                 OnapCommandRegistrar.getRegistrar().setProfile(
 227                         this.profile,
 228                         new ArrayList<>(),
 229                         new ArrayList<>());
 230             }
 231         } catch (Exception e) {
 232             this.print(e);
 233             this.exitFailure();
 234         }
 235     }
 236
 237     /**
 238      * Handles batch command. --param-file
 239      */
 240     public void handleBatchCommand() {
 241         try {
 242             if (this.paramFile != null) {
 243                 //Read YAML and loop thru it
 244                 // one
 245                 // - param-long-option-1: value
 246                 // - param-long-option-1: value
 247                 // - positional-arg1
 248                 // - positional-arg2
 249                 // two
 250                 // - param-long-option-1: value
 251                 // - param-long-option-1: value
 252                 // - positional-arg1
 253                 // - positional-arg2
 254                 try {
 255                     Map<String, Object> values = (Map<String, Object>) OnapCommandDiscoveryUtils.loadYaml(this.paramFile);
 256
 257                     for (Entry<String, Object> cmdsParam: values.entrySet()) {
 258                         for (Object param: (List)cmdsParam.getValue()) {
 259                             if (param instanceof Map) { //optional args
 260                                 Map <String, String> paramMap = (Map<String, String>) param;
 261                                 String paramName = paramMap.keySet().iterator().next();
 262                                 Object paramValue = paramMap.get(paramName);
 263                                 argsParamFile.add(this.getLongOption(paramName));
 264                                 argsParamFile.add(paramValue.toString());
 265                             } else { //positional args
 266                                 argsParamFile.add(param.toString());
 267                             }
 268                         }
 269                     }
 270
 271                 } catch (Exception e) { // NOSONAR
 272                     this.print("Failed to read param file " + this.paramFile);
 273                     this.print(e);
 274                 }
 275             }
 276         } catch (Exception e) {
 277             this.print(e);
 278             this.exitFailure();
 279         }
 280     }
 281
 282     public void verifyCommand(OnapCommand cmd) throws OnapCommandException {
 283
 284         OnapCliArgsParser.populateParams(cmd.getParameters(), args);
 285
 286         Optional<OnapCommandParameter> contextOptArg = cmd.getParameters().stream()
 287                 .filter(e -> e.getName().equals(OnapCommandConstants.VERIFY_CONTEXT_PARAM))
 288                 .findFirst();
 289
 290         List<Map<String, Object>> testSuite = OnapCommandRegistrar.getRegistrar().getTestSuite(
 291                 cmd.getName(),
 292                 cmd.getInfo().getProduct());
 293
 294         OnapCommandResult testSuiteResult = new OnapCommandResult();
 295         testSuiteResult.setType(OnapCommandResultType.TABLE);
 296         testSuiteResult.setPrintDirection(OnapCommandPrintDirection.LANDSCAPE);
 297         testSuiteResult.setIncludeTitle(true);
 298
 299         OnapCommandResultAttribute sampleFileAtt = new OnapCommandResultAttribute();
 300         OnapCommandResultAttribute sampleIdAtt = new OnapCommandResultAttribute();
 301         OnapCommandResultAttribute resultAtt = new OnapCommandResultAttribute();
 302
 303         sampleFileAtt.setName("Test");
 304         sampleIdAtt.setName("SampleId");
 305         resultAtt.setName("Result");
 306
 307         testSuiteResult.setRecords(Arrays.asList(sampleFileAtt,
 308                 sampleIdAtt,
 309                 resultAtt));
 310
 311         for (Map<String, ?> sampleTest : testSuite) {
 312
 313             sampleFileAtt.getValues().add((String) sampleTest.get(OnapCommandConstants.VERIFY_SAMPLE_FILE_ID));
 314             sampleIdAtt.getValues().add((String) sampleTest.get(OnapCommandConstants.VERIFY_SAMPLE_ID));
 315
 316             cmd = OnapCommandRegistrar.getRegistrar().get(cmd.getName(),
 317                     cmd.getInfo().getProduct());
 318             List<String> arguments = (List<String>) sampleTest.get(OnapCommandConstants.VERIFY_INPUT);
 319
 320             OnapCliArgsParser.populateParams(cmd.getParameters(), arguments);
 321             this.print("\n***************Test Command: \n" + sampleTest.get(OnapCommandConstants.VERIFY_INPUT).toString());
 322
 323             cmd.getParametersMap().get(OnapCommandConstants.DEFAULT_PARAMETER_DEBUG).setValue(Boolean.TRUE);
 324
 325             Optional<OnapCommandParameter> contextOpt = cmd.getParameters().stream()
 326                     .filter(e -> e.getName().equals(OnapCommandConstants.VERIFY_CONTEXT_PARAM))
 327                     .findFirst();
 328
 329             if (contextOpt.isPresent()) {
 330                 HashMap<String, Object> map = new HashMap<>();
 331
 332                 Object moco = sampleTest.get(OnapCommandConstants.VERIFY_MOCO);
 333                 if (moco == null) {
 334                     continue;
 335                 }
 336                 map.put(OnapCommandConstants.VERIFY_MOCO, moco);
 337
 338                 if (contextOptArg.isPresent()) {
 339                     OnapCommandParameter contextArg = contextOptArg.get();
 340                     map.putAll((Map) contextArg.getValue());
 341                 }
 342
 343                 contextOpt.get().setValue(map);
 344             }
 345
 346             OnapCommandResult testResult = cmd.execute();
 347             String actualOutput = testResult.print().trim();
 348             String expectedOutput = (String) sampleTest.get(OnapCommandConstants.VERIFY_OUPUT);
 349             expectedOutput = expectedOutput == null ? "" : expectedOutput.trim();
 350
 351             if (actualOutput.equals(expectedOutput)) {
 352                 resultAtt.getValues().add(OnapCommandConstants.VERIFY_RESULT_PASS);
 353             } else {
 354                 resultAtt.getValues().add(OnapCommandConstants.VERIFY_RESULT_FAIL);
 355             }
 356             this.printerr(testResult.getDebugInfo());
 357             this.print("\n***************Expected Output: \n" + expectedOutput);
 358             this.print("\n***************Actual Output: \n" + actualOutput);
 359         }
 360
 361         this.print(testSuiteResult.print());
 362     }
 363
 364     /**
 365      * Handles Interactive Mode.
 366      */
 367     public void handleInteractive() { // NOSONAR
 368         if (this.cmdName == null) {
 369             ConsoleReader console = null;
 370             try {
 371                 OnapCommandRegistrar.getRegistrar().setInteractiveMode(true);
 372                 console = createConsoleReader();
 373                 String line = null;
 374
 375                 while ((line = console.readLine()) != null) {
 376                     line = line.trim();
 377                     if (OnapCliConstants.PARAM_INTERACTIVE_EXIT.equalsIgnoreCase(line)) {
 378                         break;
 379                     } else if (OnapCliConstants.PARAM_INTERACTIVE_CLEAR.equalsIgnoreCase(line)) {
 380                         console.clearScreen();
 381                     } else {
 382                         this.args = new ArrayList<>();
 383                         this.args.addAll(Arrays.asList(line.split(OnapCliConstants.PARAM_INTERACTIVE_ARG_SPLIT_PATTERN)));
 384
 385                     if (!args.isEmpty() && this.args.get(0).equals(OnapCliConstants.PARAM_INTERACTIVE_USE)) {
 386                         if (args.size() == 1) {
 387                             this.print("Please use it in the form of use <product-version>.\nSupported versions: " +
 388                                     OnapCommandRegistrar.getRegistrar().getAvailableProductVersions());
 389                         } else {
 390                             try {
 391                                 OnapCommandRegistrar.getRegistrar().setEnabledProductVersion(args.get(1));
 392                                 console = createConsoleReader();
 393                             } catch (OnapCommandException e) {
 394                                 this.print(e);
 395                             }
 396                         }
 397
 398                     } else if (!args.isEmpty() && this.args.get(0).equals(OnapCliConstants.PARAM_INTERACTIVE_HELP)) {
 399                         this.print(OnapCommandRegistrar.getRegistrar().getHelpForEnabledProductVersion());
 400                         this.print(OnapCli.getDirectiveHelp());
 401                         } else if (!args.isEmpty() && this.args.get(0).equals(OnapCliConstants.PARAM_INTERACTIVE_VERSION)) {
 402                             this.printVersion = true;
 403                             handleVersion();
 404
 405                         } else if (!args.isEmpty() && this.args.get(0).equals(OnapCliConstants.PARAM_INTERACTIVE_PROFILE)) {
 406                             if (args.size() == 1) {
 407                                 this.print("Please use it in the form of 'profile <profile-name>'\n");
 408                                 this.print("Available profiles: ");
 409                                 this.print(OnapCommandRegistrar.getRegistrar().getUserProfiles().toString());
 410                             } else {
 411                                 this.profile = args.get(1);
 412                                 handleProfile();
 413                             }
 414
 415                         } else if (!args.isEmpty() && this.args.get(0).equals(OnapCliConstants.PARAM_INTERACTIVE_SET)) {
 416                             if (args.size() > 1) {
 417                                 String[] paramEntry = args.get(1).trim().split("=", 2);
 418                                 if (paramEntry.length == 2) {
 419                                     OnapCommandRegistrar.getRegistrar().addParamCache(paramEntry[0].trim(), paramEntry[1].trim());
 420                                 } else {
 421                                     this.print("Please use it in the form of 'set <param-name>=<param-value>'");
 422                                 }
 423                             } else {
 424                                 this.print(OnapCommandRegistrar.getRegistrar().getParamCache().toString());
 425                             }
 426
 427                         } else if (!args.isEmpty() && this.args.get(0).equals(OnapCliConstants.PARAM_INTERACTIVE_UNSET)) {
 428                             if (args.size() > 1) {
 429                                 for (int i = 1; i < args.size(); i++) {
 430                                     OnapCommandRegistrar.getRegistrar().removeParamCache(args.get(i));
 431                                 }
 432                             }
 433                         } else {
 434                             if (!(args.size() == 1 && args.get(0).trim().isEmpty())) {
 435                                 this.setArgs(this.args.toArray(new String[]{}));
 436                                 handleCommand();
 437                             }
 438                         }
 439                     }
 440                 }
 441             } catch (IOException e) { // NOSONAR
 442                 this.print("Failed to read console, " + e.getMessage());
 443             } catch (OnapCommandException e) {
 444                 this.print(e);
 445                 this.exitFailure();
 446             } finally {
 447                 try {
 448                     TerminalFactory.get().restore();
 449                 } catch (Exception e) { // NOSONAR
 450                 }
 451                 this.exitSuccessfully();
 452             }
 453         }
 454     }
 455
 456     /**
 457      * Handles command.
 458      */
 459     public void handleCommand() {
 460         OnapCommand cmd = null;
 461         if (this.cmdName != null) {
 462             try {
 463                 if (this.product != null) {
 464                     cmd = OnapCommandRegistrar.getRegistrar().get(this.cmdName, this.product);
 465                 } else {
 466                     cmd = OnapCommandRegistrar.getRegistrar().get(this.cmdName);
 467                 }
 468             } catch (Exception e) {
 469                 this.print(e);
 470                 this.exitFailure();
 471                 return;
 472             }
 473
 474             ExecutionStoreContext executionStoreContext = null;
 475
 476             try {
 477                 //Registrar identified this command marked with rpc as true and it will make direct RPC command call...
 478                 if (cmd.isRpc() && !this.cmdName.equals("schema-rpc")) {
 479                     this.handleRpcCommand(cmd);
 480                     return;
 481                 }
 482
 483                 // verify
 484                 if(args.contains(OnapCommandConstants.VERIFY_LONG_OPTION)
 485                         || args.contains(OnapCommandConstants.VERIFY_SHORT_OPTION)) {
 486                     verifyCommand(cmd);
 487                     this.exitSuccessfully();
 488                     return;
 489                 }
 490                 // check for help or version
 491                 if (args.size() == 2) {
 492                     if (this.getLongOption(OnapCliConstants.PARAM_HELP_LOGN).equals(args.get(1))
 493                             || this.getShortOption(OnapCliConstants.PARAM_HELP_SHORT).equals(args.get(1))) {
 494                         String help = cmd.printHelp();
 495                         this.print(help);
 496                         this.exitSuccessfully();
 497                         return;
 498                     } else if (this.getLongOption(OnapCliConstants.PARAM_VERSION_LONG).equals(args.get(1))
 499                             || this.getShortOption(OnapCliConstants.PARAM_VERSION_SHORT).equals(args.get(1))) {
 500                         String version = cmd.printVersion();
 501                         this.print(version);
 502                         this.exitSuccessfully();
 503                         return;
 504                     }
 505                 }
 506
 507                 //refer params from profile
 508                 if (this.profile != null) {
 509
 510                     Map<String, String> paramCache = new HashMap<>(); //NOSONAR
 511                     if (this.product == null)
 512                         paramCache = OnapCommandRegistrar.getRegistrar().getParamCache();
 513                     else
 514                         paramCache = OnapCommandRegistrar.getRegistrar().getParamCache(this.product);
 515
 516                     for (OnapCommandParameter param: cmd.getParameters()) {
 517                         if (paramCache.containsKey(
 518                                 cmd.getInfo().getService() + ":" + cmd.getName() + ":" + param.getLongOption())) {
 519                             param.setValue(paramCache.get(
 520                                     cmd.getInfo().getService() + ":" + cmd.getName() + ":" + param.getLongOption()));
 521                         } else if (paramCache.containsKey(
 522                                 cmd.getInfo().getService() + ":" + param.getLongOption())) {
 523                             param.setValue(paramCache.get(
 524                                     cmd.getInfo().getService() + ":" + param.getLongOption()));
 525                         } else if (paramCache.containsKey(param.getLongOption())) {
 526                             param.setValue(paramCache.get(param.getLongOption()));
 527                         }
 528                     }
 529                 }
 530
 531                 //load the parameters value from the map read from param-file
 532                 if (!this.argsParamFile.isEmpty()) {
 533                     OnapCliArgsParser.populateParams(cmd.getParameters(), this.argsParamFile);
 534                 }
 535
 536                 OnapCliArgsParser.populateParams(cmd.getParameters(), this.args);
 537
 538                 //start the execution
 539                     if (this.requestId != null && this.product != null && !this.requestId.isEmpty()&& !(this.product.equalsIgnoreCase("open-cli") &&
 540                             this.cmdName.equalsIgnoreCase("execution-list"))) {
 541                         String input = cmd.getArgsJson(true);
 542                         executionStoreContext = OnapCommandExecutionStore.getStore().storeExectutionStart(
 543                                 this.requestId,
 544                                 cmd.getInfo().getProduct(),
 545                                 cmd.getInfo().getService(),
 546                                 this.cmdName,
 547                                 this.profile,
 548                                 input);
 549                     }
 550
 551                 cmd.setExecutionContext(executionStoreContext);
 552                 OnapCommandResult result = cmd.execute();
 553
 554                 this.handleTracking(cmd);
 555
 556                 if (result.isPassed()) {
 557                     this.exitSuccessfully();
 558                     generateSmapleYaml(cmd);
 559                 } else {
 560                     this.exitFailure();
 561                 }
 562             } catch (OnapCommandWarning w) {
 563                 if (cmd.getExecutionContext() != null) {
 564                     OnapCommandExecutionStore.getStore().storeExectutionEnd(
 565                             cmd.getExecutionContext(),
 566                             w.getMessage(),
 567                             null,
 568                             cmd.getResult().getDebugInfo(),
 569                             cmd.getResult().isPassed());
 570                 }
 571
 572                 this.print(w);
 573                 this.printerr(cmd.getResult().getDebugInfo());
 574                 this.exitSuccessfully();
 575             } catch (Exception e) {
 576                 if (executionStoreContext != null) {
 577                     OnapCommandExecutionStore.getStore().storeExectutionEnd(
 578                             executionStoreContext,
 579                             null,
 580                             e.getMessage(),
 581                             cmd.getResult().getDebugInfo(),
 582                             false);
 583                 }
 584
 585                 this.print(e);
 586                 this.printerr(cmd.getResult().getDebugInfo());
 587                 this.exitFailure();
 588             }
 589         }
 590     }
 591
 592     public void handleTracking(OnapCommand cmd) throws OnapCommandException {
 593         if (cmd.getResult().isDebug())
 594             this.printerr(cmd.getResult().getDebugInfo());
 595
 596         String printOut = cmd.getResult().print();
 597         this.print(printOut);
 598
 599         if (cmd.getExecutionContext() != null) {
 600             OnapCommandExecutionStore.getStore().storeExectutionEnd(
 601                     cmd.getExecutionContext(),
 602                     printOut,
 603                     null,
 604                     cmd.getResult().getDebugInfo(),
 605                     cmd.getResult().isPassed());
 606         }
 607     }
 608     /**
 609      * When user invokes cli with RPC arguments...
 610      */
 611     public void handleRpc() {
 612         if (this.rpcHost != null && this.rpcPort != null && this.product != null) {
 613             try {
 614                 OnapCommand cmd = OnapCommandRegistrar.getRegistrar().get("schema-rpc", "open-cli");
 615                 cmd.getParametersMap().get(OnapCommandConstants.RPC_HOST).setValue(this.rpcHost);
 616                 cmd.getParametersMap().get(OnapCommandConstants.RPC_PORT).setValue(this.rpcPort);
 617                 cmd.getParametersMap().get(OnapCommandConstants.RPC_PRODUCT).setValue(this.product);
 618                 cmd.getParametersMap().get(OnapCommandConstants.RPC_CMD).setValue(this.cmdName);
 619
 620                 this.handleRpcCommand(cmd);
 621             } catch (Exception e) {
 622                 this.print(e);
 623                 this.exitFailure();
 624             }
 625         }
 626     }
 627
 628     private void handleRpcCommand(OnapCommand cmd) throws OnapCommandException {
 629         Map<String, List<String>> argsMap = new HashMap<>();
 630         argsMap.put(OnapCommandConstants.RPC_ARGS, this.args);
 631         if (this.profile != null )
 632             cmd.getParametersMap().get(OnapCommandConstants.RPC_PROFILE).setValue(this.profile);
 633         cmd.getParametersMap().get(OnapCommandConstants.RPC_REQID).setValue(this.requestId);
 634         cmd.getParametersMap().get(OnapCommandConstants.RPC_MODE).setValue(OnapCommandConstants.RPC_MODE_RUN_CLI);
 635         cmd.getParametersMap().get(OnapCommandConstants.RPC_ARGS).setValue(argsMap);
 636
 637         OnapCommandResult result = cmd.execute();
 638         Result output = (Result) result.getOutput();
 639
 640         this.exitCode = output.getExitCode();
 641         this.print(output.getOutput());
 642     }
 643     /**
 644      * Handles all client input.
 645      */
 646     public void handle() {
 647         this.handleRpc();
 648
 649         if (this.exitCode == -1) {
 650             this.handleHelp();
 651         }
 652
 653         if (this.exitCode == -1) {
 654             this.handleVersion();
 655         }
 656
 657         if (this.exitCode == -1) {
 658             this.handleProfile();
 659         }
 660
 661         if (this.exitCode == -1) {
 662             this.handleBatchCommand();
 663         }
 664
 665         if (this.exitCode == -1) {
 666             this.handleInteractive();
 667         }
 668
 669         if (this.exitCode == -1) {
 670             this.handleCommand();
 671         }
 672     }
 673
 674     public static String getDirectiveHelp() throws OnapCommandHelpFailed {
 675         OnapCommandResult help = new OnapCommandResult();
 676         help.setType(OnapCommandResultType.TABLE);
 677         help.setPrintDirection(OnapCommandPrintDirection.LANDSCAPE);
 678
 679         OnapCommandResultAttribute attr = new OnapCommandResultAttribute();
 680         attr.setName(OnapCommandConstants.NAME.toUpperCase());
 681         attr.setDescription(OnapCommandConstants.DESCRIPTION);
 682         attr.setScope(OnapCommandResultAttributeScope.SHORT);
 683         help.getRecords().add(attr);
 684
 685         OnapCommandResultAttribute attrDesc = new OnapCommandResultAttribute();
 686         attrDesc.setName(OnapCommandConstants.DESCRIPTION.toUpperCase());
 687         attrDesc.setDescription(OnapCommandConstants.DESCRIPTION);
 688         attrDesc.setScope(OnapCommandResultAttributeScope.SHORT);
 689         help.getRecords().add(attrDesc);
 690
 691         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_CLEAR);
 692         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_CLEAR_MSG);
 693
 694         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_EXIT);
 695         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_EXIT_MSG);
 696
 697         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_VERSION);
 698         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_VERSION_MSG);
 699
 700         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_USE);
 701         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_USE_MSG);
 702
 703         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_SET);
 704         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_SET_MSG);
 705
 706         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_UNSET);
 707         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_UNSET_MSG);
 708
 709         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_HELP);
 710         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_HELP_MSG);
 711
 712         attr.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_PROFILE);
 713         attrDesc.getValues().add(OnapCliConstants.PARAM_INTERACTIVE_PROFILE_MSG);
 714         try {
 715             return "\n\nDirectives:\n" + help.print();
 716         } catch (OnapCommandException e) {
 717             throw new OnapCommandHelpFailed(e);
 718         }
 719     }
 720
 721     /**
 722      * Creates console reader object.
 723      *
 724      * @return ConsoleReader
 725      * @throws IOException
 726      *             exception
 727      */
 728     private ConsoleReader createConsoleReader() throws IOException {
 729         ConsoleReader console = new ConsoleReader(); // NOSONAR
 730         try {
 731             //ignore system commands
 732             StringCompleter strCompleter = new StringCompleter(OnapCommandRegistrar.getRegistrar().listCommandsForEnabledProductVersion());
 733             strCompleter.add(OnapCliConstants.PARAM_INTERACTIVE_EXIT,
 734                     OnapCliConstants.PARAM_INTERACTIVE_CLEAR,
 735                     OnapCliConstants.PARAM_INTERACTIVE_USE,
 736                     OnapCliConstants.PARAM_INTERACTIVE_HELP,
 737                     OnapCliConstants.PARAM_INTERACTIVE_VERSION,
 738                     OnapCliConstants.PARAM_INTERACTIVE_SET,
 739                     OnapCliConstants.PARAM_INTERACTIVE_UNSET,
 740                     OnapCliConstants.PARAM_INTERACTIVE_PROFILE);
 741             console.addCompleter(strCompleter);
 742             console.setPrompt(OnapCliConstants.PARAM_INTERACTIVE_PROMPT + ":" + OnapCommandRegistrar.getRegistrar().getEnabledProductVersion() + ">");
 743         } catch (OnapCommandException e) { // NOSONAR
 744             this.print("Failed to load oclip commands," + e.getMessage());
 745         }
 746
 747         return console;
 748     }
 749
 750
 751     private void generateSmapleYaml(OnapCommand cmd) throws OnapCommandException {
 752         if (Boolean.parseBoolean(OnapCommandConfig.getPropertyValue(OnapCommandConstants.SAMPLE_GEN_ENABLED)) && this.getExitCode() == OnapCliConstants.EXIT_SUCCESS) {
 753             try {
 754                 SampleYamlGenerator.generateSampleYaml(cmd.getName(), args, cmd.getResult().print(),
 755                         cmd.getInfo().getProduct(),
 756                         OnapCommandConfig.getPropertyValue(OnapCommandConstants.SAMPLE_GEN_TARGET_FOLDER) + "/" + cmd.getSchemaName().replaceAll(".yaml", "") + "-sample.yaml",
 757                         cmd.getResult().isDebug(),
 758                         OnapCommandConfig.getPropertyValue(OnapCommandConstants.SAMPLE_GEN_NAME));
 759             } catch (IOException error) {
 760                 throw new OnapCommandInvalidSample(this.cmdName, error);
 761             }
 762         }
 763     }
 764
 765     /**
 766      * Main method.
 767      *
 768      * @param args
 769      *            array
 770      */
 771     public static void main(String[] args) {
 772         OnapCli cli = new OnapCli(args); //NOSONAR
 773         cli.handle();
 774         System.exit(cli.getExitCode());
 775     }
 776
 777 }