Added APT repository handlers

Author: cmaglie

handler_apt_repositories.go

@@ -0,0 +1,111 @@
+//
+//  This file is part of arduino-connector
+//
+//  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 main
+
+import (
+	"encoding/json"
+	"fmt"
+
+	apt "github.com/arduino/go-apt-client"
+	mqtt "github.com/eclipse/paho.mqtt.golang"
+)
+
+// AptRepositoryListEvent sends a list of available repositories
+func (s *Status) AptRepositoryListEvent(client mqtt.Client, msg mqtt.Message) {
+	// Get packages from system
+	all, err := apt.ParseAPTConfigFolder("/etc/apt")
+	if err != nil {
+		s.Error("/apt/repos/list/error", fmt.Errorf("Retrieving repositories: %s", err))
+		return
+	}
+
+	// Send result
+	data, err := json.Marshal(all)
+	if err != nil {
+		s.Error("/apt/repos/list/error", fmt.Errorf("Json marshal result: %s", err))
+		return
+	}
+
+	//var out bytes.Buffer
+	//json.Indent(&out, data, "", "  ")
+	//fmt.Println(string(out.Bytes()))
+
+	s.Info("/apt/repos/list", string(data)+"\n")
+}
+
+// AptRepositoryAddEvent adds a repository to the apt configuration
+func (s *Status) AptRepositoryAddEvent(client mqtt.Client, msg mqtt.Message) {
+	var params struct {
+		Repository *apt.Repository `json:"repository"`
+	}
+	err := json.Unmarshal(msg.Payload(), &params)
+	if err != nil {
+		s.Error("/apt/repos/add/error", fmt.Errorf("Unmarshal '%s': %s", msg.Payload(), err))
+		return
+	}
+
+	err = apt.AddRepository(params.Repository, "/etc/apt")
+	if err != nil {
+		s.Error("/apt/repos/add/error", fmt.Errorf("Adding repository '%s': %s", msg.Payload(), err))
+		return
+	}
+
+	s.InfoCommandOutput("/apt/repos/add", []byte("OK"), nil)
+}
+
+// AptRepositoryRemoveEvent removes a repository from the apt configuration
+func (s *Status) AptRepositoryRemoveEvent(client mqtt.Client, msg mqtt.Message) {
+	var params struct {
+		Repository *apt.Repository `json:"repository"`
+	}
+	err := json.Unmarshal(msg.Payload(), &params)
+	if err != nil {
+		s.Error("/apt/repos/remove/error", fmt.Errorf("Unmarshal '%s': %s", msg.Payload(), err))
+		return
+	}
+
+	err = apt.RemoveRepository(params.Repository, "/etc/apt")
+	if err != nil {
+		s.Error("/apt/repos/remove/error", fmt.Errorf("Removing repository '%s': %s", msg.Payload(), err))
+		return
+	}
+
+	s.InfoCommandOutput("/apt/repos/remove", []byte("OK"), nil)
+}
+
+// AptRepositoryEditEvent modifies a repository definition in the apt configuration
+func (s *Status) AptRepositoryEditEvent(client mqtt.Client, msg mqtt.Message) {
+	var params struct {
+		OldRepository *apt.Repository `json:"old_repository"`
+		NewRepository *apt.Repository `json:"new_repository"`
+	}
+	err := json.Unmarshal(msg.Payload(), &params)
+	if err != nil {
+		s.Error("/apt/repos/edit/error", fmt.Errorf("Unmarshal '%s': %s", msg.Payload(), err))
+		return
+	}
+
+	err = apt.EditRepository(params.OldRepository, params.NewRepository, "/etc/apt")
+	if err != nil {
+		s.Error("/apt/repos/edit/error", fmt.Errorf("Changing repository '%s': %s", msg.Payload(), err))
+		return
+	}
+
+	s.InfoCommandOutput("/apt/repos/edit", []byte("OK"), nil)
+}

main.go

@@ -218,6 +218,11 @@ func subscribeTopics(mqttClient mqtt.Client, id string, status *Status) {
 	mqttClient.Subscribe("$aws/things/"+id+"/apt/update/post", 1, status.AptUpdateEvent)
 	mqttClient.Subscribe("$aws/things/"+id+"/apt/upgrade/post", 1, status.AptUpgradeEvent)
 	mqttClient.Subscribe("$aws/things/"+id+"/apt/Remove/post", 1, status.AptRemoveEvent)
+
+	mqttClient.Subscribe("$aws/things/"+id+"/apt/repos/list/post", 1, status.AptRepositoryListEvent)
+	mqttClient.Subscribe("$aws/things/"+id+"/apt/repos/add/post", 1, status.AptRepositoryAddEvent)
+	mqttClient.Subscribe("$aws/things/"+id+"/apt/repos/remove/post", 1, status.AptRepositoryRemoveEvent)
+	mqttClient.Subscribe("$aws/things/"+id+"/apt/repos/edit/post", 1, status.AptRepositoryEditEvent)
 }
 
 func addFileToSketchDB(file os.FileInfo, status *Status) *SketchStatus {