universe@34: /*
universe@34:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
universe@34:  *
universe@34:  * Copyright 2018 Mike Becker. All rights reserved.
universe@34:  *
universe@34:  * Redistribution and use in source and binary forms, with or without
universe@34:  * modification, are permitted provided that the following conditions are met:
universe@34:  *
universe@34:  *   1. Redistributions of source code must retain the above copyright
universe@34:  *      notice, this list of conditions and the following disclaimer.
universe@34:  *
universe@34:  *   2. Redistributions in binary form must reproduce the above copyright
universe@34:  *      notice, this list of conditions and the following disclaimer in the
universe@34:  *      documentation and/or other materials provided with the distribution.
universe@34:  *
universe@34:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@34:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@34:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@34:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@34:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@34:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@34:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@34:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@34:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@34:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@34:  * POSSIBILITY OF SUCH DAMAGE.
universe@34:  *
universe@34:  */
universe@34: package de.uapcore.lightpit.dao.postgres;
universe@34: 
universe@34: import de.uapcore.lightpit.LightPITModule;
universe@34: import de.uapcore.lightpit.dao.AbstractDao;
universe@34: import de.uapcore.lightpit.dao.ModuleDao;
universe@34: import de.uapcore.lightpit.entities.Module;
universe@34: 
universe@34: import java.sql.Connection;
universe@34: import java.sql.PreparedStatement;
universe@34: import java.sql.ResultSet;
universe@34: import java.sql.SQLException;
universe@34: import java.util.Map;
universe@34: import java.util.Set;
universe@34: 
universe@34: public final class PGModuleDao extends AbstractDao<Module> implements ModuleDao {
universe@34: 
universe@34:     @Override
universe@34:     protected PreparedStatement listQuery(Connection connection) throws SQLException {
universe@34:         return connection.prepareStatement("select * from lpitcore_module");
universe@34:     }
universe@34: 
universe@34:     @Override
universe@34:     protected Module mapColumns(ResultSet result) throws SQLException {
universe@34:         final var mod = new Module();
universe@34:         mod.setModID(result.getInt("modid"));
universe@34:         mod.setClassname(result.getString("classname"));
universe@34:         mod.setVisible(result.getBoolean("visible"));
universe@34:         mod.setPriority(result.getInt("priority"));
universe@34:         return mod;
universe@34:     }
universe@34: 
universe@34:     @Override
universe@34:     public void syncRegisteredModuleClasses(Connection conn, Set<Map.Entry<String, LightPITModule>> moduleSet) throws SQLException {
universe@34: 
universe@34:         var check = conn.prepareStatement("select visible from lpitcore_module where classname = ?");
universe@34:         var insert = conn.prepareStatement("insert into lpitcore_module (classname, visible, priority) values (?, ?, ?)");
universe@34:         insert.setBoolean(2, true);
universe@34:         // update/delete not required, we do this in the module management UI
universe@34: 
universe@34:         for (Map.Entry<String, LightPITModule> modEntry : moduleSet) {
universe@34:             if (modEntry.getValue().systemModule()) continue;
universe@34: 
universe@34:             check.setString(1, modEntry.getKey());
universe@34:             try (ResultSet r = check.executeQuery()) {
universe@34:                 if (!r.next()) {
universe@34:                     insert.setString(1, modEntry.getKey());
universe@34:                     insert.setInt(3, modEntry.getValue().defaultPriority());
universe@34:                     insert.executeUpdate();
universe@34:                 }
universe@34:             }
universe@34:         }
universe@34:     }
universe@34: }