Added AddRepository function

Author: cmaglie

repos.go

@@ -23,6 +23,7 @@ import (
 	"bytes"
 	"fmt"
 	"io/ioutil"
+	"os"
 	"path/filepath"
 	"regexp"
 	"strings"
@@ -160,3 +161,31 @@ func ParseAPTConfigFolder(folderPath string) (RepositoryList, error) {
 	}
 	return res, nil
 }
+
+// AddRepository adds the specified repository to the specified APT
+// config folder (usually /etc/apt). The new repository is saved into
+// a file named "managed.list"
+func AddRepository(repo *Repository, configFolderPath string) error {
+	repos, err := ParseAPTConfigFolder(configFolderPath)
+	if err != nil {
+		return fmt.Errorf("parsing APT config: %s", err)
+	}
+	if repos.Contains(repo) {
+		return fmt.Errorf("The repository is already configured")
+	}
+
+	// Add to the "managed.list" file
+	managedPath := filepath.Join(configFolderPath, "sources.list.d", "managed.list")
+	f, err := os.OpenFile(managedPath, os.O_APPEND|os.O_WRONLY, 0644)
+	if os.IsNotExist(err) {
+		f, err = os.OpenFile(managedPath, os.O_CREATE|os.O_WRONLY, 0644)
+	}
+	if err != nil {
+		return fmt.Errorf("Opening %s: %s", managedPath, err)
+	}
+	defer f.Close()
+	if _, err = f.WriteString(repo.APTConfigLine() + "\n"); err != nil {
+		return fmt.Errorf("Writing repo data to config file %s: %s", managedPath, err)
+	}
+	return nil
+}

repos_test.go

@@ -21,6 +21,7 @@ package apt
 import (
 	"encoding/json"
 	"io/ioutil"
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -40,3 +41,37 @@ func TestParseAPTConfigFolder(t *testing.T) {
 		require.EqualValues(t, expected[i], repo, "Comparing element %d", i)
 	}
 }
+
+func TestAddRepository(t *testing.T) {
+	// test cleanup
+	defer os.Remove("testdata/apt2/sources.list.d/managed.list")
+
+	repo1 := &Repository{
+		Enabled:      true,
+		SourceRepo:   false,
+		URI:          "http://ppa.launchpad.net/webupd8team/java/ubuntu",
+		Distribution: "zesty",
+		Components:   "main",
+		Comment:      "",
+	}
+	repo2 := &Repository{
+		Enabled:      false,
+		SourceRepo:   true,
+		URI:          "http://ppa.launchpad.net/webupd8team/java/ubuntu",
+		Distribution: "zesty",
+		Components:   "main",
+		Comment:      "",
+	}
+	err := AddRepository(repo1, "testdata/apt2")
+	require.NoError(t, err, "Adding repository")
+	err = AddRepository(repo2, "testdata/apt2")
+	require.NoError(t, err, "Adding repository")
+
+	repos, err := ParseAPTConfigFolder("testdata/apt2")
+	require.NoError(t, err, "running List command")
+	require.True(t, repos.Contains(repo1), "Configuration contains: %#v", repo1)
+	require.True(t, repos.Contains(repo1), "Configuration contains: %#v", repo2)
+
+	err = AddRepository(repo2, "testdata/apt2")
+	require.Error(t, err, "Adding repository again")
+}

testdata/apt2/sources.list

@@ -0,0 +1,3 @@
+deb http://it.archive.ubuntu.com/ubuntu/ zesty main restricted
+deb-src http://it.archive.ubuntu.com/ubuntu/ zesty main restricted
+