fix mainclass issue

Author: chandu0101

build.sbt

@@ -1,6 +1,6 @@
 name := "sbt-sri-platform"
 
-//version := "2017.3.23-beta"
+//version := "2017.3.26.1-beta"
 
 scalaVersion := "2.10.6"
 

src/main/scala/sri/sbt/platform/ConfigBuilder.scala

@@ -1,9 +1,11 @@
 package sri.sbt.platform
 
+import org.scalajs.core.tools.linker.backend.ModuleKind
 import sbt.Keys._
 import sbt._
 import org.scalajs.sbtplugin.ScalaJSPlugin.AutoImport._
 import org.scalajs.sbtplugin.ScalaJSPluginInternal
+import org.scalajs.sbtplugin.ScalaJSPluginInternal.scalaJSProjectBaseSettings
 
 object ConfigBuilder {
 
@@ -70,8 +72,15 @@ object ConfigBuilder {
         dev := {
           val indexFile = baseDirectory.value / entryFile
           val indexContent = IO.read(indexFile)
-          (fastOptJS in config).value.data
-          val launcher = s"""require("./$aPath");"""
+          println(s"value : ${scalaJSUseMainModuleInitializer.value}")
+          (fastOptJS).value.data
+          val mainClassRef = (mainClass in Compile).value
+            .getOrElse("")
+            .split('.')
+            .map(s => s"""["$s"]""")
+            .mkString("")
+            .concat("().main()")
+          val launcher = s"""require("./$aPath")${mainClassRef};"""
           if (!indexContent.contains(launcher)) IO.append(indexFile, launcher)
 //          if (!isServerStarted) {
 //            if ((npmStart !) == 0) {
@@ -84,7 +93,13 @@ object ConfigBuilder {
           val indexFile = baseDirectory.value / entryFile
           val indexContent = IO.read(indexFile)
           (fullOptJS in config).value.data
-          val launcher = s"""require("./$aPath");"""
+          val mainClassRef = (mainClass in Compile).value
+            .getOrElse("")
+            .split('.')
+            .map(s => s"""["$s"]""")
+            .mkString("")
+            .concat("().main()")
+          val launcher = s"""require("./$aPath")${mainClassRef};"""
           if (!indexContent.contains(launcher)) IO.append(indexFile, launcher)
         }
       ))

src/main/scala/sri/sbt/platform/SriPlatFormPlugin.scala

@@ -25,11 +25,11 @@ object SriPlatFormPlugin extends AutoPlugin {
     Seq(
       defaultConfiguration := Some(common),
       scalaJSModuleKind := ModuleKind.CommonJSModule,
-      scalaJSUseMainModuleInitializer := true,
       ivyConfigurations := overrideConfigs(ios,
                                            android,
                                            common,
                                            CustomCompile)(
-        ivyConfigurations.value)
+        ivyConfigurations.value),
+      scalaJSUseMainModuleInitializer := true
     ) ++ buildConfig(android) ++ buildConfig(ios)
 }