支持限定游客可阅读的章节数量

Author: TruthHun88

controllers/DocumentController.go

@@ -228,6 +228,7 @@ func (this *DocumentController) indexWithPassword() {
 
 //阅读文档.
 func (this *DocumentController) Read() {
+
 	identify := this.Ctx.Input.Param(":key")
 	token := this.GetString("token")
 	id := this.GetString(":id")
@@ -272,6 +273,13 @@ func (this *DocumentController) Read() {
 		this.Abort404(bookName, bookLink)
 	}
 
+	// 是否允许阅读
+	isAllowRead := true
+	percent := 100
+	if this.Member.MemberId == 0 {
+		isAllowRead, percent = doc.IsAllowReadChapter(doc.BookId, doc.DocumentId)
+	}
+
 	bodyText := ""
 	authHTTPS := strings.ToLower(models.GetOptionValue("AUTO_HTTPS", "false")) == "true"
 	if doc.Release != "" {
@@ -387,16 +395,22 @@ func (this *DocumentController) Read() {
 	models.NewBookCounter().Increase(bookResult.BookId, true)
 	comments, _ := models.NewComments().Comments(1, 1000, models.CommentOpt{DocId: doc.DocumentId, Status: []int{1}})
 
+	if !isAllowRead {
+		doc.Release = ""
+	}
+
 	if this.IsAjax() {
 		var data struct {
-			Id        int                         `json:"doc_id"`
-			DocTitle  string                      `json:"doc_title"`
-			Body      string                      `json:"body"`
-			Title     string                      `json:"title"`
-			Bookmark  bool                        `json:"bookmark"` //是否已经添加了书签
-			View      int                         `json:"view"`
-			UpdatedAt string                      `json:"updated_at"`
-			Comments  []models.BookCommentsResult `json:"comments"`
+			Id          int                         `json:"doc_id"`
+			DocTitle    string                      `json:"doc_title"`
+			Body        string                      `json:"body"`
+			Title       string                      `json:"title"`
+			Bookmark    bool                        `json:"bookmark"` //是否已经添加了书签
+			View        int                         `json:"view"`
+			UpdatedAt   string                      `json:"updated_at"`
+			Comments    []models.BookCommentsResult `json:"comments"`
+			IsAllowRead bool                        `json:"is_allow_read"`
+			Percent     int                         `json:"percent"`
 		}
 		data.DocTitle = doc.DocumentName
 		data.Body = doc.Release
@@ -405,9 +419,9 @@ func (this *DocumentController) Read() {
 		data.Bookmark = existBookmark
 		data.View = doc.Vcnt
 		data.UpdatedAt = doc.ModifyTime.Format("2006-01-02 15:04:05")
-		//data.Body = doc.Markdown
 		data.Comments = comments
-
+		data.IsAllowRead = isAllowRead
+		data.Percent = percent
 		this.JsonResult(0, "ok", data)
 	}
 
@@ -474,6 +488,8 @@ func (this *DocumentController) Read() {
 	this.Data["UpdatedAt"] = doc.ModifyTime.Format("2006-01-02 15:04:05")
 	this.Data["ExistWeCode"] = strings.TrimSpace(models.GetOptionValue("DOWNLOAD_WECODE", "")) != ""
 	this.Data["Comments"] = comments
+	this.Data["IsAllowRead"] = isAllowRead
+	this.Data["Percent"] = percent
 }
 
 //编辑文档.

models/document.go

@@ -2,6 +2,7 @@ package models
 
 import (
 	"os"
+	"strconv"
 	"strings"
 	"time"
 
@@ -737,3 +738,39 @@ func (m *Document) SplitMarkdownAndStore(seg string, markdown string, docId int)
 	}
 	return
 }
+
+// IsAllowReadChapter 书籍章节是否允许阅读
+func (m *Document) IsAllowReadChapter(bookId int, docId int) (yes bool, percent int) {
+	percent = 100
+	if bookId == 0 || docId == 0 {
+		return
+	}
+
+	percent, _ = strconv.Atoi(GetOptionValue("VISITOR_ALLOW_READED_PERCENT", "0"))
+	if percent <= 0 {
+		return false, percent
+	}
+
+	if percent >= 100 {
+		return true, percent
+	}
+
+	var docs []Document
+
+	orm.NewOrm().QueryTable(m).Filter("book_id", bookId).OrderBy("parent_id", "order_sort").All(&docs, "document_id", "identify")
+	length := len(docs)
+	if length == 0 {
+		return true, percent
+	}
+
+	index := int(float64(length*percent) / 100)
+	if index <= 0 {
+		index = 1
+	}
+	for _, doc := range docs[:index] {
+		if doc.DocumentId == docId {
+			return true, percent
+		}
+	}
+	return false, percent
+}

models/options.go

@@ -321,6 +321,11 @@ func (m *Option) Init() error {
 			OptionName:  "DEFAULT_HOME_PAGE",
 			OptionTitle: "默认首页",
 		},
+		{
+			OptionValue: "100",
+			OptionName:  "VISITOR_ALLOW_READED_PERCENT",
+			OptionTitle: "游客允许阅读百分比", // 游客未登录情况下允许阅读的章节百分比
+		},
 	}
 
 	for _, op := range options {

static/css/bookstack.css

@@ -1497,4 +1497,10 @@ form .btn-row .col-md-10{padding-left: 0;}
     left: 18px;
     top: 6px;
     position: absolute;
+}
+
+.locked-reading{padding-bottom: 100px;}
+.locked-reading .locked-icon{
+    width: 128px;height: 128px;margin: 100px auto 15px;
+    background: url("/static/images/lock.png") no-repeat center center;
 }

static/images/lock.png

@@ -48,6 +48,9 @@ function load_doc(url,wd,without_history) {
             var doc_title = res.data.doc_title;
             var title = res.data.title;
             var $body = body;
+            try {
+                if(res.data.is_allow_read==false){$body='<div class="locked-reading text-center"><div class="locked-icon"></div><div class="help-block">未登录游客仅可阅读 <span style="color:red">'+res.data.percent+'%</span> 章节，请您 <a href="/login" title="登录">登录</a> 再阅读</div>'            }
+            } catch (error) {}
             $("#page-content").html($body);
             // RenderByMarkdown($body);
             $("title").text(title);

static/js/bookstack.js

@@ -129,7 +129,14 @@ <h1 id="article-title">{{.Title}}</h1>
                         </div>
                     {{end}}
                     <article class="article-body  {{if eq .Model.Editor "markdown"}}markdown-body editormd-preview-container{{else}}editor-content{{end}}"  id="page-content">
-                    {{.Content}}
+                        {{if .IsAllowRead}}
+                            {{.Content}}
+                        {{else}}
+                            <div class="locked-reading text-center">
+                                <div class="locked-icon"></div>
+                                <div class="help-block">未登录游客仅可阅读 <span style="color:red">{{.Percent}}%</span> 章节，请您 <a href="/login" title="登录">登录</a> 再阅读</div>
+                            </div>
+                        {{end}}
                     </article>
                     {{if (ads "content-bottom" $.IsMobile)}}
                         <div class="ap ap-content-bottom mgt-15px">

views/document/default_read.html

@@ -380,6 +380,11 @@
                         {{end}}
 
                         {{if eq .Tab "operation"}}
+                            <div class="form-group">
+                                <label>游客允许阅读章节百分比</label>
+                                <input type="number" min="0" max="100" name="VISITOR_ALLOW_READED_PERCENT" class="form-control" value="{{.VISITOR_ALLOW_READED_PERCENT.OptionValue}}">
+                                <div class="help-block">未登录用户，允许阅读的书籍章节百分比。0表示不允许，100表示所有章节游客都可以阅读</div>
+                            </div>
                             <div class="form-group">
                                 <label>是否需要登录才能下载电子书</label>
                                 <input type="text" name="DOWNLOAD_LIMIT" class="form-control" value="{{.DOWNLOAD_LIMIT.OptionValue}}">