commit c1b8fd838617c683d999686c16b1d412290a32a6 · veryroundbird.house/core

+19 -3
guard/guard.go
···

       2
       2
        
       

     

       3
       3
        
       import (

     

       4
       4
        
       	"context"

     

       5
       5
       +
       	"errors"

     

       5
       6
        
       	"fmt"

     

       7
       7
       +
       	"io"

     

       6
       8
        
       	"log/slog"

     

       7
       9
        
       	"net/http"

     

       8
       10
        
       	"net/url"

     
···

       43
       45
        
       				Usage: "internal API endpoint",

     

       44
       46
        
       				Value: "http://localhost:5444",

     

       45
       47
        
       			},

     

       48
       48
       +
       			&cli.StringFlag{

     

       49
       49
       +
       				Name:  "motd-file",

     

       50
       50
       +
       				Usage: "path to message of the day file",

     

       51
       51
       +
       				Value: "/home/git/motd",

     

       52
       52
       +
       			},

     

       46
       53
        
       		},

     

       47
       54
        
       	}

     

       48
       55
        
       }

     
···

       54
       61
        
       	gitDir := cmd.String("git-dir")

     

       55
       62
        
       	logPath := cmd.String("log-path")

     

       56
       63
        
       	endpoint := cmd.String("internal-api")

     

       64
       64
       +
       	motdFile := cmd.String("motd-file")

     

       57
       65
        
       

     

       58
       66
        
       	logFile, err := os.OpenFile(logPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)

     

       59
       67
        
       	if err != nil {

     
···

       149
       157
        
       		"fullPath", fullPath,

     

       150
       158
        
       		"client", clientIP)

     

       151
       159
        
       

     

       152
       152
       -
       	if gitCommand == "git-upload-pack" {

     

       153
       153
       -
       		fmt.Fprintf(os.Stderr, "\x02%s\n", "Welcome to this knot!")

     

       160
       160
       +
       	var motdReader io.Reader

     

       161
       161
       +
       	if reader, err := os.Open(motdFile); err != nil {

     

       162
       162
       +
       		if !errors.Is(err, os.ErrNotExist) {

     

       163
       163
       +
       			l.Error("failed to read motd file", "error", err)

     

       164
       164
       +
       		}

     

       165
       165
       +
       		motdReader = strings.NewReader("Welcome to this knot!\n")

     

       154
       166
        
       	} else {

     

       155
       155
       -
       		fmt.Fprintf(os.Stderr, "%s\n", "Welcome to this knot!")

     

       167
       167
       +
       		motdReader = reader

     

       168
       168
       +
       	}

     

       169
       169
       +
       	if gitCommand == "git-upload-pack" {

     

       170
       170
       +
       		io.WriteString(os.Stderr, "\x02")

     

       156
       171
        
       	}

     

       172
       172
       +
       	io.Copy(os.Stderr, motdReader)

     

       157
       173
        
       

     

       158
       174
        
       	gitCmd := exec.Command(gitCommand, fullPath)

     

       159
       175
        
       	gitCmd.Stdout = os.Stdout