Function startCallbackServerWithContext

cmd/login.go:427–487

(ctx context.Context)

}

func startCallbackServerWithContext(ctx context.Context) (int, <-chan AuthResponse, <-chan error, func(), error) {
	listener, err := net.Listen("tcp", "127.0.0.1:0")
	if err != nil {
		return 0, nil, nil, nil, err
	}

	port := listener.Addr().(*net.TCPAddr).Port

	authChan := make(chan AuthResponse, 1)
	errChan := make(chan error, 1)

	mux := http.NewServeMux()
	server := &http.Server{
		Handler: mux,
	}

	mux.HandleFunc("/callback", func(w http.ResponseWriter, r *http.Request) {
		token := r.URL.Query().Get("token")
		refreshToken := r.URL.Query().Get("refresh_token")
		errorParam := r.URL.Query().Get("error")

		if errorParam != "" {
			errChan <- fmt.Errorf("authentication error: %s", errorParam)
			w.Header().Set("Content-Type", "text/html")
			w.WriteHeader(http.StatusUnauthorized)
			_ = authFailedTemplate.Execute(w, map[string]string{"Error": errorParam}) //nolint:errcheck // template execution error is non-fatal
			return
		}

		if token == "" {
			errChan <- fmt.Errorf("no token received")
			w.WriteHeader(http.StatusBadRequest)
			fmt.Fprintf(w, "No token received")
			return
		}

		authResp := AuthResponse{
			Token:        token,
			RefreshToken: refreshToken,
		}

		authChan <- authResp

		w.Header().Set("Content-Type", "text/html")
		_ = authSuccessTemplate.Execute(w, nil) //nolint:errcheck // template execution error is non-fatal
	})

	go func() {
		if err := server.Serve(listener); err != nil && err != http.ErrServerClosed {
			errChan <- err
		}
	}()

	cleanup := func() {
		shutdownCtx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
		defer cancel()
		_ = server.Shutdown(shutdownCtx) //nolint:errcheck // shutdown error is non-fatal
	}