cli/record/StartRecord.java

0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL/*
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * The contents of this file are subject to the terms of the Common Development and
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * Distribution License (the License). You may not use this file except in compliance with the
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * License.
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL *
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * specific language governing permission and limitations under the License.
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL *
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * When distributing Covered Software, include this CDDL Header Notice in each file and include
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * Header, with the fields enclosed by brackets [] replaced by your own identifying
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * information: "Portions copyright [year] [name of copyright owner]".
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL *
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * Copyright 2015 ForgeRock AS.
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL */
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELpackage com.sun.identity.cli.record;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.iplanet.sso.SSOException;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.iplanet.sso.SSOToken;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.AuthenticatedCommand;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.CLIException;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.CLIUtil;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.IArgument;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.IOutput;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.LogWriter;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.RequestContext;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.cli.rest.RestCommand;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.common.configuration.ServerConfiguration;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.shared.debug.Debug;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport com.sun.identity.sm.SMSException;
a35224ef1ee8c02d389ffeeb676b4de432294fb6Phill Cunningtonimport org.forgerock.openam.core.rest.record.RecordConstants;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport java.io.IOException;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport java.net.URL;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport java.util.HashMap;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport java.util.Map;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELimport java.util.logging.Level;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL/**
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL * Start recording OpenAM by calling the records REST endpoint.
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL */
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTELpublic class StartRecord extends AuthenticatedCommand {
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL    private final Debug debug = Debug.getInstance(RecordConstants.DEBUG_INSTANCE_NAME);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL    /**
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL     * Start recording handle
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL     * @param rc Request Context.
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL     * @throws CLIException
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL     */
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL    public void handleRequest(RequestContext rc) throws CLIException {
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        super.handleRequest(rc);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        ldapLogin();
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        IOutput outputWriter = getOutputWriter();
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        String jsonFileName = getStringOptionValue(IArgument.JSON_FILE);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        String jsonContent = CLIUtil.getFileContent(getCommandManager(), jsonFileName, false);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        String serverName = getStringOptionValue(IArgument.SERVER_NAME);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        debug.message("Start recording: serverURL : '{}', data '{}'", serverName, jsonContent);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        String[] argsAttempt = {serverName, jsonContent};
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        writeLog(LogWriter.LOG_ACCESS, Level.INFO, "ATTEMPT_START_RECORD", argsAttempt);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        SSOToken adminSSOToken = getAdminSSOToken();
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        try {
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            if (!ServerConfiguration.isServerInstanceExist(adminSSOToken, serverName)) {
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                String message = "ServerName '" + serverName + "' doesn't exist";
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                debug.error(message);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                outputWriter.printlnMessage(message);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                String[] args = {serverName, jsonContent , message};
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                writeLog(LogWriter.LOG_ERROR, Level.INFO, "FAILED_START_RECORD", args);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                return;
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            }
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            Map<String, String> headers = new HashMap<String, String>();
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            headers.put("Content-Type", "application/json");
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            RestCommand restCommand = new RestCommand();
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            String result = restCommand.sendRestCommand(adminSSOToken.getTokenID(), new URL(serverName + "/json/"
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                            + RecordConstants.RECORD_REST_ENDPOINT + "?_action=" + RecordConstants.START_ACTION),
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                    "POST", headers, jsonContent);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            debug.message("Recording started with success. Result : '{}'", result);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            if (result.isEmpty()) {
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                outputWriter.printlnMessage("Result from server is empty. An error occurred. See debug logs for more " +
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                        "information");
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                String[] args = {serverName, jsonContent, "Result from server is empty. An error occurred."};
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL                writeLog(LogWriter.LOG_ERROR, Level.INFO, "FAILED_START_RECORD", args);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            }
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            String[] args = {serverName, jsonContent, result};
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            writeLog(LogWriter.LOG_ACCESS, Level.INFO, "SUCCESS_START_RECORD", args);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            outputWriter.printlnMessage(result);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        } catch (SSOException | SMSException | IOException e) {
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            debug.error("An error occurred", e);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            outputWriter.printlnMessage(e.getMessage());
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            String[] args = {serverName, jsonContent, e.getMessage()};
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL            writeLog(LogWriter.LOG_ERROR, Level.INFO, "FAILED_START_RECORD", args);
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL        }
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL    }
0f8e6a91301bd55c0b27889e849ac4e6ca905753Quentin CASTEL}