templatefile/pkg/templater/repo-apk.go

package templater

import (
	"bufio"
	"fmt"
	"io/ioutil"
	"os"
	"strings"

	"forge.cadoles.com/pcaseiro/templatefile/pkg/utils"
)

var APKConfigFile = "/etc/apk/repositories"

type APKRepository struct {
	Repository
}

func (hr *APKRepository) urlIsPresent() (bool, error) {
	f, err := os.Open(APKConfigFile)
	if err != nil {
		return false, err
	}
	defer f.Close()

	// Splits on newlines by default.
	scanner := bufio.NewScanner(f)

	line := 1
	for scanner.Scan() {
		if strings.Contains(scanner.Text(), hr.URL) {
			return true, nil
		}

		line++
	}

	if err := scanner.Err(); err != nil {
		return false, err
	}
	return false, nil

}

func (hr *APKRepository) Add() error {

	URLIsPresent, err := hr.urlIsPresent()
	if err != nil {
		return err
	}

	if URLIsPresent {
		return nil
	} else {
		data := fmt.Sprintf("%s", hr.URL)
		file, err := os.OpenFile("/etc/apk/repositories", os.O_APPEND|os.O_WRONLY, 0644)
		if err != nil {
			return err
		}
		defer file.Close()
		if _, err := file.WriteString(data); err != nil {
			return err
		}
		return nil
	}

}

func (hr *APKRepository) Update() error {
	if _, stdErr, err := utils.RunSystemCommand("apk", "update"); err != nil {
		return fmt.Errorf("%s [%s]", stdErr, err)
	}

	return nil
}

func (hr *APKRepository) Delete() error {
	fileBytes, err := ioutil.ReadFile("/etc/apk/repositories")
	if err != nil {
		return err
	}
	lines := strings.Split(string(fileBytes), "\n")
	for _, line := range lines {
		fmt.Println(line)
	}
	return nil
}

func (hr *APKRepository) Manage() error {
	if hr.Enabled {
		return hr.Add()
	} else {
		return hr.Delete()
	}
}