Events handling

This commit is contained in:
JW
2023-08-16 17:36:40 +02:00
parent 59befcd6f9
commit 4b9e531c95
121 changed files with 4368 additions and 613 deletions

View File

@@ -0,0 +1,99 @@
package io.github.jwdeveloper.tiktok.intefacee;
import com.squareup.javapoet.*;
import io.github.jwdeveloper.tiktok.TikTokLiveClientBuilder;
import io.github.jwdeveloper.tiktok.events.TikTokEvent;
import io.github.jwdeveloper.tiktok.events_generator.EventGeneratorSettings;
import org.reflections.Reflections;
import javax.lang.model.element.Modifier;
import java.util.Set;
public class EventsInterfaceGenerator {
public void compile(EventGeneratorSettings settings) {
Reflections reflections = new Reflections("io.github.jwdeveloper.tiktok.events.messages");
// Get all types (i.e., classes) in the specified package
var classes = reflections.getSubTypesOf(TikTokEvent.class);
classes.add(TikTokEvent.class);
for (Class<?> clazz : classes) {
// System.out.println(clazz.getName());
}
// var result = generateInterface("io.github.jwdeveloper.tiktok.events", classes);System.out.println(result);
var result = getBuilderImplementation("x",classes); System.out.println(result);
}
public String generateInterface(String packageName, Set<Class<? extends TikTokEvent>> eventsClasses) {
TypeSpec.Builder classBuilder = TypeSpec.interfaceBuilder("TikTokEvents");
classBuilder.addModifiers(Modifier.PUBLIC);
classBuilder.addTypeVariable(TypeVariableName.get("T"));
// Generate constructors
for (var clazz : eventsClasses) {
var clazzName = clazz.getSimpleName();
var methodName = clazzName.replace("TikTok", "");
methodName = methodName.replace("Event", "");
MethodSpec.Builder constructorBuilder = MethodSpec.methodBuilder("on" + methodName);
var name = "Consumer<" + clazzName + ">";
constructorBuilder.addModifiers(Modifier.ABSTRACT, Modifier.PUBLIC);
constructorBuilder.addParameter(ClassName.bestGuess(name), "event");
constructorBuilder.returns(TypeVariableName.get("T"));
classBuilder.addMethod(constructorBuilder.build());
}
// Generate Java class
TypeSpec javaClass = classBuilder.build();
var result = JavaFile.builder(packageName, javaClass).build();
return result.toString();
}
public String getBuilderImplementation(String packageName, Set<Class<? extends TikTokEvent>> eventsClasses) {
TypeSpec.Builder classBuilder = TypeSpec.classBuilder("TikTokEvents");
classBuilder.addModifiers(Modifier.PUBLIC);
/*
public TikTokClientBuilder onLinkMicFanTicket(Consumer<TikTokLinkMicFanTicketEvent> event) {
tikTokEventHandler.subscribe(TikTokEventHandler.class, event);
return this;
}
*/
// Generate constructors
for (var clazz : eventsClasses) {
var clazzName = clazz.getSimpleName();
var methodName = clazzName.replace("TikTok", "");
methodName ="on" + methodName.replace("Event", "");
MethodSpec.Builder constructorBuilder = MethodSpec.methodBuilder( methodName);
var name = "Consumer<" + clazzName + ">";
constructorBuilder.addModifiers( Modifier.PUBLIC);
constructorBuilder.addParameter(ClassName.bestGuess(name), "event");
constructorBuilder.addStatement("tikTokEventHandler.subscribe("+clazzName+".class,event)");
constructorBuilder.addStatement("return this");
constructorBuilder.returns(TikTokLiveClientBuilder.class);
classBuilder.addMethod(constructorBuilder.build());
}
// Generate Java class
TypeSpec javaClass = classBuilder.build();
var result = JavaFile.builder(packageName, javaClass).build();
return result.toString();
}
}