Josh Hunt
71e1305364
* Use saved dashboard model to create library panel connections when importing * Rename variables in dashboard import for clarity
112 lines
3.6 KiB
Go
112 lines
3.6 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/api/routing"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
"github.com/grafana/grafana/pkg/services/accesscontrol"
|
|
"github.com/grafana/grafana/pkg/services/dashboardimport"
|
|
"github.com/grafana/grafana/pkg/services/dashboardimport/api"
|
|
"github.com/grafana/grafana/pkg/services/dashboardimport/utils"
|
|
"github.com/grafana/grafana/pkg/services/dashboards"
|
|
"github.com/grafana/grafana/pkg/services/librarypanels"
|
|
"github.com/grafana/grafana/pkg/services/plugindashboards"
|
|
"github.com/grafana/grafana/pkg/services/quota"
|
|
"github.com/grafana/grafana/pkg/services/schemaloader"
|
|
)
|
|
|
|
func ProvideService(routeRegister routing.RouteRegister,
|
|
quotaService *quota.QuotaService, schemaLoaderService *schemaloader.SchemaLoaderService,
|
|
pluginDashboardService plugindashboards.Service, pluginStore plugins.Store,
|
|
libraryPanelService librarypanels.Service, dashboardService dashboards.DashboardService,
|
|
ac accesscontrol.AccessControl,
|
|
) *ImportDashboardService {
|
|
s := &ImportDashboardService{
|
|
pluginDashboardService: pluginDashboardService,
|
|
dashboardService: dashboardService,
|
|
libraryPanelService: libraryPanelService,
|
|
}
|
|
|
|
dashboardImportAPI := api.New(s, quotaService, schemaLoaderService, pluginStore, ac)
|
|
dashboardImportAPI.RegisterAPIEndpoints(routeRegister)
|
|
|
|
return s
|
|
}
|
|
|
|
type ImportDashboardService struct {
|
|
pluginDashboardService plugindashboards.Service
|
|
dashboardService dashboards.DashboardService
|
|
libraryPanelService librarypanels.Service
|
|
}
|
|
|
|
func (s *ImportDashboardService) ImportDashboard(ctx context.Context, req *dashboardimport.ImportDashboardRequest) (*dashboardimport.ImportDashboardResponse, error) {
|
|
var draftDashboard *models.Dashboard
|
|
if req.PluginId != "" {
|
|
loadReq := &plugindashboards.LoadPluginDashboardRequest{
|
|
PluginID: req.PluginId,
|
|
Reference: req.Path,
|
|
}
|
|
if resp, err := s.pluginDashboardService.LoadPluginDashboard(ctx, loadReq); err != nil {
|
|
return nil, err
|
|
} else {
|
|
draftDashboard = resp.Dashboard
|
|
}
|
|
} else {
|
|
draftDashboard = models.NewDashboardFromJson(req.Dashboard)
|
|
}
|
|
|
|
evaluator := utils.NewDashTemplateEvaluator(draftDashboard.Data, req.Inputs)
|
|
generatedDash, err := evaluator.Eval()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
saveCmd := models.SaveDashboardCommand{
|
|
Dashboard: generatedDash,
|
|
OrgId: req.User.OrgId,
|
|
UserId: req.User.UserId,
|
|
Overwrite: req.Overwrite,
|
|
PluginId: req.PluginId,
|
|
FolderId: req.FolderId,
|
|
}
|
|
|
|
dto := &dashboards.SaveDashboardDTO{
|
|
OrgId: saveCmd.OrgId,
|
|
Dashboard: saveCmd.GetDashboardModel(),
|
|
Overwrite: saveCmd.Overwrite,
|
|
User: req.User,
|
|
}
|
|
|
|
savedDashboard, err := s.dashboardService.ImportDashboard(ctx, dto)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = s.libraryPanelService.ImportLibraryPanelsForDashboard(ctx, req.User, savedDashboard, req.FolderId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = s.libraryPanelService.ConnectLibraryPanelsForDashboard(ctx, req.User, savedDashboard)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &dashboardimport.ImportDashboardResponse{
|
|
UID: savedDashboard.Uid,
|
|
PluginId: req.PluginId,
|
|
Title: savedDashboard.Title,
|
|
Path: req.Path,
|
|
Revision: savedDashboard.Data.Get("revision").MustInt64(1),
|
|
FolderId: savedDashboard.FolderId,
|
|
ImportedUri: "db/" + savedDashboard.Slug,
|
|
ImportedUrl: savedDashboard.GetUrl(),
|
|
ImportedRevision: savedDashboard.Data.Get("revision").MustInt64(1),
|
|
Imported: true,
|
|
DashboardId: savedDashboard.Id,
|
|
Slug: savedDashboard.Slug,
|
|
}, nil
|
|
}
|