|
@@ -5,6 +5,7 @@ import (
|
|
"log"
|
|
"log"
|
|
"net/http"
|
|
"net/http"
|
|
"os"
|
|
"os"
|
|
|
|
+ "path/filepath"
|
|
"time"
|
|
"time"
|
|
|
|
|
|
"github.com/gorilla/handlers"
|
|
"github.com/gorilla/handlers"
|
|
@@ -66,6 +67,27 @@ func main() {
|
|
generator.Generators = make(map[string]generator.Generator)
|
|
generator.Generators = make(map[string]generator.Generator)
|
|
generator.Generators["list"] = list.NewListGenerator(config.Config)
|
|
generator.Generators["list"] = list.NewListGenerator(config.Config)
|
|
|
|
|
|
|
|
+ files, err := filepath.Glob("generator/generators/*")
|
|
|
|
+ log.Println("Found the following generators...")
|
|
|
|
+ gTypes := make([]*orm.GeneratorType, 0)
|
|
|
|
+ for _, g := range files {
|
|
|
|
+ gTypes = append(gTypes, &orm.GeneratorType{Name: filepath.Base(g)})
|
|
|
|
+ }
|
|
|
|
+ log.Println("Create generator_type table...")
|
|
|
|
+ for _, gt := range gTypes {
|
|
|
|
+ var genType orm.GeneratorType
|
|
|
|
+ if err := orm.DB().Where("name = ?", gt.Name).First(&genType).Error; err != nil {
|
|
|
|
+ panic(err)
|
|
|
|
+ }
|
|
|
|
+ log.Println(genType)
|
|
|
|
+ if genType.Name == "" {
|
|
|
|
+ err := orm.DB().Debug().Create(gt).Error
|
|
|
|
+ if err != nil {
|
|
|
|
+ panic(err)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
log.Println("Starting cron jobs...")
|
|
log.Println("Starting cron jobs...")
|
|
|
|
|
|
c := cron.New()
|
|
c := cron.New()
|