-
-
Notifications
You must be signed in to change notification settings - Fork 111
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
213 additions
and
86 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,156 @@ | ||
| package actions | ||
|
|
||
| import ( | ||
| "github.com/rs/zerolog/log" | ||
| "github.com/thomiceli/opengist/internal/config" | ||
| "github.com/thomiceli/opengist/internal/db" | ||
| "github.com/thomiceli/opengist/internal/git" | ||
| "os" | ||
| "path/filepath" | ||
| "strings" | ||
| "sync" | ||
| ) | ||
|
|
||
| type ActionStatus struct { | ||
| Running bool | ||
| } | ||
|
|
||
| const ( | ||
| SyncReposFromFS = iota | ||
| SyncReposFromDB = iota | ||
| GitGcRepos = iota | ||
| SyncGistPreviews = iota | ||
| ResetHooks = iota | ||
| ) | ||
|
|
||
| var ( | ||
| mutex sync.Mutex | ||
| actions = make(map[int]ActionStatus) | ||
| ) | ||
|
|
||
| func updateActionStatus(actionType int, running bool) { | ||
| actions[actionType] = ActionStatus{ | ||
| Running: running, | ||
| } | ||
| } | ||
|
|
||
| func IsRunning(actionType int) bool { | ||
| mutex.Lock() | ||
| defer mutex.Unlock() | ||
| return actions[actionType].Running | ||
| } | ||
|
|
||
| func Run(actionType int) { | ||
| mutex.Lock() | ||
|
|
||
| if actions[actionType].Running { | ||
| mutex.Unlock() | ||
| return | ||
| } | ||
|
|
||
| updateActionStatus(actionType, true) | ||
| mutex.Unlock() | ||
|
|
||
| defer func() { | ||
| mutex.Lock() | ||
| updateActionStatus(actionType, false) | ||
| mutex.Unlock() | ||
| }() | ||
|
|
||
| var functionToRun func() | ||
| switch actionType { | ||
| case SyncReposFromFS: | ||
| functionToRun = syncReposFromFS | ||
| case SyncReposFromDB: | ||
| functionToRun = syncReposFromDB | ||
| case GitGcRepos: | ||
| functionToRun = gitGcRepos | ||
| case SyncGistPreviews: | ||
| functionToRun = syncGistPreviews | ||
| case ResetHooks: | ||
| functionToRun = resetHooks | ||
| default: | ||
| panic("unhandled default case") | ||
| } | ||
|
|
||
| functionToRun() | ||
| } | ||
|
|
||
| func syncReposFromFS() { | ||
| log.Info().Msg("Syncing repositories from filesystem...") | ||
| gists, err := db.GetAllGistsRows() | ||
| if err != nil { | ||
| log.Error().Err(err).Msg("Cannot get gists") | ||
| return | ||
| } | ||
| for _, gist := range gists { | ||
| // if repository does not exist, delete gist from database | ||
| if _, err := os.Stat(git.RepositoryPath(gist.User.Username, gist.Uuid)); err != nil && !os.IsExist(err) { | ||
| if err2 := gist.Delete(); err2 != nil { | ||
| log.Error().Err(err2).Msgf("Cannot delete gist %d", gist.ID) | ||
| return | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func syncReposFromDB() { | ||
| log.Info().Msg("Syncing repositories from database...") | ||
| entries, err := filepath.Glob(filepath.Join(config.GetHomeDir(), "repos", "*", "*")) | ||
| if err != nil { | ||
| log.Error().Err(err).Msg("Cannot read repos directories") | ||
| return | ||
| } | ||
|
|
||
| for _, e := range entries { | ||
| path := strings.Split(e, string(os.PathSeparator)) | ||
| gist, _ := db.GetGist(path[len(path)-2], path[len(path)-1]) | ||
|
|
||
| if gist.ID == 0 { | ||
| if err := git.DeleteRepository(path[len(path)-2], path[len(path)-1]); err != nil { | ||
| log.Error().Err(err).Msgf("Cannot delete repository %s/%s", path[len(path)-2], path[len(path)-1]) | ||
| return | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func gitGcRepos() { | ||
| log.Info().Msg("Garbage collecting all repositories...") | ||
| if err := git.GcRepos(); err != nil { | ||
| log.Error().Err(err).Msg("Error garbage collecting repositories") | ||
| } | ||
| } | ||
|
|
||
| func syncGistPreviews() { | ||
| log.Info().Msg("Syncing all Gist previews...") | ||
|
|
||
| gists, err := db.GetAllGistsRows() | ||
| if err != nil { | ||
| log.Error().Err(err).Msg("Cannot get gists") | ||
| return | ||
| } | ||
| for _, gist := range gists { | ||
| if err = gist.UpdatePreviewAndCount(false); err != nil { | ||
| log.Error().Err(err).Msgf("Cannot update preview and count for gist %d", gist.ID) | ||
| return | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func resetHooks() { | ||
| log.Info().Msg("Resetting Git server hooks for all repositories...") | ||
| entries, err := filepath.Glob(filepath.Join(config.GetHomeDir(), "repos", "*", "*")) | ||
| if err != nil { | ||
| log.Error().Err(err).Msg("Cannot read repos directories") | ||
| return | ||
| } | ||
|
|
||
| for _, e := range entries { | ||
| path := strings.Split(e, string(os.PathSeparator)) | ||
| if err := git.CreateDotGitFiles(path[len(path)-2], path[len(path)-1]); err != nil { | ||
| log.Error().Err(err).Msgf("Cannot reset hooks for repository %s/%s", path[len(path)-2], path[len(path)-1]) | ||
| return | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.