FilesExpand file tree

main

/

main.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

79 lines (60 loc) · 1.56 KB

main

/

main.go

Copy path

Top

File metadata and controls

• Code
• Blame

79 lines (60 loc) · 1.56 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

package main

import (

"embed"

"log"

"net/http"

"os"

"os/signal"

"syscall"

"gorm.io/driver/postgres"

"gorm.io/gorm"

"github.com/semanser/ai-coder/assets"

"github.com/semanser/ai-coder/executor"

"github.com/semanser/ai-coder/models"

"github.com/semanser/ai-coder/router"

"github.com/semanser/ai-coder/services"

)

const defaultPort = "8080"

//go:embed templates/prompts/*.tmpl

var promptTemplates embed.FS

func main() {

sigChan := make(chan os.Signal, 1)

signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)

dsn := os.Getenv("DATABASE_URL")

if dsn == "" {

log.Fatal("failed to read DB env variable")

}

db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})

if err != nil {

log.Fatalf("failed to connect database: %v", err)

}

// Migrate the schema

db.AutoMigrate(&models.Flow{}, &models.Task{})

port := os.Getenv("PORT")

if port == "" {

port = defaultPort

}

r := router.New(db)

assets.Init(promptTemplates)

services.Init()

err = executor.InitDockerClient()

if err != nil {

log.Fatalf("failed to initialize Docker client: %v", err)

}

executor.ProcessQueue(db)

// Run the server in a separate goroutine

go func() {

log.Printf("connect to http://localhost:%s/playground for GraphQL playground", port)

if err := http.ListenAndServe(":"+port, r); err != nil {

log.Fatalf("HTTP server error: %v", err)

}

}()

// Wait for termination signal

<-sigChan

log.Println("Shutting down...")

// Cleanup resources

if err := executor.Cleanup(); err != nil {

log.Printf("Error during cleanup: %v", err)

}

log.Println("Shutdown complete")

}