move some code around to make it public

api.go

@@ -7,6 +7,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"hash/crc32"
 	"net"
 	"strings"
 	"time"
@@ -403,3 +404,9 @@ func GeneratePassword() string {
 	rand.Read(b)
 	return base64.StdEncoding.EncodeToString(b)
 }
+
+// GenerateUsername returns a username somehow related to the name of
+// the mount, possibly unique (but not actually guaranteed to be so).
+func GenerateUsername(path string) string {
+	return fmt.Sprintf("source%d", crc32.ChecksumIEEE([]byte(path)))
+}

cmd/radioctl/radioctl.go

@@ -4,7 +4,6 @@ import (
 	"encoding/json"
 	"flag"
 	"fmt"
-	"hash/crc32"
 	"log"
 	"net/url"
 	"os"
@@ -140,14 +139,10 @@ func getClient() *autoradio.Client {
 	return autoradio.NewClient(autoradio.NewEtcdClient(false))
 }
 
-func generateUsername(path string) string {
-	return fmt.Sprintf("source%d", crc32.ChecksumIEEE([]byte(path)))
-}
-
 func setRelay(m *autoradio.Mount, relayUrl string) {
 	if relayUrl == "" {
 		// Randomly generate source credentials.
-		m.Username = generateUsername(m.Name)
+		m.Username = autoradio.GenerateUsername(m.Name)
 		m.Password = autoradio.GeneratePassword()
 	} else {
 		// Validate the given relay URL.

debug.go

+package autoradio
+
+import "sort"
+
+// MountStatus reports the configuration and status of a mount,
+// including eventual transcoded mounts that source it.
+type MountStatus struct {
+	Mount       *Mount
+	Listeners   int
+	TransMounts []*MountStatus
+}
+
+func newMountStatus(m *Mount, nodes []*NodeStatus) *MountStatus {
+	var listeners int
+	for _, n := range nodes {
+		for _, ims := range n.Mounts {
+			if ims.Name == m.Name {
+				listeners += ims.Listeners
+				break
+			}
+		}
+	}
+	return &MountStatus{
+		Mount:     m,
+		Listeners: listeners,
+	}
+}
+
+type mountStatusList []*MountStatus
+
+func (l mountStatusList) Len() int      { return len(l) }
+func (l mountStatusList) Swap(i, j int) { l[i], l[j] = l[j], l[i] }
+func (l mountStatusList) Less(i, j int) bool {
+	return l[i].Mount.Name < l[j].Mount.Name
+}
+
+// MountsToStatus converts a list of mounts (and eventually the
+// current list of nodes) to a nicely sorted and tree-aggregated list
+// of MountStatus objects. The list of nodes can be nil, in which case
+// listener statistics will be omitted.
+func MountsToStatus(mounts []*Mount, nodes []*NodeStatus) []*MountStatus {
+	// Aggregate stats, and create a tree of transcoding mounts.
+	ms := make(map[string]*MountStatus)
+	for _, m := range mounts {
+		if m.HasTranscoder() {
+			continue
+		}
+		ms[m.Name] = newMountStatus(m, nodes)
+	}
+	for _, m := range mounts {
+		if !m.HasTranscoder() {
+			continue
+		}
+		src := ms[m.Transcoding.SourceName]
+		if src == nil {
+			continue
+		}
+		src.TransMounts = append(src.TransMounts, newMountStatus(m, nodes))
+	}
+	msl := make([]*MountStatus, 0, len(ms))
+	for _, m := range ms {
+		msl = append(msl, m)
+	}
+
+	// Sort everything (including transcoded mounts).
+	sort.Sort(mountStatusList(msl))
+	for _, s := range msl {
+		if s.TransMounts != nil {
+			sort.Sort(mountStatusList(s.TransMounts))
+		}
+	}
+	return msl
+}

fe/http.go

@@ -11,7 +11,6 @@ import (
 	"net/http"
 	"net/url"
 	"path/filepath"
-	"sort"
 	"strconv"
 	"strings"
 	"time"
@@ -266,76 +265,16 @@ func (h *HTTPRedirector) serveSource(mount *autoradio.Mount, w http.ResponseWrit
 	proxy.ServeHTTP(w, r)
 }
 
-type mountStatus struct {
-	Mount       *autoradio.Mount
-	Listeners   int
-	TransMounts []*mountStatus
-}
-
-func newMountStatus(m *autoradio.Mount, nodes []*autoradio.NodeStatus) *mountStatus {
-	var listeners int
-	for _, n := range nodes {
-		for _, ims := range n.Mounts {
-			if ims.Name == m.Name {
-				listeners += ims.Listeners
-				break
-			}
-		}
-	}
-	return &mountStatus{
-		Mount:     m,
-		Listeners: listeners,
-	}
-}
-
-type mountStatusList []*mountStatus
-
-func (l mountStatusList) Len() int      { return len(l) }
-func (l mountStatusList) Swap(i, j int) { l[i], l[j] = l[j], l[i] }
-func (l mountStatusList) Less(i, j int) bool {
-	return l[i].Mount.Name < l[j].Mount.Name
-}
-
 // Serve our cluster status page.
 func (h *HTTPRedirector) serveStatusPage(w http.ResponseWriter, r *http.Request) {
 	nodes, _ := h.client.GetNodes()
 	mounts, _ := h.client.ListMounts()
 
-	// Aggregate stats, and create a tree of transcoding mounts.
-	ms := make(map[string]*mountStatus)
-	for _, m := range mounts {
-		if m.HasTranscoder() {
-			continue
-		}
-		ms[m.Name] = newMountStatus(m, nodes)
-	}
-	for _, m := range mounts {
-		if !m.HasTranscoder() {
-			continue
-		}
-		src := ms[m.Transcoding.SourceName]
-		if src == nil {
-			continue
-		}
-		src.TransMounts = append(src.TransMounts, newMountStatus(m, nodes))
-	}
-	msl := make([]*mountStatus, 0, len(ms))
-	for _, m := range ms {
-		msl = append(msl, m)
-	}
-
-	// Sort everything.
-	sort.Sort(mountStatusList(msl))
-	for _, s := range msl {
-		if s.TransMounts != nil {
-			sort.Sort(mountStatusList(s.TransMounts))
-		}
-	}
-
+	msl := autoradio.MountsToStatus(mounts, nodes)
 	ctx := struct {
 		Domain string
 		Nodes  []*autoradio.NodeStatus
-		Mounts []*mountStatus
+		Mounts []*autoradio.MountStatus
 	}{h.domain, nodes, msl}
 
 	var buf bytes.Buffer