UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-05-21 19:56:06 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Add spec-dirs argument to cdi list

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2024-02-08 14:44:23 +01:00
parent 15d905def0
commit b6ab444529
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
cmd/nvidia-ctk/cdi/list/list.go
View File

@ -17,6 +17,7 @@
package list package list
import ( import (
"errors"
"fmt" "fmt"
"github.com/urfave/cli/v2" "github.com/urfave/cli/v2"
@ -29,7 +30,9 @@ type command struct {
logger logger.Interface logger logger.Interface
} }
type config struct{} type config struct {
cdiSpecDirs cli.StringSlice
}
// NewCommand constructs a cdi list command with the specified logger // NewCommand constructs a cdi list command with the specified logger
func NewCommand(logger logger.Interface) *cli.Command { func NewCommand(logger logger.Interface) *cli.Command {
@ -55,19 +58,29 @@ func (m command) build() *cli.Command {
}, },
} }
c.Flags = []cli.Flag{} c.Flags = []cli.Flag{
&cli.StringSliceFlag{
Name: "spec-dir",
Usage: "specify the directories to scan for CDI specifications",
Value: cli.NewStringSlice(cdi.DefaultSpecDirs...),
Destination: &cfg.cdiSpecDirs,
},
}
return &c return &c
} }
func (m command) validateFlags(c *cli.Context, cfg *config) error { func (m command) validateFlags(c *cli.Context, cfg *config) error {
if len(cfg.cdiSpecDirs.Value()) == 0 {
return errors.New("at least one CDI specification directory must be specified")
}
return nil return nil
} }
func (m command) run(c *cli.Context, cfg *config) error { func (m command) run(c *cli.Context, cfg *config) error {
registry, err := cdi.NewCache( registry, err := cdi.NewCache(
cdi.WithAutoRefresh(false), cdi.WithAutoRefresh(false),
cdi.WithSpecDirs(cdi.DefaultSpecDirs...), cdi.WithSpecDirs(cfg.cdiSpecDirs.Value()...),
) )
if err != nil { if err != nil {
return fmt.Errorf("failed to create CDI cache: %v", err) return fmt.Errorf("failed to create CDI cache: %v", err)