收录入口

Author: TruthHun88

commands/command.go

@@ -76,6 +76,7 @@ func RegisterModel() {
 		new(models.ReadCount),
 		new(models.ReadRecord),
 		new(models.Bookmark),
+		new(models.SubmitBooks),
 		models.NewRelateBook(),
 		models.NewAuth(),
 		models.NewBanner(),

controllers/SubmitController.go

@@ -1,5 +1,12 @@
 package controllers
 
+import (
+	"strings"
+
+	"github.com/TruthHun/BookStack/models"
+	"github.com/astaxie/beego"
+)
+
 type SubmitController struct {
 	BaseController
 }
@@ -11,5 +18,29 @@ func (this *SubmitController) Index() {
 }
 
 func (this *SubmitController) Post() {
+	uid := this.Member.MemberId
+	if uid <= 0 {
+		this.JsonResult(1, "请先登录")
+	}
+
+	form := &models.SubmitBooks{}
+	err := this.ParseForm(form)
+	if err != nil {
+		beego.Error(err.Error())
+		this.JsonResult(1, "数据解析失败")
+	}
+
+	lowerURL := strings.ToLower(form.Url)
+	if !(strings.HasPrefix(lowerURL, "https://") || strings.HasPrefix(lowerURL, "http://")) {
+		this.JsonResult(1, "URL链接地址格式不正确")
+	}
 
+	if form.Url == "" || form.Title == "" {
+		this.JsonResult(1, "请填写必填项")
+	}
+	form.Uid = uid
+	if err = form.Add(); err != nil {
+		this.JsonResult(1, err.Error())
+	}
+	this.JsonResult(0, "提交成功，感谢您的分享。")
 }

models/submit_books.go

@@ -1,15 +1,49 @@
 package models
 
-import "time"
+import (
+	"time"
+
+	"github.com/astaxie/beego"
+	"golang.org/x/tools/go/ssa/interp/testdata/src/errors"
+
+	"github.com/TruthHun/gotil/cryptil"
+	"github.com/astaxie/beego/orm"
+)
 
 type SubmitBooks struct {
 	Id        int
-	Uid       int
-	Title     string
-	Url       string
+	Uid       int    `orm:"index"`
+	Title     string `form:"title"`
+	Url       string `form:"url"`
 	UrlMd5    string `orm:"size(32);unique"`
-	Message   string `xorm:"size(512)"`
+	Message   string `orm:"size(512)" form:"message"`
 	Status    bool
 	CreatedAt time.Time
 	UpdatedAt time.Time
 }
+
+func (m *SubmitBooks) Add() (err error) {
+	o := orm.NewOrm()
+	m.UrlMd5 = cryptil.Md5Crypt(m.Url)
+	m.CreatedAt = time.Now()
+	m.UpdatedAt = m.CreatedAt
+	exist := &SubmitBooks{}
+	o.QueryTable(m).Filter("url_md5", m.UrlMd5).One(exist)
+	if exist.Id > 0 { //已存在，直接返回
+		return
+	}
+
+	o.QueryTable(m).Filter("uid", m.Uid).OrderBy("-id").One(exist)
+	if exist.Id > 0 && m.CreatedAt.Unix()-exist.CreatedAt.Unix() < 60 { // TODO: 后期从配置中查询
+		err = errors.New("提交频率过快，1分钟内仅限提交一次")
+		return
+	}
+
+	_, err = o.Insert(m)
+	if err != nil {
+		beego.Error(err.Error())
+		err = errors.New("数据写入失败")
+	}
+
+	return
+}

utils/util.go

@@ -47,7 +47,7 @@ import (
 
 //更多存储类型有待扩展
 const (
-	Version           = "2.0 beta1"
+	Version           = "2.0 beta2"
 	StoreLocal string = "local"
 	StoreOss   string = "oss"
 )

views/submit/index.html

@@ -40,12 +40,12 @@ <h4>收录说明</h4>
                     <div>您在阅读外文开发文档，网速太慢或网络不稳定打不开？</div>
                     <div><strong>我们帮您解决！</strong></div>
                     <div>您只需要提交相关链接，我们来负责整理和排版，给你我以及每一位读者良好的阅读体验。</div>
-                    <div>书栈网内容转载分发，遵循 <strong>CC-BY-NC</strong>（<strong>知识共享-署名-非商用</strong>）协议，在转载分发过程中，注明来源，不作商用，并且保留内容中的打赏二维码和赞助链接。</div>
+                    <div>书栈网内容转载分发，遵循 <strong>CC-BY-NC</strong>（<strong>知识共享-署名-非商用</strong>）协议，在转载分发过程中，注明来源，不作商用，并且保留内容中的打赏二维码和赞助链接等。</div>
                 </div>
                 <h4>书栈开源</h4>
                 <div class="help-block">
                     <strong>书栈网 BookStack</strong>
-                    <p>Gitee: <a href="https://gitee.com/truthhun/bookstack" target="_blank">https://gitee.com/truthhun/bookstack</a></p>
+                    <p>Gitee: <a href="https://gitee.com/truthhun/BookStack" target="_blank">https://gitee.com/truthhun/BookStack</a></p>
                     <p>GitHub: <a href="https://github.com/truthhun/bookstack" target="_blank">https://github.com/truthhun/bookstack</a></p>
                     <div><strong>配套微信小程序 BookChat</strong></div>
                     <p>Gitee: <a href="https://gitee.com/truthhun/BookChat" target="_blank">https://gitee.com/truthhun/BookChat</a></p>
@@ -56,5 +56,22 @@ <h4>书栈开源</h4>
     </div>
 </div>
 {{template "widgets/footer.html" .}}
+<script src="{{$.StaticDomain}}/static/layer/layer.js" type="text/javascript"></script>
+<script>
+    $(function () {
+        $(".submit-book [type=submit]").click(function (e) {
+            e.preventDefault()
+            var form = $(this).parents("form"),action=form.attr("action");
+            $.post(action,form.serialize(),function (res) {
+                layer.msg(res.message);
+                if(res.errcode==0){
+                    setTimeout(function () {
+                        location.href="/"
+                    },1500)
+                }
+            })
+        });
+    })
+</script>
 </body>
 </html>