update with new library

build.gradle

@@ -6,7 +6,7 @@ plugins {
 }
 
 group = 'io.nexilis'
-version = '3.3.4'
+version = '3.3.7'
 
 compileJava {
     sourceCompatibility = '1.8'

src/main/java/io/nexilis/transformer/InjectProtectionTask.java

@@ -36,7 +36,7 @@ public class InjectProtectionTask extends DefaultTask {
         Path inputPath = inputDir.getAsFile().get().toPath();
         String fileName = inputPath.getFileName().toString();
         try (Stream<Path> stream = Files.walk(getProject().getLayout().getBuildDirectory().get().getAsFile().toPath())) {
-            stream.filter(path -> path.toString().endsWith(fileName.replace(".java",".class")))
+            stream.filter(path -> path.toString().endsWith(fileName.replace(fileName.split("\\.")[1],".class")))
                     .forEach(classFile -> {
                         try {
                             injectProtectionIntoClass(classFile);

src/main/java/io/nexilis/transformer/ProtectionInjector.java

@@ -59,7 +59,7 @@ public class ProtectionInjector extends ClassVisitor {
             mv.visitVarInsn(ALOAD, 0);
             mv.visitLdcInsn(project != null ? (project.getProperties().get("nexilis.floatingMode") == null ? 0 : project.getProperties().get("nexilis.floatingMode")) : 0);
             mv.visitVarInsn(ALOAD, 0);
-            mv.visitMethodInsn(INVOKESTATIC, "io/nexilis/ui/API", "connect", "(Ljava/lang/String;Ljava/lang/String;Landroid/app/Activity;Ljava/lang/Integer;Lio/nexilis/ui/Callback;)V", false);
+            mv.visitMethodInsn(INVOKESTATIC, "io/nexilis/ui/API", "connect", "(Ljava/lang/String;Ljava/lang/String;Landroid/app/Activity;ILio/nexilis/ui/Callback;)V", false);
         }
     }
 

src/main/java/io/nexilis/util/Util.java

@@ -19,7 +19,7 @@ public class Util {
             Files.walkFileTree(Paths.get(dir), new SimpleFileVisitor<Path>() {
                 @Override
                 public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {
-                    if (file.getFileName().toString().equals(mainActivity + ".java")) {
+                    if (file.getFileName().toString().startsWith(mainActivity)) {
                         mainPath[0] = file;
                         return FileVisitResult.TERMINATE;
                     }