Merge keystore dao to kernel core

core/org.wso2.carbon.core/pom.xml

@@ -94,6 +94,10 @@
             <groupId>org.wso2.carbon</groupId>
             <artifactId>org.wso2.carbon.utils</artifactId>
         </dependency>
+        <dependency>
+            <groupId>org.wso2.carbon.utils</groupId>
+            <artifactId>org.wso2.carbon.database.utils</artifactId>
+        </dependency>
         <dependency>
             <groupId>org.wso2.carbon.crypto</groupId>
             <artifactId>org.wso2.carbon.crypto.api</artifactId>
@@ -248,6 +252,7 @@
                             org.wso2.carbon.registry.core.service,
                             org.wso2.carbon.user.core.*; version=0.0.0,
                             org.bouncycastle.*; version="${imp.pkg.version.bcp}",
+                            org.wso2.carbon.database.utils.*;version="${org.wso2.carbon.database.utils.version.range}",
                             *;resolution:=optional
                         </Import-Package>
                         <Embed-Dependency>

core/org.wso2.carbon.core/src/main/java/org/wso2/carbon/core/internal/KeyStoreMgtComponent.java

@@ -0,0 +1,132 @@
+/*
+ * Copyright (c) 2023, WSO2 LLC. (https://www.wso2.com).
+ *
+ * WSO2 LLC. licenses this file to you 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 org.wso2.carbon.core.internal;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.osgi.framework.BundleContext;
+import org.osgi.service.component.ComponentContext;
+import org.osgi.service.component.annotations.Activate;
+import org.osgi.service.component.annotations.Component;
+import org.osgi.service.component.annotations.Deactivate;
+import org.osgi.service.component.annotations.Reference;
+import org.osgi.service.component.annotations.ReferenceCardinality;
+import org.osgi.service.component.annotations.ReferencePolicy;
+import org.wso2.carbon.base.ServerConfiguration;
+import org.wso2.carbon.core.keystore.KeyStoreManagementException;
+import org.wso2.carbon.core.keystore.KeyStoreManagementService;
+import org.wso2.carbon.core.keystore.KeyStoreManagementServiceImpl;
+import org.wso2.carbon.user.core.service.RealmService;
+import org.wso2.carbon.utils.ConfigurationContextService;
+
+import javax.naming.Context;
+import javax.naming.InitialContext;
+import javax.naming.NamingException;
+import javax.sql.DataSource;
+
+import static org.wso2.carbon.core.keystore.constants.KeyStoreConstants.KEYSTORE_DATASOURCE;
+
+@Component(
+        name = "keystore.mgt.service.component",
+        immediate = true
+)
+public class KeyStoreMgtComponent {
+
+    private static final Log log = LogFactory.getLog(KeyStoreMgtComponent.class);
+
+    @Activate
+    protected void activate(ComponentContext ctxt) {
+
+        try {
+            BundleContext bundleCtx = ctxt.getBundleContext();
+            bundleCtx.registerService(KeyStoreManagementService.class.getName(), new KeyStoreManagementServiceImpl(),
+                    null);
+            initDataSource();
+            log.debug("Security Mgt bundle is activated");
+        } catch (KeyStoreManagementException e) {
+            log.error("Failed to activate SecurityMgtServiceComponent", e);
+        }
+    }
+
+    @Deactivate
+    protected void deactivate(ComponentContext ctxt) {
+
+        log.debug("Security Mgt bundle is deactivated");
+    }
+
+    @Reference(
+            name = "config.context.service",
+            service = ConfigurationContextService.class,
+            cardinality = ReferenceCardinality.MANDATORY,
+            policy = ReferencePolicy.DYNAMIC,
+            unbind = "unsetConfigurationContextService"
+    )
+    protected void setConfigurationContextService(ConfigurationContextService contextService) {
+
+        if (log.isDebugEnabled()) {
+            log.debug("Setting the ConfigurationContext");
+        }
+        KeyStoreMgtDataHolder.setConfigurationContextService(contextService);
+    }
+
+    @Reference(
+            name = "user.realmservice.default",
+            service = RealmService.class,
+            cardinality = ReferenceCardinality.MANDATORY,
+            policy = ReferencePolicy.DYNAMIC,
+            unbind = "unsetRealmService"
+    )
+    protected void setRealmService(RealmService realmService) {
+
+        if (log.isDebugEnabled()) {
+            log.debug("Setting the RealmService");
+        }
+        KeyStoreMgtDataHolder.setRealmService(realmService);
+    }
+
+    protected void unsetRealmService(RealmService realmService) {
+
+        if (log.isDebugEnabled()) {
+            log.debug("Unsetting the RealmService");
+        }
+        KeyStoreMgtDataHolder.setRealmService(null);
+    }
+
+    protected void unsetConfigurationContextService(ConfigurationContextService contextService) {
+
+        if (log.isDebugEnabled()) {
+            log.debug("Unsetting the ConfigurationContext");
+        }
+        KeyStoreMgtDataHolder.setConfigurationContextService(null);
+    }
+
+    private void initDataSource() throws KeyStoreManagementException {
+
+        String dataSourceName = ServerConfiguration.getInstance()
+                .getFirstProperty(KEYSTORE_DATASOURCE);
+        Context ctx = null;
+        try {
+            ctx = new InitialContext();
+            DataSource dataSource = (DataSource) ctx.lookup(dataSourceName);
+            KeyStoreMgtDataHolder.setDataSource(dataSource);
+        } catch (NamingException e) {
+            throw new KeyStoreManagementException(e.getMessage());
+        }
+    }
+}

core/org.wso2.carbon.core/src/main/java/org/wso2/carbon/core/internal/KeyStoreMgtDataHolder.java

@@ -0,0 +1,83 @@
+/*
+ * Copyright (c) 2023, WSO2 LLC. (https://www.wso2.com).
+ *
+ * WSO2 LLC. licenses this file to you 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 org.wso2.carbon.core.internal;
+
+import org.apache.axis2.context.ConfigurationContext;
+import org.wso2.carbon.core.keystore.KeyStoreManagementException;
+import org.wso2.carbon.registry.core.Resource;
+import org.wso2.carbon.user.core.service.RealmService;
+import org.wso2.carbon.utils.ConfigurationContextService;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.sql.DataSource;
+
+public class KeyStoreMgtDataHolder {
+
+    private static RealmService realmService;
+    private static ConfigurationContextService ccService;
+    private static DataSource dataSource;
+    private static Map<String, Resource> policyResourceMap = new HashMap<>();
+
+    private KeyStoreMgtDataHolder() {
+
+    }
+
+    public static RealmService getRealmService() throws KeyStoreManagementException {
+
+        if (realmService == null) {
+            throw new KeyStoreManagementException("The main user realm is null");
+        }
+        return realmService;
+    }
+
+    public static void setRealmService(RealmService realmService) {
+
+        KeyStoreMgtDataHolder.realmService = realmService;
+    }
+
+    public static ConfigurationContext getConfigurationContext() throws Exception {
+
+        if (ccService == null) {
+            throw new KeyStoreManagementException("CC service is null");
+        }
+        return ccService.getClientConfigContext();
+    }
+
+    public static void setConfigurationContextService(ConfigurationContextService ccService) {
+
+        KeyStoreMgtDataHolder.ccService = ccService;
+    }
+
+    public static void addPolicyResource(String location, Resource resource) {
+
+        policyResourceMap.put(location, resource);
+    }
+
+    public static DataSource getDataSource() {
+
+        return dataSource;
+    }
+
+    public static void setDataSource(DataSource dataSource) {
+
+        KeyStoreMgtDataHolder.dataSource = dataSource;
+    }
+}