firebase send and multi-plexed auth

Author: carterharrison

.vscode/settings.json

@@ -32,6 +32,7 @@
     }
   ],
   "cSpell.words": [
-    "Geohash"
+    "Geohash",
+    "firestore"
   ]
 }

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@xyo-network/sdk-diviner-nodejs",
-  "version": "0.7.1",
+  "version": "0.7.2",
   "description": "XYO Diviner SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -33,6 +33,7 @@
     "dist/payment/lightning/plugin/index.js",
     "dist/payment/plugin/index.js",
     "dist/payment/repository/dynammodb/plugin/index.js",
+    "dist/payment/repository/firestore/plugin/index.js",
     "dist/query/plugin/query/index.js",
     "dist/query/plugin/query-chain-tracer/index.js",
     "dist/query/plugin/query-geohash/index.js",
@@ -73,6 +74,8 @@
     "aws-sdk": "^2.441.0",
     "bs58": "^4.0.1",
     "eth-sig-util": "1.4.2",
+    "firebase": "^6.2.0",
+    "firebase-admin": "^8.1.0",
     "graphql": "^14.2.1",
     "graphql-tag": "^2.10.1",
     "graphql-type-json": "^0.3.0",

src/payment/plugin/index.ts

@@ -27,7 +27,7 @@ class QueryPaymentPlugin implements IXyoPlugin {
     const creditEndpoint = new XyoCreditEndpoint(store)
     const spendEndpoint = new XyoSpendEndpoint(store)
     const scan = delegate.deps.QUERY as XyoQuery
-    scan.auth = new XyoQueryAuth(store)
+    scan.auth.authProviders.payment = new XyoQueryAuth(store)
 
     delegate.graphql.addQuery(XyoCreditEndpoint.query)
     delegate.graphql.addResolver(XyoCreditEndpoint.queryName, creditEndpoint)

src/payment/repository/firestore/firestore-coin-auth.ts

@@ -0,0 +1,59 @@
+import { IXyoAuth } from '../../../query'
+import { XyoCoinAuth } from './firestore-coin-spend-store'
+import firebaseAdmin from 'firebase-admin'
+
+interface IFirestoreCoinAuthConfig {
+  token: string
+}
+
+export class FirestoreCoinAuth implements IXyoAuth {
+  public name = 'firestore-coin-auth'
+  private store: XyoCoinAuth
+
+  constructor (store: XyoCoinAuth) {
+    this.store = store
+  }
+
+  public async getCredits (token: string): Promise<number> {
+    const userId = await this.getUserIdFromFirebaseToken(token)
+
+    if (!userId) {
+      return 0
+    }
+
+    return await this.store.getCreditsForKey(userId) || 0
+  }
+
+  public async didComplete(config: any): Promise<void> {
+    const authConfig = config.coin as IFirestoreCoinAuthConfig
+    const userId = await this.getUserIdFromFirebaseToken(authConfig.token)
+
+    if (userId) {
+      const constAmount = (await this.store.getCreditsForKey(userId) || 0)
+      await this.store.setCreditsForKey(userId, constAmount - 1)
+      await this.store.incrementSpentForUser(userId)
+      return
+    }
+
+    throw new Error('No user font to spend tokens')
+  }
+
+  public async auth(config: any): Promise<boolean> {
+    const authConfig = config.coin as IFirestoreCoinAuthConfig
+    const userId = await this.getUserIdFromFirebaseToken(authConfig.token)
+
+    if (userId) {
+      const canSpend = (await this.store.getCreditsForKey(userId) || 0)
+      return canSpend > 1
+    }
+
+    return false
+
+  }
+
+  private async getUserIdFromFirebaseToken (token: string): Promise<string | undefined> {
+    const decodedToken = await firebaseAdmin.auth().verifyIdToken(token)
+    return decodedToken.uid
+  }
+
+}

src/payment/repository/firestore/firestore-coin-spend-store.ts

@@ -0,0 +1,65 @@
+
+import { IXyoPaymentStore } from '../../../payment'
+import { firestore } from 'firebase'
+
+export class XyoCoinAuth implements IXyoPaymentStore {
+  private firestore: firestore.Firestore
+
+  constructor(firestoreInstance: firestore.Firestore) {
+    this.firestore = firestoreInstance
+  }
+
+  public async getCreditsForKey(key: string): Promise<number | undefined> {
+    const docRef = await this.firestore.collection('coin_users').doc(key).get()
+
+    if (!docRef.exists) {
+      return 0
+    }
+
+    const data = docRef.data()
+
+    if (data) {
+      return data.xyoCollected
+    }
+
+    return 0
+  }
+
+  public async incrementSpentForUser(key: string): Promise<void> {
+    const docRef = await this.firestore.collection('coin_users').doc(key).get()
+
+    if (!docRef.exists) {
+      return
+    }
+
+    const data = docRef.data()
+
+    if (!data) {
+      return
+    }
+
+    await this.firestore.collection('coin_users').doc(key).update({
+      xyoQuerySpent: (data.xyoQuerySpent || 0) + 1.
+    })
+  }
+
+  public async setCreditsForKey(key: string, credits: number): Promise<void> {
+    const docRef = await this.firestore.collection('coin_users').doc(key).get()
+
+    if (!docRef.exists) {
+      return
+    }
+
+    await this.firestore.collection('coin_users').doc(key).update({
+      xyoCollected: credits
+    })
+  }
+
+  public spent(creditKey: string): Promise<void> {
+    throw new Error('Method not implemented: didSpend.')
+  }
+
+  public didSpend(creditKey: string): Promise<boolean> {
+    throw new Error('Method not implemented: didSpend.')
+  }
+}

src/payment/repository/firestore/plugin/firestore-credit.endpoint.ts

@@ -0,0 +1,15 @@
+import { FirestoreCoinAuth } from '../firestore-coin-auth'
+
+export class XyoFirestoreCreditEndpoint {
+  public static query = 'coinCredits(token: String): Float'
+  public static queryName = 'coinCredits'
+  private spendStore: FirestoreCoinAuth
+
+  constructor(spend: FirestoreCoinAuth) {
+    this.spendStore = spend
+  }
+
+  public async resolve(obj: any, args: any): Promise<any> {
+    return this.spendStore.getCredits(args.token)
+  }
+}

src/payment/repository/firestore/plugin/index.ts

@@ -0,0 +1,44 @@
+import { IXyoPlugin, XyoPluginProviders, IXyoPluginDelegate } from '@xyo-network/sdk-base-nodejs'
+import { XyoQuery } from '../../../../query/xyo-query'
+import { XyoCoinAuth } from '../firestore-coin-spend-store'
+import { FirestoreCoinAuth } from '../firestore-coin-auth'
+import { XyoFirestoreCreditEndpoint } from './firestore-credit.endpoint'
+import firebaseAdmin from 'firebase-admin'
+import firebase from 'firebase'
+
+class CoinPaymentPlugin implements IXyoPlugin {
+
+  public getName(): string {
+    return 'coin-payment'
+  }
+
+  public getProvides(): string[] {
+    return []
+  }
+
+  public getPluginDependencies(): string[] {
+    return [
+      XyoPluginProviders.QUERY
+    ]
+  }
+
+  public async initialize(delegate: IXyoPluginDelegate): Promise<boolean> {
+    firebaseAdmin.initializeApp({
+      databaseURL: 'https://xyo-network-1522800011804.firebaseio.com',
+    })
+
+    const scan = delegate.deps.QUERY as XyoQuery
+    const firestoreDb = new XyoCoinAuth(firebaseAdmin.firestore() as any)
+    const auth = new FirestoreCoinAuth(firestoreDb)
+    scan.auth.authProviders.coin = auth
+
+    const creditEndpoint = new XyoFirestoreCreditEndpoint(auth)
+    delegate.graphql.addQuery(XyoFirestoreCreditEndpoint.query)
+    delegate.graphql.addResolver(XyoFirestoreCreditEndpoint.queryName, creditEndpoint)
+
+    return true
+  }
+
+}
+
+export = new CoinPaymentPlugin()

src/query/auth/xyo-query-auth-multiplex.ts

@@ -0,0 +1,36 @@
+import { IXyoAuth } from '..'
+
+export class XyoMultiplexedQueryAuth implements IXyoAuth {
+  public name = 'xyo-multiplex-auth'
+  public authProviders: {[key: string]: IXyoAuth} = {}
+
+  public async auth(config: any): Promise<boolean> {
+    const keys = Object.keys(config)
+
+    for (const key of keys) {
+      const authProvider = this.authProviders[key]
+
+      if (authProvider) {
+        return authProvider.auth(config)
+      }
+    }
+
+    return false
+  }
+
+  public async didComplete(config: any): Promise<void> {
+    const keys = Object.keys(config)
+
+    for (const key of keys) {
+      const authProvider = this.authProviders[key]
+
+      if (authProvider) {
+        await authProvider.didComplete(config)
+        return
+      }
+    }
+
+    throw new Error('Auth not found')
+  }
+
+}

src/query/auth/xyo-query-auth.ts

@@ -16,12 +16,14 @@ export class XyoQueryAuth implements IXyoAuth {
 
   public async didComplete(config: any): Promise<void> {
     const authConfig = config.payment as IXyoQueryAuthConfig
+
     const constAmount = (await this.store.getCreditsForKey(authConfig.apiKey) || 0)
     await this.store.setCreditsForKey(authConfig.apiKey, constAmount - 1)
   }
 
   public async auth(config: any): Promise<boolean> {
     const authConfig = config.payment as IXyoQueryAuthConfig
+
     const canSpend = (await this.store.getCreditsForKey(authConfig.apiKey) || 0)
 
     if (canSpend < 1) {

src/query/xyo-query.ts

@@ -1,16 +1,13 @@
 import { IXyoSelecterCreator, IXyoFilterCreator, IXyoMutatorCreater, IXyoQuery, IXyoConfig, IXyoAuth } from '.'
+import { XyoMultiplexedQueryAuth } from './auth/xyo-query-auth-multiplex'
 
 export class XyoQuery {
-  public auth?: IXyoAuth
+  public auth = new XyoMultiplexedQueryAuth()
   private selectorCreators: Map<string, IXyoSelecterCreator> = new Map()
   private filterCreators: Map<string, IXyoFilterCreator> = new Map()
   private mutatorCreators: Map<string, IXyoMutatorCreater> = new Map()
   private finishNotifiers: Map<string, (query: IXyoQuery, blocks: Buffer[]) => void> = new Map()
 
-  constructor(auth?: IXyoAuth) {
-    this.auth = auth
-  }
-
   public addSelector (selector: IXyoSelecterCreator) {
     this.selectorCreators.set(selector.name, selector)
   }