mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
116 lines
2.6 KiB
Go
116 lines
2.6 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package git
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"slices"
|
|
"strings"
|
|
|
|
"github.com/hanzoai/git/modules/util"
|
|
)
|
|
|
|
// hookNames is a list of Git server hooks' name that are supported.
|
|
var hookNames = []string{
|
|
"pre-receive",
|
|
"update",
|
|
"post-receive",
|
|
}
|
|
|
|
// ErrNotValidHook error when a git hook is not valid
|
|
var ErrNotValidHook = errors.New("not a valid Git hook")
|
|
|
|
// IsValidHookName returns true if given name is a valid Git hook.
|
|
func IsValidHookName(name string) bool {
|
|
return slices.Contains(hookNames, name)
|
|
}
|
|
|
|
// Hook represents a Git hook.
|
|
type Hook struct {
|
|
name string
|
|
IsActive bool // Indicates whether repository has this hook.
|
|
Content string // Content of hook if it's active.
|
|
Sample string // Sample content from Git.
|
|
path string // Hook file path.
|
|
}
|
|
|
|
// GetHook returns a Git hook by given name and repository.
|
|
func GetHook(repoPath, name string) (*Hook, error) {
|
|
if !IsValidHookName(name) {
|
|
return nil, ErrNotValidHook
|
|
}
|
|
h := &Hook{
|
|
name: name,
|
|
path: filepath.Join(repoPath, filepath.Join("hooks", name+".d", name)),
|
|
}
|
|
if data, err := os.ReadFile(h.path); err == nil {
|
|
h.IsActive = true
|
|
h.Content = string(data)
|
|
return h, nil
|
|
} else if !os.IsNotExist(err) {
|
|
return nil, err
|
|
}
|
|
|
|
samplePath := filepath.Join(repoPath, "hooks", name+".sample")
|
|
if data, err := os.ReadFile(samplePath); err == nil {
|
|
h.Sample = string(data)
|
|
}
|
|
return h, nil
|
|
}
|
|
|
|
// Name return the name of the hook
|
|
func (h *Hook) Name() string {
|
|
return h.name
|
|
}
|
|
|
|
// Update updates hook settings.
|
|
func (h *Hook) Update() error {
|
|
if len(strings.TrimSpace(h.Content)) == 0 {
|
|
exist, err := util.IsExist(h.path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if exist {
|
|
err := util.Remove(h.path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
h.IsActive = false
|
|
return nil
|
|
}
|
|
d := filepath.Dir(h.path)
|
|
if err := os.MkdirAll(d, os.ModePerm); err != nil {
|
|
return err
|
|
}
|
|
|
|
err := os.WriteFile(h.path, []byte(strings.ReplaceAll(h.Content, "\r", "")), os.ModePerm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
h.IsActive = true
|
|
return nil
|
|
}
|
|
|
|
// ListHooks returns a list of Git hooks of given repository.
|
|
func ListHooks(repoPath string) (_ []*Hook, err error) {
|
|
exist, err := util.IsDir(filepath.Join(repoPath, "hooks"))
|
|
if err != nil {
|
|
return nil, err
|
|
} else if !exist {
|
|
return nil, errors.New("hooks path does not exist")
|
|
}
|
|
|
|
hooks := make([]*Hook, len(hookNames))
|
|
for i, name := range hookNames {
|
|
hooks[i], err = GetHook(repoPath, name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return hooks, nil
|
|
}
|