Added repository management

Author: cmaglie

repos.go

@@ -0,0 +1,127 @@
+//
+//  This file is part of go-apt-client library
+//
+//  Copyright (C) 2017  Arduino AG (http://www.arduino.cc/)
+//
+//  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 apt
+
+import (
+	"bufio"
+	"bytes"
+	"fmt"
+	"io/ioutil"
+	"path/filepath"
+	"regexp"
+	"strings"
+)
+
+// Repository is a repository installed in the system
+type Repository struct {
+	Enabled      bool
+	SourceRepo   bool
+	Options      string
+	URI          string
+	Distribution string
+	Components   string
+	Comment      string
+}
+
+// APTConfigLine returns the source.list config line for the Repository
+func (r *Repository) APTConfigLine() string {
+	res := ""
+	if !r.Enabled {
+		res = "# "
+	}
+	if r.SourceRepo {
+		res += "deb-src "
+	} else {
+		res += "deb "
+	}
+	if strings.TrimSpace(r.Options) != "" {
+		res += "[" + r.Options + "]"
+	}
+	res += r.URI + " " + r.Distribution + " " + r.Components
+	if strings.TrimSpace(r.Comment) != "" {
+		res += " # " + r.Comment
+	}
+	return res
+}
+
+var aptConfigLineRegexp = regexp.MustCompile(`^(# )?(deb|deb-src)(?: \[(.*)\])? ([^ ]+) ([^ ]+) ([^#\n]+)(?: +# *(.*))?$`)
+
+func parseAPTConfigLine(line string) *Repository {
+	match := aptConfigLineRegexp.FindAllStringSubmatch(line, -1)
+	if len(match) == 0 || len(match[0]) < 6 {
+		return nil
+	}
+	fields := match[0]
+	//fmt.Printf("%+v\n", fields)
+	return &Repository{
+		Enabled:      fields[1] != "# ",
+		SourceRepo:   fields[2] == "deb-src",
+		Options:      fields[3],
+		URI:          fields[4],
+		Distribution: fields[5],
+		Components:   fields[6],
+		Comment:      fields[7],
+	}
+}
+
+func parseAPTConfigFile(configPath string) ([]*Repository, error) {
+	data, err := ioutil.ReadFile(configPath)
+	if err != nil {
+		return nil, fmt.Errorf("Reading %s: %s", configPath, err)
+	}
+	scanner := bufio.NewScanner(bytes.NewReader(data))
+
+	res := []*Repository{}
+	for scanner.Scan() {
+		line := scanner.Text()
+		repo := parseAPTConfigLine(line)
+		//fmt.Printf("%+v\n", repo)
+		if repo != nil {
+			res = append(res, repo)
+		}
+	}
+	return res, nil
+}
+
+// ParseAPTConfigFolder scans an APT config folder (usually /etc/apt) to
+// get information about configured repositories
+func ParseAPTConfigFolder(folderPath string) ([]*Repository, error) {
+	sources := []string{filepath.Join(folderPath, "sources.list")}
+
+	sourcesFolder := filepath.Join(folderPath, "sources.list.d")
+	list, err := ioutil.ReadDir(sourcesFolder)
+	if err != nil {
+		return nil, fmt.Errorf("Reading %s folder: %s", sourcesFolder, err)
+	}
+	for _, l := range list {
+		if strings.HasSuffix(l.Name(), ".list") {
+			sources = append(sources, filepath.Join(sourcesFolder, l.Name()))
+		}
+	}
+
+	res := []*Repository{}
+	for _, source := range sources {
+		repos, err := parseAPTConfigFile(source)
+		if err != nil {
+			return nil, fmt.Errorf("Parsing %s: %s", source, err)
+		}
+		res = append(res, repos...)
+	}
+	return res, nil
+}

repos_test.go

@@ -0,0 +1,42 @@
+//
+//  This file is part of go-apt-client library
+//
+//  Copyright (C) 2017  Arduino AG (http://www.arduino.cc/)
+//
+//  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 apt
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestParseAPTConfigFolder(t *testing.T) {
+	repos, err := ParseAPTConfigFolder("testdata/apt")
+	require.NoError(t, err, "running List command")
+
+	expectedData, err := ioutil.ReadFile("testdata/TestParseAPTConfigFolder.json")
+	require.NoError(t, err, "Reading test data")
+	expected := []*Repository{}
+	err = json.Unmarshal(expectedData, &expected)
+	require.NoError(t, err, "Decoding expected data")
+
+	for i, repo := range repos {
+		require.EqualValues(t, expected[i], repo, "Comparing element %d", i)
+	}
+}