2014-04-10 22:20:58 +04:00
// 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 base
import (
"bytes"
"container/list"
"encoding/json"
2014-09-17 08:03:03 +04:00
"errors"
2014-04-10 22:20:58 +04:00
"fmt"
"html/template"
2014-05-22 05:37:13 +04:00
"runtime"
2014-04-10 22:20:58 +04:00
"strings"
"time"
2014-05-26 04:11:25 +04:00
2014-09-17 22:22:51 +04:00
"github.com/gogits/gogs/modules/mahonia"
2014-05-26 04:11:25 +04:00
"github.com/gogits/gogs/modules/setting"
2014-09-17 08:03:03 +04:00
"github.com/saintfish/chardet"
2014-04-10 22:20:58 +04:00
)
func Str2html ( raw string ) template . HTML {
return template . HTML ( raw )
}
func Range ( l int ) [ ] int {
return make ( [ ] int , l )
}
func List ( l * list . List ) chan interface { } {
e := l . Front ( )
c := make ( chan interface { } )
go func ( ) {
for e != nil {
c <- e . Value
e = e . Next ( )
}
close ( c )
} ( )
return c
}
func ShortSha ( sha1 string ) string {
if len ( sha1 ) == 40 {
return sha1 [ : 10 ]
}
return sha1
}
2014-09-17 08:03:03 +04:00
func ToUtf8WithErr ( content [ ] byte ) ( error , string ) {
detector := chardet . NewTextDetector ( )
result , err := detector . DetectBest ( content )
if err != nil {
return err , ""
}
if result . Charset == "utf8" {
return nil , string ( content )
}
decoder := mahonia . NewDecoder ( result . Charset )
if decoder != nil {
return nil , decoder . ConvertString ( string ( content ) )
}
return errors . New ( "unknow char decoder" ) , string ( content )
}
func ToUtf8 ( content string ) string {
_ , res := ToUtf8WithErr ( [ ] byte ( content ) )
return res
}
2014-04-10 22:20:58 +04:00
var mailDomains = map [ string ] string {
"gmail.com" : "gmail.com" ,
}
var TemplateFuncs template . FuncMap = map [ string ] interface { } {
2014-05-22 05:37:13 +04:00
"GoVer" : func ( ) string {
2014-07-26 08:24:27 +04:00
return strings . Title ( runtime . Version ( ) )
2014-05-22 05:37:13 +04:00
} ,
2014-04-10 22:20:58 +04:00
"AppName" : func ( ) string {
2014-05-26 04:11:25 +04:00
return setting . AppName
2014-04-10 22:20:58 +04:00
} ,
2014-09-20 04:11:34 +04:00
"AppSubUrl" : func ( ) string {
return setting . AppSubUrl
2014-09-14 21:35:22 +04:00
} ,
2014-04-10 22:20:58 +04:00
"AppVer" : func ( ) string {
2014-05-26 04:11:25 +04:00
return setting . AppVer
2014-04-10 22:20:58 +04:00
} ,
"AppDomain" : func ( ) string {
2014-05-26 04:11:25 +04:00
return setting . Domain
2014-04-10 22:20:58 +04:00
} ,
2014-04-27 11:05:13 +04:00
"CdnMode" : func ( ) bool {
2014-05-26 04:11:25 +04:00
return setting . ProdMode && ! setting . OfflineMode
2014-04-10 22:20:58 +04:00
} ,
"LoadTimes" : func ( startTime time . Time ) string {
return fmt . Sprint ( time . Since ( startTime ) . Nanoseconds ( ) / 1e6 ) + "ms"
} ,
"AvatarLink" : AvatarLink ,
2014-07-26 08:24:27 +04:00
"str2html" : Str2html , // TODO: Legacy
"Str2html" : Str2html ,
2014-04-10 22:20:58 +04:00
"TimeSince" : TimeSince ,
"FileSize" : FileSize ,
"Subtract" : Subtract ,
2014-04-11 20:14:11 +04:00
"Add" : func ( a , b int ) int {
return a + b
} ,
2014-04-10 22:20:58 +04:00
"ActionIcon" : ActionIcon ,
"ActionDesc" : ActionDesc ,
"DateFormat" : DateFormat ,
"List" : List ,
"Mail2Domain" : func ( mail string ) string {
if ! strings . Contains ( mail , "@" ) {
return "try.gogits.org"
}
suffix := strings . SplitN ( mail , "@" , 2 ) [ 1 ]
domain , ok := mailDomains [ suffix ]
if ! ok {
return "mail." + suffix
}
return domain
} ,
"SubStr" : func ( str string , start , length int ) string {
return str [ start : start + length ]
} ,
"DiffTypeToStr" : DiffTypeToStr ,
"DiffLineTypeToStr" : DiffLineTypeToStr ,
"ShortSha" : ShortSha ,
2014-07-26 08:24:27 +04:00
"Md5" : EncodeMd5 ,
"ActionContent2Commits" : ActionContent2Commits ,
"Oauth2Icon" : Oauth2Icon ,
"Oauth2Name" : Oauth2Name ,
2014-09-17 08:03:03 +04:00
"ToUtf8" : ToUtf8 ,
2014-04-10 22:20:58 +04:00
}
type Actioner interface {
GetOpType ( ) int
GetActUserName ( ) string
GetActEmail ( ) string
2014-05-09 10:42:50 +04:00
GetRepoUserName ( ) string
2014-04-10 22:20:58 +04:00
GetRepoName ( ) string
GetBranch ( ) string
GetContent ( ) string
}
// ActionIcon accepts a int that represents action operation type
// and returns a icon class name.
func ActionIcon ( opType int ) string {
switch opType {
2014-09-26 00:36:19 +04:00
case 1 , 8 : // Create, transfer repository.
2014-07-26 08:24:27 +04:00
return "repo"
2014-04-14 05:00:12 +04:00
case 5 , 9 : // Commit repository.
2014-07-26 08:24:27 +04:00
return "git-commit"
2014-04-10 22:20:58 +04:00
case 6 : // Create issue.
2014-07-26 08:24:27 +04:00
return "issue-opened"
2014-05-06 21:47:47 +04:00
case 10 : // Comment issue.
return "comment"
2014-04-10 22:20:58 +04:00
default :
return "invalid type"
}
}
2014-09-26 00:36:19 +04:00
// FIXME: Legacy
2014-04-10 22:20:58 +04:00
const (
2014-09-14 21:35:22 +04:00
TPL_CREATE_REPO = ` <a href="%s/user/%s">%s</a> created repository <a href="%s">%s</a> `
TPL_COMMIT_REPO = ` <a href="%s/user/%s">%s</a> pushed to <a href="%s/src/%s">%s</a> at <a href="%s">%s</a>%s `
TPL_COMMIT_REPO_LI = ` <div><img src="%s?s=16" alt="user-avatar"/> <a href="%s/commit/%s" rel="nofollow">%s</a> %s</div> `
TPL_CREATE_ISSUE = ` < a href = "%s/user/%s" > % s < / a > opened issue < a href = "%s/issues/%s" > % s # % s < / a >
2014-04-10 22:20:58 +04:00
< div > < img src = "%s?s=16" alt = "user-avatar" / > % s < / div > `
2014-09-14 21:35:22 +04:00
TPL_TRANSFER_REPO = ` <a href="%s/user/%s">%s</a> transfered repository <code>%s</code> to <a href="%s">%s</a> `
TPL_PUSH_TAG = ` <a href="%s/user/%s">%s</a> pushed tag <a href="%s/src/%s" rel="nofollow">%s</a> at <a href="%s">%s</a> `
TPL_COMMENT_ISSUE = ` < a href = "%s/user/%s" > % s < / a > commented on issue < a href = "%s/issues/%s" > % s # % s < / a >
2014-05-06 21:47:47 +04:00
< div > < img src = "%s?s=16" alt = "user-avatar" / > % s < / div > `
2014-04-10 22:20:58 +04:00
)
type PushCommit struct {
Sha1 string
Message string
AuthorEmail string
AuthorName string
}
type PushCommits struct {
Len int
Commits [ ] * PushCommit
}
2014-07-26 08:24:27 +04:00
func ActionContent2Commits ( act Actioner ) * PushCommits {
var push * PushCommits
if err := json . Unmarshal ( [ ] byte ( act . GetContent ( ) ) , & push ) ; err != nil {
return nil
}
return push
}
2014-09-26 00:36:19 +04:00
// FIXME: Legacy
2014-04-10 22:20:58 +04:00
// ActionDesc accepts int that represents action operation type
// and returns the description.
func ActionDesc ( act Actioner ) string {
actUserName := act . GetActUserName ( )
email := act . GetActEmail ( )
2014-05-09 10:42:50 +04:00
repoUserName := act . GetRepoUserName ( )
2014-04-10 22:20:58 +04:00
repoName := act . GetRepoName ( )
2014-05-09 10:42:50 +04:00
repoLink := repoUserName + "/" + repoName
2014-04-10 22:20:58 +04:00
branch := act . GetBranch ( )
content := act . GetContent ( )
switch act . GetOpType ( ) {
case 1 : // Create repository.
2014-09-20 04:11:34 +04:00
return fmt . Sprintf ( TPL_CREATE_REPO , setting . AppSubUrl , actUserName , actUserName , repoLink , repoName )
2014-04-10 22:20:58 +04:00
case 5 : // Commit repository.
var push * PushCommits
if err := json . Unmarshal ( [ ] byte ( content ) , & push ) ; err != nil {
return err . Error ( )
}
buf := bytes . NewBuffer ( [ ] byte ( "\n" ) )
for _ , commit := range push . Commits {
buf . WriteString ( fmt . Sprintf ( TPL_COMMIT_REPO_LI , AvatarLink ( commit . AuthorEmail ) , repoLink , commit . Sha1 , commit . Sha1 [ : 7 ] , commit . Message ) + "\n" )
}
if push . Len > 3 {
2014-09-14 21:35:22 +04:00
buf . WriteString ( fmt . Sprintf ( ` <div><a href=" {{ AppRootSubUrl }} /%s/%s/commits/%s" rel="nofollow">%d other commits >></a></div> ` , actUserName , repoName , branch , push . Len ) )
2014-04-10 22:20:58 +04:00
}
2014-09-20 04:11:34 +04:00
return fmt . Sprintf ( TPL_COMMIT_REPO , setting . AppSubUrl , actUserName , actUserName , repoLink , branch , branch , repoLink , repoLink ,
2014-04-10 22:20:58 +04:00
buf . String ( ) )
case 6 : // Create issue.
infos := strings . SplitN ( content , "|" , 2 )
2014-09-20 04:11:34 +04:00
return fmt . Sprintf ( TPL_CREATE_ISSUE , setting . AppSubUrl , actUserName , actUserName , repoLink , infos [ 0 ] , repoLink , infos [ 0 ] ,
2014-04-10 22:20:58 +04:00
AvatarLink ( email ) , infos [ 1 ] )
case 8 : // Transfer repository.
newRepoLink := content + "/" + repoName
2014-09-20 04:11:34 +04:00
return fmt . Sprintf ( TPL_TRANSFER_REPO , setting . AppSubUrl , actUserName , actUserName , repoLink , newRepoLink , newRepoLink )
2014-04-14 05:00:12 +04:00
case 9 : // Push tag.
2014-09-20 04:11:34 +04:00
return fmt . Sprintf ( TPL_PUSH_TAG , setting . AppSubUrl , actUserName , actUserName , repoLink , branch , branch , repoLink , repoLink )
2014-05-06 21:47:47 +04:00
case 10 : // Comment issue.
infos := strings . SplitN ( content , "|" , 2 )
2014-09-20 04:11:34 +04:00
return fmt . Sprintf ( TPL_COMMENT_ISSUE , setting . AppSubUrl , actUserName , actUserName , repoLink , infos [ 0 ] , repoLink , infos [ 0 ] ,
2014-05-06 21:47:47 +04:00
AvatarLink ( email ) , infos [ 1 ] )
2014-04-10 22:20:58 +04:00
default :
return "invalid type"
}
}
func DiffTypeToStr ( diffType int ) string {
diffTypes := map [ int ] string {
1 : "add" , 2 : "modify" , 3 : "del" ,
}
return diffTypes [ diffType ]
}
func DiffLineTypeToStr ( diffType int ) string {
switch diffType {
case 2 :
return "add"
case 3 :
return "del"
case 4 :
return "tag"
}
return "same"
}
2014-04-14 05:00:12 +04:00
func Oauth2Icon ( t int ) string {
switch t {
case 1 :
return "fa-github-square"
case 2 :
return "fa-google-plus-square"
case 3 :
return "fa-twitter-square"
case 4 :
2014-08-10 05:01:29 +04:00
return "fa-qq"
2014-04-14 05:00:12 +04:00
case 5 :
return "fa-weibo"
}
return ""
}
2014-05-06 00:21:43 +04:00
func Oauth2Name ( t int ) string {
switch t {
case 1 :
return "GitHub"
case 2 :
2014-08-10 05:01:29 +04:00
return "Google+"
2014-05-06 00:21:43 +04:00
case 3 :
return "Twitter"
case 4 :
2014-08-10 05:01:29 +04:00
return "腾讯 QQ"
2014-05-06 00:21:43 +04:00
case 5 :
return "Weibo"
}
return ""
}