// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// Package proxy implements the HTTP protocols for serving a Go module proxy.
package sumdb

import (
	"fmt"
	"io"
	"net/http"
	"net/url"
	"strings"
)

var enableGoogleSumDB bool
var supportedSumDB = []string{
	"sum.golang.org",
	"gosum.io",
}

func init() {
	go func() {
		p := "https://sum.google.com"
		_, err := http.Get(p)
		if err == nil {
			enableGoogleSumDB = true
		}
	}()
}

//Handler handles sumdb request
func Handler(w http.ResponseWriter, r *http.Request) {
	if !enableGoogleSumDB {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	for _, supported := range supportedSumDB {
		uri := fmt.Sprintf("/sumdb/%s/supported", supported)
		if r.URL.Path == uri {
			w.WriteHeader(http.StatusOK)
			return
		}
	}

	p := "https://" + strings.TrimPrefix(r.URL.Path, "/sumdb/")
	_, err := url.Parse(p)
	if err != nil {
		w.WriteHeader(http.StatusGone)
		fmt.Fprintf(w, err.Error())
		return
	}

	resp, err := http.Get(p)
	if err != nil {
		w.WriteHeader(http.StatusGone)
		fmt.Fprintf(w, err.Error())
		return
	}
	defer resp.Body.Close()
	w.WriteHeader(resp.StatusCode)
	if _, err := io.Copy(w, resp.Body); err != nil {
		fmt.Fprintf(w, err.Error())
		return
	}

	return
}