Building Plugins¶

func (p *DigitaloceanProvider) Scan(req *pb.ScanRequest, stream pb.Provider_ScanServer) error {
    // 1. Initialize your cloud SDK client.
    token := req.Config["DIGITALOCEAN_TOKEN"]
    client := godo.NewFromToken(token)

    // 2. List resources from the cloud API.
    droplets, _, _ := client.Droplets.List(stream.Context(), nil)

    // 3. For each resource, stream a node back to the engine.
    for _, d := range droplets {
        node := &pb.Node{
            Crn: fmt.Sprintf("crn:digitalocean::%s:compute:droplet:%s:%s",
                d.Region.Slug, "account-id", d.Name),
            Properties: map[string]string{
                "instance_type": d.SizeSlug,
                "vcpus":         fmt.Sprint(d.Vcpus),
                "memory_mb":     fmt.Sprint(d.Memory),
            },
            Cost: estimateCost(d.SizeSlug),
        }
        stream.Send(&pb.ScanResponse{Nodes: []*pb.Node{node}})
    }

    return nil
}