podman
84 строки · 1.3 Кб
1//go:build darwin
2
3package main
4
5import (
6"fmt"
7"io"
8"io/fs"
9"os"
10"time"
11
12"github.com/spf13/cobra"
13)
14
15const (
16trigger = "GO\n"
17fail = "NO"
18success = "OK"
19)
20
21var serviceCmd = &cobra.Command{
22Use: "service",
23Short: "Service requests",
24Long: "Service requests",
25PreRun: silentUsage,
26Run: serviceRun,
27Hidden: true,
28}
29
30func init() {
31rootCmd.AddCommand(serviceCmd)
32}
33
34func serviceRun(cmd *cobra.Command, args []string) {
35info, err := os.Stdin.Stat()
36if err != nil || info.Mode()&fs.ModeSocket == 0 {
37fmt.Fprintln(os.Stderr, "This is an internal command that is not intended for standard terminal usage")
38os.Exit(1)
39}
40
41os.Exit(service())
42}
43
44func service() int {
45defer os.Stdout.Close()
46defer os.Stdin.Close()
47defer os.Stderr.Close()
48if len(os.Args) < 3 {
49fmt.Print(fail)
50return 1
51}
52target := os.Args[2]
53
54request := make(chan bool)
55go func() {
56buf := make([]byte, 3)
57_, err := io.ReadFull(os.Stdin, buf)
58request <- err == nil && string(buf) == trigger
59}()
60
61valid := false
62select {
63case valid = <-request:
64case <-time.After(5 * time.Second):
65}
66
67if !valid {
68fmt.Println(fail)
69return 2
70}
71
72err := os.Remove(dockerSock)
73if err == nil || os.IsNotExist(err) {
74err = os.Symlink(target, dockerSock)
75}
76
77if err != nil {
78fmt.Print(fail)
79return 3
80}
81
82fmt.Print(success)
83return 0
84}
85