From d30b4142d1f9c6b6d43868dadd2f452be724ed1b Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Mon, 18 Jun 2018 11:27:10 +0900
Subject: [PATCH] Implement basic pagination

---
 doccano/app/server/static/annotation.1.js | 38 +++++++++++++++++++----
 doccano/app/server/views.py               | 13 ++++++--
 2 files changed, 43 insertions(+), 8 deletions(-)

diff --git a/doccano/app/server/static/annotation.1.js b/doccano/app/server/static/annotation.1.js
index 93715d0b..28a8f596 100644
--- a/doccano/app/server/static/annotation.1.js
+++ b/doccano/app/server/static/annotation.1.js
@@ -26,6 +26,11 @@ var vm = new Vue({
         total: 0,
         remaining: 0,
         searchQuery: '',
+        hasNext: false,
+        hasPrevious: false,
+        nextPageNum: 1,
+        prevPageNum: 1,
+        page: 1,
     },
 
     methods: {
@@ -70,14 +75,31 @@ var vm = new Vue({
             this.items[this.cur]['labels'].splice(index, 1)
         },
         nextPage: function () {
-            this.cur = Math.min(this.cur + 1, this.items.length - 1);
             this.remaining -= 1;
-            console.log('nextPage');
+            this.cur += 1;
+            if (this.cur == this.items.length) {
+              if (this.hasNext) {
+                this.page = this.nextPageNum;
+                this.submit();
+                this.cur = 0;
+              } else {
+                this.cur = this.items.length - 1;
+              }
+            }
             this.showMessage(this.cur);
         },
         prevPage: function () {
-            this.cur = Math.max(this.cur - 1, 0);
             this.remaining += 1;
+            this.cur -= 1;
+            if (this.cur == -1) {
+              if (this.hasPrevious) {
+                this.page = this.prevPageNum;
+                this.submit();
+                this.cur = this.items.length - 1;
+              } else {
+                this.cur = 0;
+              }
+            }
             this.showMessage(this.cur);
         },
         activeLearn: function () {
@@ -86,12 +108,16 @@ var vm = new Vue({
         submit: function () {
             console.log('submit' + this.searchQuery);
             var self = this;
-            axios.get('/' + base_url + '/apis/search?keyword=' + this.searchQuery)
+            axios.get('/' + base_url + '/apis/search?keyword=' + this.searchQuery + '&page=' + this.page)
                 .then(function (response) {
                     console.log('search response');
-                    console.log(response.data['data']);
+                    console.log(response.data);
                     self.items = response.data['data'];
-                    self.searchQuery = '';
+                    self.hasNext = response.data['has_next']
+                    self.nextPageNum = response.data['next_page_number']
+                    self.hasPrevious = response.data['has_previous']
+                    self.prevPageNum = response.data['previous_page_number']
+                    //self.searchQuery = '';
                 })
                 .catch(function (error) {
                     console.log('ERROR!! happend by Backend.')
diff --git a/doccano/app/server/views.py b/doccano/app/server/views.py
index 4cba995b..1ec6bb9f 100644
--- a/doccano/app/server/views.py
+++ b/doccano/app/server/views.py
@@ -5,6 +5,7 @@ from django.shortcuts import render
 from django.views import View
 from django.views.generic.list import ListView
 from django.views.generic.detail import DetailView
+from django.core.paginator import Paginator
 
 from .models import Annotation, Label, Document, Project
 
@@ -85,8 +86,16 @@ class SearchAPI(View):
         if not docs:
             docs = [{'id': None, 'labels': [], 'text': ''}]
         # Annotation.objects.select_related('data').all().filter(data__text__contains=keyword)
-
-        return JsonResponse({'data': docs})
+        paginator = Paginator(docs, 5)
+        page = request.GET.get('page', 1)
+        page = paginator.get_page(page)
+        docs = page.object_list
+
+        return JsonResponse({'data': docs,
+                             'has_next': page.has_next(),
+                             'has_previous': page.has_previous(),
+                             'previous_page_number': page.previous_page_number() if page.has_previous() else None,
+                             'next_page_number': page.next_page_number() if page.has_next() else None})
 
 
 class LabelAPI(View):