summaryrefslogtreecommitdiff
path: root/lib/db/index.js
blob: 468d9beebdbc6fa565cfdcbb5d25aa67fc022d6c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
var db = module.exports

var connection = require("./bookshelf")
var bookshelf = connection.bookshelf
var knex = connection.knex

var Folder = db.Folder = bookshelf.Model.extend({
  tableName: 'folders',
  hasTimestamps: true,
})
var File = db.File = bookshelf.Model.extend({
  tableName: 'files',
  hasTimestamps: true,
})
var Job = db.Job = bookshelf.Model.extend({
  tableName: 'job',
  hasTimestamps: true,
})
var Task = db.Task = bookshelf.Model.extend({
  tableName: 'tasks',
  hasTimestamps: true,
})

db.crud = function(type, model) {
  return {
    index: (q) => {
      return model.query( (qb) => {
        const limit = q.limit || 100
        const offset = q.offset || 0
        if (limit) {
          delete q.limit
        }
        if (q.offset) {
          delete q.offset
        }
        if (Object.keys(q).length > 0) qb.where(q)
        qb.orderBy("id", "desc")
        if (limit) qb.limit( limit )
        if (offset) qb.offset( offset )
        return qb
      }).fetchAll()
    },
    show: (id) => {
      return new model({'id': id}).fetch()
    },
    create: (data) => {
      return new model(data).save()
    },
    update: (id, data) => {
      return new model({'id': id}).save(data)
    },
    destroy: (id) => {
      return new model({'id': id}).destroy(data)
    },
  }
}
db.getFolders = function(id) {
  var model = new Folder({'id': id})
  return model.fetch()
}
db.getFolderFiles = function(id) {
  return File.query(function(qb){
    qb.where('folder_id', '=', id).orderBy("id", "desc")
  }).fetch()
}
db.getLatest = function () {
  return Image.query(function(qb){
    qb.orderBy("id", "desc").limit(1)
  }).fetch()
}
db.getRandom = function () {
  return Image.query(function(qb){
    qb.orderBy(knex.raw('RAND()')).limit(1)
  }).fetch()
}

db.createImage = function(url){
  return new Image({ url: url, shorturl: "" }).save()
}