aiforge/services/repository/repository.go

// Copyright 2019 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package repository

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strings"

	"code.gitea.io/gitea/models"
	"code.gitea.io/gitea/modules/log"
	"code.gitea.io/gitea/modules/notification"
	repo_module "code.gitea.io/gitea/modules/repository"
	"code.gitea.io/gitea/modules/setting"
	pull_service "code.gitea.io/gitea/services/pull"
)

// CreateRepository creates a repository for the user/organization.
func CreateRepository(doer, owner *models.User, opts models.CreateRepoOptions) (*models.Repository, error) {
	repo, err := repo_module.CreateRepository(doer, owner, opts)
	if err != nil {
		if repo != nil {
			if errDelete := models.DeleteRepository(doer, owner.ID, repo.ID); errDelete != nil {
				log.Error("Rollback deleteRepository: %v", errDelete)
			}
		}
		return nil, err
	}

	notification.NotifyCreateRepository(doer, owner, repo)

	return repo, nil
}

// ForkRepository forks a repository
func ForkRepository(doer, u *models.User, oldRepo *models.Repository, name, desc, alias string) (*models.Repository, error) {
	repo, err := repo_module.ForkRepository(doer, u, oldRepo, name, desc, alias)
	if err != nil {
		if repo != nil {
			if errDelete := models.DeleteRepository(doer, u.ID, repo.ID); errDelete != nil {
				log.Error("Rollback deleteRepository: %v", errDelete)
			}
		}
		return nil, err
	}

	notification.NotifyForkRepository(doer, oldRepo, repo)

	return repo, nil
}

// DeleteRepository deletes a repository for a user or organization.
func DeleteRepository(doer *models.User, repo *models.Repository) error {
	if err := pull_service.CloseRepoBranchesPulls(doer, repo); err != nil {
		log.Error("CloseRepoBranchesPulls failed: %v", err)
	}

	if err := models.DeleteRepository(doer, repo.OwnerID, repo.ID); err != nil {
		return err
	}

	notification.NotifyDeleteRepository(doer, repo)

	return nil
}

// PushCreateRepo creates a repository when a new repository is pushed to an appropriate namespace
func PushCreateRepo(authUser, owner *models.User, repoName string) (*models.Repository, error) {
	if !authUser.IsAdmin {
		if owner.IsOrganization() {
			if ok, err := owner.CanCreateOrgRepo(authUser.ID); err != nil {
				return nil, err
			} else if !ok {
				return nil, fmt.Errorf("cannot push-create repository for org")
			}
		} else if authUser.ID != owner.ID {
			return nil, fmt.Errorf("cannot push-create repository for another user")
		}
	}

	repo, err := CreateRepository(authUser, owner, models.CreateRepoOptions{
		Name:      repoName,
		IsPrivate: true,
	})
	if err != nil {
		return nil, err
	}

	return repo, nil
}

func GetRecommendCourseKeyWords() ([]string, error) {

	url := setting.RecommentRepoAddr + "course_keywords"
	result, err := RecommendFromPromote(url)

	if err != nil {
		return []string{}, err
	}
	return result, err

}

func GetRecommendRepoFromPromote(filename string) ([]map[string]interface{}, error) {
	resultRepo := make([]map[string]interface{}, 0)
	url := setting.RecommentRepoAddr + filename
	result, err := RecommendFromPromote(url)

	if err != nil {

		return resultRepo, err
	}

	//resultRepo := make([]*models.Repository, 0)
	for _, repoName := range result {
		tmpIndex := strings.Index(repoName, "/")
		if tmpIndex == -1 {
			log.Info("error repo name format.")
		} else {
			ownerName := strings.Trim(repoName[0:tmpIndex], " ")
			repoName := strings.Trim(repoName[tmpIndex+1:], " ")
			repo, err := models.GetRepositoryByOwnerAndAlias(ownerName, repoName)
			if err == nil {
				repoMap := make(map[string]interface{})
				repoMap["ID"] = fmt.Sprint(repo.ID)
				repoMap["Name"] = repo.Name
				repoMap["Alias"] = repo.Alias
				if repo.RepoType == models.RepoCourse {
					//Load creator
					repo.GetCreator()
					repoMap["Creator"] = repo.Creator
				}

				repoMap["OwnerName"] = repo.OwnerName
				repoMap["NumStars"] = repo.NumStars
				repoMap["NumForks"] = repo.NumForks
				repoMap["Description"] = repo.Description
				repoMap["NumWatchs"] = repo.NumWatches
				repoMap["Topics"] = repo.Topics
				repoMap["Avatar"] = repo.RelAvatarLink()
				resultRepo = append(resultRepo, repoMap)
			} else {
				log.Info("query repo error," + err.Error())
			}
		}
	}
	return resultRepo, nil
}

