Netlify functions example

examples/ecommerce-netlify-functions/.netlify/edge-functions-import-map.json

@@ -0,0 +1 @@
+{"imports":{"netlify:edge":"https://edge-bootstrap.netlify.app/v1/index.ts"}}

examples/ecommerce-netlify-functions/functions/addToCart.js

@@ -0,0 +1,32 @@
+'use strict';
+
+const { Cart } = require('../models');
+
+const handler = async(event) => {
+  try {
+    if (event.body.cartId) {
+      // get the document containing the specified cartId
+      const cart = await Cart.findOne({ _id: event.body.cartId });
+      for (const product of event.body.product) {
+        const exists = cart.items.find(item =>
+          item.productId.toString() === product.productId.toString()
+        );
+        if (!exists) {
+          cart.items.push(product);
+        } else {
+          exists.quantity += product.quantity;
+        }
+      }
+      await cart.save();
+      return { statusCode: 200, body: cart };
+    } else {
+      // If no cartId, create a new cart
+      const cart = await Cart.create({ items: event.body.product });
+      return { statusCode: 200, body: cart };
+    }
+  } catch (error) {
+    return { statusCode: 500, body: error.toString() };
+  }
+};
+
+module.exports = { handler };

examples/ecommerce-netlify-functions/functions/checkout.js

@@ -0,0 +1,57 @@
+'use strict';
+const stripe = require('stripe')(process.env.STRIPE_KEY);
+
+const { Cart, Order, Product } = require('../models');
+
+const handler = async(event) => {
+  try {
+    const stripeProducts = { line_items: [] };
+    const total = 0;
+    for (let i = 0; i < event.body.products.length; i++) {
+      const product = await Product.findOne({_id: event.body.products[i].productId});
+      stripeProducts.line_items.push({
+        price_data: {
+          currency: 'usd',
+          product_data: {
+            name: product.productName
+          },
+          unit_amount: product.productPrice,
+        },
+        quantity: event.body.products[i].quantity
+      });
+      total = total + (product.productPrice * event.body.products[i].quantity);
+    }
+    const salesTax = total * .06;
+    const session = await stripe.checkout.sessions.create({
+      line_items: stripeProducts.line_items,
+      mode: 'payment',
+      success_url: 'insert a url here',
+      cancel_url: 'insert a url here'
+    });
+    const orders = await Order.find();
+    const orderNumber = orders.length ? orders.length + 1 : 1;
+    const order = await Order.create({
+      items: event.body.products,
+      total: total,
+      orderNumber: orderNumber,
+      name: event.body.name,
+      email: event.body.email,
+      address1: event.body.address1,
+      city: event.body.city,
+      state: event.body.state,
+      zip: event.body.zip,
+      location: { lat: event.body.lat, lng: event.body.lng },
+      shipping: event.body.shipping,
+      paymentMethod: { id: event.body.paymentMethod.id, brand: event.body.paymentMethod.brand, last4: event.body.paymentMethod.last4 },
+      salesTax: salesTax
+    });
+    const cart = await Cart.findOne({ _id: event.body.cartId });
+    cart.orderId = order._id;
+    await cart.save();
+    return { statusCode: 200, body: { order: order, cart: cart }, headers: { Location: session.url } };
+  } catch (error) {
+    return { statusCode: 500, body: error.toString() };
+  }
+};
+
+module.exports = { handler };

examples/ecommerce-netlify-functions/functions/getCart.js

@@ -0,0 +1,15 @@
+'use strict';
+
+const { Cart } = require('../models');
+
+const handler = async(event) => {
+  try {
+    // get the document containing the specified cartId
+    const cart = await Cart.findOne({ _id: event.queryStringParameters.cartId }).setOptions({ sanitizeFilter: true });
+    return { statusCode: 200, body: cart };
+  } catch (error) {
+    return { statusCode: 500, body: error.toString() };
+  }
+};
+
+module.exports = { handler };

examples/ecommerce-netlify-functions/functions/getProducts.js

@@ -0,0 +1,14 @@
+'use strict';
+
+const { Product } = require('../models');
+
+const handler = async(event) => {
+  try {
+    const products = await Product.find();
+    return { statusCode: 200, body: products };
+  } catch (error) {
+    return { statusCode: 500, body: error.toString() };
+  }
+};
+
+module.exports = { handler };

examples/ecommerce-netlify-functions/functions/removeFromCart.js

@@ -0,0 +1,27 @@
+'use strict';
+
+const { Cart } = require('../models');
+
+const handler = async(event) => {
+  try {
+    const cart = await Cart.findOne({ _id: event.body.cartId });
+    const index = cart.items.findIndex((item) =>
+      item.productId == event.body.product.productId
+    );
+    if (index == -1) {
+      return { statusCode: 500, body: 'Product not found' };
+    }
+    if (event.body.product.quantity) {
+      cart.items[index].quantity -= event.body.product.quantity;
+      await cart.save();
+    } else {
+      cart.items.splice(index, 1);
+      await cart.save();
+    }
+    return { statusCode: 200, body: cart };
+  } catch (error) {
+    return { statusCode: 500, body: error.toString() };
+  }
+};
+
+module.exports = { handler };

examples/ecommerce-netlify-functions/models.js

