Add some basic auth

This commit is contained in:
Gabriel Adrian Samfira 2022-04-28 16:13:20 +00:00
parent 66b46ae0ab
commit 0883fcd5cd
24 changed files with 1687 additions and 674 deletions

View file

@ -1,107 +1,214 @@
package main
import (
"context"
"flag"
"fmt"
"log"
"os/signal"
"runner-manager/config"
"runner-manager/database/sql"
"runner-manager/params"
"runner-manager/util"
)
// import (
// "context"
// "flag"
// "fmt"
// "log"
// "os/signal"
// "runner-manager/config"
// "runner-manager/database/sql"
// "runner-manager/params"
// "runner-manager/util"
// )
var (
conf = flag.String("config", config.DefaultConfigFilePath, "runner-manager config file")
version = flag.Bool("version", false, "prints version")
)
// var (
// conf = flag.String("config", config.DefaultConfigFilePath, "runner-manager config file")
// version = flag.Bool("version", false, "prints version")
// )
var Version string
// var Version string
func main() {
flag.Parse()
if *version {
fmt.Println(Version)
return
}
ctx, stop := signal.NotifyContext(context.Background(), signals...)
defer stop()
fmt.Println(ctx)
// func main() {
// flag.Parse()
// if *version {
// fmt.Println(Version)
// return
// }
// ctx, stop := signal.NotifyContext(context.Background(), signals...)
// defer stop()
// fmt.Println(ctx)
cfg, err := config.NewConfig(*conf)
if err != nil {
log.Fatalf("Fetching config: %+v", err)
}
// cfg, err := config.NewConfig(*conf)
// if err != nil {
// log.Fatalf("Fetching config: %+v", err)
// }
db, err := sql.NewSQLDatabase(ctx, cfg.Database)
if err != nil {
log.Fatal(err)
}
// db, err := sql.NewSQLDatabase(ctx, cfg.Database)
// if err != nil {
// log.Fatal(err)
// }
fmt.Println(db)
// fmt.Println(db)
txt := "ana are mere prune și alune"
// txt := "ana are mere prune și alune"
enc, err := util.Aes256EncodeString(txt, "pamkotepAyksemfeghoibidEwCivbaut")
if err != nil {
log.Fatal(err)
}
// enc, err := util.Aes256EncodeString(txt, "pamkotepAyksemfeghoibidEwCivbaut")
// if err != nil {
// log.Fatal(err)
// }
fmt.Printf("encrypted: %d\n", len(enc))
// fmt.Printf("encrypted: %d\n", len(enc))
dec, err := util.Aes256DecodeString(enc, "pamkotepAyksemfeghoibidEwCivbaut")
if err != nil {
log.Fatal(err)
}
// dec, err := util.Aes256DecodeString(enc, "pamkotepAyksemfeghoibidEwCivbaut")
// if err != nil {
// log.Fatal(err)
// }
fmt.Println(dec)
// fmt.Println(dec)
repo, err := db.CreateRepository(ctx, "gabriel-samfira", "scripts", "")
if err != nil {
log.Fatal(err)
}
// repo, err := db.CreateRepository(ctx, "gabriel-samfira", "", "scripts", "")
// if err != nil {
// log.Fatal(err)
// }
pool, err := db.CreateRepositoryPool(ctx, repo.ID, params.CreatePoolParams{
ProviderName: "lxd_local",
MaxRunners: 10,
MinIdleRunners: 1,
Image: "ubuntu:20.04",
Flavor: "default",
Tags: []string{
"myrunner",
"superAwesome",
},
OSType: config.Linux,
OSArch: config.Amd64,
})
if err != nil {
log.Fatal(err)
}
fmt.Println(pool)
// pool, err := db.CreateRepositoryPool(ctx, repo.ID, params.CreatePoolParams{
// ProviderName: "lxd_local",
// MaxRunners: 10,
// MinIdleRunners: 1,
// Image: "ubuntu:20.04",
// Flavor: "default",
// Tags: []string{
// "myrunner",
// "superAwesome",
// },
// OSType: config.Linux,
// OSArch: config.Amd64,
// })
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(pool)
pool2, err := db.CreateRepositoryPool(ctx, repo.ID, params.CreatePoolParams{
ProviderName: "lxd_local2",
MaxRunners: 10,
MinIdleRunners: 1,
Image: "ubuntu:20.04",
Flavor: "default",
Tags: []string{
"myrunner",
"superAwesome2",
},
OSType: config.Linux,
OSArch: config.Amd64,
})
if err != nil {
log.Fatal(err)
}
fmt.Println(pool2)
// pool2, err := db.CreateRepositoryPool(ctx, repo.ID, params.CreatePoolParams{
// ProviderName: "lxd_local2",package main
pool3, err := db.FindRepositoryPoolByTags(ctx, repo.ID, []string{"myrunner", "superAwesome2"})
if err != nil {
log.Fatal(err)
}
// import (
// "context"
// "flag"
// "fmt"
// "log"
// "os/signal"
// "runner-manager/config"
// "runner-manager/database/sql"
// "runner-manager/params"
// "runner-manager/util"
// )
fmt.Println(pool3)
}
// var (
// conf = flag.String("config", config.DefaultConfigFilePath, "runner-manager config file")
// version = flag.Bool("version", false, "prints version")
// )
// var Version string
// func main() {
// flag.Parse()
// if *version {
// fmt.Println(Version)
// return
// }
// ctx, stop := signal.NotifyContext(context.Background(), signals...)
// defer stop()
// fmt.Println(ctx)
// cfg, err := config.NewConfig(*conf)
// if err != nil {
// log.Fatalf("Fetching config: %+v", err)
// }
// db, err := sql.NewSQLDatabase(ctx, cfg.Database)
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(db)
// txt := "ana are mere prune și alune"
// enc, err := util.Aes256EncodeString(txt, "pamkotepAyksemfeghoibidEwCivbaut")
// if err != nil {
// log.Fatal(err)
// }
// fmt.Printf("encrypted: %d\n", len(enc))
// dec, err := util.Aes256DecodeString(enc, "pamkotepAyksemfeghoibidEwCivbaut")
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(dec)
// repo, err := db.CreateRepository(ctx, "gabriel-samfira", "", "scripts", "")
// if err != nil {
// log.Fatal(err)
// }
// pool, err := db.CreateRepositoryPool(ctx, repo.ID, params.CreatePoolParams{
// ProviderName: "lxd_local",
// MaxRunners: 10,
// MinIdleRunners: 1,
// Image: "ubuntu:20.04",
// Flavor: "default",
// Tags: []string{
// "myrunner",
// "superAwesome",
// },
// OSType: config.Linux,
// OSArch: config.Amd64,
// })
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(pool)
// pool2, err := db.CreateRepositoryPool(ctx, repo.ID, params.CreatePoolParams{
// ProviderName: "lxd_local2",
// MaxRunners: 10,
// MinIdleRunners: 1,
// Image: "ubuntu:20.04",
// Flavor: "default",
// Tags: []string{
// "myrunner",
// "superAwesome2",
// },
// OSType: config.Linux,
// OSArch: config.Amd64,
// })
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(pool2)
// pool3, err := db.FindRepositoryPoolByTags(ctx, repo.ID, []string{"myrunner", "superAwesome2"})
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(pool3)
// }
// MaxRunners: 10,
// MinIdleRunners: 1,
// Image: "ubuntu:20.04",
// Flavor: "default",
// Tags: []string{
// "myrunner",
// "superAwesome2",
// },
// OSType: config.Linux,
// OSArch: config.Amd64,
// })
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(pool2)
// pool3, err := db.FindRepositoryPoolByTags(ctx, repo.ID, []string{"myrunner", "superAwesome2"})
// if err != nil {
// log.Fatal(err)
// }
// fmt.Println(pool3)
// }

