package middleware

import (
	"fmt"
	"strings"

	"github.com/grafana/grafana/pkg/bus"
	m "github.com/grafana/grafana/pkg/models"
	"github.com/grafana/grafana/pkg/setting"
	"gopkg.in/macaron.v1"
)

func getDashboardURLBySlug(orgID int64, slug string) (string, error) {
	query := m.GetDashboardQuery{Slug: slug, OrgId: orgID}

	if err := bus.Dispatch(&query); err != nil {
		return "", m.ErrDashboardNotFound
	}

	return m.GetDashboardUrl(query.Result.Uid, query.Result.Slug), nil
}

func RedirectFromLegacyDashboardURL() macaron.Handler {
	return func(c *m.ReqContext) {
		slug := c.Params("slug")

		if slug == "" {
			return
		}
		if url, err := getDashboardURLBySlug(c.OrgId, slug); err == nil {
			url = fmt.Sprintf("%s?%s", url, c.Req.URL.RawQuery)
			c.Redirect(url, 301)
		}
	}
}

func RedirectFromLegacyDashboardSoloURL() macaron.Handler {
	return func(c *m.ReqContext) {
		slug := c.Params("slug")
		renderRequest := c.QueryBool("render")

		if slug == "" {
			return
		}
		if url, err := getDashboardURLBySlug(c.OrgId, slug); err == nil {
			if renderRequest && strings.Contains(url, setting.AppSubUrl) {
				url = strings.Replace(url, setting.AppSubUrl, "", 1)
			}
			url = strings.Replace(url, "/d/", "/d-solo/", 1)
			url = fmt.Sprintf("%s?%s", url, c.Req.URL.RawQuery)
			c.Redirect(url, 301)
		}
	}
}