Merge 787575d into 19c1457

src/filters/ids.js

@@ -0,0 +1,12 @@
+const filter = require('lodash/filter')
+const includes = require('lodash/includes')
+const curry = require('lodash/curry')
+
+function idsFilter (val, db) {
+  if (val == null) return db
+
+  const idArray = val.split('|').map(Number)
+  return filter(db, (b) => includes(idArray, b.id))
+}
+
+module.exports = curry(idsFilter)

src/filters/index.js

@@ -10,6 +10,7 @@ const foodFilter = require('./food')
 const hopsFilter = require('./hops')
 const maltFilter = require('./malt')
 const yeastFilter = require('./yeast')
+const idsFilter = require('./ids')
 
 function filters (db, opts) {
   const {
@@ -25,7 +26,8 @@ function filters (db, opts) {
     brewed_after,
     hops,
     malt,
-    food
+    food,
+    ids
   } = opts
 
   return pipe(
@@ -41,7 +43,8 @@ function filters (db, opts) {
     foodFilter(food),
     hopsFilter(hops),
     maltFilter(malt),
-    yeastFilter(yeast)
+    yeastFilter(yeast),
+    idsFilter(ids)
   )(db)
 }
 

test/filters/ids.spec.js

@@ -0,0 +1,13 @@
+const db = require('punkapi-db')
+const idsFilter = require('../../dist/filters/ids')
+
+describe('idsFilter', function() {
+  it('should return beers with ids 2,8,20', function () {
+    idsFilter('2|8|20', db).should.containDeep([{id: 2}, {id: 8}, {id: 20}])
+  })
+
+  it('should not return beers with ids 300', function () {
+    idsFilter('2|8|300', db).should.containDeep([{id: 2}, {id: 8}])
+    idsFilter('2|8|300', db).should.not.containDeep([{id: 300}])
+  })
+})

test/index.js

@@ -173,4 +173,15 @@ describe('.beers()', function() {
     beers(opts)[0].ingredients.yeast.should.match(/Wyeast 1056/i)
     done()
   })
+
+  it('should return beers with ids of 1,4,29', function (done) {
+    const opts = {
+      ids: '1|4|29'
+    }
+    beers(opts).should.be.a.Array()
+    beers(opts)[0].id.should.be.equal(1)
+    beers(opts)[1].id.should.be.equal(4)
+    beers(opts)[2].id.should.be.equal(29)
+    done()
+  })
 })