Subscription Payments #2: Keeping track of Customer Billing information using Mongo and Stripe Webhooks

mongod

const mongoose = require('mongoose')

const db = 'mongodb://localhost:27017/users'
mongoose.connect(
  db,
  {
    useUnifiedTopology: true,
    useNewUrlParser: true,
    useFindAndModify: false,
    useCreateIndex: true
  },
  (error) => {
    if (error) console.log(error)
  }
)

const mongoose = require('mongoose')
const Schema = mongoose.Schema

const userSchema = new Schema({
  email: String,
  billingID: String,
  plan: { type: String, enum: ['none', 'basic', 'pro'], default: 'none' },
  hasTrial: { type: Boolean, default: false },
  endDate: { type: Date, default: null }
})

const userModel = mongoose.model('user', userSchema, 'user')

module.exports = userModel

const addUser = (User) => ({ email, billingID, plan, endDate }) => {
  if (!email || !billingID || !plan) { throw new Error('Missing Data. Please provide values for email, billingID, plan') }

  const user = new User({ email, billingID, plan, endDate })
  return user.save()
}

const getUsers = (User) => () => {
  return User.find({})
}

const getUserByEmail = (User) => async (email) => {
  return await User.findOne({ email })
}

const getUserByBillingID = (User) => async (billingID) => {
  return await User.findOne({ billingID })
}

const updatePlan = (User) => (email, plan) => {
  return User.findOneAndUpdate({ email, plan })
}

module.exports = (User) => {
  return {
    addUser: addUser(User),
    getUsers: getUsers(User),
    getUserByEmail: getUserByEmail(User),
    updatePlan: updatePlan(User),
    getUserByBillingID: getUserByBillingID(User)
  }
}

const UserService = require('./src/user')

app.post("/login", async function (req, res) {
  const { email } = req.body;
  console.log("email", email);

  let customer = await UserService.getUserByEmail(email);
  let customerInfo = {};

  if (!customer) {
    console.log(`email ${email} does not exist. Making one. `);
    try {
      customerInfo = await Stripe.addNewCustomer(email);

      customer = await UserService.addUser({
        email: customerInfo.email,
        billingID: customerInfo.id,
				hasTrial: false,
        plan: "none",
        endDate: null
      });

      console.log(
        `A new user signed up and addded to DB. The ID for ${email} is ${JSON.stringify(
          customerInfo
        )}`
      );

      console.log(`User also added to DB. Information from DB: ${customer}`);
    } catch (e) {
      console.log(e);
      res.status(200).json({ e });
      return;
    }
  }

  req.session.email = email;
  res.redirect('/account')
})

stripe login
stripe listen --forward-to localhost:4242/webhook

app.use("/webhook", bodyParser.raw({ type: "application/json" }))

app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: true }))

app.post('/webhook', async (req, res) => {
  let event

  try {
    event = Stripe.createWebhook(req.body, req.header('Stripe-Signature'))
  } catch (err) {
    console.log(err)
    return res.sendStatus(400)
  }

  const data = event.data.object
  console.log(event.type, data)

  res.sendStatus(200)
})

npm start

app.post('/webhook', async (req, res) => {
  let event

  try {
    event = Stripe.createWebhook(req.body, req.header('Stripe-Signature'))
  } catch (err) {
    console.log(err)
    return res.sendStatus(400)
  }

  const data = event.data.object
  console.log(event.type, data)

  switch (event.type) {
    case 'customer.subscription.created': {
      console.log('new subscription added' + JSON.stringify(data))
      const user = await UserService.getUserByBillingID(data.customer)

      if (data.plan.id === productToPriceMap.BASIC) {
        console.log('You are talking about basic product')
        user.plan = 'basic'
      }

      if (data.plan.id === productToPriceMap.PRO) {
        console.log('You are talking about pro product')
        user.plan = 'pro'
      }

      const isOnTrial = data.status === 'trialing'

      if (isOnTrial) {
        user.hasTrial = true
        user.endDate = new Date(data.current_period_end * 1000)
      } else if (data.status === 'active') {
        user.hasTrial = false
        user.endDate = new Date(data.current_period_end * 1000)
      }

      await user.save()

      break
    }
    default:
  }

  res.sendStatus(200)
})

app.post("/login", async function (req, res) {
  const { email } = req.body;
  console.log("email", email);

  let customer = await UserService.getUserByEmail(email);
  let customerInfo = {};

  if (!customer) {
	  // ..
  } else {
    const isTrialExpired =
      customer.plan !== 'none' && customer.endDate < new Date().getTime()

    if (isTrialExpired) {
      console.log('trial expired')
      customer.hasTrial = false
      customer.save()
    } else {
      console.log(
        'no trial information',
        customer.hasTrial,
        customer.plan !== 'none',
        customer.endDate < new Date().getTime()
      )
    }
  }
  req.session.email = email
  res.send(customerInfo)
})