mirror of
https://github.com/0rangebananaspy/authelia.git
synced 2024-09-14 22:47:21 +07:00
01b77384f9
Allow users to select and save the preferred duo device and method, depending on availability in the duo account. A default enrollment URL is provided and adjusted if returned by the duo API. This allows auto-enrollment if enabled by the administrator. Closes #594. Closes #1039.
106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package suites
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/go-rod/rod"
|
|
)
|
|
|
|
// GetLoginBaseURL returns the URL of the login portal and the path prefix if specified.
|
|
func GetLoginBaseURL() string {
|
|
if PathPrefix != "" {
|
|
return LoginBaseURL + PathPrefix
|
|
}
|
|
|
|
return LoginBaseURL
|
|
}
|
|
|
|
func (rs *RodSession) collectCoverage(page *rod.Page) {
|
|
coverageDir := "../../web/.nyc_output"
|
|
now := time.Now()
|
|
|
|
resp, err := page.Eval("JSON.stringify(window.__coverage__)")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
coverageData := fmt.Sprintf("%v", resp.Value)
|
|
|
|
_ = os.MkdirAll(coverageDir, 0775)
|
|
|
|
if coverageData != "<nil>" {
|
|
err = ioutil.WriteFile(fmt.Sprintf("%s/coverage-%d.json", coverageDir, now.Unix()), []byte(coverageData), 0664) //nolint:gosec
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = filepath.Walk("../../web/.nyc_output", fixCoveragePath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (rs *RodSession) collectScreenshot(err error, page *rod.Page) {
|
|
if err == context.DeadlineExceeded && os.Getenv("CI") == t {
|
|
base := "/buildkite/screenshots"
|
|
build := os.Getenv("BUILDKITE_BUILD_NUMBER")
|
|
suite := strings.ToLower(os.Getenv("SUITE"))
|
|
job := os.Getenv("BUILDKITE_JOB_ID")
|
|
path := filepath.Join(fmt.Sprintf("%s/%s/%s/%s", base, build, suite, job)) //nolint: gocritic
|
|
|
|
if err := os.MkdirAll(path, 0755); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
pc, _, _, _ := runtime.Caller(2)
|
|
fn := runtime.FuncForPC(pc)
|
|
p := "github.com/authelia/authelia/v4/internal/suites."
|
|
r := strings.NewReplacer(p, "", "(", "", ")", "", "*", "", ".", "-")
|
|
|
|
page.MustScreenshotFullPage(fmt.Sprintf("%s/%s.jpg", path, r.Replace(fn.Name())))
|
|
}
|
|
}
|
|
|
|
func fixCoveragePath(path string, file os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if file.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
coverage, err := filepath.Match("*.json", file.Name())
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if coverage {
|
|
read, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
wd, _ := os.Getwd()
|
|
ciPath := strings.TrimSuffix(wd, "internal/suites")
|
|
content := strings.ReplaceAll(string(read), "/node/src/app/", ciPath+"web/")
|
|
|
|
err = ioutil.WriteFile(path, []byte(content), 0)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|