pass different image classes to getImage

1 files changed, 6 insertions, 2 deletions

diff --git a/bog.js b/bog.js
index 1aff354..a9e2cd3 100644
--- a/bog.js
+++ b/bog.js
@@ -98,8 +98,12 @@ async function get (key) {
 
 // bog.getImage
 
-function getImage (id, keys) {
-  var image = h('img', {classList: 'avatar'})
+function getImage (id, keys, classList) {
+  if (classList) {
+    var image = h('img', {classList: classList})
+  } else {
+    var image = h('img', {classList: 'avatar'})
+  }
 
   localforage.getItem('image:' + id).then(cache => {
     if (cache) {