forked from tektoncd/pipeline
/
decode_script.go
71 lines (60 loc) · 2.24 KB
/
decode_script.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
/*
Copyright 2020 The Tekton Authors
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package subcommands
import (
"bytes"
"encoding/base64"
"fmt"
"io"
"os"
)
// DecodeScriptCommand is the command name for decoding scripts.
const DecodeScriptCommand = "decode-script"
// decodeScript rewrites a script file from base64 back into its original content from
// the Step definition.
func decodeScript(scriptPath string) error {
decodedBytes, permissions, err := decodeScriptFromFile(scriptPath)
if err != nil {
return fmt.Errorf("error decoding script file %q: %w", scriptPath, err)
}
err = os.WriteFile(scriptPath, decodedBytes, permissions)
if err != nil {
return fmt.Errorf("error writing decoded script file %q: %w", scriptPath, err)
}
return nil
}
// decodeScriptFromFile reads the script at scriptPath, decodes it from
// base64, and returns the decoded bytes w/ the permissions to use when re-writing
// or an error.
func decodeScriptFromFile(scriptPath string) ([]byte, os.FileMode, error) {
scriptFile, err := os.Open(scriptPath)
if err != nil {
return nil, 0, fmt.Errorf("error reading from script file %q: %w", scriptPath, err)
}
defer scriptFile.Close()
encoded := bytes.NewBuffer(nil)
if _, err = io.Copy(encoded, scriptFile); err != nil {
return nil, 0, fmt.Errorf("error reading from script file %q: %w", scriptPath, err)
}
fileInfo, err := scriptFile.Stat()
if err != nil {
return nil, 0, fmt.Errorf("error statting script file %q: %w", scriptPath, err)
}
perms := fileInfo.Mode().Perm()
decoded := make([]byte, base64.StdEncoding.DecodedLen(encoded.Len()))
n, err := base64.StdEncoding.Decode(decoded, encoded.Bytes())
if err != nil {
return nil, 0, fmt.Errorf("error decoding script file %q: %w", scriptPath, err)
}
return decoded[0:n], perms, nil
}