-
Notifications
You must be signed in to change notification settings - Fork 128
/
create.go
123 lines (100 loc) · 3.5 KB
/
create.go
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
// Copyright (c) 2021 Apptainer a Series of LF Projects LLC
// For website terms of use, trademark policy, privacy policy and other
// project policies see https://lfprojects.org/policies
// 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 sources of this project regarding your
// rights to use or distribute this software.
package plugin
import (
"errors"
"fmt"
"io/fs"
"io/ioutil"
"os"
"path/filepath"
"github.com/apptainer/apptainer/internal/pkg/buildcfg"
"github.com/apptainer/apptainer/pkg/sylog"
)
const goMod = `module %s
go 1.16
require github.com/apptainer/apptainer v0.0.0
replace github.com/apptainer/apptainer => ./apptainer_source
`
const mainGo = `package main
import (
pluginapi "github.com/apptainer/apptainer/pkg/plugin"
)
// 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: "%s",
Author: "Put your name or mail here",
Version: "0.1.0",
Description: "Put a nice description",
},
Callbacks: []pluginapi.Callback{},
Install: installCallback,
}
func installCallback(path string) error {
// Create required stuff during "plugin install"
// (eg: configuration file, setup ...). Be careful
// during setup as this callback is executed with
// root privileges.
return nil
}
// Write plugin callbacks here and register them in Callbacks
`
const gitIgnore = `apptainer_source
*.sif
*.o
*.a
`
// Create creates a skeleton plugin directory structure
// to start development of a new plugin.
func Create(path, name string) error {
dir, err := filepath.Abs(path)
if err != nil {
return fmt.Errorf("could not determine absolute path for %s: %s", path, err)
}
if err := os.MkdirAll(dir, 0o755); err != nil {
return fmt.Errorf("while creating plugin directory %s: %s", dir, err)
}
// create go.mod skeleton
filename := filepath.Join(dir, "go.mod")
content := fmt.Sprintf(goMod, name)
if err := ioutil.WriteFile(filename, []byte(content), 0o644); err != nil {
return fmt.Errorf("while creating plugin %s: %s", filename, err)
}
// create main.go skeleton
filename = filepath.Join(dir, "main.go")
content = fmt.Sprintf(mainGo, name)
if err := ioutil.WriteFile(filename, []byte(content), 0o644); err != nil {
return fmt.Errorf("while creating plugin %s: %s", filename, err)
}
// create .gitignore skeleton
filename = filepath.Join(dir, ".gitignore")
if err := ioutil.WriteFile(filename, []byte(gitIgnore), 0o644); err != nil {
return fmt.Errorf("while creating plugin %s: %s", filename, err)
}
// create symlink to apptainer source directory
sourceLink := filepath.Join(dir, ApptainerSource)
if _, err := os.Stat(buildcfg.SOURCEDIR); os.IsNotExist(err) {
ls := fmt.Sprintf("ln -s /path/to/apptainer/source %s", sourceLink)
sylog.Warningf("Apptainer source %s doesn't exist, you would have to execute manually %q", buildcfg.SOURCEDIR, ls)
return nil
} else if err != nil {
return fmt.Errorf("while getting %s information: %s", sourceLink, err)
}
// delete it first if already present
if err := os.Remove(sourceLink); err != nil {
if !errors.Is(err, fs.ErrNotExist) {
return fmt.Errorf("while removing %s symlink: %s", sourceLink, err)
}
}
if err := os.Symlink(buildcfg.SOURCEDIR, sourceLink); err != nil {
return fmt.Errorf("while creating symlink %s: %s", sourceLink, err)
}
return nil
}