diff --git a/biliapi/pom.xml b/biliapi/pom.xml new file mode 100644 index 0000000..6c77166 --- /dev/null +++ b/biliapi/pom.xml @@ -0,0 +1,26 @@ + + + 4.0.0 + + com.yutou + bilibili + 0.0.1-SNAPSHOT + + + biliapi + + + 21 + 21 + UTF-8 + + + + io.reactivex.rxjava3 + rxjava + 3.1.8 + + + \ No newline at end of file diff --git a/biliapi/src/main/java/com/yutou/bili/api/LoginApi.java b/biliapi/src/main/java/com/yutou/bili/api/LoginApi.java new file mode 100644 index 0000000..54ecfa2 --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/api/LoginApi.java @@ -0,0 +1,18 @@ +package com.yutou.bili.api; + +import com.yutou.bili.bean.login.LoginInfoBean; +import com.yutou.bili.bean.login.QRCodeGenerateBean; +import com.yutou.okhttp.HttpBody; +import retrofit2.Call; +import retrofit2.http.GET; +import retrofit2.http.Query; + +public interface LoginApi { + @GET("/x/passport-login/web/qrcode/generate") + Call> getQRCodeGenerate(); + + @GET("/x/passport-login/web/qrcode/poll") + Call> loginQRCode(@Query("qrcode_key") String qrcode_key); + + +} diff --git a/biliapi/src/main/java/com/yutou/bili/api/UserApi.java b/biliapi/src/main/java/com/yutou/bili/api/UserApi.java new file mode 100644 index 0000000..20440e4 --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/api/UserApi.java @@ -0,0 +1,11 @@ +package com.yutou.bili.api; + +import com.yutou.bili.bean.login.UserInfoBean; +import com.yutou.okhttp.HttpBody; +import retrofit2.Call; +import retrofit2.http.GET; + +public interface UserApi { + @GET("/x/web-interface/nav") + Call> getUserInfo(); +} diff --git a/biliapi/src/main/java/com/yutou/bili/bean/login/LoginCookie.java b/biliapi/src/main/java/com/yutou/bili/bean/login/LoginCookie.java new file mode 100644 index 0000000..1b9b436 --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/bean/login/LoginCookie.java @@ -0,0 +1,34 @@ +package com.yutou.bili.bean.login; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.google.gson.annotations.SerializedName; +import com.yutou.databases.AbsDatabasesBean; +import lombok.Data; +import lombok.EqualsAndHashCode; + +@EqualsAndHashCode(callSuper = true) +@Data +public class LoginCookie extends AbsDatabasesBean { + @JSONField(name = "SESSDATA") + String sessdta; + @JSONField(name = "Path") + String path; + @JSONField(name = "DedeUserID") + String dedeUserID; + @JSONField(name = "DedeUserID__ckMd5") + String dedeUserIDCkMd5; + @JSONField(name = "bili_jct") + String biliJct; + @JSONField(name = "Expires") + String expires; + @JSONField(name = "Domain") + String domain; + @JSONField(name = "sid") + String sid; + @JSONField(name = "gourl") + String gourl; + + public LoginCookie() { + setTableName("login_cookie"); + } +} diff --git a/biliapi/src/main/java/com/yutou/bili/bean/login/LoginInfoBean.java b/biliapi/src/main/java/com/yutou/bili/bean/login/LoginInfoBean.java new file mode 100644 index 0000000..c66de42 --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/bean/login/LoginInfoBean.java @@ -0,0 +1,17 @@ +package com.yutou.bili.bean.login; + +import com.yutou.okhttp.BaseBean; +import lombok.Data; +import lombok.EqualsAndHashCode; + +@EqualsAndHashCode(callSuper = true) +@Data +public class LoginInfoBean extends BaseBean { + private String url; + private String refresh_token; + private long timestamp; + private int code; + private String message; + + +} diff --git a/biliapi/src/main/java/com/yutou/bili/bean/login/QRCodeGenerateBean.java b/biliapi/src/main/java/com/yutou/bili/bean/login/QRCodeGenerateBean.java new file mode 100644 index 0000000..02fa07f --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/bean/login/QRCodeGenerateBean.java @@ -0,0 +1,15 @@ +package com.yutou.bili.bean.login; + + +import com.yutou.okhttp.BaseBean; +import lombok.Data; +import lombok.EqualsAndHashCode; + +@EqualsAndHashCode(callSuper = true) +@Data +public class QRCodeGenerateBean extends BaseBean { + + private String url; + private String qrcode_key; + +} diff --git a/biliapi/src/main/java/com/yutou/bili/bean/login/UserInfoBean.java b/biliapi/src/main/java/com/yutou/bili/bean/login/UserInfoBean.java new file mode 100644 index 0000000..72482a8 --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/bean/login/UserInfoBean.java @@ -0,0 +1,288 @@ +package com.yutou.bili.bean.login; + + +import com.alibaba.fastjson2.annotation.JSONField; +import com.google.gson.annotations.SerializedName; +import com.yutou.okhttp.BaseBean; +import lombok.Data; +import lombok.EqualsAndHashCode; + +import java.util.Date; + +@EqualsAndHashCode(callSuper = true) +@Data +public class UserInfoBean extends BaseBean { + + @JSONField(name = "isLogin") + private boolean isLogin; + + @JSONField(name = "email_verified") + private int emailVerified; + + @JSONField(name = "face") + private String face; + + @JSONField(name = "face_nft") + private int faceNft; + + @JSONField(name = "face_nft_type") + private int faceNftType; + + @JSONField(name = "level_info") + private LevelInfo levelInfo; + + @JSONField(name = "mid") + private long mid; + + @JSONField(name = "mobile_verified") + private int mobileVerified; + + @JSONField(name = "money") + private double money; + + @JSONField(name = "moral") + private int moral; + + @JSONField(name = "official") + private Official official; + + @JSONField(name = "officialVerify") + private OfficialVerify officialVerify; + + @JSONField(name = "pendant") + private Pendant pendant; + + @JSONField(name = "scores") + private int scores; + + @JSONField(name = "uname") + private String uname; + + @JSONField(name = "vipDueDate") + private Date vipDueDate; + + @JSONField(name = "vipStatus") + private int vipStatus; + + @JSONField(name = "vipType") + private int vipType; + + @JSONField(name = "vip_pay_type") + private int vipPayType; + + @JSONField(name = "vip_theme_type") + private int vipThemeType; + + @JSONField(name = "vip_label") + private VipLabel vipLabel; + + @JSONField(name = "vip_avatar_subscript") + private int vipAvatarSubscript; + + @JSONField(name = "vip_nickname_color") + private String vipNicknameColor; + + @JSONField(name = "vip") + private Vip vip; + + @JSONField(name = "wallet") + private Wallet wallet; + + @JSONField(name = "has_shop") + private boolean hasShop; + + @JSONField(name = "shop_url") + private String shopUrl; + + @JSONField(name = "allowance_count") + private int allowanceCount; + + @JSONField(name = "answer_status") + private int answerStatus; + + @JSONField(name = "is_senior_member") + private boolean isSeniorMember; + + @JSONField(name = "wbi_img") + private WbiImg wbiImg; + + @JSONField(name = "is_jury") + private boolean isJury; + + // 省略了getters和setters + + // level_info的内部数据结构 + @Data + public static class LevelInfo { + @JSONField(name = "current_level") + private int currentLevel; + + @JSONField(name = "current_min") + private int currentMin; + + @JSONField(name = "current_exp") + private int currentExp; + + @JSONField(name = "next_exp") + private String nextExp; + + // getters and setters + } + + // official的结构 + @Data + public static class Official { + @JSONField(name = "role") + private int role; + + @JSONField(name = "title") + private String title; + + @JSONField(name = "desc") + private String desc; + + @JSONField(name = "type") + private int type; + + // getters and setters + } + + // officialVerify的结构 + @Data + public static class OfficialVerify { + @JSONField(name = "type") + private int type; + + @JSONField(name = "desc") + private String desc; + + // getters and setters + } + + // pendant的结构 + @Data + public static class Pendant { + @JSONField(name = "pid") + private int pid; + + @JSONField(name = "name") + private String name; + + @JSONField(name = "image") + private String image; + + @JSONField(name = "expire") + private int expire; + + @JSONField(name = "image_enhance") + private String imageEnhance; + + @JSONField(name = "image_enhance_frame") + private String imageEnhanceFrame; + + @JSONField(name = "n_pid") + private int nPid; + + // getters and setters + } + + // vipLabel的结构 + @Data + public static class VipLabel { + @JSONField(name = "path") + private String path; + + @JSONField(name = "text") + private String text; + + @JSONField(name = "label_theme") + private String labelTheme; + + @JSONField(name = "text_color") + private String textColor; + + @JSONField(name = "bg_style") + private String bgStyle; + + @JSONField(name = "bg_color") + private String bgColor; + + @JSONField(name = "border_color") + private String borderColor; + + @JSONField(name = "use_img_label") + private boolean useImgLabel; + + @JSONField(name = "img_label_uri_hans") + private String imgLabelUriHans; + + @JSONField(name = "img_label_uri_hant") + private String imgLabelUriHant; + + @JSONField(name = "img_label_uri_hans_static") + private String imgLabelUriHansStatic; + + @JSONField(name = "img_label_uri_hant_static") + private String imgLabelUriHantStatic; + + // getters and setters + } + + // vip的结构，和vipLabel类似 + @Data + public static class Vip { + @JSONField(name = "type") + private int type; + + @JSONField(name = "status") + private int status; + + @JSONField(name = "due_date") + private Date dueDate; + + @JSONField(name = "vip_pay_type") + private int vipPayType; + + @JSONField(name = "theme_type") + private int themeType; + + @JSONField(name = "label") + private VipLabel label; + + @JSONField(name = "avatar_subscript") + private int avatarSubscript; + + @JSONField(name = "nickname_color") + private String nicknameColor; + // getters and setters + } + + // wallet的结构 + @Data + public static class Wallet { + @JSONField(name = "mid") + private long mid; + + @JSONField(name = "bcoin_balance") + private int bcoinBalance; + + @JSONField(name = "coupon_balance") + private int couponBalance; + + @JSONField(name = "coupon_due_time") + private int couponDueTime; + // getters and setters + } + + // WbiImg的结构 + @Data + public static class WbiImg { + @JSONField(name = "img_url") + private String imgUrl; + + @JSONField(name = "sub_url") + private String subUrl; + + // getters and setters + } +} diff --git a/biliapi/src/main/java/com/yutou/bili/databases/BiliBiliLoginDatabase.java b/biliapi/src/main/java/com/yutou/bili/databases/BiliBiliLoginDatabase.java new file mode 100644 index 0000000..983c0df --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/databases/BiliBiliLoginDatabase.java @@ -0,0 +1,49 @@ +package com.yutou.bili.databases; + +import com.yutou.bili.bean.login.LoginCookie; +import com.yutou.databases.SQLiteManager; + +import java.util.List; + +public class BiliBiliLoginDatabase extends SQLiteManager { + LoginCookie cookie; + private static BiliBiliLoginDatabase instance; + + private BiliBiliLoginDatabase(Class tClass) { + try { + cookie = new LoginCookie(); + } catch (Exception e) { + throw new RuntimeException(e); + } + init(); + } + + public static BiliBiliLoginDatabase getInstance() { + if (instance == null) { + instance = new BiliBiliLoginDatabase(LoginCookie.class); + } + return instance; + } + + + public BiliBiliLoginDatabase initData(LoginCookie cookie) { + this.cookie = cookie; + for (BuildSqlTable table : build.getTable()) { + cookie.setTableName(table.getName()); + add(cookie); + } + return this; + } + + public List get() { + return super.get(cookie.getTableName(), LoginCookie.class); + } + + @Override + protected LoginCookie getDataBean() { + return new LoginCookie(); + } +} +/** + * + */ \ No newline at end of file diff --git a/biliapi/src/main/java/com/yutou/bili/net/BiliLoginNetApiManager.java b/biliapi/src/main/java/com/yutou/bili/net/BiliLoginNetApiManager.java new file mode 100644 index 0000000..dbc8b90 --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/net/BiliLoginNetApiManager.java @@ -0,0 +1,118 @@ +package com.yutou.bili.net; + +import com.alibaba.fastjson2.JSONObject; +import com.yutou.bili.api.LoginApi; +import com.yutou.bili.bean.login.LoginCookie; +import com.yutou.bili.bean.login.LoginInfoBean; +import com.yutou.bili.bean.login.QRCodeGenerateBean; +import com.yutou.bili.databases.BiliBiliLoginDatabase; +import com.yutou.databases.AbsDatabasesBean; +import com.yutou.okhttp.HttpBody; +import com.yutou.okhttp.HttpCallback; +import com.yutou.okhttp.api.BaseApi; +import okhttp3.Headers; +import org.springframework.util.StringUtils; +import retrofit2.Response; + +import java.io.IOException; +import java.text.ParsePosition; +import java.text.SimpleDateFormat; +import java.util.*; + +public class BiliLoginNetApiManager extends BaseApi { + public static final int LOGIN_QRCODE = 100; + public static final int LOGIN_SUCCESS = 101; + + private static BiliLoginNetApiManager instance; + LoginApi loginApi; + + private BiliLoginNetApiManager() { + super("https://passport.bilibili.com"); + loginApi = createApi(LoginApi.class); + } + + public static BiliLoginNetApiManager getInstance() { + if (instance == null) { + instance = new BiliLoginNetApiManager(); + } + return instance; + } + + public void login(HttpCallback callback) { + List cookie = BiliBiliLoginDatabase.getInstance().get(); + if (!cookie.isEmpty()) { + callback.onResponse(null, LOGIN_SUCCESS, null, cookie.get(0), null); + return; + } + loginApi.getQRCodeGenerate().enqueue(new HttpCallback() { + @Override + public void onResponse(Headers headers, int code, String status, QRCodeGenerateBean response, String rawResponse) { + String oauthKey = response.getQrcode_key(); + String url = response.getUrl(); + String bd = "gourl=https%3A%2F%2Fpassport.bilibili.com%2Fajax%2FminiLogin%2Fredirect&oauthKey=" + oauthKey; + callback.onResponse(headers, LOGIN_QRCODE, null, null, url); + new Thread(() -> waitLogin(oauthKey, callback)).start(); + } + + @Override + public void onFailure(Throwable throwable) { + + } + }); + } + + private void waitLogin(String oauthKey, HttpCallback callback) { + long time = System.currentTimeMillis(); + new Timer().schedule(new TimerTask() { + @Override + public void run() { + String bd = "gourl=https%3A%2F%2Fpassport.bilibili.com%2Fajax%2FminiLogin%2Fredirect&oauthKey=" + oauthKey; + if ((System.currentTimeMillis() - time) > 5 * 60 * 1000) { + cancel(); + return; + } + + try { + Response> response = loginApi.loginQRCode(oauthKey).execute(); + Headers headers = response.headers(); + HttpBody httpBody = response.body(); + if (httpBody.getData().getCode() == 0) { + JSONObject ck = new JSONObject(); + List list = headers.values("set-cookie"); + for (String cookie : list) { + String[] split = cookie.split(";"); + for (String string : split) { + if (!ck.containsKey(string) && !StringUtils.isEmpty(string) && string.contains("=")) { + String key = string.split("=")[0].trim(); + String value = string.split("=")[1].trim(); + if (key.contains("Expires")) { + SimpleDateFormat sdf = new SimpleDateFormat("EEE, dd MMMM yyyy HH:mm:ss z", Locale.ENGLISH); + sdf.setTimeZone(TimeZone.getDefault()); + Date date = sdf.parse(value, new ParsePosition(0)); + value = String.valueOf(date.getTime() / 1000); + } + ck.put(key, value); + } + } + } + if (!list.isEmpty()) { + ck.put("gourl", bd); + LoginCookie cookie = JSONObject.parseObject(ck.toString(), LoginCookie.class); + cancel(); + callback.onResponse(headers, LOGIN_SUCCESS, "ok", cookie, ck.toString()); + + } else { + callback.onFailure(new RuntimeException("cookie为空")); + cancel(); + } + } + } catch (IOException e) { + cancel(); + throw new RuntimeException(e); + } + + } + }, 1000, 3000); + } + +} diff --git a/biliapi/src/main/java/com/yutou/bili/net/BiliUserNetApiManager.java b/biliapi/src/main/java/com/yutou/bili/net/BiliUserNetApiManager.java new file mode 100644 index 0000000..c4a9688 --- /dev/null +++ b/biliapi/src/main/java/com/yutou/bili/net/BiliUserNetApiManager.java @@ -0,0 +1,47 @@ +package com.yutou.bili.net; + +import com.alibaba.fastjson2.JSONObject; +import com.yutou.bili.api.UserApi; +import com.yutou.bili.bean.login.LoginCookie; +import com.yutou.bili.databases.BiliBiliLoginDatabase; +import com.yutou.inter.IHttpApiCheckCallback; +import com.yutou.okhttp.api.BaseApi; + +import java.util.HashMap; +import java.util.List; + +public class BiliUserNetApiManager extends BaseApi { + public static final int LOGIN_LOGOUT = 1; + + public static BiliUserNetApiManager manager; + + private BiliUserNetApiManager() { + super("https://api.bilibili.com"); + } + + public static BiliUserNetApiManager getInstance() { + if (manager == null) { + manager = new BiliUserNetApiManager(); + } + return manager; + } + + public void getUserApi(IHttpApiCheckCallback callback) { + List list = BiliBiliLoginDatabase.getInstance().get(); + if (!list.isEmpty()) { + HashMap headers = new HashMap<>(); + LoginCookie cookie = list.get(0); + JSONObject json = JSONObject.parseObject(JSONObject.toJSONString(cookie)); + StringBuilder ck=new StringBuilder(); + for (String key : json.keySet()) { + ck.append(key).append("=").append(json.getString(key)).append("; "); + } + headers.put("Cookie", ck.toString()); + setHeaders(headers); + callback.onSuccess(createApi(UserApi.class)); + } else { + callback.onError(LOGIN_LOGOUT, "未登录"); + } + } + +} diff --git a/common/pom.xml b/common/pom.xml new file mode 100644 index 0000000..6b8bcf9 --- /dev/null +++ b/common/pom.xml @@ -0,0 +1,34 @@ + + + 4.0.0 + + com.yutou + bilibili + 0.0.1-SNAPSHOT + + + common + + + 21 + 21 + UTF-8 + + + + + com.squareup.okhttp3 + okhttp + 4.12.0 + + + + + com.squareup.retrofit2 + retrofit + 2.11.0 + + + \ No newline at end of file diff --git a/common/src/main/java/com/yutou/databases/AbsDatabasesBean.java b/common/src/main/java/com/yutou/databases/AbsDatabasesBean.java new file mode 100644 index 0000000..fcf5cdb --- /dev/null +++ b/common/src/main/java/com/yutou/databases/AbsDatabasesBean.java @@ -0,0 +1,17 @@ +package com.yutou.databases; + +import com.alibaba.fastjson2.JSONObject; +import com.alibaba.fastjson2.JSONWriter; +import lombok.Data; + +@Data +public class AbsDatabasesBean { + String tableName; + + + public JSONObject toJson() { + JSONObject json = JSONObject.parseObject(JSONObject.toJSONString(this, JSONWriter.Feature.WriteMapNullValue)); + json.remove("tableName"); + return json; + } +} diff --git a/common/src/main/java/com/yutou/databases/SQLiteManager.java b/common/src/main/java/com/yutou/databases/SQLiteManager.java new file mode 100644 index 0000000..11f9e7f --- /dev/null +++ b/common/src/main/java/com/yutou/databases/SQLiteManager.java @@ -0,0 +1,302 @@ +package com.yutou.databases; + + +import com.alibaba.fastjson2.JSONObject; +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.utils.Log; +import lombok.Data; +import lombok.Getter; + +import java.io.File; +import java.lang.reflect.Field; +import java.sql.*; +import java.util.ArrayList; +import java.util.Date; +import java.util.List; + +public abstract class SQLiteManager { + protected Connection conn; + private String url = "jdbc:sqlite:"; + private File sql; + @Getter + public BuildSql build; + + public void init() { + AbsDatabasesBean data = getDataBean(); + List items = new ArrayList<>(); + Field[] fields = data.getClass().getDeclaredFields(); + for (Field field : fields) { + String name = field.getAnnotation(JSONField.class).name(); + if (name.equals("tableName")) continue; + String type = BuildSqlItem.TYPE_STRING; + if (field.getType() == int.class) { + type = BuildSqlItem.TYPE_INT; + } else if (field.getType() == String.class) { + type = BuildSqlItem.TYPE_STRING; + } else if (field.getType() == Date.class) { + type = BuildSqlItem.TYPE_TIME; + } + items.add(BuildSqlItem.create().setName(name).setType(type).setNull(true).setKey(false)); + } + BuildSql sql = BuildSql.create() + .setFileName("bilibili_login.db") + .setTable(BuildSqlTable.create().setName(data.getTableName()).setItem(items)); + build(sql); + } + + + public void startBatch() { + try { + conn.setAutoCommit(false); + } catch (SQLException e) { + Log.e(e); + } + } + + public void closeBatch() { + try { + conn.setAutoCommit(true); + } catch (SQLException e) { + Log.e(e); + } + } + + public void close() { + try { + conn.close(); + } catch (SQLException throwables) { + throwables.printStackTrace(); + } + } + + public void commit() { + try { + conn.commit(); + } catch (SQLException e) { + Log.e(e); + } + } + + public void add(T t) { + try { + Statement statement = conn.createStatement(); + StringBuilder sb = new StringBuilder(); + StringBuilder value = new StringBuilder(); + sb.append("INSERT INTO `").append(t.getTableName()).append("` "); + sb.append("("); + value.append("("); + for (String key : t.toJson().keySet()) { + sb.append("`").append(key).append("`,"); + value.append("'").append(t.toJson().get(key)).append("',"); + } + sb.deleteCharAt(sb.length() - 1); + value.deleteCharAt(value.length() - 1); + value.append(")"); + sb.append(") VALUES "); + sb.append(value); + statement.executeUpdate(sb.toString()); + statement.close(); + } catch (SQLException e) { + throw new RuntimeException(e); + } + } + + public List get(String table, Class tClass) { + List list = new ArrayList<>(); + try { + Statement statement = conn.createStatement(); + String sql = "SELECT * FROM `" + table + "`"; + ResultSet resultSet = statement.executeQuery(sql); + while (resultSet.next()) { + JSONObject json = new JSONObject(); + for (int i = 0; i < resultSet.getMetaData().getColumnCount(); i++) { + String name = resultSet.getMetaData().getColumnName(i + 1); + json.put(name, resultSet.getObject(name)); + } + list.add(json.to(tClass)); + } + resultSet.close(); + statement.closeOnCompletion(); + } catch (Exception e) { + throw new RuntimeException(e); + } + return list; + } + + + private void createSql(BuildSql build) { + try { + sql.mkdirs(); + sql.delete(); + conn = DriverManager.getConnection(url + sql.getAbsolutePath()); + } catch (Exception e) { + Log.e(e); + } + startBatch(); + for (BuildSqlTable table : build.getTable()) { + Log.i("创建表:" + table.getName()); + createSqlOfTable(table); + } + closeBatch(); + } + + private void createSqlOfTable(BuildSqlTable table) { + String tableName = table.getName(); + try { + Statement statement = conn.createStatement(); + List items = table.getItem(); + StringBuilder sql = new StringBuilder(); + sql.append("CREATE TABLE `") + .append(tableName) + .append("` ("); + for (BuildSqlItem item : items) { + StringBuilder builder = new StringBuilder(); + String type = switch (item.getType()) { + case "int" -> " INTEGER "; + case "TIME" -> " NUMERIC "; + default -> " TEXT "; + }; + builder.append("`") + .append(item.getName()) + .append("`") + .append(type) + .append(item.isNull() ? "" : " NOT NULL ") + .append(item.isKey() ? " PRIMARY KEY AUTOINCREMENT " : "") + .append(","); + sql.append(builder.toString()); + } + sql.append(");"); + System.out.println("创建表 > " + sql.toString()); + statement.execute(sql.toString().replace(",);", ");")); + statement.closeOnCompletion(); + } catch (Exception e) { + Log.e(e); + } + } + + protected void build(BuildSql buildSql) { + try { + this.build = buildSql; + Class.forName("org.sqlite.JDBC"); + sql = new File("databases" + File.separator + buildSql.getFileName()); + if (!sql.exists()) { + createSql(buildSql); + } else { + conn = DriverManager.getConnection(url + sql.getAbsolutePath()); + } + } catch (Exception e) { + Log.e(e); + } + } + + public boolean setDB(String fileName) { + try { + Class.forName("org.sqlite.JDBC"); + sql = new File("db" + File.separator + fileName); + if (sql.exists()) { + if (conn != null && !conn.isClosed()) { + conn.close(); + } + conn = DriverManager.getConnection(url + sql.getAbsolutePath()); + return true; + } + } catch (Exception e) { + Log.e(e); + } + return false; + } + + + protected abstract AbsDatabasesBean getDataBean(); + + public static void main(String[] args) { + + } + + @Data + public static class BuildSql { + String fileName; + List table; + + public static BuildSql create() { + return new BuildSql(); + } + + public BuildSql setFileName(String fileName) { + this.fileName = fileName; + return this; + } + + public BuildSql setTable(List table) { + this.table = table; + return this; + } + + public BuildSql setTable(BuildSqlTable... table) { + this.table = List.of(table); + return this; + } + } + + @Data + public static class BuildSqlTable { + String name; + List item; + + public static BuildSqlTable create() { + return new BuildSqlTable(); + } + + public BuildSqlTable setName(String name) { + this.name = name; + return this; + } + + public BuildSqlTable setItem(List item) { + this.item = item; + return this; + } + + public BuildSqlTable setItem(BuildSqlItem... item) { + this.item = List.of(item); + return this; + } + } + + @Data + public static class BuildSqlItem { + public static final String TYPE_INT = "init"; + public static final String TYPE_STRING = "String"; + public static final String TYPE_TIME = "TIME"; + + + String name; + String type; + boolean isNull; + boolean isKey; + + public static BuildSqlItem create() { + return new BuildSqlItem(); + } + + public BuildSqlItem setName(String name) { + this.name = name; + return this; + } + + public BuildSqlItem setType(String type) { + this.type = type; + return this; + } + + public BuildSqlItem setNull(boolean aNull) { + isNull = aNull; + return this; + } + + public BuildSqlItem setKey(boolean key) { + isKey = key; + return this; + } + } +} diff --git a/common/src/main/java/com/yutou/inter/IHttpApiCheckCallback.java b/common/src/main/java/com/yutou/inter/IHttpApiCheckCallback.java new file mode 100644 index 0000000..17a7588 --- /dev/null +++ b/common/src/main/java/com/yutou/inter/IHttpApiCheckCallback.java @@ -0,0 +1,7 @@ +package com.yutou.inter; + +public interface IHttpApiCheckCallback { + void onSuccess(T api); + + void onError(int code, String error); +} diff --git a/common/src/main/java/com/yutou/okhttp/BaseBean.java b/common/src/main/java/com/yutou/okhttp/BaseBean.java new file mode 100644 index 0000000..ffca9c9 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/BaseBean.java @@ -0,0 +1,6 @@ +package com.yutou.okhttp; + +import java.io.Serializable; + +public class BaseBean implements Serializable { +} diff --git a/common/src/main/java/com/yutou/okhttp/GetRequestParams.java b/common/src/main/java/com/yutou/okhttp/GetRequestParams.java new file mode 100644 index 0000000..c574f91 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/GetRequestParams.java @@ -0,0 +1,32 @@ +package com.yutou.okhttp; + + +import okhttp3.Headers; +import okhttp3.HttpUrl; +import okhttp3.Request; + +import java.util.HashMap; + +public class GetRequestParams implements IRequestParam { + /** + * 构建Request + * + * @param request + * @return + */ + @Override + public Request getRequest(HashMap headerMap, HashMap map, Request request) { + Headers.Builder headerBuild = request.headers().newBuilder(); + if (!headerMap.isEmpty()) { + for (String key : headerMap.keySet()) { + headerBuild.add(key, headerMap.get(key)); + } + } + //添加公共参数 + HttpUrl.Builder builder = request.url().newBuilder(); + for (String key : map.keySet()) { + builder.addQueryParameter(key, String.valueOf(map.get(key))); + } + return request.newBuilder().url(builder.build()).headers(headerBuild.build()).build(); + } +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/HttpBody.java b/common/src/main/java/com/yutou/okhttp/HttpBody.java new file mode 100644 index 0000000..7d8588e --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/HttpBody.java @@ -0,0 +1,14 @@ +package com.yutou.okhttp; + +import lombok.Data; + +@Data +public class HttpBody { + private String msg; + private String status; + private int code; + private int retcode; + private T data; + private String src; + +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/HttpCallback.java b/common/src/main/java/com/yutou/okhttp/HttpCallback.java new file mode 100644 index 0000000..147a502 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/HttpCallback.java @@ -0,0 +1,35 @@ +package com.yutou.okhttp; + +import okhttp3.Headers; +import retrofit2.Call; +import retrofit2.Callback; +import retrofit2.Response; + +public abstract class HttpCallback implements Callback> { + + public abstract void onResponse(Headers headers,int code, String status, T response, String rawResponse); + + public abstract void onFailure(Throwable throwable); + + @Override + public void onResponse(Call> call, Response> response) { + if (response.body() != null) { + onResponse( + response.headers(), + response.body().getRetcode(), + response.body().getStatus(), + response.body().getData(), + response.body().getSrc() + ); + } else { + onFailure(new NullPointerException("response body is null")); + } + call.cancel(); + } + + @Override + public void onFailure(Call> call, Throwable throwable) { + onFailure(throwable); + call.cancel(); + } +} diff --git a/common/src/main/java/com/yutou/okhttp/HttpLoggingInterceptor.java b/common/src/main/java/com/yutou/okhttp/HttpLoggingInterceptor.java new file mode 100644 index 0000000..ac6e501 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/HttpLoggingInterceptor.java @@ -0,0 +1,209 @@ +package com.yutou.okhttp; + + +import com.yutou.utils.Log; +import okhttp3.*; +import okhttp3.internal.http.HttpHeaders; +import okio.Buffer; + +import java.io.IOException; +import java.nio.charset.Charset; +import java.util.concurrent.TimeUnit; +import java.util.logging.Logger; + +public class HttpLoggingInterceptor implements Interceptor { + private static final String TAG = "HttpLogging"; + + private static final Charset UTF8 = Charset.forName("UTF-8"); + + private volatile Level printLevel = Level.NONE; + private java.util.logging.Level colorLevel; + private Logger logger; + + private static boolean prLog; + + public static void setLog(boolean log) { + prLog = log; + } + + public enum Level { + NONE, //不打印log + BASIC, //只打印 请求首行 和 响应首行 + HEADERS, //打印请求和响应的所有 Header + BODY //所有数据全部打印 + } + + public HttpLoggingInterceptor(String tag) { + logger = Logger.getLogger(tag); + colorLevel = java.util.logging.Level.INFO; + } + + public void setPrintLevel(Level level) { + if (printLevel == null) + throw new NullPointerException("printLevel == null. Use Level.NONE instead."); + printLevel = level; + } + + public void setColorLevel(java.util.logging.Level level) { + colorLevel = level; + } + + private void log(String message) { + //logger.log(colorLevel, message); + if (prLog) { + Log.i(TAG, message); + } + //Log.e(TAG,message); + } + + @Override + public Response intercept(Chain chain) throws IOException { + Request request = chain.request(); + if (request.body() != null && request.body().contentLength() == 0) { + request = chain.call().request(); + } + if (printLevel == Level.NONE) { + return chain.proceed(request); + } + + //请求日志拦截 + logForRequest(request, chain.connection()); + + //执行请求，计算请求时间 + long startNs = System.nanoTime(); + Response response; + try { + response = chain.proceed(request); + } catch (Exception e) { + log("<-- HTTP FAILED: " + e); + throw e; + } + long tookMs = TimeUnit.NANOSECONDS.toMillis(System.nanoTime() - startNs); + + //响应日志拦截 + return logForResponse(response, tookMs); + } + + private void logForRequest(Request request, Connection connection) throws IOException { + boolean logBody = (printLevel == Level.BODY); + boolean logHeaders = (printLevel == Level.BODY || printLevel == Level.HEADERS); + RequestBody requestBody = request.body(); + boolean hasRequestBody = requestBody != null; + Protocol protocol = connection != null ? connection.protocol() : Protocol.HTTP_1_1; + + try { + String requestStartMessage = "--> " + request.method() + ' ' + request.url() + ' ' + protocol; + log(requestStartMessage); + + if (logHeaders) { + if (hasRequestBody) { + // Request body headers are only present when installed as a network interceptor. Force + // them to be included (when available) so there values are known. + if (requestBody.contentType() != null) { + log("\tContent-Type: " + requestBody.contentType()); + } + if (requestBody.contentLength() != -1) { + log("\tContent-Length: " + requestBody.contentLength()); + } + } + Headers headers = request.headers(); + for (int i = 0, count = headers.size(); i < count; i++) { + String name = headers.name(i); + // Skip headers from the request body as they are explicitly logged above. + if (!"Content-Type".equalsIgnoreCase(name) && !"Content-Length".equalsIgnoreCase(name)) { + log("\t" + name + ": " + headers.value(i)); + } + } + + log(" "); + if (logBody && hasRequestBody) { + if (isPlaintext(requestBody.contentType())) { + bodyToString(request); + } else { + log("\tbody: maybe [binary body], omitted!"); + } + } + } + } catch (Exception e) { + logger.log(java.util.logging.Level.WARNING, e.getMessage(), e); + } finally { + log("--> END " + request.method()); + } + } + + private Response logForResponse(Response response, long tookMs) { + Response.Builder builder = response.newBuilder(); + Response clone = builder.build(); + ResponseBody responseBody = clone.body(); + boolean logBody = (printLevel == Level.BODY); + boolean logHeaders = (printLevel == Level.BODY || printLevel == Level.HEADERS); + + try { + log("<-- " + clone.code() + ' ' + clone.message() + ' ' + clone.request().url() + " (" + tookMs + "ms）"); + if (logHeaders) { + Headers headers = clone.headers(); + for (int i = 0, count = headers.size(); i < count; i++) { + log("\t" + headers.name(i) + ": " + headers.value(i)); + } + log(" "); + if (logBody && HttpHeaders.hasBody(clone)) { + if (responseBody == null) return response; + + if (isPlaintext(responseBody.contentType())) { + byte[] bytes = responseBody.byteStream().readAllBytes(); + MediaType contentType = responseBody.contentType(); + String body = new String(bytes, getCharset(contentType)); + log("\tbody:" + body); + responseBody = ResponseBody.create(responseBody.contentType(), bytes); + return response.newBuilder().body(responseBody).build(); + } else { + log("\tbody: maybe [binary body], omitted!"); + } + } + } + } catch (Exception e) { + logger.log(java.util.logging.Level.WARNING, e.getMessage(), e); + } finally { + log("<-- END HTTP"); + } + return response; + } + + private static Charset getCharset(MediaType contentType) { + Charset charset = contentType != null ? contentType.charset(UTF8) : UTF8; + if (charset == null) charset = UTF8; + return charset; + } + + /** + * Returns true if the body in question probably contains human readable text. Uses a small sample + * of code points to detect unicode control characters commonly used in binary file signatures. + */ + private static boolean isPlaintext(MediaType mediaType) { + if (mediaType == null) return false; + if (mediaType.type() != null && mediaType.type().equals("text")) { + return true; + } + String subtype = mediaType.subtype(); + if (subtype != null) { + subtype = subtype.toLowerCase(); + if (subtype.contains("x-www-form-urlencoded") || subtype.contains("json") || subtype.contains("xml") || subtype.contains("html")) // + return true; + } + return false; + } + + private void bodyToString(Request request) { + try { + Request copy = request.newBuilder().build(); + RequestBody body = copy.body(); + if (body == null) return; + Buffer buffer = new Buffer(); + body.writeTo(buffer); + Charset charset = getCharset(body.contentType()); + log("\tbody:" + buffer.readString(charset)); + } catch (Exception e) { + logger.log(java.util.logging.Level.WARNING, e.getMessage(), e); + } + } +} diff --git a/common/src/main/java/com/yutou/okhttp/IRequestParam.java b/common/src/main/java/com/yutou/okhttp/IRequestParam.java new file mode 100644 index 0000000..c1ac1e5 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/IRequestParam.java @@ -0,0 +1,9 @@ +package com.yutou.okhttp; + +import okhttp3.Request; + +import java.util.HashMap; + +public interface IRequestParam { + Request getRequest(HashMap header, HashMap map, Request request); +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/ParamsContext.java b/common/src/main/java/com/yutou/okhttp/ParamsContext.java new file mode 100644 index 0000000..b54f919 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/ParamsContext.java @@ -0,0 +1,43 @@ +package com.yutou.okhttp; + +import okhttp3.Request; + +import java.util.HashMap; + +public class ParamsContext { + private IRequestParam iRequestParam; + private Request request; + private HashMap map; + private HashMap headerMap; + + public ParamsContext(HashMap map, Request request) { + if (map == null) { + map = new HashMap<>(); + } + this.map = map; + this.request = request; + this.headerMap = new HashMap<>(); + } + + public ParamsContext(HashMap headerMap, HashMap map, Request request) { + if (map == null) { + map = new HashMap<>(); + } + this.map = map; + this.request = request; + this.headerMap = headerMap; + } + + public Request getInRequest() { + switch (request.method()) { + case "GET": + iRequestParam = new GetRequestParams(); + break; + case "POST": + iRequestParam = new PostRequestParams(); + break; + } + headerMap.remove("tableName"); + return iRequestParam.getRequest(headerMap, map, request); + } +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/PostRequestParams.java b/common/src/main/java/com/yutou/okhttp/PostRequestParams.java new file mode 100644 index 0000000..6145eaa --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/PostRequestParams.java @@ -0,0 +1,73 @@ +package com.yutou.okhttp; + + +import com.alibaba.fastjson2.JSONObject; +import okhttp3.*; + +import java.net.URLEncoder; +import java.util.HashMap; +import java.util.Map; +import java.util.Set; + +public class PostRequestParams implements IRequestParam { + @Override + public Request getRequest(HashMap headerMap, HashMap map, Request request) { + Headers.Builder headerBuilder = request.headers().newBuilder(); + if (!headerMap.isEmpty()) { + for (String key : headerMap.keySet()) { + headerBuilder.add(key, headerMap.get(key)); + } + } + if (request.body() instanceof FormBody) { + FormBody.Builder bodyBuilder = new FormBody.Builder(); + FormBody formBody = (FormBody) request.body(); + for (int i = 0; i < formBody.size(); i++) { + bodyBuilder.addEncoded(formBody.encodedName(i), formBody.encodedValue(i)); + } + for (String key : map.keySet()) { + bodyBuilder.addEncoded(key, String.valueOf(map.get(key))); + } + formBody = bodyBuilder.build(); + request = request.newBuilder().headers(headerBuilder.build()).post(formBody).build(); + } else if (request.body() != null) { + RequestBody requestBody = RequestBody.create(MediaType.parse("application/json; charset=utf-8"), toUrlParams(map)); + request = request.newBuilder() + .headers(headerBuilder.build()) + .post(request.body()) + .post(requestBody).build(); + } + return request; + } + + public static String toUrlParams(JSONObject json) { + StringBuilder string = new StringBuilder(); + Set keys = json.keySet(); + for (String key : keys) { + try { + string.append("&").append(key).append("=").append(URLEncoder.encode(json.getString(key), "UTF-8")); + } catch (Exception e) { + e.printStackTrace(); + try { + string.append("&").append(URLEncoder.encode(key, "UTF-8")).append("="); + // string += "&" + key + "="; + } catch (Exception e1) { + string.append("&").append(key).append("="); + } + } + } + + string = new StringBuilder(string.substring(1, string.length()).replaceAll(" ", "")); + return string.toString(); + } + + public static String toUrlParams(Map map) { + if (map.isEmpty()) { + return ""; + } + StringBuilder builder = new StringBuilder(); + for (String key : map.keySet()) { + builder.append(key).append("=").append(map.get(key)).append("&"); + } + return builder.substring(0, builder.length() - 1); + } +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/api/BaseApi.java b/common/src/main/java/com/yutou/okhttp/api/BaseApi.java new file mode 100644 index 0000000..8fb2f7e --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/api/BaseApi.java @@ -0,0 +1,107 @@ +package com.yutou.okhttp.api; + +import com.google.gson.Gson; +import com.google.gson.GsonBuilder; +import com.google.gson.internal.bind.DateTypeAdapter; +import com.yutou.okhttp.HttpLoggingInterceptor; +import com.yutou.okhttp.ParamsContext; +import com.yutou.okhttp.converter.JsonCallAdapter; +import com.yutou.okhttp.converter.JsonConverterFactory; +import okhttp3.Interceptor; +import okhttp3.OkHttpClient; +import okhttp3.Request; +import okhttp3.Response; +import retrofit2.CallAdapter; +import retrofit2.Converter; +import retrofit2.Retrofit; + +import java.io.IOException; +import java.util.Date; +import java.util.HashMap; + +public class BaseApi { + private String URL; + private HashMap params; + private HashMap headers; + + public BaseApi(String URL) { + this.URL = URL; + } + + public BaseApi setURL(String URL) { + this.URL = URL; + return this; + } + + public BaseApi setHeaders(HashMap headers) { + this.headers = headers; + return this; + } + + public BaseApi setParams(HashMap params) { + this.params = params; + return this; + } + + /** + * 创建一个接口方法 + * + * @param okHttpClient okhttp客户端 + * @param converterFactory 处理工厂类 + * @param callAdapterFactory 请求适配器工厂 + * @param baseUrl 基础地质 + * @param service 接口 + * @param 接口泛型 + * @return 接口 + */ + public T create(OkHttpClient okHttpClient, Converter.Factory converterFactory, CallAdapter.Factory callAdapterFactory, String baseUrl, Class service) { + Retrofit.Builder builder = new Retrofit.Builder() + //基础url + .baseUrl(baseUrl) + //客户端OKHttp + .client(okHttpClient); + //添加转换工厂 + if (null != converterFactory) { + builder.addConverterFactory(converterFactory); + } + //添加请求工厂 + if (null != callAdapterFactory) { + builder.addCallAdapterFactory(callAdapterFactory); + } + //创建retrofit对象 + Retrofit retrofit = builder.build(); + + //返回创建的api + return retrofit.create(service); + } + + public T createApi(Class apiClass) { + Gson gson = new GsonBuilder() + .registerTypeAdapter(Date.class, new DateTypeAdapter()) + .create(); + HttpLoggingInterceptor loggingInterceptor = new HttpLoggingInterceptor("http"); + loggingInterceptor.setPrintLevel(HttpLoggingInterceptor.Level.BODY); + OkHttpClient.Builder builder = new OkHttpClient() + .newBuilder() + .addInterceptor(initQuery()) + .addInterceptor(loggingInterceptor); + + return create(builder.build(), + JsonConverterFactory.create(gson), + JsonCallAdapter.create(), + URL, + apiClass); + } + public Interceptor initQuery() { + Interceptor addQueryParameterInterceptor = new Interceptor() { + @Override + public Response intercept(Chain chain) throws IOException { + Request request = chain.request(); + //配置公共参数 + request = new ParamsContext(headers,params,request).getInRequest(); + return chain.proceed(request); + } + }; + return addQueryParameterInterceptor; + } +} diff --git a/common/src/main/java/com/yutou/okhttp/converter/JsonCallAdapter.java b/common/src/main/java/com/yutou/okhttp/converter/JsonCallAdapter.java new file mode 100644 index 0000000..8fe0ff9 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/converter/JsonCallAdapter.java @@ -0,0 +1,19 @@ +package com.yutou.okhttp.converter; + +import org.jetbrains.annotations.Nullable; +import retrofit2.CallAdapter; +import retrofit2.Retrofit; + +import java.lang.annotation.Annotation; +import java.lang.reflect.Type; + +public class JsonCallAdapter extends CallAdapter.Factory{ + public static JsonCallAdapter create(){ + return new JsonCallAdapter(); + } + @Nullable + @Override + public CallAdapter get(Type type, Annotation[] annotations, Retrofit retrofit) { + return null; + } +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/converter/JsonConverterFactory.java b/common/src/main/java/com/yutou/okhttp/converter/JsonConverterFactory.java new file mode 100644 index 0000000..c2d3a58 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/converter/JsonConverterFactory.java @@ -0,0 +1,40 @@ +package com.yutou.okhttp.converter; + +import com.google.gson.Gson; +import com.google.gson.TypeAdapter; +import com.google.gson.reflect.TypeToken; +import okhttp3.RequestBody; +import okhttp3.ResponseBody; +import org.jetbrains.annotations.Nullable; +import retrofit2.Converter; +import retrofit2.Retrofit; + +import java.lang.annotation.Annotation; +import java.lang.reflect.Type; + +public class JsonConverterFactory extends Converter.Factory { + Gson gson; + public static JsonConverterFactory create(Gson gson) { + return new JsonConverterFactory(gson); + } + + private JsonConverterFactory(Gson gson) { + this.gson = gson; + } + + @Nullable + @Override + public Converter requestBodyConverter(Type type, Annotation[] parameterAnnotations, Annotation[] methodAnnotations, Retrofit retrofit) { + // return super.requestBodyConverter(type, parameterAnnotations, methodAnnotations, retrofit); + TypeAdapter adapter = gson.getAdapter(TypeToken.get(type)); + return new JsonRequestBodyConverter<>(gson,adapter); + } + + @Nullable + @Override + public Converter responseBodyConverter(Type type, Annotation[] annotations, Retrofit retrofit) { + // return super.responseBodyConverter(type, annotations, retrofit); + TypeAdapter adapter = gson.getAdapter(TypeToken.get(type)); + return new JsonResponseBodyConverter<>(gson,adapter,type); + } +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/converter/JsonRequestBodyConverter.java b/common/src/main/java/com/yutou/okhttp/converter/JsonRequestBodyConverter.java new file mode 100644 index 0000000..6c7df58 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/converter/JsonRequestBodyConverter.java @@ -0,0 +1,36 @@ +package com.yutou.okhttp.converter; + +import com.google.gson.Gson; +import com.google.gson.TypeAdapter; +import com.google.gson.stream.JsonWriter; +import okhttp3.MediaType; +import okhttp3.RequestBody; +import okio.Buffer; +import org.jetbrains.annotations.Nullable; +import retrofit2.Converter; + +import java.io.IOException; +import java.io.OutputStreamWriter; +import java.io.Writer; +import java.nio.charset.StandardCharsets; + +public class JsonRequestBodyConverter implements Converter { + Gson gson; + TypeAdapter adapter; + public JsonRequestBodyConverter(Gson gson, TypeAdapter adapter) { + this.gson=gson; + this.adapter=adapter; + } + + @Nullable + @Override + public RequestBody convert(T value) throws IOException { + Buffer buffer = new Buffer(); + Writer writer = new OutputStreamWriter(buffer.outputStream(), StandardCharsets.UTF_8); + JsonWriter jsonWriter = gson.newJsonWriter(writer); + adapter.write(jsonWriter, value); + jsonWriter.close(); + byte[] bytes = buffer.readByteArray(); + return RequestBody.create(MediaType.parse("application/json; charset=UTF-8"),bytes ); + } +} \ No newline at end of file diff --git a/common/src/main/java/com/yutou/okhttp/converter/JsonResponseBodyConverter.java b/common/src/main/java/com/yutou/okhttp/converter/JsonResponseBodyConverter.java new file mode 100644 index 0000000..69c06d1 --- /dev/null +++ b/common/src/main/java/com/yutou/okhttp/converter/JsonResponseBodyConverter.java @@ -0,0 +1,45 @@ +package com.yutou.okhttp.converter; + +import com.alibaba.fastjson2.JSONObject; +import com.alibaba.fastjson2.JSONReader; +import com.alibaba.fastjson2.JSONWriter; +import com.google.gson.Gson; +import com.google.gson.TypeAdapter; +import com.yutou.okhttp.HttpBody; +import okhttp3.ResponseBody; +import org.jetbrains.annotations.Nullable; +import retrofit2.Converter; + +import java.io.IOException; +import java.lang.reflect.Type; + +public class JsonResponseBodyConverter implements Converter { + Gson gson; + TypeAdapter adapter; + Type type; + + public JsonResponseBodyConverter(Gson gson, TypeAdapter adapter, Type type) { + this.gson = gson; + this.adapter = adapter; + this.type = type; + } + + @Nullable + @Override + public T convert(ResponseBody responseBody) throws IOException { + String string = new String(responseBody.bytes()); + responseBody.close(); + HttpBody body; + try { + body = JSONObject.parseObject(string, type, JSONReader.Feature.FieldBased); + body.setSrc(string); + return (T) body; + } catch (Exception e) { + e.printStackTrace(); + body = new HttpBody(); + body.setSrc(string); + } + return (T) body; + + } +} \ No newline at end of file diff --git a/src/main/java/com/yutou/bilibili/Tools/AppTools.java b/common/src/main/java/com/yutou/utils/AppTools.java similarity index 95% rename from src/main/java/com/yutou/bilibili/Tools/AppTools.java rename to common/src/main/java/com/yutou/utils/AppTools.java index 5d705f9..1f44dfd 100644 --- a/src/main/java/com/yutou/bilibili/Tools/AppTools.java +++ b/common/src/main/java/com/yutou/utils/AppTools.java @@ -1,15 +1,16 @@ -package com.yutou.bilibili.Tools; +package com.yutou.utils; +import jakarta.servlet.http.Cookie; +import jakarta.servlet.http.HttpServletRequest; +import jakarta.servlet.http.HttpServletResponse; +import jakarta.xml.bind.DatatypeConverter; import org.springframework.core.io.FileSystemResource; import org.springframework.http.HttpHeaders; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; import org.springframework.util.StringUtils; -import javax.servlet.http.Cookie; -import javax.servlet.http.HttpServletRequest; -import javax.servlet.http.HttpServletResponse; -import javax.xml.bind.DatatypeConverter; + import java.io.*; import java.net.URLEncoder; import java.security.MessageDigest; @@ -26,7 +27,7 @@ public class AppTools { writer.close(); return true; } catch (IOException e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); return false; } } @@ -166,7 +167,7 @@ public class AppTools { public static String deleteCookie(HttpServletRequest request, HttpServletResponse response, String key) { for (Cookie cookie : request.getCookies()) { if (cookie.getName().equals(key)) { - com.yutou.bilibili.Tools.Log.i("删除key=" + key); + Log.i("删除key=" + key); cookie.setMaxAge(0); cookie.setPath("/"); cookie.setValue(null); @@ -249,7 +250,7 @@ public class AppTools { digest.update(data.getBytes()); return DatatypeConverter.printHexBinary(digest.digest()); } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } return ""; } @@ -278,7 +279,7 @@ public class AppTools { }else{ if(srcFileName.equals("Activity.smali")){ - com.yutou.bilibili.Tools.Log.i("文件夹已存在："+destFileName); + Log.i("文件夹已存在："+destFileName); } } @@ -302,10 +303,10 @@ public class AppTools { in.close(); return true; } catch (FileNotFoundException e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); return false; } catch (IOException e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); return false; } } diff --git a/common/src/main/java/com/yutou/utils/Base64Tools.java b/common/src/main/java/com/yutou/utils/Base64Tools.java new file mode 100644 index 0000000..4b9cafd --- /dev/null +++ b/common/src/main/java/com/yutou/utils/Base64Tools.java @@ -0,0 +1,34 @@ +package com.yutou.utils; + +import java.io.File; +import java.io.IOException; +import java.nio.file.Files; +import java.nio.file.Paths; +import java.util.Base64; + +public class Base64Tools { + public static String encode(String str) { + return Base64.getEncoder().encodeToString(str.getBytes()); + } + + public static String decode(String str) { + return new String(Base64.getDecoder().decode(str)); + } + + public static String encode(byte[] bytes) { + return Base64.getEncoder().encodeToString(bytes); + } + + public static String encode(File file) { + try { + byte[] fileContent = Files.readAllBytes(Paths.get(file.getAbsolutePath())); + return encode(fileContent); + } catch (IOException e) { + e.printStackTrace(); + } + return null; + } + + public static void main(String[] args) { + } +} diff --git a/src/main/java/com/yutou/bilibili/Tools/ConfigTools.java b/common/src/main/java/com/yutou/utils/ConfigTools.java similarity index 91% rename from src/main/java/com/yutou/bilibili/Tools/ConfigTools.java rename to common/src/main/java/com/yutou/utils/ConfigTools.java index 66befd4..4bbb231 100644 --- a/src/main/java/com/yutou/bilibili/Tools/ConfigTools.java +++ b/common/src/main/java/com/yutou/utils/ConfigTools.java @@ -1,6 +1,7 @@ -package com.yutou.bilibili.Tools; +package com.yutou.utils; -import com.alibaba.fastjson.JSONObject; + +import com.alibaba.fastjson2.JSONObject; import java.io.*; @@ -11,6 +12,10 @@ public class ConfigTools { public static final String CONFIG="config.json"; public static final String DATA="data.json"; public static final String SQLITE="sqlite.json"; + + public static final String NAPCAT_URL = "napcat.url"; + public static final String QQ = "qq_bot"; + static { try { File file=new File(CONFIG); @@ -23,7 +28,7 @@ public class ConfigTools { } file=null; }catch (Exception e){ - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } } @@ -68,7 +73,7 @@ public class ConfigTools { writer.close(); return true; } catch (IOException e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); return false; } } @@ -84,7 +89,7 @@ public class ConfigTools { return str.toString(); } catch (Exception e) { Log.i("error = "+file.getAbsolutePath()); - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } return null; } diff --git a/src/main/java/com/yutou/bilibili/Tools/Log.java b/common/src/main/java/com/yutou/utils/Log.java similarity index 90% rename from src/main/java/com/yutou/bilibili/Tools/Log.java rename to common/src/main/java/com/yutou/utils/Log.java index 57d8adf..6fe4ce1 100644 --- a/src/main/java/com/yutou/bilibili/Tools/Log.java +++ b/common/src/main/java/com/yutou/utils/Log.java @@ -1,4 +1,4 @@ -package com.yutou.bilibili.Tools; +package com.yutou.utils; import java.io.File; @@ -13,8 +13,9 @@ public class Log { public static void i(String tag, Object log) { i('[' + tag + ']' + log); } + public static void i(Object log) { - if (!((boolean) ConfigTools.load(ConfigTools.CONFIG, "logout"))) { + if (!((boolean) ConfigTools.load(ConfigTools.CONFIG, "logcat"))) { return; } System.out.printf("[%s]%s%n", @@ -25,7 +26,7 @@ public class Log { } public static void e(Exception e) { - if (!((boolean) ConfigTools.load(ConfigTools.CONFIG, "logout"))) { + if (!ConfigTools.load(ConfigTools.CONFIG, "logcat", Boolean.class)){ return; } i(e.getMessage()); @@ -34,7 +35,7 @@ public class Log { public static void i(String timer, int roomId, Object log) { String logFlag = RedisTools.get("live-log", 1); - if (logFlag!=null&&logFlag.startsWith("true")) { + if (logFlag != null && logFlag.startsWith("true")) { if (logFlag.contains("|")) { if (log.toString().contains(logFlag.split("\\|")[1])) { getLogger("[" + timer.replace(":", "_") + "]" + roomId).log(Level.INFO, log.toString()); diff --git a/src/main/java/com/yutou/bilibili/Tools/RedisTools.java b/common/src/main/java/com/yutou/utils/RedisTools.java similarity index 69% rename from src/main/java/com/yutou/bilibili/Tools/RedisTools.java rename to common/src/main/java/com/yutou/utils/RedisTools.java index 29ae94d..1ff6853 100644 --- a/src/main/java/com/yutou/bilibili/Tools/RedisTools.java +++ b/common/src/main/java/com/yutou/utils/RedisTools.java @@ -1,21 +1,19 @@ -package com.yutou.bilibili.Tools; +package com.yutou.utils; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.QQBot.QQBotManager; +import com.alibaba.fastjson2.JSONObject; import redis.clients.jedis.Jedis; import redis.clients.jedis.JedisPool; import redis.clients.jedis.JedisPoolConfig; import redis.clients.jedis.JedisPubSub; -import java.io.BufferedReader; import java.io.IOException; import java.io.InputStream; -import java.io.InputStreamReader; import java.util.Properties; import java.util.Set; public class RedisTools { + public static final int QQBOT_USER = 3; private static boolean isNotInstallRedis = false; private static String host; private static int port; @@ -51,8 +49,8 @@ public class RedisTools { return true; } - public static boolean set(String key, String value) { - return set(0, key, value); + public static boolean set(Object key, String value) { + return set(0, key.toString(), value); } public static boolean set(String key, String value, int timeout) { @@ -91,12 +89,12 @@ public class RedisTools { return value; } - public static String get(String key) { - return get(key, 0); + public static String get(Object key) { + return get(key.toString(), 0); } - public static boolean remove(String key) { - return remove(key,0); + public static boolean remove(Object key) { + return remove(key.toString(),0); } public static void removeLoginState(String uid) { @@ -117,12 +115,12 @@ public class RedisTools { return tmp; } - public static boolean exists(String key, String value) { + public static boolean exists(Object key, String value) { if (isNotInstallRedis) { return false; } Jedis jedis = getRedis(); - boolean flag = value.equals(jedis.get(key)); + boolean flag = value.equals(jedis.get(key.toString())); jedis.close(); return flag; } @@ -183,53 +181,7 @@ public class RedisTools { private static boolean init = false; - public static void initRedisPoolSub() { - if (init) - return; - init = true; - Log.i("初始化订阅"); - new Thread(new Runnable() { - @Override - public void run() { - Jedis jedis = getPoolRedis(); - if (jedis != null) - jedis.psubscribe(new Consumer(), "*"); - } - }).start(); - } - - protected static class Consumer extends JedisPubSub { - @Override - public void onPMessage(String pattern, String channel, String message) { - super.onPMessage(pattern, channel, message); - Log.i("onPMessage: channel=" + channel + " msg=" + message + " pattern=" + pattern); - switch (channel) { - case "bot": - bot(message); - break; - } - } - - @Override - public void onMessage(String channel, String message) { - super.onMessage(channel, message); - Log.i("onMessage: channel=" + channel + " msg=" + message); - } - - - - - public static void bot(String value) { - switch (value) { - case "getip": - JSONObject json = JSONObject.parseObject(HttpTools.get("https://api.asilu.com/ip/")); - String ip = json.getString("ip"); - QQBotManager.getInstance().sendMessage("服务器IP:\n" + ip); - break; - } - } - } public static void main(String[] args) { RedisTools.pullMsg("msg", "abc"); } diff --git a/pom.xml b/pom.xml index 73ced4c..f30b6c2 100644 --- a/pom.xml +++ b/pom.xml @@ -5,16 +5,22 @@ org.springframework.boot spring-boot-starter-parent - 2.4.3 + 3.3.0 com.yutou bilibili 0.0.1-SNAPSHOT + pom bilibili Demo project for Spring Boot + + biliapi + qqbot + common + - 1.8 + 21 1.6.10 @@ -26,9 +32,9 @@ org.springframework.boot spring-boot-starter-web-services - - mysql - mysql-connector-java + + com.mysql + mysql-connector-j runtime @@ -52,9 +58,9 @@ spring-boot-starter-websocket - com.alibaba - fastjson - 1.2.80 + com.alibaba.fastjson2 + fastjson2 + 2.0.22 org.xerial @@ -64,18 +70,13 @@ org.projectlombok lombok - 1.18.22 + annotationProcessor org.java-websocket Java-WebSocket 1.5.2 - - mysql - mysql-connector-java - runtime - org.mybatis.spring.boot mybatis-spring-boot-starter @@ -103,27 +104,25 @@ 4.1.2 - - net.mamoe - mirai-core-jvm - 2.10.0 - - - org.jetbrains.kotlinx - kotlinx-coroutines-jdk8 - 1.6.0-native-mt - - - org.jetbrains.kotlinx - kotlinx-coroutines-core-jvm - 1.6.0-native-mt - redis.clients jedis 4.2.0 + + com.yutou + common + 0.0.1-SNAPSHOT + compile + + + com.yutou + qqbot + 0.0.1-SNAPSHOT + compile + + @@ -190,6 +189,14 @@ + + org.apache.maven.plugins + maven-compiler-plugin + + 7 + 7 + + diff --git a/qqbot/pom.xml b/qqbot/pom.xml new file mode 100644 index 0000000..4c474c1 --- /dev/null +++ b/qqbot/pom.xml @@ -0,0 +1,29 @@ + + + 4.0.0 + + com.yutou + bilibili + 0.0.1-SNAPSHOT + + + qqbot + + + 21 + 21 + UTF-8 + + + + + + + \ No newline at end of file diff --git a/qqbot/src/main/java/com/yutou/napcat/NapCatQQ.java b/qqbot/src/main/java/com/yutou/napcat/NapCatQQ.java new file mode 100644 index 0000000..f1cf592 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/NapCatQQ.java @@ -0,0 +1,57 @@ +package com.yutou.napcat; + +import com.yutou.napcat.handle.MessageHandleBuild; +import com.yutou.napcat.handle.Text; +import com.yutou.napcat.http.NapCatApi; +import com.yutou.napcat.model.SendMessageResponse; +import com.yutou.okhttp.HttpCallback; +import okhttp3.Headers; + +import java.io.File; + +public class NapCatQQ { + + + private NapCatQQ() { + + } + + public static void main(String[] args) { + /* List> list = new ArrayList<>(); + list.add(new Text("1", false)); + list.add(new Text("2", false)); + list.add(new Text("3")); + list.add(new Text("4", false)); + list.add(new At(583819556L)); + list.add(new Text("5")); + QQBotManager.getInstance().sendMessage(false, 891655174L, list);*/ + NapCatApi.setLog(false); + File file = new File("C:\\Users\\58381\\Downloads\\0074TT8Yly1hp5mqidwqeg30g20f27wh.gif"); + NapCatApi.getMessageApi().sendPrivateMsg( + MessageHandleBuild.create() + .setQQNumber(583819556L) + //.add(new Image(file)) + .add(new Text("abc")) + .build() + ).enqueue(new HttpCallback() { + @Override + public void onResponse(Headers headers, int code, String status, SendMessageResponse response, String rawResponse) { + System.out.println("code = " + code + ", status = " + status + ", response = " + response + ", rawResponse = " + rawResponse); + } + + @Override + public void onFailure(Throwable throwable) { + throwable.printStackTrace(); + } + }); + } + /** + * 私聊 + {"self_id":240828363,"user_id":583819556,"time":1714472684,"message_id":376,"real_id":376,"message_type":"private","sender":{"user_id":583819556,"nickname":"魔芋","card":""},"raw_message":"123","font":14,"sub_type":"friend","message":[{"data":{"text":"123"},"type":"text"}],"message_format":"array","post_type":"message"} + */ + + /** + * 群聊 + * {"self_id":240828363,"user_id":583819556,"time":1714472695,"message_id":377,"real_id":377,"message_type":"group","sender":{"user_id":583819556,"nickname":"魔芋","card":"","role":"owner"},"raw_message":"222","font":14,"sub_type":"normal","message":[{"data":{"text":"222"},"type":"text"}],"message_format":"array","post_type":"message","group_id":891655174} + */ +} diff --git a/qqbot/src/main/java/com/yutou/napcat/QQDatabase.java b/qqbot/src/main/java/com/yutou/napcat/QQDatabase.java new file mode 100644 index 0000000..46e5e7d --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/QQDatabase.java @@ -0,0 +1,54 @@ +package com.yutou.napcat; + +import com.yutou.napcat.model.FriendBean; +import com.yutou.napcat.model.GroupBean; +import lombok.Getter; +import lombok.Setter; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +public class QQDatabase { + private static Map data = new HashMap<>(); + + @Getter + @Setter + private static FriendBean me; + + public static void addUser(Long qq, FriendBean bean) { + data.put(qq, bean); + } + + public static void addGroup(Long qq, GroupBean bean) { + data.put(qq, bean); + } + + public static boolean checkFriend(Long qq) { + if (data.containsKey(qq)) { + return data.get(qq) instanceof FriendBean; + } + return false; + } + + public static boolean checkGroup(Long qq) { + if (data.containsKey(qq)) { + return data.get(qq) instanceof GroupBean; + } + return false; + } + + public static List getGroups() { + List numbers = new ArrayList<>(); + for (Long qq : data.keySet()) { + if (data.get(qq) instanceof GroupBean) { + { + numbers.add((GroupBean) data.get(qq)); + } + } + } + return numbers; + } + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/QQNumberManager.java b/qqbot/src/main/java/com/yutou/napcat/QQNumberManager.java new file mode 100644 index 0000000..7799209 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/QQNumberManager.java @@ -0,0 +1,145 @@ +package com.yutou.napcat; + +import com.alibaba.fastjson2.JSON; +import com.alibaba.fastjson2.JSONArray; +import com.alibaba.fastjson2.JSONObject; +import com.yutou.utils.RedisTools; +import redis.clients.jedis.Jedis; + +import java.util.ArrayList; +import java.util.List; +import java.util.Set; + +public class QQNumberManager { + private static QQNumberManager manager; + private QQNumberManager(){ + + } + + public static QQNumberManager getManager() { + if(manager==null) { + manager=new QQNumberManager(); + } + return manager; + } + public void addNumber(Long qq,boolean isGroup){ + if(RedisTools.exists(qq,null)){ + return; + } + JSONObject json=new JSONObject(); + json.put("group",isGroup); + json.put("power",new JSONArray()); + json.put("model",new JSONArray()); + RedisTools.set(qq,json.toJSONString()); + } + public List getNumber(){ + List list =new ArrayList<>(); + Jedis jedis=RedisTools.getRedis(); + jedis.select(RedisTools.QQBOT_USER); + Set set=jedis.keys("*"); + for (String s : set) { + try { + list.add(Long.parseLong(s)); + }catch (Exception ignored){ + + } + } + jedis.close(); + return list; + } + public boolean addPower(Long qq, String power){ + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("power"); + array.add(power); + json.put("power",array); + return RedisTools.set(qq,json.toJSONString()); + } + return false; + } + public List getPower(Long qq){ + List list=new ArrayList<>(); + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("power"); + for (Object power : array) { + list.add((String) power); + } + return list; + } + return list; + } + public List getUseModel(long qq) { + List list=new ArrayList<>(); + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("model"); + for (Object power : array) { + list.add((String) power); + } + return list; + } + return list; + } + public boolean delPower(Long qq, String power){ + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("power"); + array.remove(power); + json.put("power",array); + return RedisTools.set(qq,json.toJSONString()); + } + return false; + } + public boolean addUseModel(Long qq,Class modelClass){ + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("model"); + array.add(modelClass.getName()); + json.put("model",array); + return RedisTools.set(qq,json.toJSONString()); + } + return false; + } + public boolean delUseModel(Long qq,Class modelClass){ + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("model"); + array.remove(modelClass.getName()); + json.put("model",array); + return RedisTools.set(qq,json.toJSONString()); + } + return false; + } + public boolean isExistsPower(Long qq, String... power){ + //1 + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("power"); + for (String key : power) { + if(!array.contains(key)){ + return false; + } + } + return true; + } + return false; + } + public boolean isGroup(Long qq){ + if(RedisTools.exists(qq,null)){ + JSONObject json= JSON.parseObject(RedisTools.get(qq)); + return json.getBooleanValue("group"); + } + return false; + } + public boolean isUseModel(Long qq,Class modelClass){ + if(RedisTools.exists(qq,null)){ + JSONObject json=JSON.parseObject(RedisTools.get(qq)); + JSONArray array=json.getJSONArray("model"); + return array.contains(modelClass.getName()); + } + return false; + } + + +} \ No newline at end of file diff --git a/qqbot/src/main/java/com/yutou/napcat/enums/MessageEnum.java b/qqbot/src/main/java/com/yutou/napcat/enums/MessageEnum.java new file mode 100644 index 0000000..9a93a93 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/enums/MessageEnum.java @@ -0,0 +1,29 @@ +package com.yutou.napcat.enums; + +public enum MessageEnum { + TEXT("text"), + IMAGE("image"), + AT("at"), + REPLY("reply"), + JSON("json"); + + String type; + + MessageEnum(String type) { + this.type = type; + } + + public String getType() { + return type; + } + + public static MessageEnum of(String type) { + for (MessageEnum messageEnum : MessageEnum.values()) { + if (messageEnum.getType().equals(type)) { + return messageEnum; + } + } + + return null; + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/enums/RecordFormatEnum.java b/qqbot/src/main/java/com/yutou/napcat/enums/RecordFormatEnum.java new file mode 100644 index 0000000..052fb0c --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/enums/RecordFormatEnum.java @@ -0,0 +1,28 @@ +package com.yutou.napcat.enums; + +public enum RecordFormatEnum { + MP3("mp3"), + AMR("amr"), + WMA("wma"), + M4A("m4a"), + SPX("spx"), + OGG("ogg"), + WAV("wav"), + FLAC("flac"); + + + private final String format; + + RecordFormatEnum(String format) { + this.format = format; + } + + public String getFormat() { + return format; + } + + @Override + public String toString() { + return format; + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/event/GroupMessageEvent.java b/qqbot/src/main/java/com/yutou/napcat/event/GroupMessageEvent.java new file mode 100644 index 0000000..0d4a9b6 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/event/GroupMessageEvent.java @@ -0,0 +1,12 @@ +package com.yutou.napcat.event; + +import com.yutou.napcat.model.GroupFrom; +import lombok.Getter; +import lombok.Setter; + +@Setter +@Getter +public class GroupMessageEvent extends MessageEvent { + private GroupFrom group; + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/event/MessageEvent.java b/qqbot/src/main/java/com/yutou/napcat/event/MessageEvent.java new file mode 100644 index 0000000..977db2b --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/event/MessageEvent.java @@ -0,0 +1,183 @@ +package com.yutou.napcat.event; + +import com.alibaba.fastjson2.JSONArray; +import com.alibaba.fastjson2.JSONObject; +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.napcat.QQDatabase; +import com.yutou.napcat.enums.MessageEnum; +import com.yutou.napcat.handle.*; +import com.yutou.napcat.model.SourceFrom; +import lombok.Data; +import lombok.val; + +import java.lang.reflect.Type; +import java.util.ArrayList; +import java.util.List; + +@Data +public class MessageEvent { + + @JSONField(name = "self_id") + private Long selfId; + + @JSONField(name = "user_id") + private Long userId; + + @JSONField(name = "time") + private Long time; + + @JSONField(name = "message_id") + private Integer messageId; + + @JSONField(name = "real_id") + private Integer realId; + + @JSONField(name = "message_type") + private String messageType; + + @JSONField(name = "sender") + private SourceFrom source; + + @JSONField(name = "raw_message") + private String rawMessage; + + @JSONField(name = "font") + private Integer font; + + @JSONField(name = "sub_type") + private String subType; + + @JSONField(name = "message") + private List> message; + + @JSONField(name = "message_format") + private String messageFormat; + + @JSONField(name = "post_type") + private String postType; + + @JSONField(name = "group_id") + private Long groupId; + + + public static MessageEvent parseHandleHttp(String jsonString) { + + return parseHandle(JSONObject.parseObject(jsonString).getJSONObject("data").toString()); + } + + public static MessageEvent parseHandle(String jsonString) { + JSONObject json = JSONObject.parseObject(jsonString); + + JSONArray array = json.getJSONArray("message"); + List> messageList = new ArrayList<>(); + for (Object o : array) { + JSONObject _json = (JSONObject) o; + Type classType = null; + MessageEnum _type = MessageEnum.of(_json.getString("type")); + classType = switch (_type) { + case TEXT -> Text.TextInfo.class; + case IMAGE -> Image.ImageInfo.class; + case AT -> At.AtData.class; + case JSON -> OtherHandle.OtherInfo.class; + case REPLY -> Reply.ReplyInfo.class; + default -> classType; + }; + BaseHandle handle = new BaseHandle<>(_type.getType()); + if (_type == MessageEnum.JSON) { + handle.setData(JSONObject.parseObject(((JSONObject) o).getJSONObject("data").getString("data"), classType)); + } else { + handle.setData(JSONObject.parseObject(((JSONObject) o).getJSONObject("data").toString(), classType)); + } + messageList.add(handle); + } + + + SourceFrom sender = new SourceFrom(); + sender.setUserId(json.getJSONObject("sender").getLong("user_id")); + sender.setNickname(json.getJSONObject("sender").getString("nickname")); + sender.setCard(json.getJSONObject("sender").getString("card")); + + MessageEvent event = new MessageEvent(); + event.setSelfId(json.getLong("self_id")); + event.setUserId(json.getLong("user_id")); + event.setTime(json.getLong("time")); + event.setMessageId(json.getInteger("message_id")); + event.setRealId(json.getInteger("real_id")); + event.setMessageType(json.getString("message_type")); + event.setSource(sender); + event.setRawMessage(json.getString("raw_message")); + event.setFont(json.getInteger("font")); + event.setSubType(json.getString("sub_type")); + event.setMessageFormat(json.getString("message_format")); + event.setPostType(json.getString("post_type")); + event.setGroupId(json.getLong("group_id")); + event.setMessage(messageList); + return event; + } + + public boolean hasType(MessageEnum messageEnum) { + for (BaseHandle handle : message) { + if (MessageEnum.of(handle.getType()) == messageEnum) { + return true; + } + } + return false; + } + + public List findAllType(Class t) { + List tmp = new ArrayList<>(); + try { + T newed = t.getDeclaredConstructor().newInstance(); + for (BaseHandle baseHandle : message) { + if (baseHandle.getType().equals(newed.getType())) { + newed.setData(baseHandle.getData()); + tmp.add(newed); + } + } + } catch (Exception e) { + e.printStackTrace(); + } + return tmp; + } + + public T findType(Class tClass) { + List list = findAllType(tClass); + if (list == null || list.isEmpty()) { + return null; + } + return list.get(0); + } + + + public boolean isAtMe() { + if (!hasType(MessageEnum.AT)) { + return false; + } + List list = findAllType(At.class); + for (At handle : list) { + if (handle.getData().getQq() == QQDatabase.getMe().getUserId()) { + return true; + } + } + return false; + } + + public String getTextMessage() { + val texts = findAllType(Text.class); + StringBuilder sb = new StringBuilder(); + for (Text text : texts) { + sb.append(text); + } + return sb.toString().trim(); + } + + public boolean isUser() { + return "private".equals(messageType); + } + + public boolean isGroup() { + return "group".equals(messageType); + } + + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/At.java b/qqbot/src/main/java/com/yutou/napcat/handle/At.java new file mode 100644 index 0000000..866f6fb --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/At.java @@ -0,0 +1,32 @@ +package com.yutou.napcat.handle; + +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +public class At extends BaseHandle { + public At() { + super("at"); + } + + public At(Long userId) { + super("at"); + data = new AtData(userId); + } + + @Override + public String toString() { + if (data == null) { + return null; + } + return String.format("[CQ:at,qq=%d]", data.qq); + } + + @Data + public class AtData extends BaseBean { + private Long qq; + + public AtData(Long qq) { + this.qq = qq; + } + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/BaseHandle.java b/qqbot/src/main/java/com/yutou/napcat/handle/BaseHandle.java new file mode 100644 index 0000000..376a6a5 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/BaseHandle.java @@ -0,0 +1,22 @@ +package com.yutou.napcat.handle; + +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +@Data +public class BaseHandle extends BaseBean { + protected String type; + protected T data; + protected String src; + + public BaseHandle(String type) { + this.type = type; + } + + + public BaseHandle(String type, Object obj) { + super(); + this.type = type; + this.data= (T) obj; + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/Image.java b/qqbot/src/main/java/com/yutou/napcat/handle/Image.java new file mode 100644 index 0000000..ab035bf --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/Image.java @@ -0,0 +1,35 @@ +package com.yutou.napcat.handle; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.utils.Base64Tools; +import lombok.Data; + +import java.io.File; + +public class Image extends BaseHandle { + + public Image() { + super("image"); + } + + public Image(String imageUrl) { + super("image"); + data = new ImageInfo(imageUrl); + } + public Image(File imageFile){ + super("image"); + data=new ImageInfo("base64://"+ Base64Tools.encode(imageFile)); + } + + @Data + public static class ImageInfo { + String file; + String url; + @JSONField(name = "file_size") + String fileSize; + + public ImageInfo(String file) { + this.file = file; + } + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/MessageHandleBuild.java b/qqbot/src/main/java/com/yutou/napcat/handle/MessageHandleBuild.java new file mode 100644 index 0000000..9486daa --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/MessageHandleBuild.java @@ -0,0 +1,53 @@ +package com.yutou.napcat.handle; + +import com.alibaba.fastjson2.JSONObject; + +import java.util.ArrayList; +import java.util.List; + +public class MessageHandleBuild { + private List> msgList = new ArrayList<>(); + private long qq; + private boolean autoEscape; + private boolean isGroup; + + public static MessageHandleBuild create() { + return new MessageHandleBuild(); + } + + private MessageHandleBuild() { + msgList = new ArrayList<>(); + } + + public MessageHandleBuild setQQNumber(long qq) { + this.qq = qq; + return this; + } + + public MessageHandleBuild setAutoEscape(boolean autoEscape) { + this.autoEscape = autoEscape; + return this; + } + + public MessageHandleBuild add(BaseHandle msg) { + msgList.add(msg); + return this; + } + + public MessageHandleBuild setGroup(boolean group) { + isGroup = group; + return this; + } + + public JSONObject build() { + JSONObject json=new JSONObject(); + if(isGroup){ + json.put("group_id", qq); + }else { + json.put("user_id", qq); + } + json.put("auto_escape", autoEscape); + json.put("message", msgList); + return json; + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/OtherHandle.java b/qqbot/src/main/java/com/yutou/napcat/handle/OtherHandle.java new file mode 100644 index 0000000..800c8b6 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/OtherHandle.java @@ -0,0 +1,19 @@ +package com.yutou.napcat.handle; + +import com.yutou.napcat.model.AppShareBean; +import lombok.Data; + +public class OtherHandle extends BaseHandle{ + + public OtherHandle() { + super("json"); + data= new OtherInfo(); + } + + + @Data + public static class OtherInfo extends AppShareBean{ + + + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/QuoteReply.java b/qqbot/src/main/java/com/yutou/napcat/handle/QuoteReply.java new file mode 100644 index 0000000..cab5cad --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/QuoteReply.java @@ -0,0 +1,14 @@ +package com.yutou.napcat.handle; + +public class QuoteReply { + private final String messageId; + + public QuoteReply(String messageId) { + this.messageId = messageId; + } + + @Override + public String toString() { + return String.format("[CQ:reply,id=%s]",messageId); + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/Record.java b/qqbot/src/main/java/com/yutou/napcat/handle/Record.java new file mode 100644 index 0000000..bec8dfd --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/Record.java @@ -0,0 +1,38 @@ +package com.yutou.napcat.handle; + +import com.yutou.utils.ConfigTools; +import lombok.Data; + +public class Record extends BaseHandle { + + public Record() { + super("record"); + } + + public Record(String file) { + super("record"); + data = new RecordInfo(file); + } + + public Record(String file, boolean proxy) { + super("record"); + data = new RecordInfo(file); + data.proxy = proxy ? "1" : "0"; + } + + + @Data + public static class RecordInfo { + String file; + int magic; + String url; + String cache; + String proxy; + String timeout; + + public RecordInfo(String file) { + this.file = file; + } + + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/Reply.java b/qqbot/src/main/java/com/yutou/napcat/handle/Reply.java new file mode 100644 index 0000000..76aaedc --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/Reply.java @@ -0,0 +1,24 @@ +package com.yutou.napcat.handle; + +import lombok.Data; + +public class Reply extends BaseHandle { + + public Reply() { + super("reply"); + } + + public Reply(long id) { + super("reply"); + data = new ReplyInfo(id); + } + + @Data + public static class ReplyInfo { + private long id; + + public ReplyInfo(long id) { + this.id = id; + } + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/handle/Text.java b/qqbot/src/main/java/com/yutou/napcat/handle/Text.java new file mode 100644 index 0000000..db82d5c --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/handle/Text.java @@ -0,0 +1,51 @@ +package com.yutou.napcat.handle; + +import lombok.Data; + +public class Text extends BaseHandle { + + public Text() { + super("text"); + } + + public Text(String text) { + super("text"); + data = new TextInfo(text + "\n"); + } + + public Text(String text, boolean isNewLine) { + super("text"); + if (isNewLine) { + data = new TextInfo(text + "\n"); + } else { + data = new TextInfo(text); + } + } + + public Text(boolean isNewLine, String... text) { + super("text"); + StringBuilder sb = new StringBuilder(); + for (String s : text) { + sb.append(s); + } + if (isNewLine) { + data = new TextInfo(sb + "\n"); + } else { + data = new TextInfo(sb.toString()); + } + } + + @Override + public String toString() { + return data.text.trim(); + } + + @Data + public static class TextInfo { + String text; + + public TextInfo(String text) { + this.text = text.trim(); + } + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/http/FriendApi.java b/qqbot/src/main/java/com/yutou/napcat/http/FriendApi.java new file mode 100644 index 0000000..324da72 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/http/FriendApi.java @@ -0,0 +1,14 @@ +package com.yutou.napcat.http; + +import com.yutou.napcat.model.FriendBean; +import com.yutou.okhttp.HttpBody; +import retrofit2.Call; +import retrofit2.http.POST; + +import java.util.List; + +public interface FriendApi { + @POST("/get_friend_list") + Call>> getFriendList( + ); +} diff --git a/qqbot/src/main/java/com/yutou/napcat/http/GroupApi.java b/qqbot/src/main/java/com/yutou/napcat/http/GroupApi.java new file mode 100644 index 0000000..1901c9c --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/http/GroupApi.java @@ -0,0 +1,70 @@ +package com.yutou.napcat.http; + +import com.yutou.napcat.model.GroupBean; +import com.yutou.napcat.model.GroupUserBean; +import com.yutou.okhttp.BaseBean; +import com.yutou.okhttp.HttpBody; +import retrofit2.Call; +import retrofit2.http.Field; +import retrofit2.http.FormUrlEncoded; +import retrofit2.http.POST; + +import java.util.List; + +public interface GroupApi { + /** + * 禁言 + * @param group 群号 + * @param user 用户 + * @param duration 禁言时长，单位秒 + * + */ + @FormUrlEncoded + @POST("/set_group_ban") + Call> groupBan( + @Field("group_id") long group, + @Field("user_id") long user, + @Field("duration") long duration + ); + + /** + * 禁言群组全体成员 + * @param group 群号 + */ + @FormUrlEncoded + @POST("/set_group_whole_ban") + Call> groupBanAll( + @Field("group_id") long group + ); + + /** + * 获取群组列表 + */ + @POST("/get_group_list") + Call>> getGroupList( + ); + + /** + * 获取群组信息 + */ + @POST("/get_group_info") + Call> getGroupInfo( + ); + + /** + * 获取群组成员信息 + */ + @POST("/get_group_member_info") + Call> getGroupUserInfo( + ); + + /** + * 获取群组成员列表 + * @param group 群号 + */ + @FormUrlEncoded + @POST("/get_group_member_list") + Call> getGroupUserList( + @Field("group_id") long group + ); +} diff --git a/qqbot/src/main/java/com/yutou/napcat/http/MessageAPI.java b/qqbot/src/main/java/com/yutou/napcat/http/MessageAPI.java new file mode 100644 index 0000000..2986cc1 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/http/MessageAPI.java @@ -0,0 +1,50 @@ +package com.yutou.napcat.http; + +import com.alibaba.fastjson2.JSONObject; +import com.yutou.napcat.model.MessageBean; +import com.yutou.napcat.model.SendMessageResponse; +import com.yutou.okhttp.BaseBean; +import com.yutou.okhttp.HttpBody; +import retrofit2.Call; +import retrofit2.http.Body; +import retrofit2.http.Field; +import retrofit2.http.FormUrlEncoded; +import retrofit2.http.POST; + +public interface MessageAPI { + /** + * 发送私聊消息 + * @param message {@link com.yutou.napcat.handle.MessageHandleBuild} + * @return 消息id + */ + @POST("/send_private_msg") + Call> sendPrivateMsg( + @Body + JSONObject message); + /** + * 发送群聊消息 + * @param message 消息内容 + * @return 消息id + */ + @POST("/send_group_msg") + Call> sendGroupMsg( + @Body + JSONObject message + ); + + /** + * 撤回消息 + * @param messageId 消息id + */ + @FormUrlEncoded + @POST("/delete_msg") + Call> delMsg( + @Field("message_id") long messageId + ); + + @FormUrlEncoded + @POST("/get_msg") + Call> getMessage( + @Field("message_id") long messageId + ); +} diff --git a/qqbot/src/main/java/com/yutou/napcat/http/NapCatApi.java b/qqbot/src/main/java/com/yutou/napcat/http/NapCatApi.java new file mode 100644 index 0000000..85287de --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/http/NapCatApi.java @@ -0,0 +1,33 @@ +package com.yutou.napcat.http; + +import com.yutou.okhttp.HttpLoggingInterceptor; +import com.yutou.okhttp.api.BaseApi; +import com.yutou.utils.ConfigTools; + +public class NapCatApi extends BaseApi { + private static final String URL; + static { + URL= ConfigTools.load(ConfigTools.CONFIG,ConfigTools.NAPCAT_URL,String.class); + } + + public NapCatApi() { + super(URL); + } + + public static void setLog(boolean log){ + HttpLoggingInterceptor.setLog(log); + } + + public static MessageAPI getMessageApi(){ + return new NapCatApi().setURL(URL).createApi(MessageAPI.class); + } + public static UtilsApi getUtilsApi(){ + return new NapCatApi().setURL(URL).createApi(UtilsApi.class); + } + public static GroupApi getGroupApi(){ + return new NapCatApi().setURL(URL).createApi(GroupApi.class); + } + public static FriendApi getFriendApi(){ + return new NapCatApi().setURL(URL).createApi(FriendApi.class); + } +} diff --git a/qqbot/src/main/java/com/yutou/napcat/http/UtilsApi.java b/qqbot/src/main/java/com/yutou/napcat/http/UtilsApi.java new file mode 100644 index 0000000..63a0ab7 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/http/UtilsApi.java @@ -0,0 +1,80 @@ +package com.yutou.napcat.http; + +import com.yutou.napcat.model.*; +import com.yutou.okhttp.BaseBean; +import com.yutou.okhttp.HttpBody; +import retrofit2.Call; +import retrofit2.http.Field; +import retrofit2.http.FormUrlEncoded; +import retrofit2.http.POST; + +public interface UtilsApi { + /** + * 获取语音 + * + * @param fileId 收到的语音文件名（消息段的 file 参数），如 0B38145AA44505000B38145AA4450500.silk + * @param format 要转换到的格式，目前支持 mp3、amr、wma、m4a、spx、ogg、wav、flac {@link com.yutou.napcat.enums.RecordFormatEnum} + * @return 转换后的语音文件路径 + */ + @FormUrlEncoded + @POST("/get_record") + Call> getMessageRecord( + @Field("file") String fileId, + @Field("out_format") String format + ); + + /** + * 获取图片 + * + * @param fileId 收到的图片文件名（消息段的 file 参数） + * @return 下载后的图片文件路径 + */ + @FormUrlEncoded + @POST("/get_image") + Call> getMessageImage( + @Field("file") String fileId + ); + + /** + * 检查是否可以发送图片 + */ + @POST("/can_send_image") + Call> checkSendImage( + ); + + /** + * 检查是否可以发送语音 + */ + @POST("/can_send_record") + Call> checkSendRecord( + ); + + /** + * 获取机器人状态 + */ + @POST("/get_status") + Call> checkQQBotStatus( + ); + + /** + * 获取机器人版本信息 + */ + @POST("/get_version_info") + Call> checkQQBotVersion( + ); + + /** + * 清理缓存 + */ + @POST("/clean_cache") + Call> cleanCache( + ); + + /** + * 获取登录信息 + */ + @POST("/get_login_info") + Call> getLoginInfo( + ); + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/AppShareBean.java b/qqbot/src/main/java/com/yutou/napcat/model/AppShareBean.java new file mode 100644 index 0000000..b6da655 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/AppShareBean.java @@ -0,0 +1,139 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +import java.util.Map; + +@Data + +public class AppShareBean extends BaseBean { + + @JSONField(name = "ver") + private String version; + + @JSONField(name = "prompt") + private String prompt; + + @JSONField(name = "config") + private Config config; + + @JSONField(name = "needShareCallBack") + private boolean needShareCallBack; + + @JSONField(name = "app") + private String app; + + @JSONField(name = "view") + private String view; + + @JSONField(name = "meta") + private Meta meta; + + // getters and setters... + + // Inner classes + + @Data + public static class Config { + @JSONField(name = "type") + private String type; + + @JSONField(name = "width") + private int width; + + @JSONField(name = "height") + private int height; + + @JSONField(name = "forward") + private int forward; + + @JSONField(name = "autoSize") + private int autoSize; + + @JSONField(name = "ctime") + private long ctime; + + @JSONField(name = "token") + private String token; + + // getters and setters... + } + + @Data + public static class Meta { + @JSONField(name = "detail_1") + private Detail detail1; + + // If there can be multiple "detail_X" entries, you might need a Map + // @JSONField(name = "detail_X") + // private Map details; + + // getters and setters... + } + + @Data + public static class Detail { + @JSONField(name = "appid") + private String appid; + + @JSONField(name = "appType") + private int appType; + + @JSONField(name = "title") + private String title; + + @JSONField(name = "desc") + private String desc; + + @JSONField(name = "icon") + private String icon; + + @JSONField(name = "preview") + private String preview; + + @JSONField(name = "url") + private String url; + + @JSONField(name = "scene") + private int scene; + + @JSONField(name = "host") + private Host host; + + @JSONField(name = "shareTemplateId") + private String shareTemplateId; + + // Since "shareTemplateData" is an empty object, it can be left as a Map or a specific class + // if you know the structure of the data that might be there + @JSONField(name = "shareTemplateData") + private Map shareTemplateData; + + @JSONField(name = "qqdocurl") + private String qqdocurl; + + @JSONField(name = "showLittleTail") + private String showLittleTail; + + @JSONField(name = "gamePoints") + private String gamePoints; + + @JSONField(name = "gamePointsUrl") + private String gamePointsUrl; + + // getters and setters... + } + + @Data + public static class Host { + @JSONField(name = "uin") + private long uin; + + @JSONField(name = "nick") + private String nick; + + // getters and setters... + } + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/CheckSendImageBean.java b/qqbot/src/main/java/com/yutou/napcat/model/CheckSendImageBean.java new file mode 100644 index 0000000..fe7823d --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/CheckSendImageBean.java @@ -0,0 +1,12 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +@Data +public class CheckSendImageBean extends BaseBean { + @JSONField(name = "yes") + private boolean yes; + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/CheckSendRecordBean.java b/qqbot/src/main/java/com/yutou/napcat/model/CheckSendRecordBean.java new file mode 100644 index 0000000..89b16ef --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/CheckSendRecordBean.java @@ -0,0 +1,12 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +@Data +public class CheckSendRecordBean extends BaseBean { + @JSONField(name = "yes") + private boolean yes; + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/FriendBean.java b/qqbot/src/main/java/com/yutou/napcat/model/FriendBean.java new file mode 100644 index 0000000..db53c52 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/FriendBean.java @@ -0,0 +1,15 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +@Data +public class FriendBean extends BaseBean { + @JSONField(name = "user_id") + private long userId;//qq号 + @JSONField(name = "nickname") + private String nickName;//昵称 + @JSONField(name = "remark") + private String remark;//备注 +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/GroupBean.java b/qqbot/src/main/java/com/yutou/napcat/model/GroupBean.java new file mode 100644 index 0000000..1d7c9ab --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/GroupBean.java @@ -0,0 +1,18 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +@Data +public class GroupBean extends BaseBean { + @JSONField(name = "group_id") + private long groupId; + @JSONField(name = "group_name") + private String groupName; + @JSONField(name = "member_count") + private int userCount; + @JSONField(name = "max_member_count") + private int userMaxCount; + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/GroupFrom.java b/qqbot/src/main/java/com/yutou/napcat/model/GroupFrom.java new file mode 100644 index 0000000..1b3b278 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/GroupFrom.java @@ -0,0 +1,8 @@ +package com.yutou.napcat.model; + +import lombok.Data; + +@Data +public class GroupFrom { + private long id; +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/GroupUserBean.java b/qqbot/src/main/java/com/yutou/napcat/model/GroupUserBean.java new file mode 100644 index 0000000..9e20250 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/GroupUserBean.java @@ -0,0 +1,69 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +@Data +public class GroupUserBean extends BaseBean { + // 群号 + @JSONField(name = "group_id") + private long groupId; + + // QQ号 + @JSONField(name = "user_id") + private long userId; + + // 昵称 + @JSONField(name = "nickname") + private String nickname; + + // 群名片/备注 + @JSONField(name = "card") + private String card; + + // 性别，male 或 female 或 unknown + @JSONField(name = "sex") + private String sex; + + // 年龄 + @JSONField(name = "age") + private int age; + + // 地区 + @JSONField(name = "area") + private String area; + + // 加群时间戳 + @JSONField(name = "join_time") + private int joinTime; + + // 最后发言时间戳 + @JSONField(name = "last_sent_time") + private int lastSentTime; + + // 成员等级 + @JSONField(name = "level") + private String level; + + // 角色，owner 或 admin 或 member + @JSONField(name = "role") + private String role; + + // 是否不良记录成员 + @JSONField(name = "unfriendly") + private boolean unfriendly; + + // 专属头衔 + @JSONField(name = "title") + private String title; + + // 专属头衔过期时间戳 + @JSONField(name = "title_expire_time") + private int titleExpireTime; + + // 是否允许修改群名片 + @JSONField(name = "card_changeable") + private boolean cardChangeable; + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/Message.java b/qqbot/src/main/java/com/yutou/napcat/model/Message.java new file mode 100644 index 0000000..962a07d --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/Message.java @@ -0,0 +1,23 @@ +package com.yutou.napcat.model; + +import lombok.Data; + +@Data +public class Message { + private String message; + private String srcMessage; + private String id; + public Message(String message, String srcMessage) { + this.message = message; + this.srcMessage = srcMessage; + } + + public String serializeToMiraiCode(){ + return srcMessage; + } + + public String contentToString(){ + return message; + } + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/MessageBean.java b/qqbot/src/main/java/com/yutou/napcat/model/MessageBean.java new file mode 100644 index 0000000..df79626 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/MessageBean.java @@ -0,0 +1,45 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +/** + * 消息 + * @see 文档 + */ +@Data +public class MessageBean extends BaseBean { + /** + *发送时间 + */ + @JSONField(name = "time") + private long time; + /** + * 消息类型，同 消息事件 + */ + @JSONField(name = "message_type") + private String type; + /** + * 消息 ID + */ + @JSONField(name = "message_id") + private int messageId; + /** + * 消息真实 ID + */ + @JSONField(name = "real_id") + private int realId; + /** + * 发送人信息，同 消息事件 + */ + @JSONField(name = "sender") + private SenderBean sender; + /** + * 消息内容 + */ + @JSONField(name = "message") + private String message; + + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/QQBotStatusBean.java b/qqbot/src/main/java/com/yutou/napcat/model/QQBotStatusBean.java new file mode 100644 index 0000000..24b0498 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/QQBotStatusBean.java @@ -0,0 +1,13 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +@Data +public class QQBotStatusBean extends BaseBean { + @JSONField(name = "online") + private String online;//当前 QQ 在线，null 表示无法查询到在线状态 + @JSONField(name = "good") + private String good;//状态符合预期，意味着各模块正常运行、功能正常，且 QQ 在线 +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/QQBotVersionBean.java b/qqbot/src/main/java/com/yutou/napcat/model/QQBotVersionBean.java new file mode 100644 index 0000000..9ba668e --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/QQBotVersionBean.java @@ -0,0 +1,17 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import lombok.Data; + +@Data +public class QQBotVersionBean { + @JSONField(name = "app_name") + private String appName;//应用标识，如 mirai-native + + @JSONField(name = "app_version") + private String appVersion;//应用版本，如 1.2.3 + + @JSONField(name = "protocol_version") + private String protocolVersion;//OneBot 标准版本，如 v11 + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/SendMessageRequest.java b/qqbot/src/main/java/com/yutou/napcat/model/SendMessageRequest.java new file mode 100644 index 0000000..feb73c7 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/SendMessageRequest.java @@ -0,0 +1,21 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.google.gson.annotations.SerializedName; +import com.yutou.napcat.handle.BaseHandle; +import com.yutou.okhttp.BaseBean; +import lombok.Data; +import lombok.EqualsAndHashCode; + +import java.util.List; +@EqualsAndHashCode(callSuper = true) +@Data +public class SendMessageRequest extends BaseBean { + @JSONField(name = "user_id") + private long userId; + @JSONField + List> message; + @SerializedName("auto_escape") + private boolean notCQCode; + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/SendMessageResponse.java b/qqbot/src/main/java/com/yutou/napcat/model/SendMessageResponse.java new file mode 100644 index 0000000..c8b0df3 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/SendMessageResponse.java @@ -0,0 +1,16 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; +import lombok.EqualsAndHashCode; + +@EqualsAndHashCode(callSuper = true) +@Data +public class SendMessageResponse extends BaseBean { + @JSONField( name = "message_id") + private int id; + + private String e; + +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/SenderBean.java b/qqbot/src/main/java/com/yutou/napcat/model/SenderBean.java new file mode 100644 index 0000000..ffc1596 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/SenderBean.java @@ -0,0 +1,24 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import com.yutou.okhttp.BaseBean; +import lombok.Data; + +/** + * 发送人结构体 + * + */ +@Data +public class SenderBean extends BaseBean { + @JSONField(name = "user_id") + private long userId; + @JSONField(name = "nickname") + private String nickName; + /** + * male 或 female 或 unknown + */ + @JSONField(name = "sex") + private String sex; + @JSONField(name = "age") + private int age; +} diff --git a/qqbot/src/main/java/com/yutou/napcat/model/SourceFrom.java b/qqbot/src/main/java/com/yutou/napcat/model/SourceFrom.java new file mode 100644 index 0000000..0dc3553 --- /dev/null +++ b/qqbot/src/main/java/com/yutou/napcat/model/SourceFrom.java @@ -0,0 +1,17 @@ +package com.yutou.napcat.model; + +import com.alibaba.fastjson2.annotation.JSONField; +import lombok.Data; + +@Data +public class SourceFrom { + @JSONField(name = "user_id") + private long userId; + private String nickname; + private String card; + private String role; + + public Long getFromId() { + return userId; + } +} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/BiliUserController.java b/src/main/java/com/yutou/bilibili/BiliBili/Controllers/BiliUserController.java deleted file mode 100644 index c31c033..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/BiliUserController.java +++ /dev/null @@ -1,82 +0,0 @@ -package com.yutou.bilibili.BiliBili.Controllers; - -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.Tools.AppTools; -import org.springframework.stereotype.Controller; -import org.springframework.util.StringUtils; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.ResponseBody; - -import java.io.File; -import java.util.Map; -import java.util.Timer; -import java.util.TimerTask; - -@Controller -@RequestMapping("/bili/") -public class BiliUserController { - @ResponseBody - @RequestMapping("/login/get/test.do") - public JSONObject testLogin(){ - JSONObject json=new JSONObject(); - if(StringUtils.isEmpty(AppTools.readFile(new File("cookies.json")))){ - json.put("code",-1); - json.put("msg","未登录"); - }else{ - JSONObject login= LiveUtils.getUserLoginInfo(); - JSONObject data=new JSONObject(); - json.put("code",login.getInteger("code")); - data.put("uname",login.getJSONObject("data").getString("uname")); - data.put("icon",login.getJSONObject("data").getString("face")); - json.put("data",data); - } - return json; - } - @ResponseBody - @RequestMapping("/login/set/login.do") - public JSONObject login(){ - JSONObject login=LiveUtils.http_get("https://passport.bilibili.com/qrcode/getLoginUrl"); - JSONObject json=new JSONObject(); - json.put("code",login.getInteger("code")); - json.put("url",login.getJSONObject("data").getString("url")); - new Thread(new Runnable() { - @Override - public void run() { - waitLogin(login.getJSONObject("data").getString("oauthKey")); - } - }).start(); - return json; - } - public void waitLogin(String oauthKey){ - long time=System.currentTimeMillis(); - String bd="gourl=https%3A%2F%2Fpassport.bilibili.com%2Fajax%2FminiLogin%2Fredirect&oauthKey="+oauthKey; - new Timer().schedule(new TimerTask() { - @Override - public void run() { - if((System.currentTimeMillis()-time)>5*60*1000){ - cancel(); - return; - } - JSONObject json=LiveUtils.http_post("https://passport.bilibili.com/qrcode/getLoginInfo",bd); - if(json.containsKey("code")&&json.getInteger("code")==0){ - String _url=json.getJSONObject("data").getString("url"); - Map map=AppTools.getUrlParams(_url); - JSONObject tmp=LiveUtils.http_post(_url,""); - JSONObject cookie=new JSONObject(); - for (String key : map.keySet()) { - cookie.put(key,map.get(key)); - } - String sid=tmp.getString("cookie"); - sid=sid.split("sid=")[1]; - sid=sid.split(";")[0]; - cookie.put("sid",sid); - cookie.put("Domain",".bilibili.com"); - AppTools.saveFile(new File("cookies.json"),cookie.toJSONString()); - cancel(); - } - } - },0,3000); - - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/BiliVideoController.java b/src/main/java/com/yutou/bilibili/BiliBili/Controllers/BiliVideoController.java deleted file mode 100644 index 7955bb4..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/BiliVideoController.java +++ /dev/null @@ -1,149 +0,0 @@ -package com.yutou.bilibili.BiliBili.Controllers; - -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.BiliBili.Tools.BiliTools; -import com.yutou.bilibili.Services.ISystemConfigService; -import com.yutou.bilibili.Services.IUserService; -import com.yutou.bilibili.Tools.*; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import com.yutou.bilibili.mybatis.model.UUser; -import org.springframework.core.io.FileSystemResource; -import org.springframework.http.ResponseEntity; -import org.springframework.stereotype.Controller; -import org.springframework.util.StringUtils; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.ResponseBody; - -import javax.annotation.Resource; -import javax.servlet.http.HttpServletRequest; -import java.io.File; -import java.io.UnsupportedEncodingException; -import java.net.URLDecoder; -import java.util.ArrayList; -import java.util.List; -import java.util.Objects; - -@Controller -public class BiliVideoController { - @Resource - IUserService service; - @Resource - IBiliBiliLiveService liveService; - @Resource - ISystemConfigService configService; - - @ResponseBody - @RequestMapping("/bili/video/get/list.do") - public JSONObject videoList(HttpServletRequest request) { - JSONObject json = new JSONObject(); - String token = AppTools.getLoginToken(request); - UUser user = service.getUserToToken(token); - if (user == null) { - json.put("code", -1); - json.put("msg", "未登录"); - json.put("data", new JSONObject()); - return json; - } - if (ServiceTools.getInstance().auth(request, user.getUser(), "/bili/video/get/")) { - File file = new File(ConfigTools.load(ConfigTools.CONFIG,"liveSavePath",String.class)); - Log.i(file.getAbsolutePath()); - if (!file.exists() || Objects.requireNonNull(file.listFiles()).length == 0) { - json.put("code", 0); - json.put("msg", "文件夹为空"); - json.put("data", new JSONObject()); - } else { - List list = AppTools.scanFilePath(file.getAbsolutePath()+File.separator+"live"); - JSONArray array = new JSONArray(); - for (File f : list) { - if(!f.getName().endsWith(".mp4")){ - continue; - } - JSONObject item = new JSONObject(); - String _time = f.getName().split(" ")[0].substring(1); - File mpeg = new File("ffmpeg_out" + File.separator + _time + File.separator + f.getName()); - if (mpeg.exists()) { - f = mpeg; - item.put("ffmpeg", true); - } else { - item.put("ffmpeg", false); - } - String date = f.getName().split(" ")[0].substring(1); - String time = f.getName().split(" ")[1].split("\\]")[0]; - String roomid = f.getName().split("\\]")[1].replace(".mp4", "").replace("_ffmpeg", "").trim(); - - BilibiliUpInfo info = liveService.queryUpToRoomId(Integer.parseInt(roomid)); - long size = f.length(); - String _size; - if (size < 1024) { - _size = size + "B"; - } else if (size < 1048576) { - _size = (size / 1024) + "KB"; - } else if (size < 1073741824) { - _size = (size / 1048576) + "MB"; - } else { - _size = (size / 1073741824) + "GB"; - } - - item.put("fileName", f.getName()); - item.put("fileSize", _size); - item.put("date", date); - item.put("time", time); - item.put("name", info == null ? roomid : info.getName()); - array.add(item); - } - json.put("code", 0); - json.put("msg", "ok"); - json.put("data", array); - } - } else { - json.put("code", -2); - json.put("msg", "未授权"); - json.put("data", new JSONObject()); - } - return json; - } - - @RequestMapping("/bili/video/download/get.do") - public ResponseEntity download(String fileName) { - String time = fileName.split(" ")[0].substring(1); - File file = new File("ffmpeg_out" + File.separator + time + File.separator + fileName); - if (!file.exists()) { - file = new File(ConfigTools.load(ConfigTools.CONFIG,"liveSavePath",String.class)+File.separator+"live" + File.separator + time + File.separator + fileName); - } - Log.i(file.getAbsolutePath()); - return AppTools.getFile(file); - } - @ResponseBody - @RequestMapping("/bili/video/set/ffmpeg.do") - public JSONObject ffmpegOut(String fileName) throws UnsupportedEncodingException { - fileName= URLDecoder.decode(fileName,"UTF-8"); - JSONObject json=new JSONObject(); - json.put("code",0); - json.put("msg","ok"); - String time = fileName.split(" ")[0].substring(1); - File file = new File("live" + File.separator + time + File.separator + fileName); - System.out.println("file "+file.getAbsolutePath()+" "+file.exists()); - String ffmpeg=configService.getConfig(Config.SYSTEM_VIDEO_FFMPEG); - if(StringUtils.isEmpty(ffmpeg)||!new File(ffmpeg).exists()){ - json.put("msg","FFmpeg未设置，请联系管理员"); - return json; - } - int i= FFmpegUtils.add(configService.getConfig(Config.SYSTEM_VIDEO_FFMPEG),file,new File("ffmpeg_out"+File.separator+time)); - json.put("model",i); - switch (i){ - case 0: - json.put("msg","已添加到后台转码"); - break; - case 1: - json.put("msg","已经转码，无须二次转码"); - break; - case 2: - json.put("msg","正在转码"); - break; - } - return json; - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/OverviewController.java b/src/main/java/com/yutou/bilibili/BiliBili/Controllers/OverviewController.java deleted file mode 100644 index 694a6b3..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/OverviewController.java +++ /dev/null @@ -1,107 +0,0 @@ -package com.yutou.bilibili.BiliBili.Controllers; - -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Datas.LiveData; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.Tools.AppTools; -import com.yutou.bilibili.Tools.ExcelUtils; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveData; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveInfo; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import org.springframework.core.io.FileSystemResource; -import org.springframework.http.ResponseEntity; -import org.springframework.stereotype.Controller; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.ResponseBody; - -import javax.annotation.Resource; -import javax.servlet.http.HttpServletRequest; -import java.io.File; -import java.text.SimpleDateFormat; -import java.util.Date; -import java.util.List; -import java.util.Map; - -@Controller -public class OverviewController { - @Resource - IBiliBiliLiveService manager; - - @RequestMapping("/overview/get/query.do") - @ResponseBody - public JSONObject queryOverview(Date startTime, Date endTime, BilibiliUpInfo upData) { - JSONObject json = new JSONObject(); - JSONObject data = new JSONObject(); - //BiliBiliLiveDatabasesManager manager = BiliBiliLiveDatabasesManager.getInstance(); - data.put("giftCount", manager.queryGiftSize(upData.getRoomid(), -1, startTime, endTime)); - data.put("giftUserCount", manager.queryGiftUserToDistinct(upData, startTime, endTime, new String[]{ - LiveData.SEND_GIFT - , LiveData.COMBO_SEND - , LiveData.GUARD_BUY})); - data.put("price", manager.queryPriceSize(upData.getRoomid(), startTime, endTime)); - long h = (endTime.getTime() - startTime.getTime()) % (1000 * 24 * 60 * 60) / (1000 * 60 * 60); - com.yutou.bilibili.Tools.Log.i("相差:" + h + "小时"); - try { - data.put("hprice", data.getInteger("price") / h); - } catch (Exception e) { - data.put("hprice", 0); - } - - - JSONArray array=new JSONArray(); - List infos=manager.queryLiveInfo(upData,startTime,endTime); - int maxUser=0; - for (BilibiliLiveInfo info : infos) { - com.yutou.bilibili.Tools.Log.i(info); - maxUser+=info.getUserindex()+info.getVipuserindex(); - JSONObject item=new JSONObject(); - String nowTime = new SimpleDateFormat("HH").format(info.getSubtime()); - item.put("time",nowTime); - item.put("popular",info.getPopular()); - item.put("user",info.getUserindex()+info.getVipuserindex()); - array.add(item); - } - try { - Map map = manager.queryUserGiftCount(upData, startTime, endTime); - int huserGift = 0; - for (String s : map.keySet()) { - huserGift += map.get(s); - } - data.put("userGift", huserGift / maxUser); - } catch (Exception e) { - data.put("userGift", 0); - } - data.put("map",array); - json.put("data",data); - json.put("code",0); - return json; - } - - @ResponseBody - @RequestMapping("/overview/get/down.do") - public JSONObject downExcel(Date startTime, Date endTime, BilibiliUpInfo upData){ - final String fileName=System.currentTimeMillis()+".xlsx"; - new Thread(new Runnable() { - @Override - public void run() { - ExcelUtils.getInstance(upData.getRoomid(),startTime,endTime,fileName); - } - }).start(); - JSONObject json=new JSONObject(); - json.put("code",0); - json.put("data",fileName); - return json; - - } - - @RequestMapping("/overview/get/download.do") - public ResponseEntity downloadExcel(HttpServletRequest request, String fname){ - File file=new File("excel"+File.separator+fname); - if(file.exists()) { - return AppTools.getFile(file); - }else{ - return null; - } - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/RealTimeDataController.java b/src/main/java/com/yutou/bilibili/BiliBili/Controllers/RealTimeDataController.java deleted file mode 100644 index bcf69d0..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/RealTimeDataController.java +++ /dev/null @@ -1,120 +0,0 @@ -package com.yutou.bilibili.BiliBili.Controllers; - -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Datas.LiveData; -import com.yutou.bilibili.BiliBili.Live; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.Tools.AppTools; -import com.yutou.bilibili.Tools.Log; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveData; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveInfo; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import org.springframework.stereotype.Controller; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.ResponseBody; - -import javax.annotation.Resource; -import java.math.BigDecimal; -import java.text.ParseException; -import java.text.SimpleDateFormat; -import java.util.Date; -import java.util.List; -import java.util.Map; - -@Controller -@RequestMapping("/realTimeData/") -public class RealTimeDataController { - @Resource - IBiliBiliLiveService service; - @ResponseBody - @RequestMapping("get/query.do") - public JSONObject queryToDayLiveData(int roomid,Date startTime,Date endTime){ - JSONObject json = new JSONObject(); - BilibiliUpInfo info = new BilibiliUpInfo(); - info.setRoomid(roomid); - Live live = LiveUtils.liveContains(info); - if(startTime==null){ - startTime=AppTools.getToDayStartTime(); - } - if(endTime==null){ - endTime=AppTools.getToDayNowTime(); - } - int userCount = service.queryLiveData(info.getRoomid(), startTime, endTime, new String[]{LiveData.INTERACT_WORD}).size(); - int vipCount = service.queryLiveData(info.getRoomid(), startTime, endTime, new String[]{LiveData.ENTRY_EFFECT}).size(); - JSONObject data = new JSONObject(); - data.put("roomid", roomid); - data.put("popular",live==null?0:live.getInfo().getPopular()); - data.put("userLength", userCount); - data.put("vipLength", vipCount); - - JSONArray price = new JSONArray(); - //改写这个sql可以按时段统计数据 - List> ps=service.queryTimeGroup(roomid, startTime, endTime); - for (Map map : ps) { - JSONObject pr = new JSONObject(); - pr.put("price", map.containsKey("price") ? map.get("price") : 0); - pr.put("time", map.get("time")); - price.add(pr); - } - - JSONArray giftArray = new JSONArray(); - Map map = service.queryGiftSize(roomid,startTime,endTime); - for (String key : map.keySet()) { - JSONObject item = new JSONObject(); - if(key==null|| "null".equals(key)) { - continue; - } - item.put("giftName",key); - item.put("size",map.get(key)); - giftArray.add(item); - } - data.put("price", price); - data.put("gift", giftArray); - json.put("data", data); - json.put("code",0); - return json; - } - @ResponseBody - @RequestMapping("data.do") - public JSONObject queryRoomData(int rid,int model){ - Date startTime = AppTools.getToDayStartTime(); - Date endTime=AppTools.getToDayNowTime(); - List giftList = service.queryLiveData(rid, startTime, endTime, new String[]{ - LiveData.SEND_GIFT - , LiveData.COMBO_SEND - , LiveData.GUARD_BUY - }); - List userList = service.queryLiveData(rid, startTime, endTime, new String[]{ - LiveData.INTERACT_WORD - , LiveData.ENTRY_EFFECT - }); - JSONObject json=new JSONObject(); - JSONArray retData=new JSONArray(); - if(model==1) { - for (BilibiliLiveData data : giftList) { - JSONObject item = new JSONObject(); - item.put("uid", data.getUid()); - item.put("msg", data.getMsg()); - json.put("size", data.getGiftindex()); - json.put("money", data.getPrice()); - item.put("time", AppTools.getToDayTimeToString(data.getSubtime())); - retData.add(item); - } - }else if(model==2) { - for (BilibiliLiveData data : userList) { - JSONObject item = new JSONObject(); - item.put("uid", data.getUid()); - item.put("msg", data.getMsg()); - item.put("time", AppTools.getToDayTimeToString(data.getSubtime())); - retData.add(item); - } - } - json.put("code",0); - json.put("count",retData.size()); - json.put("data",retData); - json.put("msg",""); - return json; - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/UpInfoController.java b/src/main/java/com/yutou/bilibili/BiliBili/Controllers/UpInfoController.java deleted file mode 100644 index ba4c6a5..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Controllers/UpInfoController.java +++ /dev/null @@ -1,163 +0,0 @@ -package com.yutou.bilibili.BiliBili.Controllers; - -import com.alibaba.fastjson.JSON; -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Live; -import com.yutou.bilibili.BiliBili.LiveController; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.BiliBili.Tools.BiliTools; -import com.yutou.bilibili.Services.IUserService; -import com.yutou.bilibili.Tools.AppTools; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import com.yutou.bilibili.mybatis.model.UUser; -import org.springframework.stereotype.Controller; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.ResponseBody; - -import javax.annotation.Resource; -import javax.servlet.http.HttpServletRequest; -import java.util.List; - -@Controller -public class UpInfoController { - @Resource - IBiliBiliLiveService service; - @Resource - IUserService userService; - - @RequestMapping("/upinfo/get/list.do") - @ResponseBody - public JSONObject upInfoList(HttpServletRequest request) { - String token = AppTools.getLoginToken(request); - UUser user = userService.getUserToToken(token); - JSONObject json = new JSONObject(); - if (user == null) { - json.put("code", -1); - json.put("msg", "未登录"); - json.put("data", "[]"); - return json; - } - List upDataList = service.getUpInfo(user.getId(), user.getPower()); - com.yutou.bilibili.Tools.Log.i("UP用户数:" + upDataList.size()); - json.put("code", 0); - json.put("data", JSONArray.toJSON(upDataList)); - return json; - } - - @ResponseBody - @RequestMapping("/upinfo/set/add.do") - public JSONObject addUpInfo(BilibiliUpInfo upData) { - JSONObject json = new JSONObject(); - if (BiliTools.checkout(upData) != null) { - upData.setEnable(0); - upData = BiliTools.checkout(upData); - com.yutou.bilibili.Tools.Log.i(upData); - boolean flag = service.addUpInfo(upData); - if (flag) { - json.put("code", 0); - json.put("msg", "添加成功"); - } else { - json.put("code", -1); - json.put("msg", "添加失败，该UP已经存在"); - } - } else { - json.put("code", -2); - json.put("msg", "添加失败，可能参数有问题或该UP已经存在"); - } - return json; - } - - @ResponseBody - @RequestMapping("/upinfo/update.do") - public JSONObject updateV1(BilibiliUpInfo upData) { - return update(upData); - } - @ResponseBody - @RequestMapping("/upinfo/set/update.do") - public JSONObject update(BilibiliUpInfo upData) { - JSONObject json = new JSONObject(); - com.yutou.bilibili.Tools.Log.i("old data =" + upData); - BilibiliUpInfo old = service.queryUp(upData); - if (!old.getUrl().equals(upData.getUrl())) { - upData = BiliTools.checkout(upData); - if (upData != null) { - upData.setEnable(0); - upData.setSavedanmu(0); - upData.setOfflinelistening(0); - } - } - com.yutou.bilibili.Tools.Log.i("save data = " + upData); - boolean flag = service.updateUpInfo(upData); - if (flag) { - LiveController.getInstance().updateUpInfo(upData); - json.put("code", 0); - json.put("msg", "更新成功"); - } else { - json.put("code", -1); - json.put("msg", "更新失败，可能参数有问题"); - } - return json; - } - - @ResponseBody - @RequestMapping("/upinfo/set/delete.do") - public JSONObject delete(BilibiliUpInfo upData) { - JSONObject json = new JSONObject(); - com.yutou.bilibili.Tools.Log.i(upData); - boolean flag = service.deleteUp(upData); - if (flag) { - json.put("code", 0); - json.put("msg", "删除成功"); - } else { - json.put("code", -1); - json.put("msg", "删除失败，可能参数有问题"); - } - return json; - } - - @ResponseBody - @RequestMapping("/upinfo/get/query.do") - public JSONObject query(BilibiliUpInfo upData) { - upData = service.queryUp(upData); - return (JSONObject) JSON.toJSON(upData); - } - - @ResponseBody - @RequestMapping("/upinfo/get/liveInfo.do") - public JSONObject liveInfo() { - List list = service.getUpInfo(); - JSONObject json = new JSONObject(); - JSONObject data = new JSONObject(); - JSONArray info = new JSONArray(); - JSONArray liveArray = new JSONArray(); - int online = 0; - JSONObject infoByUid = LiveUtils.getStatusInfoByUid(list); - JSONObject infoData = infoByUid.getJSONObject("data"); - for (BilibiliUpInfo up : list) { - JSONObject user = infoData.getJSONObject(up.getMid() + ""); - if(user==null){ - continue; - } - up.setLive(user.getInteger("live_status")); - if (up.checkLiveStatus() == 1) { - online++; - } - JSONObject item= (JSONObject) JSON.toJSON(up); - item.put("listen",Live.lives.get(up.getRoomid())!=null); - info.add(JSON.toJSON(item)); - } - for (Live live : Live.lives.values()) { - if (live.geData().getLive() == 1) { - liveArray.add(JSON.toJSON(live.geData())); - } - } - data.put("online", online); - data.put("info", info); - data.put("live", liveArray); - json.put("code", 0); - json.put("data", data); - return json; - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Live.java b/src/main/java/com/yutou/bilibili/BiliBili/Live.java deleted file mode 100644 index bbd9f41..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Live.java +++ /dev/null @@ -1,670 +0,0 @@ -package com.yutou.bilibili.BiliBili; - -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Datas.DanmuData; -import com.yutou.bilibili.BiliBili.Datas.GiftData; -import com.yutou.bilibili.BiliBili.Datas.LiveData; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.BiliBili.Tools.SaveLive; -import com.yutou.bilibili.QQBot.QQBotManager; -import com.yutou.bilibili.Tools.AppTools; -import com.yutou.bilibili.Tools.Log; -import com.yutou.bilibili.Tools.RedisTools; -import com.yutou.bilibili.Tools.Tools; -import com.yutou.bilibili.interfaces.DownloadInterface; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveData; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveInfo; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import com.yutou.bilibili.sqlite.BiliBiliLiveDatabasesManager; -import lombok.Data; -import org.java_websocket.client.WebSocketClient; -import org.java_websocket.handshake.ServerHandshake; -import org.jetbrains.annotations.NotNull; -import org.springframework.beans.BeansException; -import org.springframework.context.ApplicationContext; -import org.springframework.context.ApplicationContextAware; - -import javax.annotation.PostConstruct; -import java.io.ByteArrayOutputStream; -import java.io.File; -import java.net.URI; -import java.nio.ByteBuffer; -import java.nio.charset.StandardCharsets; -import java.text.SimpleDateFormat; -import java.util.*; - -public class Live implements ApplicationContextAware { - IBiliBiliLiveService service; - - - public static Map lives = new HashMap<>(); - - - private int roomId; - private boolean isLogin; - private boolean run; - private int userId; - private WebSocketClient client; - private BilibiliLiveInfo info; - private BilibiliUpInfo upData; - private Timer heartBeattimer; - public BiliBiliLiveDatabasesManager danmuManager = null; - private LiveInfo liveInfo; - - public LiveInfo getLiveInfo() { - return liveInfo; - } - - public void setLiveInfo(LiveInfo liveInfo) { - this.liveInfo = liveInfo; - } - - @PostConstruct - public void init() { - - } - - public Live() { - if (applicationContext != null) { - service = getBean(IBiliBiliLiveService.class); - } - info = new BilibiliLiveInfo(); - info.setGiftuser(0); - info.setPopular(0); - info.setUserindex(0); - info.setVipuserindex(0); - } - - /** - * 构造方法 - * - * @param roomId 直播间roomId，不是房间号，有可能是临时房间号，得通过 {@link com.yutou.bilibili.BiliBili.Tools.BiliTools#getBiliUpInfo(String)} 来获取roomId - * @param isLogin 是否登录用户 - */ - public void add(int roomId, boolean isLogin) { - if (Live.lives.containsKey(roomId)) { - Log.i("已经在统计:" + roomId); - return; - } - this.roomId = roomId; - this.isLogin = isLogin; - upData = new BilibiliUpInfo(); - upData.setRoomid(roomId); - info.setRoomid(roomId); - info.setGiftuser(0); - info.setVipuserindex(0); - info.setUserindex(0); - Live.lives.put(roomId, this); - updateUpInfo(); - com.yutou.bilibili.Tools.Log.i("roomId = " + roomId + ", isLogin = " + isLogin); - try { - start(); - } catch (Exception e) { - e.printStackTrace(); - } - } - - private void updateUpInfo() { - try { - upData = service.queryUp(upData); - //upData.setLive(1); - if (upData.getSavedanmu() == 1) { - danmuManager = new BiliBiliLiveDatabasesManager(); - danmuManager.init("[" + AppTools.getToDayTime() + "]" + upData.getRoomid()); - } else { - if (danmuManager != null) { - danmuManager.close(); - danmuManager = null; - } - } - } catch (Exception e) { - Log.e(e); - } - } - - private Date startTime; - - /** - * 开始监听 - * - * @throws Exception 发生异常 - */ - private void start() throws Exception { - run = true; - String url = LiveUtils.getLiveUrl(roomId); - if (url == null) { - stop(); - return; - } - upData.checkLiveStatus(); - startTime = new Date(); - HashMap header = new HashMap<>(); - header.put("Sec-WebSocket-Extensions", "permessage-deflate; client_max_window_bits"); - header.put("Sec-WebSocket-Key", "tORCZd8AI6xIyvqvgvI1Vw=="); - header.put("Sec-WebSocket-Version", "13"); - header.put("Cache-Control", "no-cache"); - header.put("Connection", "Upgrade"); - header.put("Host", "tx-gz-live-comet-02.chat.bilibili.com"); - header.put("Origin", "https://live.bilibili.com"); - header.put("Pragma", "no-cache"); - header.put("Upgrade", "websocket"); - header.put("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.190 Safari/537.36"); - if (client != null && !client.isClosed()) { - stop(); - return; - } - client = new WebSocketClient(new URI(url), header) { - private long time = 0; - private boolean init = true; - - @Override - public void onOpen(ServerHandshake serverHandshake) { - time = System.currentTimeMillis(); - com.yutou.bilibili.Tools.Log.i("ws: ok"); - try { - likeLive(); - } catch (Exception e) { - Log.e(e); - } - - } - - @Override - public void onMessage(String s) { - - } - - @Override - public void onMessage(ByteBuffer bytes) { - super.onMessage(bytes); - if (init) { - heartBeattimer = new Timer(); - heartBeattimer.schedule(new sendHeartbeat(), 2000, 20000); - init = false; - } - decompress(bytes.array()); - } - - @Override - public void onClose(int i, String s, boolean b) { - com.yutou.bilibili.Tools.Log.i("连接时间：" + (System.currentTimeMillis() - time)); - if (upData.getOfflinelistening() == 1) { - Log.i(roomId + " 断开连接，重连..."); - try { - init = true; - if(heartBeattimer!=null) { - heartBeattimer.cancel(); - } - if(client!=null) { - client.close(); - } - start(); - } catch (Exception e) { - Log.e(e); - } - return; - } - stop(); - client.close(); - run = false; - throw new RuntimeException("连接关闭: code = " + i + " msg = " + s + " b = " + b + " roomId=" + roomId); - } - - @Override - public void onClosing(int code, String reason, boolean remote) { - super.onClosing(code, reason, remote); - com.yutou.bilibili.Tools.Log.i("code = " + code + ", reason = " + reason + ", remote = " + remote); - run = false; - client.close(); - stop(); - } - - @Override - public void onError(Exception e) { - Log.e(e); - run = false; - client.close(); - stop(); - } - }; - client.connect(); - } - - public void stop() { - run = false; - client.close(); - if (danmuManager != null) { - danmuManager.close(); - } - if (SaveLive.getInstance().checkLive(roomId)) { - SaveLive.getInstance().stop(roomId); - } - Live.lives.remove(roomId); - com.yutou.bilibili.Tools.Log.i("退出" + roomId + "直播间"); - } - - /** - * 连接房间 - * - * @throws Exception e - */ - private void likeLive() throws Exception { - JSONObject tmp = LiveUtils.http_get("https://api.bilibili.com/x/web-interface/nav"); - if (tmp != null && tmp.getInteger("code") == -101) { - if (isLogin) { - new File("cookies.json").deleteOnExit(); - upData.setLive(0); - service.updateUpInfo(upData); - stop(); - return; - } else { - userId = 0; - } - } else if (tmp != null && tmp.getInteger("code") == 0) { - userId = tmp.getJSONObject("data").getInteger("mid"); - } - JSONObject http = LiveUtils.http_get("https://api.live.bilibili.com/xlive/web-room/v1/index/getDanmuInfo?id=" + roomId + "&type=0"); - if (http == null) { - stop(); - return; - } - JSONObject json = new JSONObject(); - json.put("uid", userId); - json.put("roomid", roomId); - json.put("protover", 2); - json.put("platform", "web"); - json.put("clientver", "2.6.36"); - json.put("type", 2); - json.put("key", http.getJSONObject("data").getString("token")); - ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); - outputStream.write(LiveUtils.toLH(json.toJSONString().length() + 16)); - outputStream.write(new byte[]{0, 16, 0, 1, 0, 0, 0, 7, 0, 0, 0, 1}); - outputStream.write(json.toJSONString().getBytes(StandardCharsets.UTF_8)); - outputStream.flush(); - // LiveUtils.printHex(outputStream.toByteArray()); - client.send(outputStream.toByteArray()); - checkLive(); - } - - /** - * 解压缩 - * - * @param data 待压缩的数据 - */ - public void decompress(byte[] data) { - try { - byte[] bytes = new byte[data.length - 16]; - System.arraycopy(data, 16, bytes, 0, data.length - 16); - - if (data.length > 32) { - List list = LiveUtils.getMsgList(LiveUtils.dec(bytes), new ArrayList<>(), true); - String tmp = ""; - for (int i = 0; i < list.size(); i++) { - processData(list.get(i), data); - tmp += "|||" + i + "[start] " + list.get(i) + " [end]||| "; - } - // Log.i(AppTools.getToDayTimeToString(startTime), roomId, "list size = " + list.size() + " data = " + tmp); - } else { - try { - JSONObject json = JSONObject.parseObject(new String(bytes, StandardCharsets.UTF_8)); - System.out.println("json = " + json); - com.yutou.bilibili.Tools.Log.i(json.toJSONString()); - } catch (Exception e) { - int popular = LiveUtils.bytesToInt2(bytes, 0); - info.setPopular(popular); - } - } - upData.setLive(LiveUtils.getLiveState(roomId) ? 1 : 0); - - } catch (Exception e) { - Log.e(e); - com.yutou.bilibili.Tools.Log.i("----------ERROR----------"); - com.yutou.bilibili.Tools.Log.i(new String(data, StandardCharsets.UTF_8)); - LiveUtils.printHex(LiveUtils.dec(data)); - com.yutou.bilibili.Tools.Log.i(new SimpleDateFormat("HH:mm:ss.SSS").format(new Date())); - } - } - - public void checkLive() { - boolean isLive = upData.checkLiveStatus() == 1; - if (SaveLive.getInstance().checkLive(roomId)) { - return; - } - if (isLive && upData.getSavelive() == 1) { - SaveLive.getInstance().addLive(this); - } - if (!isLive || upData.getSavelive() == 0) { - if (SaveLive.getInstance().checkLive(roomId)) { - SaveLive.getInstance().stop(roomId); - } - } - } - - - /** - * 处理数据 - * - * @param msg 数据json - * @param bytes 原始hex数据 - */ - public void processData(String msg, byte[] bytes) { - if (msg.contains("[object Object]")) { - return; - } - try { - JSONObject json = JSONObject.parseObject(msg); - JSONObject data; - BilibiliLiveData liveData = new BilibiliLiveData(); - String danmu; - GiftData giftData; - Log.i(AppTools.getToDayTime(), roomId, json.toJSONString()); - switch (json.getString("cmd")) { - case "INTERACT_WORD"://普通用户进直播间 - danmu = json.getJSONObject("data").getString("uname") + " 进入到直播间"; - liveData.setUid(json.getJSONObject("data").getInteger("uid")); - liveData.setType(LiveData.INTERACT_WORD); - liveData.setMsg(danmu); - liveData.setRoomid(roomId); - liveData.setSubtime(new Date()); - service.addLiveData(liveData); - break; - case "DANMU_MSG"://普通弹幕 - JSONArray infoData = json.getJSONArray("info"); - danmu = json.getJSONArray("info").getString(1); - DanmuData danmuData = new DanmuData(); - danmuData.setModel(infoData.getJSONArray(0).getInteger(1)); - danmuData.setFontSize(infoData.getJSONArray(0).getInteger(2)); - danmuData.setFontColor(infoData.getJSONArray(0).getInteger(3)); - danmuData.setTime(infoData.getJSONArray(0).getLong(4)); - danmuData.setUCode(infoData.getJSONArray(0).getString(7)); - danmuData.setDanmu(danmu); - danmuData.setUid(infoData.getJSONArray(2).getInteger(0)); - danmuData.setUname(infoData.getJSONArray(2).getString(1)); - - if (upData != null && upData.getSavedanmu() == 1) { - if (danmuManager != null) { - danmuManager.addDanmu(danmuData); - } - } - break; - case "SEND_GIFT"://送礼 - data = json.getJSONObject("data"); - giftData = getGiftData(data.getInteger("giftId")); - if (giftData == null) { - giftData = new GiftData(); - giftData.setName(data.getString("giftName")); - giftData.setId(data.getInteger("giftId")); - giftData.setPrice(data.getInteger("price")); - giftData.setIcon(""); - giftData.setDesc("这是阿B没有收录的礼物，金额可能不准（无法判别为免费礼物）"); - giftData.setRights("?"); - } - danmu = data.getString("uname") + " " + data.getString("action") + " " + giftData.getName(); - liveData.setType(LiveData.SEND_GIFT); - liveData.setUid(data.getInteger("uid")); - liveData.setGiftid(giftData.getId()); - liveData.setGiftindex(data.getInteger("num")); - liveData.setGiftname(giftData.getName()); - liveData.setMsg(danmu); - liveData.setPrice(giftData.getPrice()); - liveData.setPriceofcommission(giftData.getPrice() == 0 ? 0 : giftData.getPrice() / 2); - liveData.setSubtime(new Date()); - liveData.setRoomid(roomId); - info.setGiftuser(info.getGiftuser() + 1); - service.addLiveData(liveData); - break; - case "COMBO_SEND"://礼物连击 - data = json.getJSONObject("data"); - String gift = data.getString("giftName"); - if (gift == null || "null".equals(gift)) { - gift = data.getString("gift_name"); - } - danmu = data.getString("uname") + " " + data.getString("action") + " " + gift + "x" + data.getInteger("batch_combo_num"); - giftData = getGiftData(data.getInteger("gift_id")); - if (giftData == null) { - giftData = new GiftData(); - giftData.setName(data.getString("giftName")); - giftData.setId(data.getInteger("giftId")); - giftData.setPrice(data.getInteger("price")); - giftData.setIcon(""); - giftData.setDesc("这是阿B没有收录的礼物，金额可能不准（无法判别为免费礼物）"); - giftData.setRights("?"); - } - liveData.setType(LiveData.COMBO_SEND); - liveData.setUid(data.getInteger("uid")); - liveData.setGiftid(giftData.getId()); - liveData.setGiftindex(data.getInteger("batch_combo_num")); - liveData.setGiftname(giftData.getName()); - liveData.setMsg(danmu); - liveData.setPrice(giftData.getPrice() * liveData.getGiftindex()); - liveData.setPriceofcommission(giftData.getPrice() == 0 ? 0 : giftData.getPrice() / 2); - liveData.setSubtime(new Date()); - liveData.setRoomid(roomId); - info.setGiftuser(info.getGiftuser() + 1); - service.addLiveData(liveData); - break; - case "ENTRY_EFFECT"://舰长进直播间 - info.setVipuserindex(info.getVipuserindex() + 1); - danmu = "[舰长]" + json.getJSONObject("data").getString("uid") + " 进入到直播间"; - liveData.setUid(json.getJSONObject("data").getInteger("uid")); - liveData.setType(LiveData.ENTRY_EFFECT); - liveData.setMsg(danmu); - liveData.setRoomid(roomId); - liveData.setSubtime(new Date()); - service.addLiveData(liveData); - break; - case "LIVE_INTERACTIVE_GAME"://彩色弹幕？通过游戏弹幕 - break; - case "SUPER_CHAT_MESSAGE"://SC - data = json.getJSONObject("data"); - danmu = data.getJSONObject("user_info").getString("uname") + " " - + data.getJSONObject("gift").getString("gift_name") + " " + data.getInteger("price") + "元: " + data.getString("message"); - liveData.setType(LiveData.SUPER_CHAT_MESSAGE); - liveData.setUid(data.getInteger("uid")); - liveData.setMsg(danmu); - liveData.setGiftid(0); - liveData.setRoomid(roomId); - liveData.setGiftname(data.getJSONObject("gift").getString("gift_name")); - liveData.setGiftindex(1); - liveData.setPrice(data.getInteger("price") * 1000); - liveData.setPriceofcommission(liveData.getPrice() == 0 ? 0 : liveData.getPrice() / 2); - liveData.setSubtime(new Date()); - info.setGiftuser(info.getGiftuser() + 1); - service.addLiveData(liveData); - break; - case "USER_TOAST_MSG": - break; - case "GUARD_BUY"://开通/续费 牛逼的东西 - data = json.getJSONObject("data"); - liveData.setType(LiveData.GUARD_BUY); - liveData.setUid(data.getInteger("uid")); - liveData.setMsg(data.getString("gift_name")); - liveData.setGiftid(-data.getInteger("guard_level")); - liveData.setGiftindex(data.getInteger("num")); - liveData.setGiftname(data.getString("gift_name")); - liveData.setPrice(data.getInteger("price")); - liveData.setPriceofcommission(liveData.getPrice() == 0 ? 0 : liveData.getPrice() / 2); - liveData.setSubtime(new Date()); - liveData.setRoomid(roomId); - info.setGiftuser(info.getGiftuser() + 1); - service.addLiveData(liveData); - break; - case "LIVE"://开始直播,不过有在心跳包上做检测了，所以也无所谓？ - - JSONObject liveInfo = LiveUtils.getLiveInfo(upData); - if (liveInfo == null) { - break; - } - upData.setLive(liveInfo.getInteger("live_status")); - LiveInfo info = new LiveInfo(); - info.setTitle(liveInfo.getString("title")); - info.setKeyframe(liveInfo.getString("keyframe")); - info.setRoomId(liveInfo.getInteger("room_id")); - info.setShortId(liveInfo.getInteger("short_id")); - info.setInfo(liveInfo); - setLiveInfo(info); - StringBuilder builder = new StringBuilder(); - builder.append(upData.getName()).append("开播了！").append("\n"); - builder.append(info.getTitle()); - Tools.download(info.getKeyframe(), new DownloadInterface() { - @Override - public void onDownload(File file) { - super.onDownload(file); - QQBotManager.getInstance().sendMessage(file, builder.toString()); - } - }); - break; - case "PREPARING"://直播结束 - upData.setLive(0); - break; - case "WATCHED_CHANGE"://人气值 - try { - Live.this.info.setPopular(Integer.getInteger(json.getJSONObject("data").getString("text_small"))); - } catch (Exception ignored) { - } - break; - case "SUPER_CHAT_MESSAGE_JPN": - case "NOTICE_MSG": - case "HOT_RANK_CHANGED"://榜单更新等无用信息 - case "ONLINE_RANK_COUNT": - case "ONLINE_RANK_V2": - case "ONLINE_RANK_TOP3": - case "ROOM_REAL_TIME_MESSAGE_UPDATE": - case "WIDGET_BANNER"://鬼知道是啥 - case "HOT_RANK_SETTLEMENT": - case "PK_BATTLE_SETTLE_V2": - case "PK_BATTLE_END": - case "PK_BATTLE_SETTLE": - case "PK_BATTLE_PRE_NEW": - case "PK_BATTLE_PRE": - case "STOP_LIVE_ROOM_LIST"://不知道是啥，停止直播？ - - - //com.yutou.bilibili.Tools.Log.i(msg); - break; - default: - com.yutou.bilibili.Tools.Log.i("[" + roomId + "]" + msg); - liveData = new BilibiliLiveData(); - liveData.setType(LiveData.UNKNOWN_MESSAGE); - liveData.setUid(-1); - liveData.setMsg(msg); - liveData.setRoomid(roomId); - liveData.setSubtime(new Date()); - service.addLiveData(liveData); - } - checkLive(); - } catch (Exception e) { - Log.e(e); - Log.i("msg = " + msg); - // Log.i(AppTools.getToDayTimeToString(startTime), roomId, e.getLocalizedMessage()); - /*try { - JSONObject.parseObject(new String(bytes, StandardCharsets.UTF_8)); - processData(new String(bytes, StandardCharsets.UTF_8), null); - } catch (Exception e2) { - try { - JSONObject.parseObject(msg); - processData(msg, bytes); - } catch (Exception e3) { - Log.e(e2); - com.yutou.bilibili.Tools.Log.i("---------ERROR ！！ start-----"); - com.yutou.bilibili.Tools.Log.i(msg); - com.yutou.bilibili.Tools.Log.i("---------ERROR ！！ end-----"); - // LiveUtils.printHex(bytes); - } - }*/ - - } - - } - - public void clearInfo() { - info = new BilibiliLiveInfo(); - } - - public BilibiliLiveInfo getInfo() { - return info; - } - - public void setSaveDanmo(boolean saveDanmu) { - upData.setSavedanmu(saveDanmu ? 1 : 0); - } - - private static ApplicationContext applicationContext = null; - - @Override - public void setApplicationContext(@NotNull ApplicationContext applicationContext) throws BeansException { - if (Live.applicationContext == null) { - Live.applicationContext = applicationContext; - } - } - - public static T getBean(Class clazz) { - return applicationContext.getBean(clazz); - } - - public BilibiliUpInfo geData() { - return upData; - } - - /** - * 发送心跳包 - */ - public class sendHeartbeat extends TimerTask { - - public sendHeartbeat() { - - } - - @Override - public void run() { - if (!run) { - cancel(); - return; - } - updateUpInfo(); - if (upData.getOfflinelistening() != 1) { - if (upData.getLive() == 0) { - stop(); - } - } - try { - // com.yutou.bilibili.Tools.Log.i("-------发送心跳--------"); - ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); - outputStream.write(LiveUtils.toLH("[object Object]".length() + 16)); - outputStream.write(new byte[]{0, 16, 0, 1, 0, 0, 0, 2, 0, 0, 0, 1}); - outputStream.write("[object Object]".getBytes(StandardCharsets.UTF_8)); - outputStream.flush(); - client.send(outputStream.toByteArray()); - } catch (Exception e) { - Log.e(e); - com.yutou.bilibili.Tools.Log.i(client.isClosed()); - com.yutou.bilibili.Tools.Log.i(client.isOpen()); - - } - - } - } - - public GiftData getGiftData(int id) { - JSONObject item = JSONObject.parseObject(RedisTools.get("bili_gift_" + id, 2)); - GiftData data = new GiftData(); - data.setPrice(item.getInteger("price")); - data.setName(item.getString("name")); - data.setId(item.getInteger("id")); - data.setRights(item.getString("rights")); - data.setIcon(item.getString("img_basic")); - data.setDesc(item.getString("desc")); - return data; - } - - @Data - public static class LiveInfo { - private String title; - private String keyframe; - private int roomId; - private int shortId; - private JSONObject info; - - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/LiveController.java b/src/main/java/com/yutou/bilibili/BiliBili/LiveController.java deleted file mode 100644 index 51e0243..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/LiveController.java +++ /dev/null @@ -1,57 +0,0 @@ -package com.yutou.bilibili.BiliBili; - -import com.yutou.bilibili.BiliBili.Tools.SaveLive; -import com.yutou.bilibili.Tools.AppTools; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import org.springframework.util.StringUtils; - -import java.io.File; -import java.util.Timer; - -public class LiveController { - private static LiveController controller; - private Timer timer; - - public static LiveController getInstance() { - if (controller == null) { - controller = new LiveController(); - } - return controller; - } - - private LiveController() { - init(); - } - - private void init() { - - } - - public void updateUpInfo(BilibiliUpInfo data) { - Live live = LiveUtils.liveContains(data); - if (live != null) { - live.setSaveDanmo(data.getSavedanmu() == 1); - boolean saveLive=SaveLive.getInstance().checkLive(data.getRoomid()); - if(data.getSavelive()==1){ - if(!saveLive){ - SaveLive.getInstance().addLive(live); - } - }else{ - if(saveLive){ - SaveLive.getInstance().stop(data.getRoomid()); - } - } - } - if (data.getEnable()==1||data.getOfflinelistening()==1) { - if (live == null) { - live=new Live(); - live.add(data.getRoomid(), !StringUtils.isEmpty(AppTools.readFile(new File("cookies.json")))); - } - } else { - if (live != null) { - live.stop(); - } - } - - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/LiveUtils.java b/src/main/java/com/yutou/bilibili/BiliBili/LiveUtils.java deleted file mode 100644 index f055ccb..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/LiveUtils.java +++ /dev/null @@ -1,450 +0,0 @@ -package com.yutou.bilibili.BiliBili; - -import com.yutou.bilibili.BiliBili.Datas.BiliBiliUpData; -import com.yutou.bilibili.BiliBili.Datas.GiftData; -import com.alibaba.fastjson.JSON; -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.BiliBili.Tools.SaveLive; -import com.yutou.bilibili.Tools.AppTools; -import com.yutou.bilibili.Tools.HttpTools; -import com.yutou.bilibili.Tools.Log; -import com.yutou.bilibili.Tools.RedisTools; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import org.springframework.util.StringUtils; - -import javax.annotation.Resource; -import javax.xml.bind.DatatypeConverter; -import java.io.*; -import java.net.HttpURLConnection; -import java.net.URL; -import java.nio.ByteBuffer; -import java.nio.charset.StandardCharsets; -import java.util.*; -import java.util.zip.Inflater; - -public class LiveUtils { - private static String cookie = null; - private static Map liveStateMap = new HashMap<>(); - - public LiveUtils() { - - } - - - public static String getLiveUrl(int roomId) { - JSONObject json = http_get("https://api.live.bilibili.com/xlive/web-room/v1/index/getDanmuInfo?id=" + roomId + "&type=0"); - if (json != null) { - return "wss://" + json.getJSONObject("data").getJSONArray("host_list").getJSONObject(0).getString("host") + "/sub"; - } - return null; - } - - public static void printHex(byte[] bytes) { - String str = DatatypeConverter.printHexBinary(bytes); - for (int i = 0; i < str.length(); i = i + 4) { - if (i % 32 == 0 && i != 0) { - com.yutou.bilibili.Tools.Log.i("\n"); - } - if (str.length() - i > 4) { - com.yutou.bilibili.Tools.Log.i(str.substring(i, i + 4)); - } else { - com.yutou.bilibili.Tools.Log.i(str.substring(i)); - } - } - } - - public static String getHex(byte[] bytes) { - return DatatypeConverter.printHexBinary(bytes); - } - - public static byte[] dec(byte[] data) { - byte[] output = new byte[0]; - - Inflater decompresser = new Inflater(); - decompresser.reset(); - decompresser.setInput(data); - - ByteArrayOutputStream o = new ByteArrayOutputStream(data.length); - try { - byte[] buf = new byte[1024]; - while (!decompresser.finished()) { - int i = decompresser.inflate(buf); - o.write(buf, 0, i); - } - output = o.toByteArray(); - } catch (Exception e) { - // com.yutou.bilibili.Tools.Log.e(e); - try { - JSONObject json = JSONObject.parseObject(new String(data, StandardCharsets.UTF_8)); - ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); - // outputStream.write(toLH(json.toJSONString().length() + 16)); - //outputStream.write(new byte[]{0, 16, 0, 1, 0, 0, 0, 5, 0, 0, 0, 1}); - outputStream.write(json.toJSONString().getBytes(StandardCharsets.UTF_8)); - output = outputStream.toByteArray(); - } catch (Exception e1) { - output = data; - } - } finally { - try { - o.close(); - } catch (IOException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - - decompresser.end(); - return output; - } - - public static List getMsgList(byte[] bytes, List list, boolean isOne) { - int len = 16; - ByteBuffer datas = ByteBuffer.allocate(bytes.length - len); - byte[] heads = new byte[4]; - //获取到头信息 - System.arraycopy(bytes, 0, heads, 0, 4); - int size = bytesToInt2(heads, 0) - 16; - - //datas=MainApp.dec(datas); - if (size == datas.array().length) { - System.arraycopy(bytes, len, datas.array(), 0, bytes.length - len); - list.add(new String(datas.array(), StandardCharsets.UTF_8)); - } else { - try { - // com.yutou.bilibili.Tools.Log.i("datas size = " + size); - if (size > 1000000000) { - try { - JSONObject.parseObject(new String(bytes)); - list.add(new String(bytes, StandardCharsets.UTF_8)); - return list; - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - datas.clear(); - datas = ByteBuffer.allocate(size); - System.arraycopy(bytes, len, datas.array(), 0, size); - list.add(new String(datas.array(), StandardCharsets.UTF_8)); - byte[] tmps = new byte[bytes.length - size - 16]; - int length = bytes.length - (size + len); - System.arraycopy(bytes, size + (len), tmps, 0, length); - - getMsgList(tmps, list, false); - } catch (Exception e) { - list.add(new String(bytes, StandardCharsets.UTF_8)); - } - - } - return list; - } - - public static byte[] toLH(int n) { - byte[] b = new byte[4]; - b[3] = (byte) (n & 0xff); - b[2] = (byte) (n >> 8 & 0xff); - b[1] = (byte) (n >> 16 & 0xff); - b[0] = (byte) (n >> 24 & 0xff); - return b; - } - - public static int bytesToInt2(byte[] src, int offset) { - int value; - value = (int) (((src[offset] & 0xFF) << 24) - | ((src[offset + 1] & 0xFF) << 16) - | ((src[offset + 2] & 0xFF) << 8) - | (src[offset + 3] & 0xFF)); - return value; - } - - public static String getCookie() { - if (StringUtils.isEmpty(getFile("cookies.json"))) { - - return ""; - } - JSONObject json = JSONObject.parseObject(getFile("cookies.json")); - StringBuilder builder = new StringBuilder(); - for (String s : json.keySet()) { - builder.append(s).append("=").append(json.getString(s)).append(";"); - - } - return builder.toString(); - } - - /* public static String createCookie(){ - String bid="AUTO"+AppTools.randomString(16,new String[]{AppTools.numbers}); - bid="AUTO6616152607312876"; - return String.format("LIVE_BUVID=%s; _uuid=%sinfoc; buvid3=%sinfoc; sid=%s; fingerprint=%s; buvid_fp=%sinfoc; buvid_fp_plain=%sinfoc; PVID=%d;" - ,bid - ,UUID.randomUUID().toString() - ,UUID.randomUUID().toString() - ,"bm3byv4p" - ,"074eae35a01a56c40558a54ab33179e5" - ,UUID.randomUUID().toString() - ,UUID.randomUUID().toString() - ,AppTools.randomCommon(0,9,1)[0]); - }*/ - private static long oldBiliBiliHttpTime = 0; - - public synchronized static JSONObject http_get(String url) { - try { - // Log.i("调用url = "+url); - HttpURLConnection connection = getBiliHttpGet(url, getCookie()); - BufferedInputStream stream = new BufferedInputStream(connection.getInputStream()); - ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); - byte[] bytes = new byte[1024]; - int len = 0, size; - while ((len = stream.read(bytes)) != -1) { - outputStream.write(bytes, 0, len); - outputStream.flush(); - } - String str = new String(outputStream.toByteArray(), StandardCharsets.UTF_8); - outputStream.close(); - try { - JSONObject json = JSON.parseObject(str); - return json; - } catch (Exception e) { - JSONObject json = new JSONObject(); - json.put("html", str); - return json; - } finally { - - stream.close(); - connection.disconnect(); - } - - } catch (IOException e) { - //com.yutou.bilibili.Tools.Log.e(e); - Log.e(e); - Log.i("412 in " + url); - } - return null; - } - - public static JSONObject http_post(String url, String body) { - try { - if (System.currentTimeMillis() - oldBiliBiliHttpTime < 1000) { - try { - Thread.sleep(500); - } catch (InterruptedException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - oldBiliBiliHttpTime = System.currentTimeMillis(); - } - HttpURLConnection connection = getBiliHttpPost(url, getCookie()); - OutputStream connectionOutputStream = null; - if (!StringUtils.isEmpty(body)) { - connectionOutputStream = connection.getOutputStream(); - connectionOutputStream.write(body.getBytes(StandardCharsets.UTF_8)); - connectionOutputStream.flush(); - } - BufferedInputStream stream = new BufferedInputStream(connection.getInputStream()); - ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); - byte[] bytes = new byte[1024]; - int len = 0, size; - while ((len = stream.read(bytes)) != -1) { - outputStream.write(bytes, 0, len); - outputStream.flush(); - } - String str = new String(outputStream.toByteArray(), StandardCharsets.UTF_8); - outputStream.close(); - try { - JSONObject json = JSON.parseObject(str); - json.put("cookie", connection.getHeaderField("Set-Cookie")); - return json; - } catch (Exception e) { - JSONObject json = new JSONObject(); - json.put("html", str); - json.put("cookie", connection.getHeaderField("Set-Cookie")); - return json; - } finally { - stream.close(); - if (connectionOutputStream != null) { - connectionOutputStream.close(); - } - connection.disconnect(); - } - - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - return new JSONObject(); - } - - public static HttpURLConnection getBiliHttpPost(String url, String cookie) throws Exception { - if (System.currentTimeMillis() - oldBiliBiliHttpTime < 1000) { - try { - Thread.sleep(500); - } catch (InterruptedException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - oldBiliBiliHttpTime = System.currentTimeMillis(); - } - HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection(); - connection.setRequestMethod("POST"); - connection.setDoOutput(true); - connection.setDoInput(true); - connection.setRequestProperty("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8"); - connection.setRequestProperty("Accept-Language", "zh-CN,zh;q=0.8"); - connection.setRequestProperty("Cache-Control", "max-age=0"); - connection.setRequestProperty("Referer", "https://live.bilibili.com"); - connection.setRequestProperty("Connection", "keep-alive"); - connection.setRequestProperty("Upgrade-Insecure-Requests", "1"); - connection.setRequestProperty("Cookie", cookie); - connection.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/44.0.2403.157 Safari/537.36"); - return connection; - } - - public static HttpURLConnection getBiliHttpGet(String url, String cookie) throws IOException { - if (System.currentTimeMillis() - oldBiliBiliHttpTime < 1000) { - try { - Thread.sleep(500); - } catch (InterruptedException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - oldBiliBiliHttpTime = System.currentTimeMillis(); - } - HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection(); - connection.setRequestProperty("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8"); - connection.setRequestProperty("Accept-Language", "zh-CN,zh;q=0.8"); - connection.setRequestProperty("Cache-Control", "max-age=0"); - connection.setRequestProperty("Referer", "https://live.bilibili.com"); - connection.setRequestProperty("Connection", "keep-alive"); - connection.setRequestProperty("Upgrade-Insecure-Requests", "1"); - connection.setRequestProperty("Cookie", cookie); - connection.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/44.0.2403.157 Safari/537.36"); - connection.setReadTimeout(5000); - connection.setConnectTimeout(5000); - return connection; - } - - - public static String getFile(String filePath) { - File file = new File(filePath); - try { - BufferedReader reader = new BufferedReader(new FileReader(file)); - String str = "", tmp; - while ((tmp = reader.readLine()) != null) { - str += tmp; - } - reader.close(); - return str; - } catch (Exception e) { - - } - return null; - } - - - public static JSONObject getUserLoginInfo() { - return LiveUtils.http_get("https://api.bilibili.com/x/web-interface/nav"); - } - - public static Live liveContains(BilibiliUpInfo data) { - for (Live live : Live.lives.values()) { - if (live.getInfo().getRoomid().equals(data.getRoomid())) { - return live; - } - } - return null; - } - - public static long getMid(int roomId) { - JSONObject http_get = http_get("https://api.live.bilibili.com/room/v1/Room/room_init?id=" + roomId); - if (http_get == null) { - return -1; - } - if (http_get.getInteger("code") == 0) { - return http_get.getJSONObject("data").getLong("uid"); - } - return -1; - } - - public static JSONObject getLiveInfo(BilibiliUpInfo info) { - List list = new ArrayList<>(); - list.add(info); - JSONObject infoByUid = getStatusInfoByUid(list); - return infoByUid.getJSONObject("data").getJSONObject(info.getMid() + ""); - } - - public static JSONObject getLiveRoomInfo(int roomId) { - String url = "https://api.live.bilibili.com/xlive/web-room/v2/index/getRoomPlayInfo?room_id=" + roomId + "&protocol=0,1&format=0,1,2&codec=0,1&qn=10000&platform=web&ptype=8"; - return http_get(url); - } - - public static boolean getLiveState(int roomId) { - return liveStateMap.getOrDefault(roomId, false); - } - - public static JSONObject getStatusInfoByUid(List list) { - JSONArray userId = new JSONArray(); - for (BilibiliUpInfo info : list) { - userId.add(info.getMid()); - } - JSONObject json=new JSONObject(); - json.put("uids",userId); - return JSONObject.parseObject(HttpTools.post("https://api.live.bilibili.com/room/v1/Room/get_status_info_by_uids", json.toJSONString().getBytes(StandardCharsets.UTF_8))); - } - - public static void reloadLiveState(List list) { - JSONObject json = getStatusInfoByUid(list); - try { - if (json.getInteger("code") == 0) { - JSONObject data = json.getJSONObject("data"); - for (BilibiliUpInfo info : list) { - if (data.containsKey(info.getMid() + "")) { - liveStateMap.put(info.getRoomid(), data.getJSONObject(info.getMid()+"").getInteger("live_status") == 1); - } - } - } - } catch (Exception e) { - Log.i("live state = " + json); - } - } - - - public static boolean isLivePlayer(int roomId) { - return getLiveState(roomId); - } - - - public static class LiveGiftConfig { - @Resource - IBiliBiliLiveService service; - private static LiveGiftConfig config; - - public static LiveGiftConfig getInstance() { - if (config == null) { - config = new LiveGiftConfig(); - } - return config; - } - - private LiveGiftConfig() { - init(); - } - - public void init() { - JSONObject json = http_get("https://api.live.bilibili.com/xlive/web-room/v1/giftPanel/giftConfig?platform=pc"); - if (json == null) { - return; - } - JSONArray list = json.getJSONObject("data").getJSONArray("list"); - for (Object o : list) { - JSONObject item = (JSONObject) o; - RedisTools.set(2, "bili_gift_" + item.getInteger("id"), item.toJSONString()); - } - } - - } - - public static void main(String[] args) { - List list=new ArrayList<>(); - BilibiliUpInfo info=new BilibiliUpInfo(); - info.setMid(402417817L); - list.add(info); - JSONObject infoByUid = getStatusInfoByUid(list); - System.out.println(infoByUid); - } - -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Services/IBiliBiliLiveService.java b/src/main/java/com/yutou/bilibili/BiliBili/Services/IBiliBiliLiveService.java deleted file mode 100644 index 90ec788..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Services/IBiliBiliLiveService.java +++ /dev/null @@ -1,156 +0,0 @@ -package com.yutou.bilibili.BiliBili.Services; - -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveData; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveInfo; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import org.apache.ibatis.annotations.Param; - -import java.math.BigDecimal; -import java.util.Date; -import java.util.List; -import java.util.Map; - -public interface IBiliBiliLiveService { - - /** - * 获取所有up信息 - * @return up列表 - */ - List getUpInfo(); - List getUpInfo(int uid,String power); - /** - * 根据RoomId查询UP信息 - * @param data roomId - * @return UP信息 - */ - BilibiliUpInfo queryUp(BilibiliUpInfo data); - BilibiliUpInfo queryUpToUrl(String url); - - /** - * 根据RoomId查询UP信息 - * @param roomId roomId - * @return UP信息 - */ - BilibiliUpInfo queryUpToRoomId(int roomId); - /** - * 获取所有直播数据 - * @return 直播数据列表 - */ - List getLiveData(); - /** - * 获取所有直播统计 - * @return 直播统计列表 - */ - List getLiveInfo(); - /** - * 添加up信息 - * @param upData up信息 - * @return 是否成功 - */ - boolean addUpInfo(BilibiliUpInfo upData); - /** - * 更新UP信息 - * @param upData up信息 - * @return 是否成功 - */ - boolean updateUpInfo(BilibiliUpInfo upData); - /** - * 删除指定up信息 - * @param upData roomId - * @return 是否成功 - */ - boolean deleteUp(BilibiliUpInfo upData); - /** - * 新增直播数据 - * @param data 直播数据 - * @return 是否成功 - */ - boolean addLiveData(BilibiliLiveData data); - /** - * 添加直播统计 - * @param data 直播统计 - * @return 是否成功 - */ - boolean addLiveInfo(BilibiliLiveInfo data); - - /** - * 查询指定时段直播数据 - * @param roomId roomId - * @param startTime 开始时间，为空表示为当天00:00:00 - * @param endTime 结束时间，为空表示为当时 - * @param type 类型 - * @return 列表 - */ - List queryLiveData(int roomId, Date startTime, Date endTime, String[] type); - /** - * 查询指定时段直播某个礼物总数 - * @param roomId roomId - * @param giftId 礼物id,-1则为全部 - * @param startTime 开始时间 - * @param endTime 结束时间 - * @return 数量 - */ - int queryGiftSize(int roomId, int giftId, Date startTime, Date endTime); - /** - * 返回指定时段各个礼物总数 - * @param roomId roomId - * @param startTime 开始时间 - * @param endTime 结束时间 - * @return {giftName:size} - */ - Map queryGiftSize(int roomId, Date startTime, Date endTime); - /** - * 查询指定时段收益情况 - * @param roomId roomId - * @param startTime 开始时间 - * @param endTime 结束时间 - * @return 未扣除收成比例的金瓜子数量 - */ - int queryPriceSize(int roomId, Date startTime, Date endTime); - /** - * 根据礼物id查询礼物信息，该方法主要用于阿B接口未收录的礼物信息 - * @param giftId 礼物id - * @return 直播信息 - */ - BilibiliLiveData queryGiftOfId(int giftId); - /** - * 查询指定时段去重送礼人数 - * - * @param upData up信息 - * @param startTime 开始时间 - * @param endTime 结束时间 - * @param type 礼物类型 - * @return 人数 - */ - int queryGiftUserToDistinct(BilibiliUpInfo upData, Date startTime, Date endTime, String[] type); - /** - * 查询时段内人气数量 - * @param upData up信息 - * @param startTime 开始时间 - * @param endTime 结束时间 - * @return 人气 - */ - int queryPopularCount(BilibiliUpInfo upData,Date startTime, Date endTime); - /** - * 查询时段内进场人数 - * @param upData up信息 - * @param startTime 开始时间 - * @param endTime 结束时间 - * @return 人数 - */ - int queryUserCount(BilibiliUpInfo upData,Date startTime, Date endTime); - /** - * 查询时段内礼物详情 - * @param upData up信息 - * @param startTime 开始时间 - * @param endTime 结束时间 - * @return 人数 - */ - Map queryUserGiftCount(BilibiliUpInfo upData,Date startTime,Date endTime); - - List queryLiveInfo(BilibiliUpInfo upInfo,Date startTime,Date endTime); - - List>queryTimeGroup(int roomId, Date startTime, Date endTime); - - List>queryGiftTimeGroup(int roomId, Date startTime, Date endTime); -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Services/impl/BiliBiliLiveServiceImpl.java b/src/main/java/com/yutou/bilibili/BiliBili/Services/impl/BiliBiliLiveServiceImpl.java deleted file mode 100644 index 377a64a..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Services/impl/BiliBiliLiveServiceImpl.java +++ /dev/null @@ -1,244 +0,0 @@ -package com.yutou.bilibili.BiliBili.Services.impl; - -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.Services.ISystemConfigService; -import com.yutou.bilibili.Tools.Config; -import com.yutou.bilibili.mybatis.Bili.mybatis.dao.BilibiliLiveDataDao; -import com.yutou.bilibili.mybatis.Bili.mybatis.dao.BilibiliLiveInfoDao; -import com.yutou.bilibili.mybatis.Bili.mybatis.dao.BilibiliUpInfoDao; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.*; -import com.yutou.bilibili.mybatis.dao.UBiliUpDao; -import com.yutou.bilibili.mybatis.model.UBiliUp; -import com.yutou.bilibili.mybatis.model.UBiliUpExample; -import org.springframework.stereotype.Service; - -import javax.annotation.Resource; -import java.math.BigDecimal; -import java.text.SimpleDateFormat; -import java.util.*; - -@Service("BiliBiliLiveService") -public class BiliBiliLiveServiceImpl implements IBiliBiliLiveService { - @Resource - BilibiliUpInfoDao upInfoDao; - @Resource - BilibiliLiveDataDao liveDataDao; - @Resource - BilibiliLiveInfoDao liveInfoDao; - @Resource - UBiliUpDao biliUpDao; - @Resource - ISystemConfigService configService; - - private T getData(List list) { - if (list.isEmpty()) { - return null; - } - return list.get(0); - } - - @Override - public List getUpInfo() { - return upInfoDao.selectByExample(new BilibiliUpInfoExample()); - } - - @Override - public List getUpInfo(int uid, String power) { - UBiliUpExample example = new UBiliUpExample(); - List upInfos = new ArrayList<>(); - if (power.equals("[-1]")) { - upInfos = upInfoDao.selectByExample(new BilibiliUpInfoExample()); - return upInfos; - } - example.createCriteria().andUidEqualTo(uid); - List list = biliUpDao.selectByExample(example); - int[] ids = new int[list.size()]; - for (int i = 0; i < list.size(); i++) { - ids[i] = list.get(i).getRoomid(); - } - if (list.isEmpty()) { - return upInfos; - } - upInfos = upInfoDao.queryToRoomIds(ids); - com.yutou.bilibili.Tools.Log.i("list = " + upInfos.size()); - return upInfos; - } - - @Override - public BilibiliUpInfo queryUp(BilibiliUpInfo data) { - if (data.getRoomid() == null) { - return null; - } - return queryUpToRoomId(data.getRoomid()); - } - - @Override - public BilibiliUpInfo queryUpToUrl(String url) { - BilibiliUpInfoExample example = new BilibiliUpInfoExample(); - example.createCriteria().andUrlEqualTo(url); - return getData(upInfoDao.selectByExample(example)); - } - - @Override - public BilibiliUpInfo queryUpToRoomId(int roomId) { - BilibiliUpInfoExample example = new BilibiliUpInfoExample(); - example.createCriteria().andRoomidEqualTo(roomId); - return getData(upInfoDao.selectByExample(example)); - } - - - @Override - public List getLiveData() { - return liveDataDao.selectByExample(new BilibiliLiveDataExample()); - } - - @Override - public List getLiveInfo() { - return liveInfoDao.selectByExample(new BilibiliLiveInfoExample()); - } - - @Override - public boolean addUpInfo(BilibiliUpInfo upData) { - BilibiliUpInfoExample example = new BilibiliUpInfoExample(); - example.createCriteria().andRoomidEqualTo(upData.getRoomid()); - if (upInfoDao.selectByExample(example).isEmpty()) { - return upInfoDao.insert(upData) > 0; - } else { - return false; - } - } - - @Override - public boolean updateUpInfo(BilibiliUpInfo upData) { - return upInfoDao.updateByPrimaryKey(upData) > 0; - } - - @Override - public boolean deleteUp(BilibiliUpInfo upData) { - BilibiliUpInfoExample example = new BilibiliUpInfoExample(); - example.createCriteria().andRoomidEqualTo(upData.getRoomid()); - return upInfoDao.deleteByExample(example) > 0; - } - - @Override - public boolean addLiveData(BilibiliLiveData data) { - if (!isLive()) { - return false; - } - - return liveDataDao.insert(data) > 0; - } - - @Override - public boolean addLiveInfo(BilibiliLiveInfo data) { - if (!isLive()) { - return false; - } - return liveInfoDao.insert(data) > 0; - } - - @Override - public List queryLiveData(int roomId, Date startTime, Date endTime, String[] type) { - return liveDataDao.queryLiveData(roomId, startTime, endTime, type); - } - - @Override - public int queryGiftSize(int roomId, int giftId, Date startTime, Date endTime) { - int count = 0; - for (BilibiliLiveData liveData : liveDataDao.queryTimeOfRoomid(roomId, giftId, startTime, endTime)) { - count += liveData.getGiftindex() == null ? 0 : liveData.getGiftindex(); - } - return count; - } - - @Override - public Map queryGiftSize(int roomId, Date startTime, Date endTime) { - Map map = new HashMap<>(); - List list = liveDataDao.queryTimeOfRoomid(roomId, -1, startTime, endTime); - for (BilibiliLiveData liveData : list) { - if (map.containsKey(liveData.getGiftname())) { - map.put(liveData.getGiftname(), map.get(liveData.getGiftname()) + (liveData.getGiftindex() == null ? 0 : liveData.getGiftindex())); - } else { - map.put(liveData.getGiftname(), 1); - } - } - return map; - } - - @Override - public int queryPriceSize(int roomId, Date startTime, Date endTime) { - int count = 0; - for (BilibiliLiveData liveData : liveDataDao.queryTimeOfRoomid(roomId, -1, startTime, endTime)) { - count += liveData.getPrice() == null ? 0 : liveData.getPrice(); - } - return count; - } - - @Override - public BilibiliLiveData queryGiftOfId(int giftId) { - BilibiliLiveDataExample example = new BilibiliLiveDataExample(); - example.createCriteria().andGiftidEqualTo(giftId); - List list = liveDataDao.selectByExample(example); - if (list.isEmpty()) { - return null; - } else { - return list.get(0); - } - } - - @Override - public int queryGiftUserToDistinct(BilibiliUpInfo upData, Date startTime, Date endTime, String[] type) { - return liveDataDao.queryGiftUserToDistinct(upData.getRoomid(), -1, startTime, endTime, type); - } - - @Override - public int queryPopularCount(BilibiliUpInfo upData, Date startTime, Date endTime) { - int count = 0; - for (BilibiliLiveInfo info : liveInfoDao.queryTimeOfRoomid(upData.getRoomid(), startTime, endTime)) { - count += info.getPopular(); - } - return count; - } - - @Override - public int queryUserCount(BilibiliUpInfo upData, Date startTime, Date endTime) { - int count = 0; - for (BilibiliLiveInfo info : liveInfoDao.queryTimeOfRoomid(upData.getRoomid(), startTime, endTime)) { - count += (info.getUserindex() + info.getVipuserindex()); - } - return count; - } - - @Override - public Map queryUserGiftCount(BilibiliUpInfo upData, Date startTime, Date endTime) { - Map map = new HashMap<>(); - List list = liveInfoDao.queryTimeOfRoomid(upData.getRoomid(), startTime, endTime); - for (BilibiliLiveInfo info : list) { - map.put(new SimpleDateFormat("HH").format(info.getSubtime()), info.getGiftuser() == null ? 0 : info.getGiftuser()); - } - return map; - } - - @Override - public List queryLiveInfo(BilibiliUpInfo upInfo, Date startTime, Date endTime) { - return liveInfoDao.queryTimeOfRoomid(upInfo.getRoomid(), startTime, endTime); - } - - @Override - public List> queryTimeGroup(int roomId, Date startTime, Date endTime) { - return liveDataDao.queryPriceTimeGroup(roomId, startTime, endTime); - } - - @Override - public List> queryGiftTimeGroup(int roomId, Date startTime, Date endTime) { - return liveDataDao.queryGiftTimeGroup(roomId, startTime, endTime); - } - - public boolean isLive() { - String flag = configService.getConfig(Config.BILI_LIVE_FLAG); - if (flag != null) { - return !"0".equals(flag); - } - return true; - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Tools/BiliTools.java b/src/main/java/com/yutou/bilibili/BiliBili/Tools/BiliTools.java deleted file mode 100644 index 025cf1c..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Tools/BiliTools.java +++ /dev/null @@ -1,40 +0,0 @@ -package com.yutou.bilibili.BiliBili.Tools; - -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Datas.BiliBiliUpData; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import com.yutou.bilibili.sqlite.BiliBiliLiveDatabasesManager; -import org.springframework.util.StringUtils; - -public class BiliTools { - public static BilibiliUpInfo checkout(BilibiliUpInfo info) { - if (StringUtils.isEmpty(info.getUrl())) { - return null; - } - if (!info.getUrl().contains("http")) { - info.setUrl("https://live.bilibili.com/" + info.getUrl()); - } - BilibiliUpInfo data = getBiliUpInfo(info.getUrl()); - info.setUrl(data.getUrl()); - info.setRoomid(data.getRoomid()); - info.setName(data.getName()); - return info; - } - - public static BilibiliUpInfo getBiliUpInfo(String url) { - if(!url.startsWith("http")){ - url="https://live.bilibili.com/"+url; - } - BilibiliUpInfo upData = new BilibiliUpInfo(); - JSONObject json = LiveUtils.http_get("https://api.live.bilibili.com/xlive/web-room/v1/index/getInfoByRoom?room_id=" + url.replace("https://live.bilibili.com/", "").split("\\?")[0]); - com.yutou.bilibili.Tools.Log.i(json); - String roomId = json.getJSONObject("data").getJSONObject("room_info").getInteger("room_id") + ""; - String name = json.getJSONObject("data").getJSONObject("anchor_info").getJSONObject("base_info").getString("uname"); - upData.setRoomid(Integer.parseInt(roomId)); - upData.setName(name); - upData.setUrl(url); - upData.setMid(LiveUtils.getMid(upData.getRoomid())); - return upData; - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Tools/LiveT.java b/src/main/java/com/yutou/bilibili/BiliBili/Tools/LiveT.java deleted file mode 100644 index 619b61c..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Tools/LiveT.java +++ /dev/null @@ -1,64 +0,0 @@ -package com.yutou.bilibili.BiliBili.Tools; - -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveInfo; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import org.java_websocket.client.WebSocketClient; -import org.springframework.beans.BeansException; -import org.springframework.context.ApplicationContext; -import org.springframework.context.ApplicationContextAware; -import org.springframework.stereotype.Component; - -import javax.annotation.PostConstruct; -import javax.annotation.Resource; -import java.util.ArrayList; -import java.util.List; -import java.util.Timer; - -public class LiveT implements ApplicationContextAware { - private static ApplicationContext applicationContext = null; - - IBiliBiliLiveService service; - - - public static List list=new ArrayList<>(); - int v; - - @PostConstruct - public void init() { - com.yutou.bilibili.Tools.Log.i("初始化2"); - } - - public LiveT() { - com.yutou.bilibili.Tools.Log.i("初始化"); - } - - public void add(int a){ - - v=a; - BilibiliUpInfo info=new BilibiliUpInfo(); - info.setRoomid(a); - info.setOfflinelistening(0); - info.setSavedanmu(0); - info.setEnable(0); - info.setLive(0); - info.setSavelive(0); - info.setName("a"+a); - info.setUrl("#"); - service=getBean(IBiliBiliLiveService.class); - service.addUpInfo(info); - } - public int get(){ - return v; - } - - @Override - public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { - if(LiveT.applicationContext==null){ - LiveT.applicationContext=applicationContext; - } - } - public static T getBean(Class clazz){ - return applicationContext.getBean(clazz); - } -} diff --git a/src/main/java/com/yutou/bilibili/BiliBili/Tools/SaveLive.java b/src/main/java/com/yutou/bilibili/BiliBili/Tools/SaveLive.java deleted file mode 100644 index a274861..0000000 --- a/src/main/java/com/yutou/bilibili/BiliBili/Tools/SaveLive.java +++ /dev/null @@ -1,243 +0,0 @@ -package com.yutou.bilibili.BiliBili.Tools; - -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Datas.AppData; -import com.yutou.bilibili.BiliBili.Live; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.QQBot.QQBotManager; -import com.yutou.bilibili.Tools.*; - -import java.io.*; -import java.net.HttpURLConnection; -import java.net.URLEncoder; -import java.nio.charset.StandardCharsets; -import java.util.*; - -public class SaveLive { - private static SaveLive live; - private final List saveList = new ArrayList<>(); - private final Map downloads = new HashMap<>(); - private final Map heartbeats = new HashMap<>(); - - public static void main(String[] args) { - } - - private SaveLive() { - } - - public static SaveLive getInstance() { - if (live == null) { - live = new SaveLive(); - } - return live; - } - - public void addLive(Live live) { - if (saveList.contains(live.getInfo().getRoomid() + "")) { - return; - } - if (live.geData().checkLiveStatus() == 0) { - return; - } - start(live.getInfo().getRoomid()); - } - - public void startLive(int roomId) { - start(roomId); - saveList.add(roomId + ""); - } - - - public boolean checkLive(int roomId) { - return saveList.contains(roomId + ""); - } - - public File getLiveFile(int roomId) { - return downloads.get(roomId).liveFile; - } - - private long timer = 0; - - public void stop(int roomId) { - com.yutou.bilibili.Tools.Log.i("t停止录播:" + roomId + " time=" + (System.currentTimeMillis() - timer)); - saveList.remove(roomId + ""); - - if (downloads.containsKey(roomId)) { - downloads.get(roomId).isSave = false; - downloads.remove(roomId); - } - - if (heartbeats.containsKey(roomId)) { - heartbeats.get(roomId).cancel(); - heartbeats.remove(roomId); - } - - } - - private void start(int roomId) { - if (checkLive(roomId)) { - return; - } - timer = System.currentTimeMillis(); - DownloadThread thread = new DownloadThread(roomId); - downloads.put(roomId, thread); - saveList.add(roomId + ""); - } - - public List getLiveList() { - return saveList; - } - - private class DownloadThread extends Thread { - int roomId = 0; - boolean isSave = true; - Timer heartbeat; - File liveFile; - - public DownloadThread(int roomId) { - this.roomId = roomId; - isSave = true; - - start(); - - } - - private String getLiveUrl(int roomId) { - JSONObject json = LiveUtils.http_get(String.format("https://api.live.bilibili.com/xlive/web-room/v1/playUrl/playUrl?cid=%d&platform=web", roomId)); - if (json != null && json.getInteger("code") == 0) { - return json.getJSONObject("data").getJSONArray("durl").getJSONObject(0).getString("url"); - } else { - return null; - } - } - - @Override - public void run() { - super.run(); - String url = getLiveUrl(roomId); - try { - if (AppData.LIVE_SAVE_FFMPEG) { - ffmpegDownload(url); - } else { - httpDownload(url); - } - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - if (!(e instanceof FileNotFoundException)) { - Log.i("录制发生意外:" + e.getMessage()); - QQBotManager.getInstance().sendMessage("录制发生意外:" + e.getLocalizedMessage()); - } - } - SaveLive.this.stop(roomId); - - } - - private void ffmpegDownload(String url) throws Exception { - liveFile = new File(String.format("%slive%s%s%s[%s]%d.mp4", - ConfigTools.load(ConfigTools.CONFIG, "liveSavePath"), - File.separator, - AppTools.getToDayTime(), - File.separator, - AppTools.getToDayNowTimeToString().replace(":", ""), - roomId)); - if (!liveFile.exists()) { - liveFile.mkdirs(); - liveFile.delete(); - } - String exec = String.format("%s -user_agent \"%s\" -cookies \"%s\" -headers \"%s\" -i \"%s\" -threads 8 -c:v copy -y \"%s\" %s ", - AppData.FFMPEG, - AppData.BILIBILI_HEADERS, - LiveUtils.getCookie(), - "Referer:https://live.bilibili.com", - url, - liveFile.getAbsolutePath(), - "" - ); - System.out.println(exec); - boolean isPrint = false; - Process process = AppTools.exec(exec); - InputStream inputStream = process.getErrorStream(); - byte[] bytes = new byte[2048]; - while (inputStream.read(bytes) > -1 && isSave) { - //System.out.println(new String(bytes,StandardCharsets.UTF_8)); - if (!isPrint) { - com.yutou.bilibili.Tools.Log.i("开始录制:" + roomId); - QQBotManager.getInstance().sendMessage(roomId + " 已启动录制"); - isPrint = true; - } - } - process.destroy(); - System.out.println("----------------stop ffmpeg"); - inputStream.close(); - com.yutou.bilibili.Tools.Log.i("录制完成:" + roomId); - QQBotManager.getInstance().sendMessage("录制完成:" + roomId); - } - - private void httpDownload(String url) throws Exception { - HttpURLConnection connection = LiveUtils.getBiliHttpGet(url, LiveUtils.getCookie()); - connection.setReadTimeout(5000); - connection.setConnectTimeout(5000); - heartbeat = new Timer(); - //Heartbeat beat = new Heartbeat(); - heartbeat.schedule(new Heartbeat(), 0, 30000); - heartbeats.put(roomId, heartbeat); - //heartbeats.add(beat); - InputStream inputStream = connection.getInputStream(); - liveFile = new File(String.format("%slive%s%s%s[%s]%d.mp4", - ConfigTools.load(ConfigTools.CONFIG, "liveSavePath"), - File.separator, - AppTools.getToDayTime(), - File.separator, - AppTools.getToDayNowTimeToString().replace(":", ""), - roomId)); - if (!liveFile.exists()) { - liveFile.mkdirs(); - liveFile.delete(); - } - boolean isPrint = false; - FileOutputStream outputStream = new FileOutputStream(liveFile); - int len; - byte[] bytes = new byte[1024]; - while ((len = inputStream.read(bytes)) != -1 && isSave) { - outputStream.write(bytes, 0, len); - outputStream.flush(); - if (!isPrint) { - com.yutou.bilibili.Tools.Log.i("开始录制:" + roomId); - QQBotManager.getInstance().sendMessage(roomId + " 已启动录制"); - isPrint = true; - } - } - outputStream.close(); - inputStream.close(); - com.yutou.bilibili.Tools.Log.i("录制完成:" + roomId + " save = " + isSave + " len = " + len); - QQBotManager.getInstance().sendMessage("录制完成:" + roomId + " save = " + isSave + " len = " + len); - } - - private class Heartbeat extends TimerTask { - int nextInterval = 1; - - @Override - public void run() { - try { - JSONObject hearBeat; - JSONObject userHear = new JSONObject(); - JSONObject cookie = JSONObject.parseObject(LiveUtils.getFile("cookies.json")); - hearBeat = LiveUtils.http_get("https://api.live.bilibili.com/relation/v1/Feed/heartBeat"); - com.yutou.bilibili.Tools.Log.i(hearBeat + " \t"); - hearBeat = LiveUtils.http_get(String.format("https://live-trace.bilibili.com/xlive/rdata-interface/v1/heartbeat/webHeartBeat?hb=%s&pf=web", URLEncoder.encode(new String(Base64.getEncoder().encode(String.format("%d|%d|1|0", nextInterval, roomId).getBytes(StandardCharsets.UTF_8))), "UTF-8"))); - com.yutou.bilibili.Tools.Log.i(hearBeat + "\t"); - nextInterval = hearBeat.getJSONObject("data").getInteger("next_interval"); - com.yutou.bilibili.Tools.Log.i("next = " + nextInterval); - userHear.put("csrf_token", cookie.getString("bili_jct")); - userHear.put("csrf", cookie.getString("bili_jct")); - userHear.put("visit_id", ""); - hearBeat = LiveUtils.http_post("https://api.live.bilibili.com/User/userOnlineHeart", HttpTools.toUrlParams(userHear)); - com.yutou.bilibili.Tools.Log.i("[" + AppTools.getToDayNowTimeToString() + "]" + hearBeat); - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - } - - } -} diff --git a/src/main/java/com/yutou/bilibili/BilibiliApplication.java b/src/main/java/com/yutou/bilibili/BilibiliApplication.java index 2d1b912..98662cf 100644 --- a/src/main/java/com/yutou/bilibili/BilibiliApplication.java +++ b/src/main/java/com/yutou/bilibili/BilibiliApplication.java @@ -1,7 +1,5 @@ package com.yutou.bilibili; -import com.yutou.bilibili.BiliBili.Live; -import com.yutou.bilibili.BiliBili.Tools.LiveT; import com.yutou.bilibili.QQBot.QQBotManager; import com.yutou.bilibili.Tools.ExcelUtils; import com.yutou.bilibili.Tools.ServiceTools; @@ -9,14 +7,13 @@ import org.springframework.boot.SpringApplication; import org.springframework.boot.autoconfigure.SpringBootApplication; import org.springframework.context.annotation.Import; -@Import({Live.class,ServiceTools.class, ExcelUtils.class, QQBotManager.class}) +@Import({ServiceTools.class, ExcelUtils.class, QQBotManager.class}) @SpringBootApplication public class BilibiliApplication { public static String version="0.8.4"; public static void main(String[] args) { - QQBotManager.getInstance().init(); SpringApplication.run(BilibiliApplication.class, args); } diff --git a/src/main/java/com/yutou/bilibili/Controllers/SystemConfigController.java b/src/main/java/com/yutou/bilibili/Controllers/SystemConfigController.java index 5079bef..e68988b 100644 --- a/src/main/java/com/yutou/bilibili/Controllers/SystemConfigController.java +++ b/src/main/java/com/yutou/bilibili/Controllers/SystemConfigController.java @@ -1,16 +1,16 @@ package com.yutou.bilibili.Controllers; -import com.alibaba.fastjson.JSONObject; +import com.alibaba.fastjson2.JSONObject; import com.yutou.bilibili.BiliBili.Datas.AppData; import com.yutou.bilibili.QQBot.QQBotManager; import com.yutou.bilibili.Services.ISystemConfigService; import com.yutou.bilibili.Tools.Config; +import jakarta.annotation.Resource; import org.springframework.stereotype.Controller; import org.springframework.util.StringUtils; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; -import javax.annotation.Resource; import java.io.UnsupportedEncodingException; import java.net.URLDecoder; @@ -125,8 +125,6 @@ public class SystemConfigController { @ResponseBody @RequestMapping("/system/qq/login.do") public JSONObject loginQQ() { - QQBotManager.getInstance().stop(); - QQBotManager.getInstance().init(); JSONObject json = new JSONObject(); json.put("code", 0); json.put("msg", "ok"); diff --git a/src/main/java/com/yutou/bilibili/Controllers/UserController.java b/src/main/java/com/yutou/bilibili/Controllers/UserController.java index 02799f7..3439850 100644 --- a/src/main/java/com/yutou/bilibili/Controllers/UserController.java +++ b/src/main/java/com/yutou/bilibili/Controllers/UserController.java @@ -1,29 +1,24 @@ package com.yutou.bilibili.Controllers; -import com.alibaba.fastjson.JSON; -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; +import com.alibaba.fastjson2.JSON; +import com.alibaba.fastjson2.JSONArray; +import com.alibaba.fastjson2.JSONObject; import com.yutou.bilibili.Services.ISystemConfigService; -import com.yutou.bilibili.BiliBili.Tools.BiliTools; import com.yutou.bilibili.Services.IUserService; -import com.yutou.bilibili.Tools.AppTools; import com.yutou.bilibili.Tools.Config; -import com.yutou.bilibili.Tools.RedisTools; import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; import com.yutou.bilibili.mybatis.model.UBiliUp; import com.yutou.bilibili.mybatis.model.UUser; -import org.apache.tomcat.util.security.MD5Encoder; +import com.yutou.utils.AppTools; +import com.yutou.utils.RedisTools; +import jakarta.annotation.Resource; +import jakarta.servlet.http.HttpServletRequest; +import jakarta.servlet.http.HttpServletResponse; import org.springframework.stereotype.Controller; import org.springframework.util.StringUtils; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; -import javax.annotation.Resource; -import javax.servlet.http.HttpServletRequest; -import javax.servlet.http.HttpServletResponse; -import java.nio.charset.StandardCharsets; -import java.util.Base64; import java.util.Date; import java.util.UUID; @@ -31,8 +26,7 @@ import java.util.UUID; public class UserController { @Resource IUserService service; - @Resource - IBiliBiliLiveService biliService; + @Resource ISystemConfigService configService; @@ -51,7 +45,7 @@ public class UserController { user.setPower("[1,2,3,5,6,7]"); user.setSubtime(new Date()); user.setLogintoken(UUID.randomUUID().toString()); - RedisTools.set(user.getLogintoken(),user.getId()+"",30*24*60); + RedisTools.set(user.getLogintoken(), user.getId() + "", 30 * 24 * 60); AppTools.setCookie(response, "login", user.getLogintoken(), -1); boolean flag = service.reg(user); if (flag) { @@ -72,7 +66,7 @@ public class UserController { if (service.login(user.getUser(), user.getPassword())) { user = service.getUser(user.getUser()); //user.setLogintoken(UUID.randomUUID().toString()); - RedisTools.set(user.getLogintoken(),user.getId()+"",30*24*60); + RedisTools.set(user.getLogintoken(), user.getId() + "", 30 * 24 * 60); AppTools.setCookie(response, "login", user.getLogintoken(), -1); json.put("msg", "登陆成功"); json.put("power", user.getPower()); @@ -124,7 +118,7 @@ public class UserController { user.setPower(loginUser.getPower()); user.setBiliCookie(loginUser.getBiliCookie()); user.setSubtime(loginUser.getSubtime()); - user.setPassword(AppTools.getMD5 (user.getPassword())); + user.setPassword(AppTools.getMD5(user.getPassword())); service.update(user); json.put("code", 0); json.put("msg", "修改成功"); @@ -156,37 +150,6 @@ public class UserController { @RequestMapping("/user/up/set/add.do") public JSONObject addUp(HttpServletRequest request, BilibiliUpInfo info) { JSONObject json = new JSONObject(); - String token = AppTools.getLoginToken(request); - if (StringUtils.isEmpty(token)) { - json.put("code", "-1"); - json.put("msg", "未登录"); - } else { - int roomid = 0; - if (!info.getUrl().startsWith("http")) { - info.setUrl("https://live.bilibili.com/" + info.getUrl()); - } - BilibiliUpInfo tmp = biliService.queryUpToUrl(info.getUrl()); - if (tmp == null) { - info = BiliTools.checkout(info); - if (info != null) { - roomid = info.getRoomid(); - info.setEnable(0); - biliService.addUpInfo(info); - } - } else { - roomid = tmp.getRoomid(); - } - if (roomid == 0) { - json.put("code", -1); - json.put("msg", "添加失败"); - return json; - } - UBiliUp up = new UBiliUp(); - up.setUid(Integer.parseInt(RedisTools.get(token))); - up.setRoomid(roomid); - json.put("code", 0); - json.put("msg", service.addUp(up) ? "添加成功" : "添加失败"); - } return json; } @@ -199,14 +162,14 @@ public class UserController { json.put("code", -1); json.put("msg", "未登录"); } else { - UUser user = service.getUserToUid(Integer.parseInt(RedisTools.get(token)));; + UUser user = service.getUserToUid(Integer.parseInt(RedisTools.get(token))); if (user == null) { json.put("code", -2); json.put("msg", "未登录"); return json; } JSONArray powers = JSONArray.parseArray(user.getPower()); - JSONObject ujson = (JSONObject) JSON.toJSON(user); + JSONObject ujson = JSONObject.parseObject(JSONObject.toJSONString(user)); ujson.remove("logintoken"); ujson.remove("biliCookie"); ujson.remove("password"); diff --git a/src/main/java/com/yutou/bilibili/QQBot/QQBotManager.java b/src/main/java/com/yutou/bilibili/QQBot/QQBotManager.java index f09afd9..1594205 100644 --- a/src/main/java/com/yutou/bilibili/QQBot/QQBotManager.java +++ b/src/main/java/com/yutou/bilibili/QQBot/QQBotManager.java @@ -1,449 +1,212 @@ package com.yutou.bilibili.QQBot; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Controllers.RealTimeDataController; -import com.yutou.bilibili.BiliBili.Live; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.BiliBili.Tools.SaveLive; -import com.yutou.bilibili.BilibiliApplication; -import com.yutou.bilibili.Tools.*; -import com.yutou.bilibili.interfaces.DownloadInterface; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; -import kotlin.coroutines.CoroutineContext; -import net.mamoe.mirai.Bot; -import net.mamoe.mirai.BotFactory; -import net.mamoe.mirai.event.EventHandler; -import net.mamoe.mirai.event.GlobalEventChannel; -import net.mamoe.mirai.event.ListeningStatus; -import net.mamoe.mirai.event.SimpleListenerHost; -import net.mamoe.mirai.event.events.GroupMessageEvent; -import net.mamoe.mirai.event.events.MessageEvent; -import net.mamoe.mirai.message.data.Image; -import net.mamoe.mirai.message.data.MessageChainBuilder; -import net.mamoe.mirai.utils.BotConfiguration; -import net.mamoe.mirai.utils.ExternalResource; -import org.jetbrains.annotations.NotNull; -import org.springframework.beans.BeansException; -import org.springframework.context.ApplicationContext; -import org.springframework.context.ApplicationContextAware; +import com.alibaba.fastjson2.JSONObject; +import com.yutou.napcat.QQDatabase; +import com.yutou.napcat.QQNumberManager; +import com.yutou.napcat.handle.BaseHandle; +import com.yutou.napcat.handle.MessageHandleBuild; +import com.yutou.napcat.handle.Reply; +import com.yutou.napcat.handle.Text; +import com.yutou.napcat.http.NapCatApi; +import com.yutou.napcat.model.FriendBean; +import com.yutou.napcat.model.GroupBean; +import com.yutou.napcat.model.SendMessageResponse; +import com.yutou.okhttp.HttpBody; +import com.yutou.okhttp.HttpCallback; +import com.yutou.utils.ConfigTools; +import okhttp3.Headers; +import retrofit2.Response; + -import javax.annotation.Resource; import java.io.File; -import java.io.IOException; -import java.lang.reflect.Field; -import java.text.SimpleDateFormat; -import java.util.ArrayList; -import java.util.Date; +import java.util.Arrays; import java.util.List; -import java.util.Objects; -import java.util.function.Consumer; -import java.util.regex.Matcher; -import java.util.regex.Pattern; -public class QQBotManager implements ApplicationContextAware { - private static ApplicationContext applicationContext; +public class QQBotManager { - @Override - public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { - if (QQBotManager.applicationContext == null) { - QQBotManager.applicationContext = applicationContext; - } - } + public static Long defGroup = 891655174L; + public static Long defQQ = 583819556L; - private T getBean(Class tClass) { - return applicationContext.getBean(tClass); - } - - private static class QQCommands { - private final static String QQ_HELP = "!help"; - private final static String QQ_SYSTEM_LOG = "!log"; - private final static String QQ_SYSTEM_RESTART = "!restart"; - private final static String QQ_GET_IP = "!ip"; - private final static String QQ_GET_VERSION = "!version"; - private final static String QQ_LIVE_LIST = "!列表"; - private final static String QQ_LIVE_SAVE = "!录播列表"; - private final static String QQ_LIVE_TO_DAY_DATE = "!今日数据"; - private final static String QQ_LIVE_DATE = "!数据"; - private final static String QQ_LIVE_USE_SAVE = "!启动录播"; - private final static String QQ_LIVE_RELINK="!relink"; - } private static QQBotManager botManager = null; - private Bot bot; - private static long qqGroup; + private static final long qqGroup = 891655174L; private boolean isLogin = false; private static boolean isInit = false; - @Resource - RealTimeDataController realTimeDataController; - private QQBotManager() { - qqGroup = ConfigTools.load(ConfigTools.CONFIG, "qq_group",long.class,950075833L); - } - public void stop(){ - if(bot!=null){ - bot.close(); - bot=null; - } - } - public void init() { - if (!ConfigTools.load(ConfigTools.CONFIG, "qq_bot",boolean.class)) { - return; - } - new Thread(() -> { - long qq = Long.parseLong((String) ConfigTools.load(ConfigTools.CONFIG, "qq_number")); - String password = (String) ConfigTools.load(ConfigTools.CONFIG, "qq_password"); - bot = BotFactory.INSTANCE.newBot(qq, password, new BotConfiguration() { - { - setProtocol(MiraiProtocol.ANDROID_PAD); - fileBasedDeviceInfo("qq_bot_devices_info.json"); - if (!((boolean) ConfigTools.load(ConfigTools.CONFIG, "qq_debug"))) { - noBotLog(); - noNetworkLog(); - } - } - }); - //Events.registerEvents(bot, new MessageListener()); - GlobalEventChannel.INSTANCE.registerListenerHost(new MessageListener()); - Log.i("准备登陆"); - bot.login(); - Log.i("登陆成功"); + Boolean isRun = ConfigTools.load(ConfigTools.CONFIG, "qq_bot", Boolean.class); + if (isRun != null && isRun) { isLogin = true; - isInit = true; - new Thread(new Runnable() { - @Override - public void run() { - try { - Thread.sleep(1000); - } catch (InterruptedException e) { - Log.e(e); - } - String str = sendMessage("姬妻酱上线拉~"); - Log.i(str); - - } - }).start(); - bot.join(); - - }).start(); - + init(); + } } - public static QQBotManager getInstance() { + private void init() { + NapCatApi.getGroupApi().getGroupList().enqueue(new HttpCallback>() { + @Override + public void onResponse(Headers headers,int code, String status, List response, String rawResponse) { + for (GroupBean groupBean : response) { + QQDatabase.addGroup(groupBean.getGroupId(), groupBean); + QQNumberManager.getManager().addNumber(groupBean.getGroupId(), true); + } + } + + @Override + public void onFailure(Throwable throwable) { + + } + }); + NapCatApi.getFriendApi().getFriendList().enqueue(new HttpCallback>() { + @Override + public void onResponse(Headers headers,int code, String status, List response, String rawResponse) { + for (FriendBean friendBean : response) { + QQDatabase.addUser(friendBean.getUserId(), friendBean); + QQNumberManager.getManager().addNumber(friendBean.getUserId(), false); + } + } + + @Override + public void onFailure(Throwable throwable) { + + } + }); + NapCatApi.getUtilsApi().getLoginInfo().enqueue(new HttpCallback() { + @Override + public void onResponse(Headers headers,int code, String status, FriendBean response, String rawResponse) { + QQDatabase.setMe(response); + } + + @Override + public void onFailure(Throwable throwable) { + + } + }); + isInit = true; + } + + public synchronized static QQBotManager getInstance() { if (botManager == null && !isInit) { botManager = new QQBotManager(); } return botManager; } - public boolean isLogin() { - return isLogin; - } - - private Image getImage(File file) { - if (bot != null) { - ExternalResource resource=ExternalResource.create(file); - Image image; - image= Objects.requireNonNull(bot.getGroup(qqGroup)).uploadImage(resource); - try { - resource.close(); - } catch (IOException e) { - e.printStackTrace(); - } - return image; - } - return null; - } private String getNotLoginQQ() { return "没有登录QQ"; } - - public String sendMessage(String text) { - if (bot != null) { - try { - return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(text).toString(); - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - return getNotLoginQQ(); + public SendMessageResponse sendPrivateMessage(Long qq, BaseHandle... items) { + return sendMessage(true, qq, items); } - public String sendMessage(Long group, String text) { - if (bot != null) { - try { - return Objects.requireNonNull(bot.getGroup(group)).sendMessage(text).toString(); - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - return getNotLoginQQ(); + private SendMessageResponse sendGroupMessage(Long group, BaseHandle... items) { + return sendMessage(false, group, items); } - public void sendMessage(Long group, MessageChainBuilder builder) { - if (bot != null) { - Objects.requireNonNull(bot.getGroup(group)).sendMessage(builder.asMessageChain()); - } + public SendMessageResponse sendMessage(boolean user, Long qq, String msg) { + return sendMessage(user, qq, new Text(msg)); } + public SendMessageResponse sendMessage(Long qq, BaseHandle... items) { + return sendMessage(QQDatabase.checkFriend(qq), qq, Arrays.asList(items)); + } - public String sendMessage(File imageFile, String text) { + public SendMessageResponse sendMessage(boolean isUser, Long qq, BaseHandle... items) { + return sendMessage(isUser, qq, Arrays.asList(items)); + } + + public SendMessageResponse sendMessage(boolean isUser, Long qq, List> items) { try { - if (bot != null) { - Image image = getImage(imageFile); - MessageChainBuilder builder = new MessageChainBuilder(); - if (image != null) { - builder.append(image); + if(!ConfigTools.load(ConfigTools.CONFIG,ConfigTools.QQ, Boolean.class)){ + return null; + } + MessageHandleBuild handleBuild = MessageHandleBuild + .create() + .setGroup(!isUser) + .setQQNumber(qq); + for (BaseHandle item : items) { + if (item.getData() == null) { + continue; } - builder.append(text); - return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(builder.asMessageChain()).toString(); + if (item instanceof Reply) { + if (((Reply) item).getData().getId() == -1) { + continue; + } + } + handleBuild.add(item); + } + Response> response; + if (isUser) { + response = NapCatApi.getMessageApi().sendPrivateMsg( + handleBuild.build() + ).execute(); + } else { + response = NapCatApi.getMessageApi().sendGroupMsg( + handleBuild.build() + ).execute(); + } + if (response.body() != null) { + return response.body().getData(); } } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); + e.printStackTrace(); + SendMessageResponse response = new SendMessageResponse(); + response.setId(-1); + response.setE(e.getMessage()); + return response; } + return null; + } + + public String sendMessage(String text) { return getNotLoginQQ(); } - public String sendMessage(List imgs, String text) { - if (bot != null) { - MessageChainBuilder builder = new MessageChainBuilder(); - for (File img : imgs) { - builder.append(Objects.requireNonNull(getImage(img))); - } - builder.append(text); - return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(builder.asMessageChain()).toString(); - } - return getNotLoginQQ(); + public SendMessageResponse sendMessage(Long group, String text) { + return sendMessage(QQDatabase.checkFriend(group), group, new Text(text)); } - private static List getImages(String str) { - List list = new ArrayList<>(); - String regex = ""; - Pattern pattern = Pattern.compile(regex); - Matcher matcher = pattern.matcher(str); - while (matcher.find()) { - list.add(matcher.group().replace("", "") - .trim()); + public void sendMessage(Long group, StringBuilder builder) { + if (QQNumberManager.getManager().isGroup(group)) { + System.out.println("发群"); + } else { + System.out.println("发个人"); } - return list; } + public void sendMessage(File imageFile, Long qq, String text) { + sendMessage(imageFile, qq, null, text); + } + + public void sendMessage(File imageFile, Long qq, String replyMessageId, String text) { + + } + + public void sendMessage(File imageFile, String text) { + } + + public void sendMessage(List imgs, Long qq, String text) { + + } + + public static void main(String[] args) { - getInstance(); + JSONObject json = new JSONObject(); + json.put("t1", 3234567890L); + System.out.println("json = " + json); + + String tmp = json.toString(); + JSONObject json2 = JSONObject.parseObject(tmp); + System.out.println("json2 = " + json2); } - private static class MessageListener extends SimpleListenerHost { - @Override - public void handleException(@NotNull CoroutineContext context, @NotNull Throwable exception) { - // super.handleException(context, exception); - exception.printStackTrace(); - } - - - @EventHandler - public ListeningStatus onMessage(MessageEvent event) { - String message=event.getMessage().contentToString(); - String msg = message.replace("！", "!").toLowerCase(); - msg=msg.trim(); - StringBuilder builder = new StringBuilder(); - JSONObject json; - String[] cmd = new String[0]; - int _roomId = 0; - switch (msg) { - case QQCommands.QQ_GET_IP: - json = JSONObject.parseObject(HttpTools.get("https://api.asilu.com/ip/")); - String ip = json.getString("ip"); - getInstance().sendMessage("服务器IP:" + ip); - break; - case QQCommands.QQ_GET_VERSION: - sendVersion(); - break; - case QQCommands.QQ_SYSTEM_RESTART: - getInstance().sendMessage("正在重启服务"); - System.out.println("结束进程"); - try { - AppTools.exec("cd /home/yutou/public/servier/biliob && ./start.sh"); - } catch (Exception e) { - e.printStackTrace(); - } - break; - case QQCommands.QQ_LIVE_LIST: - builder.append("当前正在记录数据的直播间:"); - builder.append("\n"); - for (Live live : Live.lives.values()) { - Live.LiveInfo info=live.getLiveInfo(); - if (live.geData().getLive()==1) { - builder.append("【直播中】"); - } - builder - .append("名字:").append(live.geData().getName()).append(" ") - .append(" roomId:").append(live.geData().getRoomid()).append(" "); - if (info != null) { - builder.append("标题:").append(info.getTitle()).append("\n"); - } else { - builder.append("\n"); - } - } - getInstance().sendMessage(builder.toString()); - break; - case QQCommands.QQ_LIVE_SAVE: - builder.append("当前正在录制的直播间:"); - builder.append("\n"); - for (String roomId : SaveLive.getInstance().getLiveList()) { - BilibiliUpInfo data = new BilibiliUpInfo(); - data.setRoomid(Integer.parseInt(roomId)); - Live live = LiveUtils.liveContains(data); - if (live != null) { - File file = SaveLive.getInstance().getLiveFile(Integer.parseInt(roomId)); - builder - .append("名字:").append(live.geData().getName()).append(" ") - .append("文件大小(字节):").append(file.length()).append(" ") - .append("roomId:").append(live.geData().getRoomid()).append("\n"); - } - - } - getInstance().sendMessage(builder.toString()); - break; - case QQCommands.QQ_HELP: - for (Field field : QQCommands.class.getDeclaredFields()) { - try { - field.setAccessible(true); - builder.append(field.get(null)).append("\n"); - } catch (IllegalAccessException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - getInstance().sendMessage(builder.toString()); - break; - case QQCommands.QQ_LIVE_RELINK: - for (Live live : Live.lives.values()) { - live.stop(); - } - break; - default: - if (msg.startsWith(QQCommands.QQ_LIVE_TO_DAY_DATE)) { - try { - cmd = msg.split(" "); - _roomId = Integer.parseInt(cmd[1]); - QQBotManager.getInstance().sendMessage("请稍等，正在查询..."); - sendGiftData(_roomId, null, null); - } catch (Exception e) { - getInstance().sendMessage("参数错误。\n使用方式: " + QQCommands.QQ_LIVE_TO_DAY_DATE + "+空格+roomId"); - } - } else if (msg.startsWith(QQCommands.QQ_LIVE_DATE)) { - try { - cmd = msg.split(" "); - _roomId = Integer.parseInt(cmd[1]); - Date startTime = new SimpleDateFormat("yyyy-MM-dd_HH:mm:ss").parse(cmd[2]); - Date endTime = new SimpleDateFormat("yyyy-MM-dd_HH:mm:ss").parse(cmd[3]); - QQBotManager.getInstance().sendMessage("请稍等，正在查询..."); - sendGiftData(_roomId, startTime, endTime); - } catch (Exception e) { - getInstance().sendMessage("参数错误。" + - "\n使用方式: " + QQCommands.QQ_LIVE_DATE + "+空格+roomId+起始时间+空格+结束时间" + - "\n时间格式：年-月-日_时:分:秒 例:2021-4-1_12:00:00" + - "\n时分秒不可省略"); - } - } else if (msg.startsWith(QQCommands.QQ_LIVE_USE_SAVE)) { - try { - cmd = msg.split(" "); - _roomId = Integer.parseInt(cmd[1]); - if (SaveLive.getInstance().checkLive(_roomId)) { - SaveLive.getInstance().stop(_roomId); - } - SaveLive.getInstance().startLive(_roomId); - getInstance().sendMessage("已启动" + _roomId + "的录播"); - } catch (Exception e) { - getInstance().sendMessage("参数错误。\n使用方式: " + QQCommands.QQ_LIVE_USE_SAVE + "+空格+roomId"); - } - } else if (msg.startsWith(QQCommands.QQ_SYSTEM_LOG)) { - try { - cmd = message.replace("！","!").replace(QQCommands.QQ_SYSTEM_LOG,"").split(" "); - RedisTools.set(1,"live-log",cmd[1].trim()); - getInstance().sendMessage("日志设为:"+cmd[1].trim()); - } catch (Exception e) { - e.printStackTrace(); - } - }else { - Log.i("未知指令:"+msg); - } - } - return ListeningStatus.LISTENING; // 表示继续监听事件 - } - - public void sendGiftData(int roomId, Date startTime, Date endTime) { - StringBuilder builder = new StringBuilder(); - if (startTime == null) { - startTime = AppTools.getToDayStartTime(); - } - if (endTime == null) { - endTime = AppTools.getToDayNowTime(); - } - if (getInstance().realTimeDataController == null) { - getInstance().realTimeDataController = getInstance().getBean(RealTimeDataController.class); - } - JSONObject json = getInstance().realTimeDataController.queryToDayLiveData(roomId, startTime, endTime); - System.out.println(json); - builder.append("当前人气:").append(json.getJSONObject("data").getInteger("popular")).append("\n"); - builder.append("普通观众入场:").append(json.getJSONObject("data").getInteger("userLength")).append("\n"); - builder.append("舰长入场:").append(json.getJSONObject("data").getInteger("vipLength")).append("\n"); - int price = 0; - for (Object o : json.getJSONObject("data").getJSONArray("price")) { - price += ((JSONObject) o).getInteger("price"); - } - builder.append("金瓜子:").append(price).append(" 抽成后:").append((price / 2) / 1000).append("￥").append("\n"); - builder.append("礼物收益情况:").append("\n"); - for (Object o : json.getJSONObject("data").getJSONArray("gift")) { - builder.append(((JSONObject) o).getString("giftName")).append(":").append(((JSONObject) o).getInteger("size")).append("\n"); - } - getInstance().sendMessage(builder.toString()); - } - - private List files; - private int index = 0; - - private void sendImagesMsg(List imgs, String text) { - files = new ArrayList<>(); - index = 0; - if (imgs.size() == 0) { - getInstance().sendMessage(text); - return; - } - for (String img : imgs) { - Tools.download(img, new DownloadInterface() { - @Override - public void onDownload(File file) { - super.onDownload(file); - files.add(file); - send(imgs.size(), text); - } - - @Override - public void onError(Exception e) { - super.onError(e); - index++; - send(imgs.size(), text); - } - }); - } - } - - private void send(int size, String text) { - if ((files.size() + index) == size) { - String str = getInstance().sendMessage(files, text); - Log.i("str = " + str); - } - } - - private void sendVersion() { - String msg = "软件版本：" + BilibiliApplication.version; - QQBotManager.getInstance().sendMessage(msg); - } + public boolean isLogin() { + return true; } + + } diff --git a/src/main/java/com/yutou/bilibili/Services/impl/SystemConfigImpl.java b/src/main/java/com/yutou/bilibili/Services/impl/SystemConfigImpl.java index 339a2d8..471bfec 100644 --- a/src/main/java/com/yutou/bilibili/Services/impl/SystemConfigImpl.java +++ b/src/main/java/com/yutou/bilibili/Services/impl/SystemConfigImpl.java @@ -4,9 +4,9 @@ import com.yutou.bilibili.Services.ISystemConfigService; import com.yutou.bilibili.mybatis.dao.SConfigDao; import com.yutou.bilibili.mybatis.model.SConfig; import com.yutou.bilibili.mybatis.model.SConfigExample; +import jakarta.annotation.Resource; import org.springframework.stereotype.Service; -import javax.annotation.Resource; import java.util.List; @Service("SystemConfigService") diff --git a/src/main/java/com/yutou/bilibili/Services/impl/UserServiceImpl.java b/src/main/java/com/yutou/bilibili/Services/impl/UserServiceImpl.java index f208237..38c3ad4 100644 --- a/src/main/java/com/yutou/bilibili/Services/impl/UserServiceImpl.java +++ b/src/main/java/com/yutou/bilibili/Services/impl/UserServiceImpl.java @@ -1,20 +1,17 @@ package com.yutou.bilibili.Services.impl; import com.yutou.bilibili.Services.IUserService; -import com.yutou.bilibili.Tools.AppTools; import com.yutou.bilibili.mybatis.dao.UBiliUpDao; import com.yutou.bilibili.mybatis.dao.UUserDao; import com.yutou.bilibili.mybatis.model.UBiliUp; import com.yutou.bilibili.mybatis.model.UBiliUpExample; import com.yutou.bilibili.mybatis.model.UUser; import com.yutou.bilibili.mybatis.model.UUserExample; -import org.apache.tomcat.util.security.MD5Encoder; +import com.yutou.utils.AppTools; +import jakarta.annotation.Resource; import org.springframework.stereotype.Service; import org.springframework.util.StringUtils; -import javax.annotation.Resource; -import java.nio.charset.StandardCharsets; -import java.util.Base64; import java.util.List; @Service("UserService") @@ -26,7 +23,7 @@ public class UserServiceImpl implements IUserService { @Override public boolean login(String uname, String password) { UUserExample example=new UUserExample(); - String _password=AppTools.getMD5 (password); + String _password= AppTools.getMD5 (password); example.createCriteria().andUserEqualTo(uname).andPasswordEqualTo(_password); return !userDao.selectByExample(example).isEmpty(); } diff --git a/src/main/java/com/yutou/bilibili/Test/TestController.java b/src/main/java/com/yutou/bilibili/Test/TestController.java index d4ec070..a7bef08 100644 --- a/src/main/java/com/yutou/bilibili/Test/TestController.java +++ b/src/main/java/com/yutou/bilibili/Test/TestController.java @@ -1,40 +1,8 @@ package com.yutou.bilibili.Test; -import com.alibaba.fastjson.JSON; -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.Live; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.BiliBili.Tools.LiveT; -import com.yutou.bilibili.BiliBili.Tools.SaveLive; import org.springframework.stereotype.Controller; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.ResponseBody; @Controller public class TestController { - @ResponseBody - @RequestMapping("/add.do") - public String test(int a) { - - return "ok"; - } - @ResponseBody - @RequestMapping("/get.do") - public JSONObject get(){ - JSONObject json=new JSONObject(); - JSONArray array=new JSONArray(); - json.put("size", Live.lives.size()); - for (Live live :Live.lives.values()) { - JSONObject item=new JSONObject(); - item.put("info", JSON.toJSON(live.getInfo())); - item.put("data",JSON.toJSON(live.geData())); - item.put("isLive", live.geData().getLive()==1); - array.add(item); - } - json.put("array",array); - json.put("saveVideo",JSONArray.toJSON(SaveLive.getInstance().getLiveList())); - return json; - } } diff --git a/src/main/java/com/yutou/bilibili/Tools/APIFilter.java b/src/main/java/com/yutou/bilibili/Tools/APIFilter.java index c28c95d..d014ffa 100644 --- a/src/main/java/com/yutou/bilibili/Tools/APIFilter.java +++ b/src/main/java/com/yutou/bilibili/Tools/APIFilter.java @@ -1,20 +1,23 @@ package com.yutou.bilibili.Tools; -import com.alibaba.fastjson.JSONArray; +import com.alibaba.fastjson2.JSONArray; import com.yutou.bilibili.Services.IUserService; import com.yutou.bilibili.mybatis.dao.PermissionDao; import com.yutou.bilibili.mybatis.model.Permission; import com.yutou.bilibili.mybatis.model.PermissionExample; import com.yutou.bilibili.mybatis.model.UUser; +import com.yutou.utils.AppTools; +import com.yutou.utils.Log; +import com.yutou.utils.RedisTools; +import jakarta.annotation.Resource; +import jakarta.servlet.*; +import jakarta.servlet.annotation.WebFilter; +import jakarta.servlet.http.Cookie; +import jakarta.servlet.http.HttpServletRequest; +import jakarta.servlet.http.HttpServletResponse; import org.springframework.stereotype.Component; import org.springframework.util.StringUtils; -import javax.annotation.Resource; -import javax.servlet.*; -import javax.servlet.annotation.WebFilter; -import javax.servlet.http.Cookie; -import javax.servlet.http.HttpServletRequest; -import javax.servlet.http.HttpServletResponse; import java.io.IOException; import java.util.List; @@ -59,8 +62,8 @@ public class APIFilter implements Filter { try { url = tmp.split(tmp.split("/")[tmp.split("/").length - 1])[0]; } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - com.yutou.bilibili.Tools.Log.i("无权限请求:" + tmp); + Log.e(e); + Log.i("无权限请求:" + tmp); return; } JSONArray powers = JSONArray.parseArray(user.getPower()); diff --git a/src/main/java/com/yutou/bilibili/Tools/ApplicationClose.java b/src/main/java/com/yutou/bilibili/Tools/ApplicationClose.java index b487faa..be8228c 100644 --- a/src/main/java/com/yutou/bilibili/Tools/ApplicationClose.java +++ b/src/main/java/com/yutou/bilibili/Tools/ApplicationClose.java @@ -1,5 +1,6 @@ package com.yutou.bilibili.Tools; +import com.yutou.utils.Log; import org.springframework.context.ApplicationListener; import org.springframework.context.event.ContextClosedEvent; import org.springframework.stereotype.Component; diff --git a/src/main/java/com/yutou/bilibili/Tools/ApplicationInit.java b/src/main/java/com/yutou/bilibili/Tools/ApplicationInit.java index 61ec705..89e9291 100644 --- a/src/main/java/com/yutou/bilibili/Tools/ApplicationInit.java +++ b/src/main/java/com/yutou/bilibili/Tools/ApplicationInit.java @@ -2,18 +2,14 @@ package com.yutou.bilibili.Tools; import com.yutou.bilibili.BiliBili.Datas.AppData; import com.yutou.bilibili.BiliBili.Datas.LiveData; -import com.yutou.bilibili.BiliBili.Live; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.BiliBili.Services.IBiliBiliLiveService; import com.yutou.bilibili.Services.ISystemConfigService; import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveInfo; import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliUpInfo; +import jakarta.annotation.Resource; import org.springframework.boot.ApplicationArguments; import org.springframework.boot.ApplicationRunner; import org.springframework.stereotype.Component; -import org.springframework.util.StringUtils; -import javax.annotation.Resource; import java.io.File; import java.text.ParsePosition; import java.text.SimpleDateFormat; @@ -27,8 +23,6 @@ import java.util.TimerTask; */ @Component public class ApplicationInit implements ApplicationRunner { - @Resource - IBiliBiliLiveService service; @Resource ISystemConfigService configService; private Timer timer; @@ -36,7 +30,6 @@ public class ApplicationInit implements ApplicationRunner { @Override public void run(ApplicationArguments args) throws Exception { - LiveUtils.LiveGiftConfig.getInstance(); AppData.FFMPEG = configService.getConfig(Config.SYSTEM_VIDEO_FFMPEG); AppData.LIVE_SAVE_FFMPEG = configService.getConfig(Config.SYSTEM_VIDEO_SAVE_MODEL) != null && configService.getConfig(Config.SYSTEM_VIDEO_SAVE_MODEL).equals("ffmpeg"); startTimer(); @@ -59,9 +52,7 @@ public class ApplicationInit implements ApplicationRunner { oldTime = time; switch (time) { case "00:00": - for (Live live : Live.lives.values()) { - live.clearInfo(); - } + break; case "01:00": case "02:00": @@ -93,52 +84,13 @@ public class ApplicationInit implements ApplicationRunner { } private void checkLive() { - List list = service.getUpInfo(); - LiveUtils.reloadLiveState(list); - for (BilibiliUpInfo info : list) { - if (info.getOfflinelistening() == 1) { - if (info.getEnable() == 1 ) { - Live live = LiveUtils.liveContains(info); - if (live == null) { - live = new Live(); - live.add(info.getRoomid(), !StringUtils.isEmpty(AppTools.readFile(new File("cookies.json")))); - } - } - } - } + } }, 0, 2 * 60 * 1000); } } public void saveData(String time) { - Date date = new Date(); - String toDay = AppTools.getToDayTime() + " 00:00:00"; - String nowTime = AppTools.getToDayTime() + " " + time + ":00"; - for (BilibiliUpInfo info : service.getUpInfo()) { - Date startTime = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").parse(toDay, new ParsePosition(0)); - Date endTime = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").parse(nowTime, new ParsePosition(0)); - int userCount = service.queryLiveData(info.getRoomid(), startTime, endTime, new String[]{LiveData.INTERACT_WORD}).size(); - int vipCount = service.queryLiveData(info.getRoomid(), startTime, endTime, new String[]{LiveData.ENTRY_EFFECT}).size(); - int giftCount = service.queryLiveData(info.getRoomid(), startTime, endTime, new String[]{ - LiveData.SEND_GIFT, - LiveData.COMBO_SEND, - LiveData.GUARD_BUY, - LiveData.SUPER_CHAT_MESSAGE}).size(); - Live live = LiveUtils.liveContains(info); - int popularCount = 0; - if (live != null) { - popularCount = live.getInfo().getPopular(); - } - BilibiliLiveInfo liveInfo = new BilibiliLiveInfo(); - liveInfo.setRoomid(info.getRoomid()); - liveInfo.setGiftuser(giftCount); - liveInfo.setUserindex(userCount); - liveInfo.setVipuserindex(vipCount); - liveInfo.setPopular(popularCount); - liveInfo.setSubtime(date); - service.addLiveInfo(liveInfo); - } } } diff --git a/src/main/java/com/yutou/bilibili/Tools/ExcelUtils.java b/src/main/java/com/yutou/bilibili/Tools/ExcelUtils.java index cce6ae6..d8e764f 100644 --- a/src/main/java/com/yutou/bilibili/Tools/ExcelUtils.java +++ b/src/main/java/com/yutou/bilibili/Tools/ExcelUtils.java @@ -4,13 +4,15 @@ import com.yutou.bilibili.mybatis.Bili.mybatis.dao.BilibiliLiveDataDao; import com.yutou.bilibili.mybatis.Bili.mybatis.dao.BilibiliLiveInfoDao; import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveData; import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveInfo; +import com.yutou.utils.AppTools; +import com.yutou.utils.Log; +import jakarta.annotation.Resource; import org.apache.poi.ss.usermodel.*; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import org.springframework.beans.BeansException; import org.springframework.context.ApplicationContext; import org.springframework.context.ApplicationContextAware; -import javax.annotation.Resource; import java.io.File; import java.io.FileOutputStream; import java.util.Date; @@ -28,10 +30,10 @@ public class ExcelUtils implements ApplicationContextAware { public static File getInstance(int roomId, Date startTime, Date endTime,String fileName) { ExcelUtils utils = new ExcelUtils(); long timer=System.currentTimeMillis(); - com.yutou.bilibili.Tools.Log.i("开始注入bean"); + Log.i("开始注入bean"); utils.dataDao = getBean(BilibiliLiveDataDao.class); utils.infoDao = getBean(BilibiliLiveInfoDao.class); - com.yutou.bilibili.Tools.Log.i("注入完毕:"+(System.currentTimeMillis()-timer)); + Log.i("注入完毕:"+(System.currentTimeMillis()-timer)); return utils.initTable(roomId, startTime, endTime,fileName); } @@ -41,7 +43,7 @@ public class ExcelUtils implements ApplicationContextAware { private File initTable(int roomId, Date startTime, Date endTime,String fileName) { long timer=System.currentTimeMillis(); - com.yutou.bilibili.Tools.Log.i("进入统计:"+timer); + Log.i("进入统计:"+timer); Workbook workbook = new XSSFWorkbook(); Sheet liveData = workbook.createSheet("直播数据"); Sheet liveInfo = workbook.createSheet("小时统计"); @@ -66,7 +68,7 @@ public class ExcelUtils implements ApplicationContextAware { createCell(infoRow.createCell(5), "送礼人数"); createCell(infoRow.createCell(6), "记录时间"); - com.yutou.bilibili.Tools.Log.i("表头插入完毕:"+(System.currentTimeMillis()-timer)); + Log.i("表头插入完毕:"+(System.currentTimeMillis()-timer)); List dataList = dataDao.queryLiveData(roomId, startTime, endTime, new String[]{ INTERACT_WORD, ENTRY_EFFECT, @@ -104,7 +106,7 @@ public class ExcelUtils implements ApplicationContextAware { createCell(infoRow.createCell(5), info.getGiftuser()); createCell(infoRow.createCell(6), AppTools.getToDayTimeToString(info.getSubtime())); } - com.yutou.bilibili.Tools.Log.i("数据填充完毕:"+(System.currentTimeMillis()-timer)); + Log.i("数据填充完毕:"+(System.currentTimeMillis()-timer)); try { if(!new File("excel").exists()){ new File("excel").mkdirs(); @@ -113,11 +115,11 @@ public class ExcelUtils implements ApplicationContextAware { FileOutputStream fileOutputStream = new FileOutputStream(file); workbook.write(fileOutputStream); workbook.close(); - com.yutou.bilibili.Tools.Log.i("文件写入完毕:"+(System.currentTimeMillis()-timer)); + Log.i("文件写入完毕:"+(System.currentTimeMillis()-timer)); file.renameTo(new File("excel"+File.separator+fileName)); return file; } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } return null; diff --git a/src/main/java/com/yutou/bilibili/Tools/FFmpegUtils.java b/src/main/java/com/yutou/bilibili/Tools/FFmpegUtils.java index 6beb323..774118f 100644 --- a/src/main/java/com/yutou/bilibili/Tools/FFmpegUtils.java +++ b/src/main/java/com/yutou/bilibili/Tools/FFmpegUtils.java @@ -1,5 +1,8 @@ package com.yutou.bilibili.Tools; +import com.yutou.utils.AppTools; +import com.yutou.utils.Log; + import java.io.File; import java.io.InputStream; import java.util.ArrayList; @@ -38,8 +41,8 @@ public class FFmpegUtils { ffmpeg, file.getAbsolutePath(), out.getAbsolutePath() + File.separator + file.getName()); - com.yutou.bilibili.Tools.Log.i(exec); - Process process=AppTools.exec(exec); + Log.i(exec); + Process process= AppTools.exec(exec); InputStream inputStream = process.getErrorStream(); byte[] bytes = new byte[1024]; while (inputStream.read(bytes) > -1) { @@ -49,7 +52,7 @@ public class FFmpegUtils { file.delete(); new File(out.getAbsolutePath()+File.separator+file.getName()).delete(); } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } nowFFmpegList.remove(file); } @@ -61,13 +64,13 @@ public class FFmpegUtils { int i = -1; while (i != 1) { i = add("D:\\ffmpeg-4.3.1-2020-11-19-full_build\\bin\\ffmpeg.exe", file.getAbsoluteFile(), new File("ffmpeg_out")); - com.yutou.bilibili.Tools.Log.i(i); + Log.i(i); try { Thread.sleep(300); } catch (InterruptedException e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } } - com.yutou.bilibili.Tools.Log.i("转码完成"); + Log.i("转码完成"); } } diff --git a/src/main/java/com/yutou/bilibili/Tools/HttpTools.java b/src/main/java/com/yutou/bilibili/Tools/HttpTools.java index 49722c3..e341825 100644 --- a/src/main/java/com/yutou/bilibili/Tools/HttpTools.java +++ b/src/main/java/com/yutou/bilibili/Tools/HttpTools.java @@ -1,7 +1,8 @@ package com.yutou.bilibili.Tools; -import com.alibaba.fastjson.JSONObject; +import com.alibaba.fastjson2.JSONObject; import com.yutou.bilibili.interfaces.DownloadInterface; +import com.yutou.utils.Log; import org.springframework.util.StringUtils; import java.io.*; diff --git a/src/main/java/com/yutou/bilibili/Tools/ServiceTools.java b/src/main/java/com/yutou/bilibili/Tools/ServiceTools.java index ea2d9cf..88dee70 100644 --- a/src/main/java/com/yutou/bilibili/Tools/ServiceTools.java +++ b/src/main/java/com/yutou/bilibili/Tools/ServiceTools.java @@ -1,17 +1,18 @@ package com.yutou.bilibili.Tools; -import com.alibaba.fastjson.JSONArray; +import com.alibaba.fastjson2.JSONArray; import com.yutou.bilibili.Services.IUserService; import com.yutou.bilibili.mybatis.dao.PermissionDao; import com.yutou.bilibili.mybatis.model.Permission; import com.yutou.bilibili.mybatis.model.PermissionExample; import com.yutou.bilibili.mybatis.model.UUser; +import com.yutou.utils.Log; +import jakarta.annotation.Resource; +import jakarta.servlet.http.HttpServletRequest; import org.springframework.beans.BeansException; import org.springframework.context.ApplicationContext; import org.springframework.context.ApplicationContextAware; -import javax.annotation.Resource; -import javax.servlet.http.HttpServletRequest; import java.util.List; public class ServiceTools implements ApplicationContextAware { @@ -36,8 +37,8 @@ public class ServiceTools implements ApplicationContextAware { try { url = tmp.split(tmp.split("/")[tmp.split("/").length - 1])[0]; } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - com.yutou.bilibili.Tools.Log.i("无权限请求:" + tmp); + Log.e(e); + Log.i("无权限请求:" + tmp); return false; } JSONArray powers = JSONArray.parseArray(user.getPower()); diff --git a/src/main/java/com/yutou/bilibili/Tools/TestMain.java b/src/main/java/com/yutou/bilibili/Tools/TestMain.java index c4077a9..d332010 100644 --- a/src/main/java/com/yutou/bilibili/Tools/TestMain.java +++ b/src/main/java/com/yutou/bilibili/Tools/TestMain.java @@ -1,19 +1,10 @@ package com.yutou.bilibili.Tools; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.LiveUtils; +import com.alibaba.fastjson2.JSONObject; -import javax.xml.bind.DatatypeConverter; -import java.io.BufferedInputStream; -import java.io.ByteArrayOutputStream; -import java.io.IOException; import java.io.OutputStream; -import java.net.CookieHandler; -import java.net.CookieManager; -import java.net.HttpCookie; import java.net.HttpURLConnection; import java.nio.charset.StandardCharsets; -import java.util.List; public class TestMain { public static void main(String[] args) { @@ -21,43 +12,7 @@ public class TestMain { } public TestMain() { - try { - JSONObject cookie = JSONObject.parseObject(LiveUtils.getFile("cookies.json")); - JSONObject userHear = new JSONObject(); - userHear.put("csrf_token",cookie.getString("bili_jct")); - userHear.put("csrf",cookie.getString("bili_jct")); - userHear.put("visit_id",""); - HttpURLConnection connection=LiveUtils.getBiliHttpPost("https://api.live.bilibili.com/User/userOnlineHeart",LiveUtils.getCookie()); - connection.setRequestProperty("Content-Type", "application/x-www-form-urlencoded"); - OutputStream connectionOutputStream = null; - connectionOutputStream = connection.getOutputStream(); - connectionOutputStream.write(HttpTools.toUrlParams(userHear).getBytes(StandardCharsets.UTF_8)); - connectionOutputStream.flush(); - JSONObject hearBeat; - hearBeat=LiveUtils.http_post("https://api.live.bilibili.com/User/userOnlineHeart", HttpTools.toUrlParams(userHear)); - /* BufferedInputStream stream = new BufferedInputStream(connection.getInputStream()); - ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); - byte[] bytes = new byte[1024]; - int len = 0, size; - while ((len = stream.read(bytes)) != -1) { - outputStream.write(bytes, 0, len); - outputStream.flush(); - } - String str = new String(outputStream.toByteArray(), StandardCharsets.UTF_8);*/ - - com.yutou.bilibili.Tools.Log.i("["+AppTools.getToDayNowTimeToString()+"]"+hearBeat); - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - public static int byteArrayToInt(byte[] bytes) { - int value=0; - for(int i = 0; i < 4; i++) { - int shift= (3-i) * 8; - value +=(bytes[i] & 0xFF) << shift; - } - return value; } } diff --git a/src/main/java/com/yutou/bilibili/Tools/Tools.java b/src/main/java/com/yutou/bilibili/Tools/Tools.java index 0531d3d..910a312 100644 --- a/src/main/java/com/yutou/bilibili/Tools/Tools.java +++ b/src/main/java/com/yutou/bilibili/Tools/Tools.java @@ -1,23 +1,18 @@ package com.yutou.bilibili.Tools; -import com.alibaba.fastjson.JSONArray; import com.yutou.bilibili.interfaces.DownloadInterface; +import com.yutou.utils.Log; +import jakarta.servlet.http.HttpServletRequest; import org.springframework.core.io.FileSystemResource; import org.springframework.http.HttpHeaders; import org.springframework.http.MediaType; import org.springframework.http.ResponseEntity; -import org.springframework.util.StringUtils; -import org.springframework.web.multipart.MultipartFile; -import javax.servlet.http.Cookie; -import javax.servlet.http.HttpServletRequest; -import javax.servlet.http.HttpServletResponse; import java.io.*; import java.net.HttpURLConnection; import java.net.URL; import java.net.URLDecoder; import java.net.URLEncoder; -import java.nio.charset.StandardCharsets; import java.security.MessageDigest; import java.text.SimpleDateFormat; import java.util.Base64; @@ -111,7 +106,7 @@ public class Tools { inputStream.close(); downloadInterface.onDownload(file); } catch (IOException e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); downloadInterface.onError(e); } } @@ -153,7 +148,7 @@ public class Tools { } in.close(); } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); return null; } return bytesToHexString(digest.digest()); diff --git a/src/main/java/com/yutou/bilibili/mybatis/Bili/mybatis/model/BilibiliUpInfo.java b/src/main/java/com/yutou/bilibili/mybatis/Bili/mybatis/model/BilibiliUpInfo.java index f14d90b..c3d113e 100644 --- a/src/main/java/com/yutou/bilibili/mybatis/Bili/mybatis/model/BilibiliUpInfo.java +++ b/src/main/java/com/yutou/bilibili/mybatis/Bili/mybatis/model/BilibiliUpInfo.java @@ -4,7 +4,6 @@ import java.io.Serializable; import java.util.ArrayList; import java.util.List; -import com.yutou.bilibili.BiliBili.LiveUtils; import lombok.Data; /** @@ -50,7 +49,6 @@ public class BilibiliUpInfo implements Serializable { public Integer checkLiveStatus(){ List list=new ArrayList<>(); list.add(this); - live= LiveUtils.getStatusInfoByUid(list).getJSONObject("data").getJSONObject(mid+"").getInteger("live_status"); return live; } diff --git a/src/main/java/com/yutou/bilibili/sqlite/BiliBiliLiveDatabasesManager.java b/src/main/java/com/yutou/bilibili/sqlite/BiliBiliLiveDatabasesManager.java index a8dc1cf..608ff80 100644 --- a/src/main/java/com/yutou/bilibili/sqlite/BiliBiliLiveDatabasesManager.java +++ b/src/main/java/com/yutou/bilibili/sqlite/BiliBiliLiveDatabasesManager.java @@ -1,20 +1,14 @@ package com.yutou.bilibili.sqlite; -import com.yutou.bilibili.BiliBili.Datas.BiliBiliUpData; +import com.alibaba.fastjson2.JSONObject; import com.yutou.bilibili.BiliBili.Datas.DanmuData; -import com.yutou.bilibili.BiliBili.Datas.LiveData; -import com.yutou.bilibili.BiliBili.Datas.LiveInfo; -import com.alibaba.fastjson.JSONObject; -import com.yutou.bilibili.BiliBili.LiveUtils; -import com.yutou.bilibili.Tools.AppTools; -import com.yutou.bilibili.Tools.Log; -import com.yutou.bilibili.mybatis.Bili.mybatis.model.BilibiliLiveData; +import com.yutou.databases.AbsDatabasesBean; +import com.yutou.databases.SQLiteManager; +import com.yutou.utils.Log; import java.io.File; import java.sql.ResultSet; -import java.sql.SQLException; import java.sql.Statement; -import java.text.SimpleDateFormat; import java.util.*; public class BiliBiliLiveDatabasesManager extends SQLiteManager { @@ -24,6 +18,12 @@ public class BiliBiliLiveDatabasesManager extends SQLiteManager { super(); } + + @Override + protected AbsDatabasesBean getDataBean() { + return null; + } + private JSONObject getInitJSON(){ return JSONObject.parseObject("{\"file\":\"bilibili.db\",\"table\":[{\"name\":\"danmu\",\"item\":[{\"name\":\"id\",\"type\":\"int\",\"isNull\":false,\"isKey\":true},{\"name\":\"model\",\"type\":\"int\",\"isNull\":false,\"isKey\":false},{\"name\":\"fontSize\",\"type\":\"int\",\"isNull\":false,\"isKey\":false},{\"name\":\"fontColor\",\"type\":\"int\",\"isNull\":false,\"isKey\":false},{\"name\":\"time\",\"type\":\"int\",\"isNull\":false,\"isKey\":false},{\"name\":\"uCode\",\"type\":\"String\",\"isNull\":false,\"isKey\":false},{\"name\":\"danmu\",\"type\":\"String\",\"isNull\":false,\"isKey\":false},{\"name\":\"uid\",\"type\":\"int\",\"isNull\":false,\"isKey\":false},{\"name\":\"uname\",\"type\":\"String\",\"isNull\":false,\"isKey\":false}]}]}"); } @@ -31,13 +31,13 @@ public class BiliBiliLiveDatabasesManager extends SQLiteManager { JSONObject json = getInitJSON(); json.put("model","create"); json.put("file",fileName+".db"); - build(json); + //build(json); } public void init(File file){ JSONObject json = getInitJSON(); json.put("model","load"); json.put("file",file.getAbsolutePath()); - build(json); + // build(json); } /** @@ -61,7 +61,7 @@ public class BiliBiliLiveDatabasesManager extends SQLiteManager { statement.execute(sql); statement.closeOnCompletion(); }catch (Exception e){ - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } } @@ -93,7 +93,7 @@ public class BiliBiliLiveDatabasesManager extends SQLiteManager { set.close(); statement.closeOnCompletion();; }catch (Exception e){ - com.yutou.bilibili.Tools.Log.e(e); + Log.e(e); } return list; } diff --git a/src/main/java/com/yutou/bilibili/sqlite/SQLiteManager.java b/src/main/java/com/yutou/bilibili/sqlite/SQLiteManager.java deleted file mode 100644 index 67799c3..0000000 --- a/src/main/java/com/yutou/bilibili/sqlite/SQLiteManager.java +++ /dev/null @@ -1,150 +0,0 @@ -package com.yutou.bilibili.sqlite; - -import com.alibaba.fastjson.JSONArray; -import com.alibaba.fastjson.JSONObject; - -import java.io.File; -import java.sql.Connection; -import java.sql.DriverManager; -import java.sql.SQLException; -import java.sql.Statement; - -public class SQLiteManager { - protected Connection conn; - private String url = "jdbc:sqlite:"; - private File sql; - - - public void startBatch() { - try { - conn.setAutoCommit(false); - } catch (SQLException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - - public void closeBatch() { - try { - conn.setAutoCommit(true); - } catch (SQLException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - - public void close() { - try { - conn.close(); - } catch (SQLException throwables) { - throwables.printStackTrace(); - } - } - - public void commit() { - try { - conn.commit(); - } catch (SQLException e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - - protected SQLiteManager() { - - } - - private void createSql(JSONObject json) { - try { - sql.mkdirs(); - sql.delete(); - conn = DriverManager.getConnection(url + sql.getAbsolutePath()); - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - startBatch(); - JSONArray array = json.getJSONArray("table"); - for (Object o : array) { - com.yutou.bilibili.Tools.Log.i("创建表:" + ((JSONObject) o).getString("name")); - createSqlOfTable((JSONObject) o); - } - closeBatch(); - } - - private void createSqlOfTable(JSONObject table) { - String tableName = table.getString("name"); - try { - Statement statement = conn.createStatement(); - JSONArray items = table.getJSONArray("item"); - StringBuilder sql = new StringBuilder(); - sql.append("CREATE TABLE `") - .append(tableName) - .append("` ("); - for (Object item : items) { - StringBuilder builder = new StringBuilder(); - JSONObject it = (JSONObject) item; - String type; - switch (it.getString("type")) { - case "int": - type = " INTEGER "; - break; - case "TIME": - type = " NUMERIC "; - break; - default: - type = " TEXT "; - break; - } - builder.append("`") - .append(it.getString("name")) - .append("`") - .append(type) - .append(it.getBoolean("isNull") ? "" : " NOT NULL ") - .append(it.getBoolean("isKey") ? " PRIMARY KEY AUTOINCREMENT " : "") - .append(","); - sql.append(builder.toString()); - } - sql.append(");"); - statement.execute(sql.toString().replace(",);", ");")); - statement.closeOnCompletion(); - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - - protected void build(JSONObject json) { - try { - Class.forName("org.sqlite.JDBC"); - if ("load".equals(json.getString("model"))) { - sql = new File(json.getString("file")); - } else { - sql = new File("databases" + File.separator + json.getString("file")); - } - if (!sql.exists()) { - createSql(json); - } else { - conn = DriverManager.getConnection(url + sql.getAbsolutePath()); - } - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - } - - public boolean setDB(String fileName) { - try { - Class.forName("org.sqlite.JDBC"); - sql = new File("db" + File.separator + fileName); - if (sql.exists()) { - if (conn != null && !conn.isClosed()) { - conn.close(); - } - conn = DriverManager.getConnection(url + sql.getAbsolutePath()); - return true; - } - } catch (Exception e) { - com.yutou.bilibili.Tools.Log.e(e); - } - return false; - } - - public static void main(String[] args) { - - } -} diff --git a/src/main/java/com/yutou/bilibili/sqlite/YouTubeLiveDatabasesManager.java b/src/main/java/com/yutou/bilibili/sqlite/YouTubeLiveDatabasesManager.java deleted file mode 100644 index e302734..0000000 --- a/src/main/java/com/yutou/bilibili/sqlite/YouTubeLiveDatabasesManager.java +++ /dev/null @@ -1,15 +0,0 @@ -package com.yutou.bilibili.sqlite; - -public class YouTubeLiveDatabasesManager extends SQLiteManager{ - private static YouTubeLiveDatabasesManager manager; - public static YouTubeLiveDatabasesManager getInstance(){ - if(manager==null){ - manager=new YouTubeLiveDatabasesManager(); - } - return manager; - } - public YouTubeLiveDatabasesManager() { - super(); - - } -}