Merge pull request #5 from uniquelyparticular/feat/addstripe

Feature: Added Stripe Usage Tracking

Author: agrohs

README.md

@@ -2,7 +2,7 @@
 
 [![npm version](https://img.shields.io/npm/v/@particular./stripe-distinct-usage-tracking.svg)](https://www.npmjs.com/package/@particular./stripe-distinct-usage-tracking) [![semantic-release](https://img.shields.io/badge/%20%20%F0%9F%93%A6%F0%9F%9A%80-semantic--release-e10079.svg)](https://github.com/semantic-release/semantic-release) [![code style: prettier](https://img.shields.io/badge/code_style-prettier-ff69b4.svg)](https://github.com/prettier/prettier) [![CircleCI](https://img.shields.io/circleci/project/github/uniquelyparticular/stripe-distinct-usage-tracking.svg?label=circleci)](https://circleci.com/gh/uniquelyparticular/stripe-distinct-usage-tracking) ![dependency status: david](https://img.shields.io/david/uniquelyparticular/stripe-distinct-usage-tracking.svg)
 
-> Stripe usage tracking implementation to increment usage record only for distinct new entities
+> Stripe Metered Pricing unique usage tracking implementation to increment Subscriptions Usage Record only for distinct new entities
 
 Built with [Micro](https://github.com/zeit/micro)! 🤩
 

now.json

@@ -5,6 +5,7 @@
  "env": {
  "NODE_ENV": "production",
  "USAGETRACKING_ORIGIN_WHITELIST": "@demo-usage-tracking-origin-whitelist",
+ "GATEWAY_SK": "@particular-gateway-sk",
  "FIREBASE_PROJECT_ID": "@particular-firebase-project-id",
  "FIREBASE_CLIENT_ID": "@particular-firebase-client-id",
  "FIREBASE_PRIVATE_KEY_ID": "@particular-firebase-pk-id",

package.json

@@ -6,11 +6,16 @@
  "description": "Please contact Particular. via info@uniquelyparticular.com for any questions",
  "keywords": [
  "stripe",
+ "subscriptions",
+ "usage",
+ "usage records",
+ "usage tracking",
+ "metered",
+ "metered pricing",
  "google",
  "firebase",
  "firestore",
  "microservices",
- "integration",
  "particular",
  "particular."
  ],

src/index.js

@@ -1,4 +1,4 @@
-const { newThisPeriod } = require('./utils')
+const { createUsageRecords, newThisPeriod } = require('./utils')
 const { json, send } = require('micro')
 const { URL } = require('whatwg-url')
 const cors = require('micro-cors')()
@@ -25,7 +25,7 @@ process.on('unhandledRejection', (reason, p) => {
 
 const notAuthorized = async (req, res) =>
  send(res, 401, {
- error: 'Referer or Destination not whitelisted or insecure'
+ error: 'Referer or Origin not whitelisted'
  })
 const invalidSecret = async (req, res) =>
  send(res, 401, {
@@ -77,6 +77,8 @@ const getOrigin = (origin, referer) => {
  if (subOrigin) {
  origin = decodeURIComponent(subOrigin[1])
  }
+ console.log('origin', origin)
+ console.log('referer', referer)
  return origin || referer
 }
 
@@ -104,12 +106,26 @@ module.exports = cors(async (req, res) => {
 
  try {
  const body = await json(req)
- // console.log('body',body)
- return newThisPeriod(body)
+ const { applicationId, collectionId, subscription, tracked } = body
+
+ return newThisPeriod(applicationId, collectionId, subscription, tracked)
  .then(isNewThisPeriod => {
- return send(res, 200, {
- newThisPeriod: isNewThisPeriod
- })
+ if (isNewThisPeriod) {
+ return createUsageRecords(subscription.items)
+ .then(() => {
+ return send(res, 200, {
+ newThisPeriod: isNewThisPeriod
+ })
+ })
+ .catch(error => {
+ const { raw, headers, ...jsonError } = _toJSON(error)
+ return send(res, 500, jsonError)
+ })
+ } else {
+ return send(res, 200, {
+ newThisPeriod: isNewThisPeriod
+ })
+ }
  })
  .catch(error => {
  const jsonError = _toJSON(error)

src/utils.js

@@ -1,5 +1,6 @@
 const moment = require('moment-timezone')
 const admin = require('firebase-admin')
+const stripe = require('stripe')(process.env.GATEWAY_SK)
 
 const _firebaseConfig = {
  type: 'service_account',
@@ -30,46 +31,88 @@ if (!admin.apps.length) {
 }
 const firestore = admin.firestore()
 
-exports.newThisPeriod = data => {
- return new Promise((resolve, reject) => {
- const trackedRef = firestore
- .collection('applications')
- .doc(`${data.applicationId}`)
- .collection(`${data.collectionId}`)
- .doc(`${data.subscription.id}`)
- .collection(
- `${moment
- .unix(data.subscription.current_period_start)
- .format('MMDDYYYY')}_${moment
- .unix(data.subscription.current_period_end)
- .format('MMDDYYYY')}`
- )
- .doc(`${data.tracked.id}`)
+module.exports = {
+ newThisPeriod(applicationId, collectionId, subscription, tracked) {
+ return new Promise((resolve, reject) => {
+ const metadata = tracked.metadata
+ const trackedRef = firestore
+ .collection('applications')
+ .doc(`${applicationId}`)
+ .collection(`${collectionId}`)
+ .doc(`${subscription.id}`)
+ .collection(
+ `${moment
+ .unix(subscription.current_period_start)
+ .format('MMDDYYYY')}_${moment
+ .unix(subscription.current_period_end)
+ .format('MMDDYYYY')}`
+ )
+ .doc(`${tracked.id}`)
 
- return trackedRef
- .get()
- .then(billingPeriodTrackedRecord => {
- // console.log('billingPeriodTrackedRecord',billingPeriodTrackedRecord)
- return trackedRef
- .set(
- {
- value: data.tracked.value,
- usage: admin.firestore.FieldValue.arrayUnion(
- moment()
- .utc()
- .format()
- )
- },
- { merge: true }
- )
- .then(() => {
- // console.log(
- // `trackedRef, newThisPeriod: ${!billingPeriodTrackedRecord.exists}`
- // )
- resolve(!billingPeriodTrackedRecord.exists)
- })
- .catch(error => reject(error))
+ return trackedRef
+ .get()
+ .then(billingPeriodTrackedRecord => {
+ // console.log('billingPeriodTrackedRecord',billingPeriodTrackedRecord)
+ return trackedRef
+ .set(
+ {
+ metadata,
+ usage: admin.firestore.FieldValue.arrayUnion(
+ moment()
+ .utc()
+ .format()
+ )
+ },
+ { merge: true }
+ )
+ .then(() => {
+ // console.log(
+ // `trackedRef, newThisPeriod: ${!billingPeriodTrackedRecord.exists}`
+ // )
+ resolve(!billingPeriodTrackedRecord.exists)
+ })
+ .catch(error => reject(error))
+ })
+ .catch(error => reject(error))
+ })
+ },
+
+ createUsageRecords(
+ subscriptionItems,
+ quantity = 1,
+ timestamp = moment().unix()
+ ) {
+ return new Promise((resolve, reject) => {
+ const usageRecords = []
+ subscriptionItems.map(subscriptionItem => {
+ usageRecords.push(
+ stripe.usageRecords
+ .create(subscriptionItem.id, {
+ quantity,
+ timestamp
+ })
+ .then(handleError)
+ )
  })
- .catch(error => reject(error))
+
+ return Promise.all(usageRecords)
+ .then(responses => {
+ resolve(responses)
+ })
+ .catch(error => reject(error))
+ })
+ }
+}
+
+const handleError = response => {
+ return new Promise((resolve, reject) => {
+ if (response.error) {
+ console.error(
+ `handleError, response.error: ${JSON.stringify(response.error)}`
+ )
+ reject(response.error)
+ } else {
+ resolve(response)
+ }
  })
 }