@@ -0,0 +1,93 @@
+'use strict';
+const mongoose = require('mongoose');
+
+const productSchema = new mongoose.Schema({
+  productName: String,
+  productPrice: Number
+});
+
+const Product = mongoose.model('Product', productSchema);
+
+module.exports.Product = Product;
+
+const orderSchema = new mongoose.Schema({
+  items: [{ productId: { type: mongoose.ObjectId, required: true, ref: 'Product' }, quantity: { type: Number, required: true } }],
+  total: {
+    type: Number,
+    default: 0
+  },
+  status: {
+    type: String,
+    enum: ['PAID', 'IN_PROGRESS', 'SHIPPED', 'DELIVERED'],
+    default: 'PAID'
+  },
+  orderNumber: {
+    type: Number,
+    required: true
+  },
+  name: {
+    type: String,
+    required: true
+  },
+  email: {
+    type: String,
+    required: true
+  },
+  address1: {
+    type: String,
+    required: true
+  },
+  address2: {
+    type: String
+  },
+  city: {
+    type: String,
+    required: true
+  },
+  state: {
+    type: String,
+    required: true
+  },
+  zip: {
+    type: String,
+    required: true
+  },
+  location: new mongoose.Schema({
+    lat: {
+      type: Number,
+      required: true
+    },
+    lng: {
+      type: Number,
+      required: true
+    }
+  }, { _id: false }),
+  shipping: {
+    type: String,
+    required: true,
+    enum: ['standard', '2day']
+  },
+  paymentMethod: {
+    id: String,
+    brand: String,
+    last4: String
+  },
+  salesTax: {
+    type: Number,
+    default: 0
+  },
+}, { optimisticConcurrency: true });
+
+const Order = mongoose.model('Order', orderSchema);
+
+module.exports.Order = Order;
+
+const cartSchema = new mongoose.Schema({
+  items: [{ productId: { type: mongoose.ObjectId, required: true, ref: 'Product' }, quantity: { type: Number, required: true } }],
+  orderId: { type: mongoose.ObjectId, ref: 'Order' }
+}, { timestamps: true });
+
+const Cart = mongoose.model('Cart', cartSchema);
+
+module.exports.Cart = Cart;
+

examples/ecommerce-netlify-functions/package.json

@@ -0,0 +1,12 @@
+{
+    "dependencies": {
+        "mongoose": "^6.3.5",
+        "stripe": "^9.6.0"
+    },
+    "devDependencies": {
+        "mocha": "10.0.0"
+    },
+    "scripts": {
+        "test": "mocha --exit ./test/*.test.js"
+    }
+}

examples/ecommerce-netlify-functions/seed.js

@@ -0,0 +1,35 @@
+'use strict';
+const mongoose = require('mongoose');
+const { Product } = require('./models');
+
+async function run() {
+  await mongoose.connect('mongodb://localhost:27017/netlify');
+  await mongoose.connection.dropDatabase();
+
+  await Product.create({
+    productName: 'Netlify\'s First Flight',
+    productPrice: 2000,
+    quantity: 5
+  });
+
+  await Product.create({
+    productName: 'Netlify The Movie',
+    productPrice: 4000,
+    quantity: 5
+  });
+
+  await Product.create({
+    productName: 'Netlify vs The Internet',
+    productPrice: 6000,
+    quantity: 5
+  });
+
+  await Product.create({
+    productName: 'Netlify and The Wasp',
+    productPrice: 8000,
+    quantity: 5
+  });
+  console.log('done');
+}
+
+run();

examples/ecommerce-netlify-functions/test/addToCart.test.js

@@ -0,0 +1,52 @@
+'use strict';
+
+const { describe, it, before, after } = require('mocha');
+const assert = require('assert');
+const { handler: addToCart } = require('../functions/addToCart');
+const mongoose = require('mongoose');
+
+
+
+describe('Add to Cart', function() {
+  before(async() => {
+    await mongoose.connect('mongodb://localhost:27017/netlify');
+  });
+
+  after(async() => {
+    await mongoose.disconnect();
+  });
+  it('Should create a cart and add a product to the cart.', async function() {
+    const params = {
+      body: {
+        cartId: null,
+        product: [
+          { productId: '629e5f3686d82fc73b95b396', quantity: 2 },
+          { productId: '629e5f3686d82fc73b95b398', quantity: 1 }
+        ]
+      }
+    };
+    const result = await addToCart(params);
+    assert(result.body);
+    assert(result.body.items.length);
+  });
+  it('Should find the cart and add to it.', async function() {
+    const params = {
+      body: {
+        cartId: null,
+        product: [
+          { productId: '629e5f3686d82fc73b95b396', quantity: 2 },
+          { productId: '629e5f3686d82fc73b95b398', quantity: 1 }
+        ]
+      }
+    };
+    const result = await addToCart(params);
+    assert(result.body);
+    assert(result.body.items.length);
+    params.body.cartId = result.body._id;
+    const findCart = await addToCart(params);
+    assert(findCart.body);
+    assert.deepStrictEqual(findCart.body._id, result.body._id);
+    assert.equal(findCart.body.items[0].quantity, 4);
+    assert.equal(findCart.body.items[1].quantity, 2);
+  });
+});

examples/ecommerce-netlify-functions/test/checkout.test.js

@@ -0,0 +1,34 @@
+'use strict';
+
+const { describe, it, before, after } = require('mocha');
+const assert = require('assert');
+const { handler: addToCart } = require('../functions/addToCart');
+const { handler: checkout } = require('../functions/checkout');
+const mongoose = require('mongoose');
+
+
+
+describe('Add to Cart', function() {
+  before(async() => {
+    await mongoose.connect('mongodb://localhost:27017/netlify');
+  });
+
+  after(async() => {
+    await mongoose.disconnect();
+  });
+  it('Should do a successfult checkout run', async function() {
+    const params = {
+      body: {
+        cartId: null,
+        product: [
+          { productId: '629e5f3686d82fc73b95b396', quantity: 2 },
+          { productId: '629e5f3686d82fc73b95b398', quantity: 1 }
+        ]
+      }
+    };
+    const result = await addToCart(params);
+    assert(result.body);
+    assert(result.body.items.length);
+    params.body.cartId = result.body._id;
+  });
+});