// Copyright (c) 2026 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

package compliancepage

import (
	"context"
	"errors"
	"net/http"

	"github.com/99designs/gqlgen/graphql"
	"github.com/go-chi/chi/v5"
	"github.com/vektah/gqlparser/v2/gqlerror"
	"go.gearno.de/kit/httpserver"
	"go.probo.inc/probo/pkg/baseurl"
	"go.probo.inc/probo/pkg/gid"
	"go.probo.inc/probo/pkg/server/gqlutils"
	"go.probo.inc/probo/pkg/trust"
)

func NewIDMiddleware(trustSvc *trust.Service, baseURL string) func(next http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(
			func(w http.ResponseWriter, r *http.Request) {
				ctx := r.Context()
				// TODO: remove slug support
				value := chi.URLParam(r, "slugOrId")

				if id, err := gid.ParseGID(value); err == nil {
					compliancePage, err := trustSvc.Get(ctx, id)
					if err != nil {
						if errors.Is(err, trust.ErrPageNotFound) {
							next.ServeHTTP(w, r)
							return
						}

						httpserver.RenderJSON(
							w,
							http.StatusInternalServerError,
							&graphql.Response{
								Errors: gqlerror.List{
									gqlutils.Internal(ctx),
								},
							},
						)
						return
					}

					baseURL := baseurl.MustParse(baseURL).AppendPath("/trust/" + id.String()).MustString()
					ctx = context.WithValue(ctx, compliancePageBaseURLKey, &baseURL)
					r = r.WithContext(ctx)

					if !compliancePage.Active {
						next.ServeHTTP(w, r)
						return
					}

					ctx = context.WithValue(ctx, compliancePageKey, compliancePage)
					next.ServeHTTP(w, r.WithContext(ctx))
					return
				}

				compliancePage, err := trustSvc.GetBySlug(ctx, value)
				if err != nil {
					if errors.Is(err, trust.ErrPageNotFound) {
						next.ServeHTTP(w, r)
						return
					}

					httpserver.RenderJSON(
						w,
						http.StatusInternalServerError,
						&graphql.Response{
							Errors: gqlerror.List{
								gqlutils.Internal(ctx),
							},
						},
					)
					return
				}

				baseURL := baseurl.MustParse(baseURL).AppendPath("/trust/" + value).MustString()
				ctx = context.WithValue(ctx, compliancePageBaseURLKey, &baseURL)
				r = r.WithContext(ctx)

				if compliancePage.Active {
					ctx = context.WithValue(ctx, compliancePageKey, compliancePage)
					next.ServeHTTP(w, r.WithContext(ctx))
					return
				}

				next.ServeHTTP(w, r)
			},
		)
	}
}