func RecommendFromPromote(url string) ([]string, error) {
	resp, err := http.Get(url)
	if err != nil || resp.StatusCode != 200 {
		log.Info("Get organizations url error=" + err.Error())
		return nil, err
	}
	bytes, err := ioutil.ReadAll(resp.Body)
	resp.Body.Close()
	if err != nil {
		log.Info("Get organizations url error=" + err.Error())
		return nil, err
	}

	allLineStr := string(bytes)
	lines := strings.Split(allLineStr, "\n")
	result := make([]string, len(lines))
	for i, line := range lines {
		log.Info("i=" + fmt.Sprint(i) + " line=" + line)
		result[i] = strings.Trim(line, " ")
	}
	return result, nil
}

func CheckPushSizeLimit4Web(repo *models.Repository, fileIds []string) error {
	if err := CheckRepoNumOnceLimit(len(fileIds)); err != nil {
		return err
	}
	totalSize, err := CountUploadFileSizeByIds(fileIds)
	if err != nil {
		return UploadFileInvalidErr{}
	}
	if err := CheckRepoTotalSizeLimit(repo, totalSize); err != nil {
		return err
	}
	return nil
}

func CheckPushSizeLimit4Http(repo *models.Repository, uploadFileSize int64) error {
	if err := CheckRepoOnceTotalSizeLimit(uploadFileSize); err != nil {
		return err
	}
	if err := CheckRepoTotalSizeLimit(repo, uploadFileSize); err != nil {
		return err
	}
	return nil
}

func CheckRepoTotalSizeLimit(repo *models.Repository, uploadFileSize int64) error {
	if repo.Size+uploadFileSize > setting.Repository.RepoMaxSize*1024*1024 {
		return RepoTooLargeErr{}
	}
	return nil
}

func CheckRepoOnceTotalSizeLimit(uploadFileSize int64) error {
	if uploadFileSize > setting.Repository.Upload.TotalMaxSize*1024*1024 {
		return UploadFileTooLargeErr{}
	}
	return nil
}

func CheckRepoNumOnceLimit(uploadFileNum int) error {
	if uploadFileNum > setting.Repository.Upload.MaxFiles {
		return UploadFileTooMuchErr{}
	}
	return nil
}

func CountUploadFileSizeByIds(fileIds []string) (int64, error) {
	if len(fileIds) == 0 {
		return 0, nil
	}
	uploads, err := models.GetUploadsByUUIDs(fileIds)
	if err != nil {
		return 0, fmt.Errorf("CountUploadFileSizeByIds error [uuids: %v]: %v", fileIds, err)
	}
	var totalSize int64
	for _, upload := range uploads {
		size, err := GetUploadFileSize(upload)
		if err != nil {
			return 0, err
		}
		totalSize += size
	}
	return totalSize, nil
}

func GetUploadFileSize(upload *models.Upload) (int64, error) {
	info, err := os.Lstat(upload.LocalPath())

	if err != nil {
		return 0, err
	}
	return info.Size(), nil

}

type RepoTooLargeErr struct {
}

func (RepoTooLargeErr) Error() string {
	return fmt.Sprintf("Repository can not exceed %d MB. Please remove some unnecessary files and try again", setting.Repository.RepoMaxSize)
}

func IsRepoTooLargeErr(err error) bool {
	_, ok := err.(RepoTooLargeErr)
	return ok
}

type UploadFileTooLargeErr struct {
}

func (UploadFileTooLargeErr) Error() string {
	return fmt.Sprintf("Upload files can not exceed %d MB at a time", setting.Repository.Upload.TotalMaxSize)
}

func IsUploadFileTooLargeErr(err error) bool {
	_, ok := err.(UploadFileTooLargeErr)
	return ok
}

type RepoFileTooLargeErr struct {
}

func (RepoFileTooLargeErr) Error() string {
	return "repository file is too large"
}

func IsRepoFileTooLargeErr(err error) bool {
	_, ok := err.(RepoFileTooLargeErr)
	return ok
}

type UploadFileTooMuchErr struct {
}

func (UploadFileTooMuchErr) Error() string {
	return "upload files are too lmuch"
}

func IsUploadFileTooMuchErr(err error) bool {
	_, ok := err.(UploadFileTooMuchErr)
	return ok
}

type UploadFileInvalidErr struct {
}

func (UploadFileInvalidErr) Error() string {
	return "upload files are invalid"
}

func IsUploadFileInvalidErr(err error) bool {
	_, ok := err.(UploadFileInvalidErr)
	return ok
}