<commit_before>// Copyright 2016 The prometheus-operator Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package updater

import (
	"log"
	"os"
	"path/filepath"

	"github.com/coreos/prometheus-operator/contrib/grafana-watcher/grafana"
)

type GrafanaDatasourceUpdater struct {
	client grafana.DatasourcesInterface
	globs  []string
}

func NewGrafanaDatasourceUpdater(c grafana.DatasourcesInterface, g []string) Updater {
	return &GrafanaDatasourceUpdater{
		client: c,
		globs:  g,
	}
}

func (u *GrafanaDatasourceUpdater) Init() error {
	return u.updateDatasources()
}

func (u *GrafanaDatasourceUpdater) OnModify() error {
	return u.updateDatasources()
}

func (u *GrafanaDatasourceUpdater) updateDatasources() error {
	err := u.deleteAllDatasources()
	if err != nil {
		return err
	}
	err = u.createDatasourcesFromFiles()
	if err != nil {
		return err
	}

	return nil
}

func (u *GrafanaDatasourceUpdater) deleteAllDatasources() error {
	log.Println("Retrieving existing datasources")
	datasources, err := u.client.All()
	if err != nil {
		return err
	}

	log.Printf("Deleting %d datasources\n", len(datasources))
	for _, d := range datasources {
		log.Println("Deleting datasource:", d.Id)

		err := u.client.Delete(d.Id)
		if err != nil {
			return err
		}
	}

	return nil
}

func (u *GrafanaDatasourceUpdater) createDatasourcesFromFiles() error {
	for _, glob := range u.globs {
		filePaths, err := filepath.Glob(filepath.Join(glob, "*-datasource.json"))
		if err != nil {
			return err
		}

		for _, fp := range filePaths {
			err = u.createDatasourceFromFile(fp)<commit_msg>Print name instead of id when deleting datasources<commit_after>// Copyright 2016 The prometheus-operator Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package updater

import (
	"log"
	"os"
	"path/filepath"

	"github.com/coreos/prometheus-operator/contrib/grafana-watcher/grafana"
)

type GrafanaDatasourceUpdater struct {
	client grafana.DatasourcesInterface
	globs  []string
}

func NewGrafanaDatasourceUpdater(c grafana.DatasourcesInterface, g []string) Updater {
	return &GrafanaDatasourceUpdater{
		client: c,
		globs:  g,
	}
}

func (u *GrafanaDatasourceUpdater) Init() error {
	return u.updateDatasources()
}

func (u *GrafanaDatasourceUpdater) OnModify() error {
	return u.updateDatasources()
}

func (u *GrafanaDatasourceUpdater) updateDatasources() error {
	err := u.deleteAllDatasources()
	if err != nil {
		return err
	}
	err = u.createDatasourcesFromFiles()
	if err != nil {
		return err
	}

	return nil
}

func (u *GrafanaDatasourceUpdater) deleteAllDatasources() error {
	log.Println("Retrieving existing datasources")
	datasources, err := u.client.All()
	if err != nil {
		return err
	}

	log.Printf("Deleting %d datasources\n", len(datasources))
	for _, d := range datasources {
		log.Println("Deleting datasource:", d.Name)

		err := u.client.Delete(d.Id)
		if err != nil {
			return err
		}
	}

	return nil
}

func (u *GrafanaDatasourceUpdater) createDatasourcesFromFiles() error {
	for _, glob := range u.globs {
		filePaths, err := filepath.Glob(filepath.Join(glob, "*-datasource.json"))
		if err != nil {
			return err
		}

		for _, fp := range filePaths {
			err = u.createDatasourceFromFile(fp)
{'fraction_non_alphanumeric': 0.109408158453195, 'fraction_numerical': 0.00377269511907569, 'mean_word_length': 4.062052505966587, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 2, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5704840', 'n_tokens_mistral': 1479, 'n_tokens_neox': 1419, 'n_words': 416}