项目logo自动转换

polaris1119 · polaris1119 · commit 0dd5b50a90be · 2015-10-18T12:51:47.000+08:00
diff --git a/websites/code/studygolang/src/service/project.go b/websites/code/studygolang/src/service/project.go
@@ -256,6 +256,9 @@ func ParseProjectList(pUrl string) error {
 
 const OsChinaDomain = "http://www.oschina.net"
 
+// ProjectLogoPrefix 开源项目 logo 前缀
+const ProjectLogoPrefix = "plogo"
+
 var PresetUsernames = []string{"polaris", "blov", "agolangf", "xuanbao"}
 
 // ParseOneProject 处理单个 project
@@ -295,6 +298,15 @@ func ParseOneProject(projectUrl string) error {
 	logoSelection := doc.Find(".Project .PN img")
 	if logoSelection.AttrOr("title", "") != "" {
 		project.Logo = logoSelection.AttrOr("src", "")
+
+		if !strings.HasPrefix(project.Logo, "http") {
+			project.Logo = OsChinaDomain + project.Logo
+		}
+
+		project.Logo, err = UploadUrlFile(project.Logo, ProjectLogoPrefix)
+		if err != nil {
+			logger.Errorln("project logo upload error:", err)
+		}
 	}
 
 	// 获取项目相关链接
diff --git a/websites/code/studygolang/src/service/uploader.go b/websites/code/studygolang/src/service/uploader.go
@@ -2,13 +2,18 @@ package service
 
 import (
 	"encoding/json"
+	"errors"
 	gio "io"
+	"net/http"
+	"net/url"
+	"strings"
 
 	"config"
+	"logger"
+
 	"github.com/qiniu/api.v6/conf"
 	"github.com/qiniu/api.v6/io"
 	"github.com/qiniu/api.v6/rs"
-	"logger"
 )
 
 var uptoken string
@@ -101,3 +106,31 @@ func UploadMemoryFile(r gio.Reader, key string) (err error) {
 
 	return
 }
+
+// UploadUrlFile 将外站图片URL转为本站，如果失败，返回原图
+func UploadUrlFile(origUrl, prefix string) (string, error) {
+	if origUrl == "" || strings.Contains(origUrl, "studygolang") {
+		return origUrl, errors.New("origin image is empty or is studygolang.com")
+	}
+
+	resp, err := http.Get(origUrl)
+	if err != nil {
+		return origUrl, err
+	}
+	defer resp.Body.Close()
+
+	objUrl, err := url.Parse(origUrl)
+	if err != nil {
+		return origUrl, err
+	}
+
+	filename := objUrl.Path[strings.LastIndex(objUrl.Path, "/"):]
+	uri := prefix + filename
+
+	err = UploadMemoryFile(resp.Body, uri)
+	if err != nil {
+		return origUrl, err
+	}
+
+	return "http://studygolang.qiniudn.com/" + uri, nil
+}
