explicit message for malformed data URL in local items (#3537)

Author: mmetc

debian/migrate-hub.sh

@@ -1,15 +1,15 @@
 #!/usr/bin/env sh
 
 # This script is provided (only in the source distribution) as an ad-hoc solution
-# to migrate an installation from the crowdsec package maintained in the debian repositories
+# to migrate an installation from the crowdsec package maintained in the debian or ubuntu repositories
 # to the official crowdsec repository.
 
 set -eu
 
 if [ ! -d /var/lib/crowdsec/hub/ ]; then
     echo "You don't have a hub directory to migrate."
     echo
-    echo "Use this script only if you upgrade from the crowdsec package included in the debian repositories."
+    echo "Use this script only if you upgrade from the crowdsec package included in the debian or ubuntu repositories."
     exit 1
 fi
 

pkg/hubops/colorize.go

@@ -15,8 +15,10 @@ func colorizeItemName(fullname string) string {
 		bold := color.New(color.Bold)
 		author := parts[0]
 		name := parts[1]
+
 		return author + "/" + bold.Sprint(name)
 	}
+
 	return fullname
 }
 

pkg/hubops/download.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"net/url"
 	"os"
 	"time"
 
@@ -118,6 +119,13 @@ func downloadDataSet(ctx context.Context, dataFolder string, force bool, reader
 				continue
 			}
 
+			// twopenny validation
+			if u, err := url.Parse(dataS.SourceURL); err != nil {
+				return false, err
+			} else if u.Scheme == "" {
+				return false, fmt.Errorf("a valid URL was expected (note: local items can download data too): %s", dataS.SourceURL)
+			}
+
 			// XXX: check context cancellation
 			destPath, err := cwhub.SafePath(dataFolder, dataS.DestPath)
 			if err != nil {

test/bats/20_hub_items.bats

@@ -151,6 +151,29 @@ teardown() {
     assert_output --partial "Nothing to do."
 }
 
+@test "when upgrading the hub, a local item's data will be downloaded" {
+    rune -0 mkdir -p "$CONFIG_DIR/collections"
+    cat >"$CONFIG_DIR"/collections/foobar.yaml <<-EOT
+        data:
+          - source_url: https://localhost:1234/database.mmdb
+            dest_file: database.mmdb
+	EOT
+    rune -1 cscli hub upgrade
+    assert_line "downloading https://localhost:1234/database.mmdb"
+    assert_stderr --partial 'Get "https://localhost:1234/database.mmdb":'
+    assert_stderr --partial 'connect: connection refused'
+
+    # bad link, or local path
+    cat >"$CONFIG_DIR"/collections/foobar.yaml <<-EOT
+        data:
+          - source_url: /tmp/meh
+            dest_file: database.mmdb
+	EOT
+    rune -1 cscli hub upgrade
+    refute_line "downloading /tmp/meh"
+    assert_stderr --partial 'a valid URL was expected (note: local items can download data too): /tmp/meh'
+}
+
 @test "a local item cannot be removed by cscli" {
     rune -0 mkdir -p "$CONFIG_DIR/scenarios"
     rune -0 touch "$CONFIG_DIR/scenarios/foobar.yaml"