mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-11-13 04:32:43 +00:00
116 lines
3 KiB
Go
116 lines
3 KiB
Go
// Copyright 2018 Drone.IO Inc.
|
|
//
|
|
// 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 shared
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/woodpecker-ci/woodpecker/server/model"
|
|
"github.com/woodpecker-ci/woodpecker/server/remote"
|
|
"github.com/woodpecker-ci/woodpecker/server/store"
|
|
)
|
|
|
|
// UserSyncer syncs the user repository and permissions.
|
|
type UserSyncer interface {
|
|
Sync(ctx context.Context, user *model.User) error
|
|
}
|
|
|
|
type Syncer struct {
|
|
Remote remote.Remote
|
|
Store store.Store
|
|
Perms model.PermStore
|
|
Match FilterFunc
|
|
}
|
|
|
|
// FilterFunc can be used to filter which repositories are
|
|
// synchronized with the local datastore.
|
|
type FilterFunc func(*model.Repo) bool
|
|
|
|
// NamespaceFilter
|
|
func NamespaceFilter(namespaces map[string]bool) FilterFunc {
|
|
if namespaces == nil || len(namespaces) == 0 {
|
|
return noopFilter
|
|
}
|
|
return func(repo *model.Repo) bool {
|
|
if namespaces[repo.Owner] {
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
}
|
|
}
|
|
|
|
// noopFilter is a filter function that always returns true.
|
|
func noopFilter(*model.Repo) bool {
|
|
return true
|
|
}
|
|
|
|
// SetFilter sets the filter function.
|
|
func (s *Syncer) SetFilter(fn FilterFunc) {
|
|
s.Match = fn
|
|
}
|
|
|
|
func (s *Syncer) Sync(ctx context.Context, user *model.User) error {
|
|
unix := time.Now().Unix() - (3601) // force immediate expiration. note 1 hour expiration is hard coded at the moment
|
|
repos, err := s.Remote.Repos(ctx, user)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var remoteRepos []*model.Repo
|
|
var perms []*model.Perm
|
|
|
|
for _, repo := range repos {
|
|
if s.Match(repo) {
|
|
remoteRepos = append(remoteRepos, repo)
|
|
perm := model.Perm{
|
|
UserID: user.ID,
|
|
Repo: repo.FullName,
|
|
Pull: true,
|
|
Synced: unix,
|
|
}
|
|
remotePerm, err := s.Remote.Perm(ctx, user, repo.Owner, repo.Name)
|
|
if err == nil && remotePerm != nil {
|
|
perm.Push = remotePerm.Push
|
|
perm.Admin = remotePerm.Admin
|
|
}
|
|
perms = append(perms, &perm)
|
|
}
|
|
}
|
|
|
|
err = s.Store.RepoBatch(remoteRepos)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = s.Store.PermBatch(perms)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// this is here as a precaution. I want to make sure that if an api
|
|
// call to the version control system fails and (for some reason) returns
|
|
// an empty list, we don't wipe out the user repository permissions.
|
|
//
|
|
// the side-effect of this code is that a user with 1 repository whose
|
|
// access is removed will still display in the feed, but they will not
|
|
// be able to access the actual repository data.
|
|
if len(repos) == 0 {
|
|
return nil
|
|
}
|
|
|
|
return s.Perms.PermFlush(user, unix)
|
|
}
|