de.dakror.virtualhub.server.DBManager.java Source code

Introduction

Here is the source code for de.dakror.virtualhub.server.DBManager.java

Source

/*******************************************************************************
 * Copyright 2015 Maximilian Stark | Dakror <mail@dakror.de>
 * 
 * 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 de.dakror.virtualhub.server;

import java.io.File;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import org.json.JSONObject;

import de.dakror.virtualhub.data.Eticet;
import de.dakror.virtualhub.data.Tags;
import de.dakror.virtualhub.net.packet.Packet4Rename;

/**
 * @author Dakror
 */
public class DBManager {
    public static File database;

    static Connection connection;

    public static void init() {
        try {
            database = new File(Server.dir, "virtualhub.db");
            database.createNewFile();

            Class.forName("org.sqlite.JDBC");
            connection = DriverManager.getConnection("jdbc:sqlite:" + database.getPath().replace("\\", "/"));

            Statement s = connection.createStatement();
            s.executeUpdate(
                    "CREATE TABLE IF NOT EXISTS ETICETS(PATH varchar(500) NOT NULL PRIMARY KEY, CATALOG varchar(100), ETICET INT)");
            s.executeUpdate(
                    "CREATE TABLE IF NOT EXISTS TAGS(PATH varchar(500) NOT NULL PRIMARY KEY, CATALOG varchar(100), TAGS TEXT)");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static Eticet eticet(File f, String catalog, Eticet e) {
        try {
            if (e == Eticet.NULL) {
                ResultSet rs = connection.createStatement()
                        .executeQuery("SELECT ETICET FROM ETICETS WHERE PATH = \"" + f.getPath().replace("\\", "/")
                                + "\" AND CATALOG = \"" + catalog + "\"");
                if (!rs.next())
                    return Eticet.NONE;

                return Eticet.values()[rs.getInt(1)];
            } else {
                if (e == Eticet.NONE)
                    connection.createStatement().executeUpdate("DELETE FROM ETICETS WHERE PATH = \""
                            + f.getPath().replace("\\", "/") + "\" AND CATALOG = \"" + catalog + "\"");
                else
                    connection.createStatement().executeUpdate(
                            "INSERT OR REPLACE INTO ETICETS VALUES(\"" + f.getPath().replace("\\", "/") + "\","
                                    + e.ordinal() + ") AND CATALOG = \"" + catalog + "\"");
            }
        } catch (SQLException e1) {
            e1.printStackTrace();
        }
        return null;
    }

    public static Tags tags(File f, String catalog, Tags t) {
        try {
            if (t == null) {
                ResultSet rs = connection.createStatement().executeQuery("SELECT TAGS FROM TAGS WHERE PATH = \""
                        + f.getPath().replace("\\", "/") + "\" AND CATALOG = \"" + catalog + "\"");
                if (!rs.next())
                    return new Tags();

                return new Tags(rs.getString(1).split(", "));
            } else {
                if (t.getTags().length == 0)
                    connection.createStatement().executeUpdate("DELETE FROM TAGS WHERE PATH = \""
                            + f.getPath().replace("\\", "/") + "\" AND CATALOG = \"" + catalog + "\"");
                else
                    connection.createStatement()
                            .executeUpdate("INSERT OR REPLACE INTO TAGS VALUES(\"" + f.getPath().replace("\\", "/")
                                    + "\", \"" + catalog + "\", \"" + t.serialize() + "\")");
            }
        } catch (SQLException e1) {
            e1.printStackTrace();
        }
        return null;
    }

    public static JSONObject tagdata(String catalog) {
        JSONObject o = new JSONObject();

        try {
            ResultSet rs = connection.createStatement().executeQuery("SELECT * FROM TAGS");
            while (rs.next()) {
                o.put(rs.getString(1), rs.getString(3).split(", "));
            }
        } catch (Exception e) {
            e.printStackTrace();
        }

        return o;
    }

    public static void rename(Packet4Rename packet) {
        try {
            connection.createStatement()
                    .executeUpdate("UPDATE ETICETS SET PATH = \"" + packet.getNewFile().getPath().replace("\\", "/")
                            + "\" WHERE PATH = \"" + packet.getOldFile().getPath().replace("\\", "/") + "\"");
            connection.createStatement()
                    .executeUpdate("UPDATE TAGS SET PATH = \"" + packet.getNewFile().getPath().replace("\\", "/")
                            + "\" WHERE PATH = \"" + packet.getOldFile().getPath().replace("\\", "/") + "\"");
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}