Java Code Examples for org.apache.axiom.om.impl.builder.StAXOMBuilder#getDocumentElement()

public static OMElement parse_xml(InputStream is) throws FactoryConfigurationError, XdsInternalException {

		//		create the parser
		XMLStreamReader parser=null;

		try {
			parser = XMLInputFactory.newInstance().createXMLStreamReader(is);
		} catch (XMLStreamException e) {
			throw new XdsInternalException("Could not create XMLStreamReader from InputStream");
		} 

		//		create the builder
		StAXOMBuilder builder = new StAXOMBuilder(parser);

		//		get the root element (in this case the envelope)
		OMElement documentElement =  builder.getDocumentElement();	
		if (documentElement == null)
			throw new XdsInternalException("No document element");
		return documentElement;
	}
 

/**
 * Get roles using the AuthorizationProvider using the config given.
 *
 * @param authProviderConfig xml config.
 * @return role array
 * @throws DataServiceFault
 */
public static String[] getAllRolesUsingAuthorizationProvider(String authProviderConfig)
        throws DataServiceFault {
    try {
        AuthorizationProvider authorizationProvider;
        if (authProviderConfig != null && !authProviderConfig.isEmpty()) {
            StAXOMBuilder builder = new StAXOMBuilder(new ByteArrayInputStream(authProviderConfig.getBytes(StandardCharsets.UTF_8)));
            OMElement documentElement =  builder.getDocumentElement();
            authorizationProvider = generateAuthProviderFromXMLOMElement(documentElement);
        } else {
            authorizationProvider = new UserStoreAuthorizationProvider();
        }
        return authorizationProvider.getAllRoles();
    } catch (XMLStreamException e) {
        throw new DataServiceFault(e, "Error reading XML file data - " + authProviderConfig + " Error - " + e.getMessage());
    }
}
 

OMElement parse_xml_string(String input_string) {
		byte[] ba = input_string.getBytes();

//		create the parser
		XMLStreamReader parser=null;

		try {
			parser = XMLInputFactory.newInstance().createXMLStreamReader(new ByteArrayInputStream(ba));
		} catch (XMLStreamException e) {
			assertTrue("Could not create XMLStreamReader from " + "input stream", false);
		}
//		create the builder
		StAXOMBuilder builder = new StAXOMBuilder(parser);

//		get the root element (in this case the envelope)
		OMElement documentElement =  builder.getDocumentElement();

		return documentElement;
	}
 

public static OMElement getPolicyAsOMElement(Policy policy) {

        try {
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            XMLStreamWriter writer = XMLOutputFactory.newInstance().createXMLStreamWriter(baos);
            policy.serialize(writer);
            writer.flush();
            ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
            XMLStreamReader xmlStreamReader =
                    XMLInputFactory.newInstance().createXMLStreamReader(bais);
            StAXOMBuilder staxOMBuilder =
                    (StAXOMBuilder) OMXMLBuilderFactory.createStAXOMBuilder(
                            OMAbstractFactory.getOMFactory(), xmlStreamReader);
            return staxOMBuilder.getDocumentElement();

        } catch (Exception ex) {
            throw new RuntimeException("can't convert the policy to an OMElement", ex);
        }
    }
 

public static OMElement nonCoalescingStringToOm(String xmlStr) throws XMLStreamException {
	StringReader strReader = new StringReader(xmlStr);
	XMLInputFactory xmlInFac = XMLInputFactory.newInstance();
	// Non-Coalescing parsing
	xmlInFac.setProperty("javax.xml.stream.isCoalescing", false);

	XMLStreamReader parser = xmlInFac.createXMLStreamReader(strReader);
	StAXOMBuilder builder = new StAXOMBuilder(parser);

	return builder.getDocumentElement();
}
 

public void addMessageStore(DataHandler dh)
        throws IOException, XMLStreamException, org.wso2.carbon.message.store.stub.Exception {
    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement messageStore = builder.getDocumentElement();
    messageStoreAdminServiceStub.addMessageStore(messageStore.toString());
}
 

