-
Notifications
You must be signed in to change notification settings - Fork 62
/
v1alpha1.go
126 lines (99 loc) · 3.82 KB
/
v1alpha1.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
124
125
126
// SPDX-License-Identifier: BSD-3-Clause
// Copyright (c) 2022, Unikraft GmbH and The KraftKit Authors.
// Licensed under the BSD-3-Clause License (the "License").
// You may not use this file except in compliance with the License.
package ninepfs
import (
"context"
"fmt"
"os"
"path/filepath"
"k8s.io/apimachinery/pkg/util/uuid"
volumev1alpha1 "kraftkit.sh/api/volume/v1alpha1"
"kraftkit.sh/config"
"kraftkit.sh/log"
)
type v1alpha1Volume struct{}
func NewVolumeServiceV1alpha1(ctx context.Context, opts ...any) (volumev1alpha1.VolumeService, error) {
return &v1alpha1Volume{}, nil
}
// Create implements kraftkit.sh/api/volume/v1alpha1.Create
func (*v1alpha1Volume) Create(ctx context.Context, volume *volumev1alpha1.Volume) (*volumev1alpha1.Volume, error) {
var err error
if len(volume.Spec.Driver) == 0 {
volume.Spec.Driver = "9pfs"
} else if volume.Spec.Driver != "9pfs" {
return volume, fmt.Errorf("cannot use 9pfs driver when driver set to %s", volume.Spec.Driver)
}
if volume.ObjectMeta.UID == "" {
volume.ObjectMeta.UID = uuid.NewUUID()
}
if volume.ObjectMeta.Name == "" {
volume.ObjectMeta.Name = string(volume.ObjectMeta.UID)
}
if len(volume.Spec.Source) == 0 {
// If no Source is specified, create a new volume entry in the runtime store
log.G(ctx).Debugf("creating new volume entry in the runtime store %s", volume.ObjectMeta.UID)
volume.Spec.Source = filepath.Join(config.G[config.KraftKit](ctx).RuntimeDir, "volumes", string(volume.ObjectMeta.UID))
volume.Spec.Managed = true
} else {
volume.Spec.Managed = false
}
volume.Spec.Source, err = filepath.Abs(volume.Spec.Source)
if err != nil {
return volume, fmt.Errorf("cannot get absolute path for volume source: %w", err)
}
// Create the volume directory if it does not exist
if err := os.MkdirAll(volume.Spec.Source, 0o755); err != nil {
return volume, fmt.Errorf("cannot create volume directory: %w", err)
}
fileInfo, err := os.Stat(volume.Spec.Source)
if err != nil {
return volume, fmt.Errorf("cannot stat volume directory: %w", err)
}
if !fileInfo.IsDir() {
return volume, fmt.Errorf("volume source is not a directory: %s", volume.Spec.Source)
}
volume.Status.State = volumev1alpha1.VolumeStatePending
return volume, nil
}
// Delete implements kraftkit.sh/api/volume/v1alpha1.Delete
func (*v1alpha1Volume) Delete(_ context.Context, volume *volumev1alpha1.Volume) (*volumev1alpha1.Volume, error) {
if len(volume.Spec.Driver) == 0 || volume.Spec.Driver != "9pfs" {
return nil, nil
}
if len(volume.Spec.Source) == 0 {
return nil, nil
}
if volume.Status.State == volumev1alpha1.VolumeStateBound {
return volume, fmt.Errorf("cannot delete volume in state %s", volume.Status.State)
}
if volume.Spec.Managed {
if err := os.RemoveAll(volume.Spec.Source); err != nil {
return volume, fmt.Errorf("cannot remove volume directory: %w", err)
}
}
return nil, nil
}
// Get implements kraftkit.sh/api/volume/v1alpha1.Get
func (*v1alpha1Volume) Get(_ context.Context, volume *volumev1alpha1.Volume) (*volumev1alpha1.Volume, error) {
if len(volume.Spec.Driver) == 0 || volume.Spec.Driver != "9pfs" {
return nil, nil
}
if len(volume.Spec.Source) == 0 {
return nil, nil
}
return volume, nil
}
// List implements kraftkit.sh/api/volume/v1alpha1.List
func (*v1alpha1Volume) List(_ context.Context, volumes *volumev1alpha1.VolumeList) (*volumev1alpha1.VolumeList, error) {
return volumes, nil
}
// Update implements kraftkit.sh/api/volume/v1alpha1.Update
func (*v1alpha1Volume) Update(_ context.Context, volume *volumev1alpha1.Volume) (*volumev1alpha1.Volume, error) {
return volume, nil
}
// Watch implements kraftkit.sh/api/volume/v1alpha1.Watch
func (*v1alpha1Volume) Watch(context.Context, *volumev1alpha1.Volume) (chan *volumev1alpha1.Volume, chan error, error) {
panic("not implemented: kraftkit.sh/machine/volume/9pfs.v1alpha1Volume.Watch")
}