http-service: add config service

This commit is contained in:
Adam
2017-05-16 16:24:50 -04:00
parent e213d7e888
commit 45640a34be
8 changed files with 354 additions and 3 deletions

View File

@@ -29,6 +29,7 @@ import com.google.inject.Inject;
import net.runelite.http.api.RuneliteAPI;
import net.runelite.http.service.account.AccountService;
import net.runelite.http.service.account.AuthFilter;
import net.runelite.http.service.config.ConfigService;
import net.runelite.http.service.hiscore.HiscoreService;
import net.runelite.http.service.updatecheck.UpdateCheckService;
import net.runelite.http.service.worlds.WorldsService;
@@ -50,6 +51,9 @@ public class Service implements SparkApplication
@Inject
private AccountService accounts;
@Inject
private ConfigService config;
@Inject
private HiscoreService hiscores;
@@ -73,6 +77,7 @@ public class Service implements SparkApplication
{
xtea.init();
accounts.init();
config.init();
get("/version", (request, response) -> RuneliteAPI.getVersion());
get("/update-check", updateCheck::check, transformer);
@@ -88,6 +93,16 @@ public class Service implements SparkApplication
before("/logout", authFilter);
get("/logout", accounts::logout);
});
before("/config", authFilter);
path("/config", () ->
{
// Just using before(authFilter); here doesn't work
before("/*", authFilter);
get("", config::get, transformer);
put("/:key", config::setKey);
delete("/:key", config::unsetKey);
});
exception(Exception.class, (exception, request, response) -> logger.warn(null, exception));
}

View File

@@ -36,6 +36,7 @@ import javax.naming.NamingException;
import javax.sql.DataSource;
import net.runelite.http.service.account.AccountService;
import net.runelite.http.service.account.AuthFilter;
import net.runelite.http.service.config.ConfigService;
import net.runelite.http.service.hiscore.HiscoreService;
import net.runelite.http.service.updatecheck.UpdateCheckService;
import net.runelite.http.service.worlds.WorldsService;
@@ -118,6 +119,7 @@ public class ServiceModule extends AbstractModule
bind(AuthFilter.class);
bind(AccountService.class);
bind(ConfigService.class);
bind(HiscoreService.class);
bind(UpdateCheckService.class);
bind(WorldsService.class);

View File

@@ -30,6 +30,7 @@ import com.google.inject.name.Named;
import java.sql.Timestamp;
import java.time.Instant;
import java.util.UUID;
import net.runelite.http.api.RuneliteAPI;
import org.sql2o.Connection;
import org.sql2o.Sql2o;
import spark.Filter;
@@ -40,8 +41,6 @@ import static spark.Spark.halt;
public class AuthFilter implements Filter
{
private static final String RUNELITE_AUTH = "RUNELITE-AUTH";
private final Sql2o sql2o;
@Inject
@@ -53,7 +52,7 @@ public class AuthFilter implements Filter
@Override
public void handle(Request request, Response response) throws Exception
{
String runeliteAuth = request.headers(RUNELITE_AUTH);
String runeliteAuth = request.headers(RuneliteAPI.RUNELITE_AUTH);
if (runeliteAuth == null)
{
halt(401, "Access denied");

View File

@@ -0,0 +1,124 @@
/*
* Copyright (c) 2017, Adam <Adam@sigterm.info>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package net.runelite.http.service.config;
import com.google.inject.Inject;
import com.google.inject.name.Named;
import java.util.List;
import net.runelite.http.api.config.ConfigEntry;
import net.runelite.http.api.config.Configuration;
import net.runelite.http.service.account.beans.SessionEntry;
import org.sql2o.Connection;
import org.sql2o.Sql2o;
import org.sql2o.Sql2oException;
import spark.Request;
import spark.Response;
public class ConfigService
{
private static final String CREATE_CONFIG = "CREATE TABLE IF NOT EXISTS `config` (\n"
+ " `user` int(11) NOT NULL,\n"
+ " `key` tinytext NOT NULL,\n"
+ " `value` text NOT NULL,\n"
+ " UNIQUE KEY `user_key` (`user`,`key`(64))\n"
+ ") ENGINE=InnoDB;";
private static final String CONFIG_FK = "ALTER TABLE `config`\n"
+ " ADD CONSTRAINT `user_fk` FOREIGN KEY (`user`) REFERENCES `users` (`id`) ON DELETE CASCADE ON UPDATE CASCADE;";
private final Sql2o sql2o;
@Inject
public ConfigService(@Named("Runelite SQL2O") Sql2o sql2o)
{
this.sql2o = sql2o;
}
public void init()
{
try (Connection con = sql2o.open())
{
con.createQuery(CREATE_CONFIG)
.executeUpdate();
try
{
con.createQuery(CONFIG_FK)
.executeUpdate();
}
catch (Sql2oException ex)
{
// Ignore, happens when index already exists
}
}
}
public Configuration get(Request request, Response response)
{
SessionEntry session = request.session().attribute("session");
List<ConfigEntry> config;
try (Connection con = sql2o.open())
{
config = con.createQuery("select `key`, value from config where user = :user")
.addParameter("user", session.getUser())
.executeAndFetch(ConfigEntry.class);
}
return new Configuration(config);
}
public Object setKey(Request request, Response response)
{
SessionEntry session = request.session().attribute("session");
try (Connection con = sql2o.open())
{
con.createQuery("insert into config (user, `key`, value) values (:user, :key, :value) on duplicate key update value = :value")
.addParameter("user", session.getUser())
.addParameter("key", request.params("key"))
.addParameter("value", request.body())
.executeUpdate();
}
return "";
}
public Object unsetKey(Request request, Response response)
{
SessionEntry session = request.session().attribute("session");
try (Connection con = sql2o.open())
{
con.createQuery("delete from config where user = :user and `key` = :key")
.addParameter("user", session.getUser())
.addParameter("key", request.params("key"))
.executeUpdate();
}
return "";
}
}