/**
 * Adding proxy service
 *
 * @param dh Data Handler
 * @throws ProxyServiceAdminProxyAdminException Proxy service admin exception
 * @throws java.rmi.RemoteException             Remote exception
 * @throws javax.xml.stream.XMLStreamException  Exception
 */
public void addProxyService(DataHandler dh)
        throws ProxyServiceAdminProxyAdminException, IOException, XMLStreamException {

    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    //create the builder
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    //get the root element (in this case the envelope)
    OMElement documentElement = builder.getDocumentElement();

    ProxyData proxyData = getProxyData(documentElement.toString());

    log.info("Proxy Added");
}
 

/**
 * adding sequence
 *
 * @param dh
 * @throws SequenceEditorException
 * @throws java.io.IOException
 * @throws javax.xml.stream.XMLStreamException
 */
public void addSequence(DataHandler dh) throws SequenceEditorException, IOException, XMLStreamException {

    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    //create the builder
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement sequenceElem = builder.getDocumentElement();
    sequenceAdminServiceStub.addSequence(sequenceElem);
}
 

/**
 * This method checks if the given data service has a corresponding "services.xml" is available,
 * if so, the AxisService representing the data service is applied the instructions from its
 * "services.xml".
 */
private AxisService handleTransports(DeploymentFileData file, AxisService axisService) throws DataServiceFault {
	try (FileInputStream fis = new FileInputStream(file.getFile().getAbsoluteFile())) {
		StAXOMBuilder builder = new StAXOMBuilder(fis);
           OMElement documentElement =  builder.getDocumentElement();
           OMAttribute transports = documentElement.getAttribute(new QName(DBSFields.TRANSPORTS));
           if (transports != null) {
               String [] transportArr = transports.getAttributeValue().split(" ");
               axisService.setExposedTransports(Arrays.asList(transportArr));
           }
	} catch (Exception e) {
		throw new DataServiceFault(e, "Error in processing transports info");
	}
	return axisService;
}
 

public void addLocalEntry(String sessionCookie, DataHandler dh)
        throws LocalEntryAdminException, IOException, XMLStreamException {
    AuthenticateStub.authenticateStub(sessionCookie, localEntryAdminServiceStub);
    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    //create the builder
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement localEntryElem = builder.getDocumentElement();
    localEntryAdminServiceStub.addEntry(localEntryElem.toString());

}
 

public void addMessageStore(DataHandler dh)
        throws IOException, LocalEntryAdminException, XMLStreamException, Exception {
    messageStoreAdminServiceStub = this.setMessageStoreStubStub();
    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement localEntryElem = builder.getDocumentElement();
    messageStoreAdminServiceStub.addMessageStore(localEntryElem.toString());
}
 

public void addSequenceTemplate(DataHandler dh) throws IOException, XMLStreamException {
    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    //create the builder
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement template = builder.getDocumentElement();
    templateAdminStub.addTemplate(template);
}
 

public void addDynamicSequenceTemplate(String key, DataHandler dh) throws IOException, XMLStreamException {
    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    //create the builder
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement template = builder.getDocumentElement();
    templateAdminStub.addDynamicTemplate(key, template);
}
 

protected OMElement parse(InputStream in) {
    try {
        StAXOMBuilder builder = new StAXOMBuilder(in);
        return builder.getDocumentElement();
    } catch (XMLStreamException e) {
        fail("Error while parsing the XML from the input stream");
        return null;
    }
}
 

/**
 * adding sequence
 *
 * @param dh
 * @throws SequenceEditorException
 * @throws java.io.IOException
 * @throws javax.xml.stream.XMLStreamException
 */
public void addSequence(DataHandler dh) throws SequenceEditorException, IOException, XMLStreamException {

    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    //create the builder
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement sequenceElem = builder.getDocumentElement();
    sequenceAdminServiceStub.addSequence(sequenceElem);
}
 

