/*
* Copyright (c) 2010-2015 Evolveum
*
* 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.evolveum.midpoint.cli.ninja.action;
import com.beust.jcommander.JCommander;
import com.evolveum.midpoint.model.client.ModelClientUtil;
import com.evolveum.midpoint.cli.common.ToolsUtils;
import com.evolveum.midpoint.cli.ninja.command.Export;
import com.evolveum.midpoint.cli.ninja.util.LogWriter;
import com.evolveum.midpoint.cli.ninja.util.ObjectType;
import com.evolveum.midpoint.xml.ns._public.common.api_types_3.GetOperationOptionsType;
import com.evolveum.midpoint.xml.ns._public.common.api_types_3.ObjectListType;
import com.evolveum.midpoint.xml.ns._public.common.api_types_3.SelectorQualifiedGetOptionType;
import com.evolveum.midpoint.xml.ns._public.common.api_types_3.SelectorQualifiedGetOptionsType;
import com.evolveum.midpoint.xml.ns._public.common.common_3.OperationResultStatusType;
import com.evolveum.midpoint.xml.ns._public.common.common_3.OperationResultType;
import com.evolveum.midpoint.xml.ns._public.common.fault_3.FaultMessage;
import com.evolveum.midpoint.xml.ns._public.model.model_3.ModelPortType;
import com.evolveum.prism.xml.ns._public.query_3.OrderDirectionType;
import com.evolveum.prism.xml.ns._public.query_3.PagingType;
import com.evolveum.prism.xml.ns._public.query_3.QueryType;
import com.evolveum.prism.xml.ns._public.query_3.SearchFilterType;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.xml.sax.SAXException;
import javax.xml.bind.JAXBException;
import javax.xml.namespace.QName;
import javax.xml.ws.Holder;
import java.io.*;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.zip.ZipOutputStream;
/**
* @author lazyman
*/
public class ExportAction extends Action<Export> {
private static final int SEARCH_PAGE_SIZE = 100;
public ExportAction(Export params, JCommander commander) {
super(params, commander);
}
@Override
protected void executeAction() throws Exception {
ModelPortType port = createModelPort();
Writer writer = null;
try {
writer = createWriter();
writeHeader(writer);
if (StringUtils.isNotEmpty(getParams().getOid())) {
//get
executeGet(port, writer);
} else {
//search
executeSearch(port, writer);
}
writeFooter(writer);
writer.flush();
} catch (SAXException | JAXBException | IOException | FaultMessage ex) {
handleError("Couldn't export objects", ex);
} finally {
IOUtils.closeQuietly(writer);
}
}
private void executeGet(ModelPortType port, Writer writer) throws FaultMessage, IOException, JAXBException {
SelectorQualifiedGetOptionsType options = createOptions();
QName type = ObjectType.getType(getParams().getType());
if (type == null) {
type = ObjectType.OBJECT.getType();
}
Holder<com.evolveum.midpoint.xml.ns._public.common.common_3.ObjectType> object = new Holder<>();
Holder<OperationResultType> result = new Holder<>();
port.getObject(type, getParams().getOid(), options, object, result);
ToolsUtils.serializeObject(object.value, writer);
}
private SelectorQualifiedGetOptionsType createOptions() {
SelectorQualifiedGetOptionsType options = new SelectorQualifiedGetOptionsType();
if (getParams().isRaw()) {
SelectorQualifiedGetOptionType raw = new SelectorQualifiedGetOptionType();
GetOperationOptionsType option = new GetOperationOptionsType();
option.setRaw(true);
raw.setOptions(option);
options.getOption().add(raw);
}
return options;
}
private void executeSearch(ModelPortType port, Writer writer)
throws FaultMessage, IOException, JAXBException, SAXException {
SelectorQualifiedGetOptionsType options = createOptions();
QName type = ObjectType.getType(getParams().getType());
if (type == null) {
type = ObjectType.OBJECT.getType();
}
int count = 0;
int currentSize = 1;
Holder<ObjectListType> list = new Holder<>();
Holder<OperationResultType> result = new Holder<>();
while (currentSize > 0) {
QueryType query = createQuery(count);
port.searchObjects(type, query, options, list, result);
OperationResultType res = result.value;
if (!OperationResultStatusType.SUCCESS.equals(res.getStatus())
&& !getParams().isIgnore()) {
printInfoMessage("Search returned {}, reason: ", res.getStatus(), res.getMessage());
if (getParams().isVerbose()) {
printInfoMessage("Operation result:\n{}", ToolsUtils.serializeObject(res));
}
break;
}
ObjectListType objList = list.value;
if (getParams().isVerbose()) {
printInfoMessage("Search returned {}, status: {}/{}", res.getStatus(), count, objList.getCount());
}
List<com.evolveum.midpoint.xml.ns._public.common.common_3.ObjectType> objects = objList.getObject();
currentSize = objects.size();
for (com.evolveum.midpoint.xml.ns._public.common.common_3.ObjectType object : objects) {
ToolsUtils.serializeObject(object, writer);
writer.write('\n');
}
count += currentSize;
}
}
private QueryType createQuery(int from) throws IOException, SAXException, JAXBException {
QueryType query = new QueryType();
query.setFilter(loadQuery());
PagingType paging = new PagingType();
paging.setOffset(from);
paging.setMaxSize(SEARCH_PAGE_SIZE);
paging.setOrderBy(ModelClientUtil.createItemPathType("name"));
paging.setOrderDirection(OrderDirectionType.ASCENDING);
query.setPaging(paging);
return query;
}
private SearchFilterType loadQuery() throws IOException, SAXException, JAXBException {
File query = getParams().getQuery();
if (query == null) {
return null;
}
if (!query.exists() || !query.canRead()) {
String msg = "Couldn't read query file '" + query.getPath() + "'";
STD_ERR.info(msg);
throw new IOException(msg);
}
String xml = IOUtils.toString(new FileInputStream(query), StandardCharsets.UTF_8.name());
return ModelClientUtil.parseSearchFilterType(xml);
}
private Writer createWriter() throws IOException {
File file = getParams().getFile();
if (file != null) {
if (file.exists()) {
file.delete();
}
file.createNewFile();
OutputStream os = new FileOutputStream(file);
if (getParams().isZip()) {
os = new ZipOutputStream(os);
}
Writer writer = new OutputStreamWriter(os, StandardCharsets.UTF_8);
return new BufferedWriter(writer);
}
return new LogWriter(STD_OUT);
}
private void writeHeader(Writer writer) throws IOException {
String header = "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"
+ "<c:objects xmlns:c=\"http://midpoint.evolveum.com/xml/ns/public/common/common-3\">\n";
writer.write(header);
}
private void writeFooter(Writer writer) throws IOException {
writer.write("</c:objects>\n");
}
private void printInfoMessage(String message, Object... args) {
if (getParams().getFile() != null) {
STD_OUT.info(message, args);
} else {
STD_ERR.info(message, args);
}
}
}