View file

@ -1,160 +1,106 @@
package main
// import (
// "context"
// "flag"
// "fmt"
// "log"
// "net"
// "net/http"
// "os/signal"
import (
"context"
"flag"
"fmt"
"log"
"net"
"net/http"
"os/signal"
// "runner-manager/apiserver/controllers"
// "runner-manager/apiserver/routers"
// "runner-manager/config"
// "runner-manager/util"
// // "github.com/google/go-github/v43/github"
// // "golang.org/x/oauth2"
// // "gopkg.in/yaml.v3"
// )
"runner-manager/apiserver/controllers"
"runner-manager/apiserver/routers"
"runner-manager/auth"
"runner-manager/config"
"runner-manager/database"
"runner-manager/runner"
"runner-manager/util"
// "github.com/google/go-github/v43/github"
// "golang.org/x/oauth2"
// "gopkg.in/yaml.v3"
)
// var (
// conf = flag.String("config", config.DefaultConfigFilePath, "runner-manager config file")
// version = flag.Bool("version", false, "prints version")
// )
var (
conf = flag.String("config", config.DefaultConfigFilePath, "runner-manager config file")
version = flag.Bool("version", false, "prints version")
)
// var Version string
var Version string
// // var token = "super secret token"
// var token = "super secret token"
// func main() {
// flag.Parse()
// if *version {
// fmt.Println(Version)
// return
// }
// ctx, stop := signal.NotifyContext(context.Background(), signals...)
// defer stop()
// fmt.Println(ctx)
func main() {
flag.Parse()
if *version {
fmt.Println(Version)
return
}
ctx, stop := signal.NotifyContext(context.Background(), signals...)
defer stop()
fmt.Println(ctx)
// cfg, err := config.NewConfig(*conf)
// if err != nil {
// log.Fatalf("Fetching config: %+v", err)
// }
cfg, err := config.NewConfig(*conf)
if err != nil {
log.Fatalf("Fetching config: %+v", err)
}
// // ts := oauth2.StaticTokenSource(
// // &oauth2.Token{AccessToken: cfg.Github.OAuth2Token},
// // )
logWriter, err := util.GetLoggingWriter(cfg)
if err != nil {
log.Fatalf("fetching log writer: %+v", err)
}
log.SetOutput(logWriter)
// // tc := oauth2.NewClient(ctx, ts)
runner, err := runner.NewRunner(ctx, *cfg)
if err != nil {
log.Fatalf("failed to create controller: %+v", err)
}
// // ghClient := github.NewClient(tc)
db, err := database.NewDatabase(ctx, cfg.Database)
if err != nil {
log.Fatal(err)
}
// // // list all repositories for the authenticated user
// // repos, _, err := client.Repositories.List(ctx, "", nil)
authenticator := auth.NewAuthenticator(cfg.JWTAuth, db)
controller, err := controllers.NewAPIController(runner, authenticator)
if err != nil {
log.Fatalf("failed to create controller: %+v", err)
}
// // fmt.Println(repos, err)
jwtMiddleware, err := auth.NewjwtMiddleware(db, cfg.JWTAuth)
if err != nil {
log.Fatal(err)
}
// logWriter, err := util.GetLoggingWriter(cfg)
// if err != nil {
// log.Fatalf("fetching log writer: %+v", err)
// }
// log.SetOutput(logWriter)
initMiddleware, err := auth.NewInitRequiredMiddleware(db)
if err != nil {
log.Fatal(err)
}
// controller, err := controllers.NewAPIController()
// if err != nil {
// log.Fatalf("failed to create controller: %+v", err)
// }
router := routers.NewAPIRouter(controller, logWriter, jwtMiddleware, initMiddleware)
// router := routers.NewAPIRouter(controller, logWriter)
tlsCfg, err := cfg.APIServer.APITLSConfig()
if err != nil {
log.Fatalf("failed to get TLS config: %q", err)
}
// tlsCfg, err := cfg.APIServer.APITLSConfig()
// if err != nil {
// log.Fatalf("failed to get TLS config: %q", err)
// }
srv := &http.Server{
Addr: cfg.APIServer.BindAddress(),
TLSConfig: tlsCfg,
// Pass our instance of gorilla/mux in.
Handler: router,
}
// srv := &http.Server{
// Addr: cfg.APIServer.BindAddress(),
// TLSConfig: tlsCfg,
// // Pass our instance of gorilla/mux in.
// Handler: router,
// }
listener, err := net.Listen("tcp", srv.Addr)
if err != nil {
log.Fatalf("creating listener: %q", err)
}
// listener, err := net.Listen("tcp", srv.Addr)
// if err != nil {
// log.Fatalf("creating listener: %q", err)
// }
go func() {
if err := srv.Serve(listener); err != nil {
log.Fatalf("Listening: %+v", err)
}
}()
// go func() {
// if err := srv.Serve(listener); err != nil {
// log.Fatalf("Listening: %+v", err)
// }
// }()
// <-ctx.Done()
// // runner, err := runner.NewRunner(ctx, *cfg)
// // if err != nil {
// // log.Fatal(err)
// // }
// // fmt.Println(runner)
// // controllerID := "026d374d-6a8a-4241-8ed9-a246fff6762f"
// // provider, err := lxd.NewProvider(ctx, &cfg.Providers[0], controllerID)
// // if err != nil {
// // log.Fatal(err)
// // }
// // if err := provider.RemoveAllInstances(ctx); err != nil {
// // log.Fatal(err)
// // }
// // fmt.Println(provider)
// // if err := provider.DeleteInstance(ctx, "runner-manager-2fbe5354-be28-4e00-95a8-11479912368d"); err != nil {
// // log.Fatal(err)
// // }
// // instances, err := provider.ListInstances(ctx)
// // asJs, err := json.MarshalIndent(instances, "", " ")
// // fmt.Println(string(asJs), err)
// // log.Print("Fetching tools")
// // tools, _, err := ghClient.Actions.ListRunnerApplicationDownloads(ctx, cfg.Repositories[0].Owner, cfg.Repositories[0].Name)
// // if err != nil {
// // log.Fatal(err)
// // }
// // toolsAsYaml, err := yaml.Marshal(tools)
// // if err != nil {
// // log.Fatal(err)
// // }
// // log.Printf("got tools:\n%s\n", string(toolsAsYaml))
// // log.Print("fetching runner token")
// // ghRunnerToken, _, err := ghClient.Actions.CreateRegistrationToken(ctx, cfg.Repositories[0].Owner, cfg.Repositories[0].Name)
// // if err != nil {
// // log.Fatal(err)
// // }
// // log.Printf("got token %v", ghRunnerToken)
// // bootstrapArgs := params.BootstrapInstance{
// // Tools: tools,
// // RepoURL: cfg.Repositories[0].String(),
// // GithubRunnerAccessToken: *ghRunnerToken.Token,
// // RunnerType: cfg.Repositories[0].Pool.Runners[0].Name,
// // CallbackURL: "",
// // InstanceToken: "",
// // SSHKeys: []string{
// // "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAACAQC2oT7j/+elHY9U2ibgk2RYJgCvqIwewYKJTtHslTQFDWlHLeDam93BBOFlQJm9/wKX/qjC8d26qyzjeeeVf2EEAztp+jQfEq9OU+EtgQUi589jxtVmaWuYED8KVNbzLuP79SrBtEZD4xqgmnNotPhRshh3L6eYj4XzLWDUuOD6kzNdsJA2QOKeMOIFpBN6urKJHRHYD+oUPUX1w5QMv1W1Srlffl4m5uE+0eJYAMr02980PG4+jS4bzM170wYdWwUI0pSZsEDC8Fn7jef6QARU2CgHJYlaTem+KWSXislOUTaCpR0uhakP1ezebW20yuuc3bdRNgSlZi9B7zAPALGZpOshVqwF+KmLDi6XiFwG+NnwAFa6zaQfhOxhw/rF5Jk/wVjHIHkNNvYewycZPbKui0E3QrdVtR908N3VsPtLhMQ59BEMl3xlURSi0fiOU3UjnwmOkOoFDy/WT8qk//gFD93tUxlf4eKXDgNfME3zNz8nVi2uCPvG5NT/P/VWR8NMqW6tZcmWyswM/GgL6Y84JQ3ESZq/7WvAetdc1gVIDQJ2ejYbSHBcQpWvkocsiuMTCwiEvQ0sr+UE5jmecQvLPUyXOhuMhw43CwxnLk1ZSeYeCorxbskyqIXH71o8zhbPoPiEbwgB+i9WEoq02u7c8CmCmO8Y9aOnh8MzTKxIgQ==",
// // },
// // }
// // instance, err := provider.CreateInstance(ctx, bootstrapArgs)
// // if err != nil {
// // log.Fatal(err)
// // }
// // fmt.Println(instance)
// }
<-ctx.Done()
}