Merge pull request #17 from chaitin/image-caller-interface

允许直接使用 ID 列表触发插件扫描

go/cmd/images.go

@@ -41,6 +41,20 @@ func ScanImage(
 	return ScanImages(ctx, rang, []api.Image{image}, opts...)
 }
 
+// ScanImageIDs with a runtime and a list of IDs provided.
+func ScanImageIDs(
+	ctx context.Context, rang plugin.ExecRange,
+	runtime api.Runtime, ids []string, opts ...plugin.ExecOption,
+) error {
+	iter, err := plugin.IterateTyped(rang, "image")
+	if err != nil {
+		return err
+	}
+	return ScanIDs(ctx, iter, runtime, ids,
+		plugin.WithPrependArgs("--id"),
+		plugin.WithExecOptions(opts...))
+}
+
 // imageExactIDs specifies whether the argument list specifies
 // ID instead of searchable names.
 var imageExactIDs bool

go/cmd/root.go

@@ -123,3 +123,27 @@ func Scan(
 	})
 	return err
 }
+
+// ScanIDs attempts to load a root object while passing a list
+// of IDs that will be acceptable.
+func ScanIDs(
+	ctx context.Context, iter plugin.ExecIterator,
+	obj interface{}, ids []string, opts ...plugin.ExecOption,
+) error {
+	objVal := reflect.ValueOf(obj)
+	objTyp := objVal.Type()
+	val, ok := partitioners.Load(objTyp)
+	if !ok {
+		panic(fmt.Sprintf("undefined partition %q", objTyp))
+	}
+	root, pids := val.(partitioner)(objVal)
+	if len(pids) > 0 {
+		panic(fmt.Sprintf("invalid root object with ID"))
+	}
+	if len(ids) == 0 {
+		return nil
+	}
+	return plugin.Exec(ctx, iter, ids,
+		plugin.WithPrependArgs("--mode", root.Mode()),
+		root.Options(), plugin.WithExecOptions(opts...))
+}