1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
// Copyright (c) 2020, Sylabs Inc. All rights reserved.
// This software is licensed under a 3-clause BSD license. Please consult the
// LICENSE.md file distributed with the URIs of this project regarding your
// rights to use or distribute this software.
package main
import (
"os"
"strconv"
"github.com/spf13/cobra"
"github.com/sylabs/singularity/v4/pkg/cmdline"
pluginapi "github.com/sylabs/singularity/v4/pkg/plugin"
clicallback "github.com/sylabs/singularity/v4/pkg/plugin/callback/cli"
"github.com/sylabs/singularity/v4/pkg/runtime/engine/config"
)
// Plugin is the only variable which a plugin MUST export.
// This symbol is accessed by the plugin framework to initialize the plugin.
var Plugin = pluginapi.Plugin{
Manifest: pluginapi.Manifest{
Name: "github.com/sylabs/singularity/e2e-cli-plugin",
Author: "Sylabs Team",
Version: "0.1.0",
Description: "E2E CLI plugin",
},
Callbacks: []pluginapi.Callback{
(clicallback.Command)(callbackExitCmd),
(clicallback.SingularityEngineConfig)(callbackSingularity),
},
}
func callbackExitCmd(manager *cmdline.CommandManager) {
manager.RegisterCmd(&cobra.Command{
DisableFlagsInUseLine: true,
Args: cobra.ExactArgs(1),
Use: "exit <code>",
Short: "Exit with code",
Long: "Exit with code",
Example: "singularity exit 42",
Run: func(cmd *cobra.Command, args []string) {
code, err := strconv.Atoi(args[0])
if err != nil {
os.Exit(255)
}
os.Exit(code)
},
TraverseChildren: true,
})
}
func callbackSingularity(_ *config.Common) {
os.Exit(43)
}
|
