Create public interface by exporting internal types.

And also pass options in Start instead of relying on viper in the
implementation.

Author: clems4ever

.gitignore

@@ -20,3 +20,5 @@
 *.tar.gz
 go-graphkb
 importer-csv
+
+.config.yml

cmd/importer-csv/main.go

@@ -98,7 +98,14 @@ func main() {
 	rootCmd := &cobra.Command{
 		Use: "source-csv [opts]",
 		Run: func(cmd *cobra.Command, args []string) {
-			if err := graphkb.Start(NewCSVSource(), nil); err != nil {
+
+			options := graphkb.ImporterOptions{
+				URL:        viper.GetString("graphkb.url"),
+				AuthToken:  viper.GetString("graphkb.auth_token"),
+				SkipVerify: viper.GetBool("graphkb.skip_verify"),
+			}
+
+			if err := graphkb.Start(NewCSVSource(), options); err != nil {
 				log.Fatal(err)
 			}
 		},

graphkb/graph_importer.go

@@ -3,3 +3,5 @@ package graphkb
 import "github.com/clems4ever/go-graphkb/internal/knowledge"
 
 type GraphImporter = knowledge.GraphImporter
+
+type Transaction = knowledge.Transaction

graphkb/importer.go

@@ -2,28 +2,28 @@ package graphkb
 
 import (
 	"fmt"
+
 	"github.com/clems4ever/go-graphkb/internal/knowledge"
+	"github.com/clems4ever/go-graphkb/internal/schema"
 	"github.com/clems4ever/go-graphkb/internal/sources"
-	"github.com/spf13/viper"
 )
 
 type ImporterOptions struct {
-	CacheGraph bool
+	URL        string
+	AuthToken  string
+	SkipVerify bool
 }
 
-func Start(source sources.Source, options *ImporterOptions) error {
-	url := viper.GetString("graphkb.url")
-	if url == "" {
+func Start(source sources.Source, options ImporterOptions) error {
+	if options.URL == "" {
 		return fmt.Errorf("Please provide graphkb URL in configuration file")
 	}
-
-	authToken := viper.GetString("graphkb.auth_token")
-	if authToken == "" {
+	if options.AuthToken == "" {
 		return fmt.Errorf("Please provide a graphkb auth token to communicate with GraphKB")
 	}
 
 	observableSource := sources.NewObservableSource(source)
-	api := knowledge.NewGraphAPI(url, authToken)
+	api := knowledge.NewGraphAPI(options.URL, options.AuthToken, options.SkipVerify)
 	graphImporter := knowledge.NewGraphImporter(api)
 
 	if err := observableSource.Start(graphImporter); err != nil {
@@ -32,3 +32,15 @@ func Start(source sources.Source, options *ImporterOptions) error {
 
 	return nil
 }
+
+func CreateRelation(fromType schema.AssetType, relation, toType schema.AssetType) RelationType {
+	return schema.RelationType{
+		FromType: fromType,
+		Type:     RelationKeyType(relation),
+		ToType:   toType,
+	}
+}
+
+func CreateAsset(fromType string) AssetType {
+	return schema.AssetType(fromType)
+}

graphkb/tasks.go

@@ -3,3 +3,5 @@ package graphkb
 import "github.com/clems4ever/go-graphkb/internal/utils"
 
 type RecurrentTask = utils.RecurrentTask
+
+var NewRecurrentTask = utils.NewRecurrentTask