bench-forgejo/models/admin.go

144 lines
3.6 KiB
Go
Raw Normal View History

2014-10-09 03:59:18 +05:30
// Copyright 2014 The Gogs 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 models
import (
"fmt"
"os"
"os/exec"
"strings"
2014-10-09 03:59:18 +05:30
"time"
"github.com/Unknwon/com"
"github.com/go-xorm/xorm"
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
2014-10-09 03:59:18 +05:30
)
2016-11-25 02:24:00 +05:30
//NoticeType describes the notice type
2014-10-09 03:59:18 +05:30
type NoticeType int
const (
2016-11-25 02:24:00 +05:30
//NoticeRepository type
2016-11-07 21:54:59 +05:30
NoticeRepository NoticeType = iota + 1
2014-10-09 03:59:18 +05:30
)
// Notice represents a system notice for admin.
type Notice struct {
ID int64 `xorm:"pk autoincr"`
2014-10-09 03:59:18 +05:30
Type NoticeType
Description string `xorm:"TEXT"`
Created time.Time `xorm:"-"`
2017-01-06 20:44:33 +05:30
CreatedUnix int64 `xorm:"INDEX"`
}
2016-11-25 02:24:00 +05:30
// BeforeInsert is invoked from XORM before inserting an object of this type.
func (n *Notice) BeforeInsert() {
n.CreatedUnix = time.Now().Unix()
}
2016-11-25 02:24:00 +05:30
// AfterSet is invoked from XORM after setting the value of a field of this object.
func (n *Notice) AfterSet(colName string, _ xorm.Cell) {
switch colName {
case "created_unix":
n.Created = time.Unix(n.CreatedUnix, 0).Local()
}
2014-10-09 03:59:18 +05:30
}
// TrStr returns a translation format string.
func (n *Notice) TrStr() string {
return "admin.notices.type_" + com.ToStr(n.Type)
}
// CreateNotice creates new system notice.
func CreateNotice(tp NoticeType, desc string) error {
return createNotice(x, tp, desc)
}
func createNotice(e Engine, tp NoticeType, desc string) error {
2014-10-09 03:59:18 +05:30
n := &Notice{
Type: tp,
Description: desc,
}
_, err := e.Insert(n)
2014-10-09 03:59:18 +05:30
return err
}
2016-11-07 21:54:59 +05:30
// CreateRepositoryNotice creates new system notice with type NoticeRepository.
2014-10-09 03:59:18 +05:30
func CreateRepositoryNotice(desc string) error {
return createNotice(x, NoticeRepository, desc)
2014-10-09 03:59:18 +05:30
}
// RemoveAllWithNotice removes all directories in given path and
// creates a system notice when error occurs.
func RemoveAllWithNotice(title, path string) {
removeAllWithNotice(x, title, path)
}
func removeAllWithNotice(e Engine, title, path string) {
var err error
2016-05-24 01:54:40 +05:30
// workaround for Go not being able to remove read-only files/folders: https://github.com/golang/go/issues/9606
// this bug should be fixed on Go 1.7, so the workaround should be removed when Gogs don't support Go 1.6 anymore:
// https://github.com/golang/go/commit/2ffb3e5d905b5622204d199128dec06cefd57790
if setting.IsWindows {
2016-05-24 01:54:40 +05:30
// converting "/" to "\" in path on Windows
path = strings.Replace(path, "/", "\\", -1)
err = exec.Command("cmd", "/C", "rmdir", "/S", "/Q", path).Run()
} else {
err = os.RemoveAll(path)
}
if err != nil {
desc := fmt.Sprintf("%s [%s]: %v", title, path, err)
log.Warn(desc)
if err = createNotice(e, NoticeRepository, desc); err != nil {
log.Error(4, "CreateRepositoryNotice: %v", err)
}
}
}
2014-10-09 03:59:18 +05:30
// CountNotices returns number of notices.
func CountNotices() int64 {
count, _ := x.Count(new(Notice))
return count
}
2017-01-09 23:56:05 +05:30
// Notices returns notices in given page.
func Notices(page, pageSize int) ([]*Notice, error) {
notices := make([]*Notice, 0, pageSize)
2016-11-10 20:46:32 +05:30
return notices, x.
Limit(pageSize, (page-1)*pageSize).
Desc("id").
Find(&notices)
2014-10-09 03:59:18 +05:30
}
// DeleteNotice deletes a system notice by given ID.
func DeleteNotice(id int64) error {
_, err := x.Id(id).Delete(new(Notice))
return err
}
// DeleteNotices deletes all notices with ID from start to end (inclusive).
func DeleteNotices(start, end int64) error {
sess := x.Where("id >= ?", start)
if end > 0 {
sess.And("id <= ?", end)
}
_, err := sess.Delete(new(Notice))
return err
}
// DeleteNoticesByIDs deletes notices by given IDs.
func DeleteNoticesByIDs(ids []int64) error {
if len(ids) == 0 {
return nil
}
2016-11-10 20:46:32 +05:30
_, err := x.
2016-11-12 13:59:18 +05:30
In("id", ids).
2016-11-10 20:46:32 +05:30
Delete(new(Notice))
return err
}