private OMElement getRealmElement(InputStream inputStream) throws XMLStreamException,
        org.wso2.carbon.user.core.UserStoreException {

    try {
        inputStream = CarbonUtils.replaceSystemVariablesInXml(inputStream);
        StAXOMBuilder builder = new StAXOMBuilder(inputStream);
        OMElement documentElement = builder.getDocumentElement();
        setSecretResolver(documentElement);
        return documentElement;
    } catch (CarbonException e) {
        throw new org.wso2.carbon.user.core.UserStoreException(e.getMessage(), e);
    }
}
 

/**
 * Returns the document OMElement from the internal-apis.xml file.
 *
 * @param fileInputStream input stream of internal-apis.xml
 * @return OMelement of internal-apis.xml
 */
private static OMElement getOMElementFromFile(InputStream fileInputStream) throws CarbonException,
        XMLStreamException {
    InputStream inputStream = MicroIntegratorBaseUtils.replaceSystemVariablesInXml(fileInputStream);
    StAXOMBuilder builder = new StAXOMBuilder(inputStream);
    return builder.getDocumentElement();
}
 

public void addSequenceTemplate(DataHandler dh) throws IOException, XMLStreamException {
    XMLStreamReader parser = XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    //create the builder
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement template = builder.getDocumentElement();
    templateAdminStub.addTemplate(template);
}
 

private OMElement updateSynapseConfiguration(File synapseFile)
        throws IOException, XMLStreamException {

    OMElement synapseContent;
    BufferedInputStream bufferedInputStream = new BufferedInputStream(new FileInputStream(synapseFile));
    XMLStreamReader xmlStreamReader = XMLInputFactory.newInstance().createXMLStreamReader(bufferedInputStream);
    StAXOMBuilder stAXOMBuilder = new StAXOMBuilder(xmlStreamReader);
    synapseContent = stAXOMBuilder.getDocumentElement();
    synapseContent.build();
    bufferedInputStream.close();

    OMElement targetElement = synapseContent.getFirstChildWithName(new QName("http://ws.apache.org/ns/synapse", "target"));
    OMElement outSequenceElement = targetElement.getFirstChildWithName(new QName("http://ws.apache.org/ns/synapse", "outSequence"));
    OMElement dbReportElement = outSequenceElement.getFirstChildWithName(new QName("http://ws.apache.org/ns/synapse", "dbreport"));
    OMElement connectionElement = dbReportElement.getFirstChildWithName(
            new QName("http://ws.apache.org/ns/synapse", "connection"));
    OMElement poolElement = connectionElement.getFirstElement();
    OMElement driverElemnt = poolElement.getFirstChildWithName(new QName("http://ws.apache.org/ns/synapse", "driver"));
    OMElement urlElemnt = poolElement.getFirstChildWithName(new QName("http://ws.apache.org/ns/synapse", "url"));
    OMElement userElemnt = poolElement.getFirstChildWithName(new QName("http://ws.apache.org/ns/synapse", "user"));
    OMElement passwordElemnt = poolElement.getFirstChildWithName(new QName("http://ws.apache.org/ns/synapse", "password"));

    driverElemnt.setText(JDBC_DRIVER);
    urlElemnt.setText(JDBC_URL);
    userElemnt.setText(DB_USER);
    passwordElemnt.setText(DB_PASSWORD);
    return synapseContent;
}
 

/**
 * Add Local entry by DataHandler
 *
 * @param dh
 * @return
 * @throws java.io.IOException
 * @throws LocalEntryAdminException
 * @throws javax.xml.stream.XMLStreamException
 */
public boolean addLocalEntery(DataHandler dh)
        throws IOException, LocalEntryAdminException, XMLStreamException {
    XMLStreamReader parser =
            XMLInputFactory.newInstance().createXMLStreamReader(dh.getInputStream());
    StAXOMBuilder builder = new StAXOMBuilder(parser);
    OMElement localEntryElem = builder.getDocumentElement();
    return (localEntryAdminServiceStub.addEntry(localEntryElem.toString()));
}