diff --git a/src/main/java/love/sola/netsupport/api/API.java b/src/main/java/love/sola/netsupport/api/API.java index cf90934..778d39c 100644 --- a/src/main/java/love/sola/netsupport/api/API.java +++ b/src/main/java/love/sola/netsupport/api/API.java @@ -1,5 +1,18 @@ package love.sola.netsupport.api; +import love.sola.netsupport.api.manager.GetUser; +import love.sola.netsupport.api.manager.TicketPush; +import love.sola.netsupport.api.root.DashBoard; +import love.sola.netsupport.api.root.FlushCache; +import love.sola.netsupport.api.root.SetPassword; +import love.sola.netsupport.api.stuff.TicketLog; +import love.sola.netsupport.api.stuff.TicketLookup; +import love.sola.netsupport.api.stuff.TicketTrack; +import love.sola.netsupport.api.stuff.TicketUpdate; +import love.sola.netsupport.api.user.ProfileModify; +import love.sola.netsupport.api.user.Register; +import love.sola.netsupport.api.user.TicketQuery; +import love.sola.netsupport.api.user.TicketSubmit; import love.sola.netsupport.enums.Access; import love.sola.netsupport.wechat.Command; import me.chanjar.weixin.common.session.WxSession; @@ -29,4 +42,21 @@ public abstract class API { '}'; } + public static final Class[] LIST = new Class[]{ + GetUser.class, + TicketPush.class, + DashBoard.class, + FlushCache.class, + SetPassword.class, + TicketLog.class, + TicketLookup.class, + TicketTrack.class, + TicketUpdate.class, + ProfileModify.class, + Register.class, + TicketQuery.class, + TicketSubmit.class, + CheckSession.class + }; + } diff --git a/src/main/java/love/sola/netsupport/api/APIRouter.java b/src/main/java/love/sola/netsupport/api/APIRouter.java index ad836bc..2488ca2 100644 --- a/src/main/java/love/sola/netsupport/api/APIRouter.java +++ b/src/main/java/love/sola/netsupport/api/APIRouter.java @@ -1,6 +1,5 @@ package love.sola.netsupport.api; -import com.google.common.reflect.ClassPath; import com.google.gson.Gson; import love.sola.netsupport.enums.Access; import love.sola.netsupport.enums.Attribute; @@ -21,7 +20,6 @@ import java.io.PrintWriter; import java.text.ParseException; import java.util.HashMap; import java.util.Map; -import java.util.Set; /** * *********************************************** @@ -37,10 +35,11 @@ public class APIRouter extends HttpServlet { public APIRouter() { try { - ClassPath path = ClassPath.from(getServletContext().getClassLoader()); - Set classes = path.getTopLevelClasses(); - for (ClassPath.ClassInfo info : classes) { - Class clz = info.load(); +// ClassPath path = ClassPath.from(getServletContext().getClassLoader()); +// Set classes = path.getTopLevelClasses(); +// for (ClassPath.ClassInfo info : classes) { + for (Class clz : API.LIST) { +// Class clz = info.load(); if (!API.class.equals(clz) && API.class.isAssignableFrom(clz)) { try { System.out.print("Loading API: " + clz.getName()); @@ -68,21 +67,7 @@ public class APIRouter extends HttpServlet { try { API api = nodes.get(req.getPathInfo()); if (api == null) { -// resp.sendError(HttpServletResponse.SC_FORBIDDEN); -// obj = nodes; - - ClassPath path = ClassPath.from(getServletContext().getClassLoader().getParent()); - obj = new Object[]{ - Thread.currentThread().getContextClassLoader().toString(), - getServletContext().getClassLoader().toString(), - getClass().getClassLoader().toString(), - getClass().getPackage().getName(), - API.class.getClassLoader().toString(), - CheckSession.class.getClassLoader().toString(), - path.getResources().toString(), - path.getTopLevelClassesRecursive(getClass().getPackage().getName()).toString(), - path.getAllClasses().toString() - }; + resp.sendError(HttpServletResponse.SC_FORBIDDEN); return; } WxSession session = getSession(req);