support apple store

Extension/src/background/background.ts

@@ -14,6 +14,11 @@ chrome.runtime.onInstalled.addListener(() => {
 chrome.tabs.onUpdated.addListener(function(tabId, changeInfo, tab) {
  if(tab.url && tab.url.includes(SupportedSitesUrls[SupportedSites.INSTACART])){
  setCurrentStore(SupportedSites.INSTACART);
+ } else if(tab.url && tab.url.includes(SupportedSitesUrls[SupportedSites.APPLESTORE])){
+ setCurrentStore(SupportedSites.APPLESTORE);
+ if(tab.url && tab.url.includes("apple.com/ca/shop/bag")){
+ chrome.tabs.sendMessage(tab.id, Messages.ENABLE_CHECKOUT_STATE); 
+ }
  }
 });
 

Extension/src/components/CheckoutDialog/CheckoutDialog.tsx

@@ -1,11 +1,12 @@
 import React, { Component } from 'react'
 import "./CheckoutDialog.scss"
 import { Tearsheet } from "@carbon/ibm-cloud-cognitive"
-import { getInstaCart, getCartAverageRating } from "../../stores/getInstaCart" 
+import { getCartAverageRating, getCart } from "../../stores/storesCommon"
 import { Accordion, AccordionItem, Tag, TagTypeName } from "carbon-components-react"
 
 interface CheckoutDialogProps {
  closeCheckoutDialog: () => void;
+ currentStore: string;
 }
 
 interface CheckoutDialogState {
@@ -58,7 +59,7 @@ class CheckoutDialog extends Component<CheckoutDialogProps, CheckoutDialogState>
  }
 
  render() {
- const cart = getInstaCart();
+ const cart = getCart(this.props.currentStore);
  const cartRank = getCartAverageRating(cart);
  console.log(cart);
  return (

Extension/src/containers/CheckoutContainer.tsx

@@ -7,6 +7,7 @@ interface CheckoutContainerProps {
  isPermissionGranted: boolean;
  setPermission: () => void;
  endCheckoutSequence: () => void;
+ currentStore: string;
 }
 
 interface CheckoutContainerState {
@@ -55,7 +56,8 @@ class CheckoutContainer extends Component<CheckoutContainerProps, CheckoutContai
  message={"Take2 is analyzing your cart...\n Thanks for trying your best to make smart choices."}
  />
  : <CheckoutDialog
- closeCheckoutDialog={() => {this.props.endCheckoutSequence()}}
+ currentStore={this.props.currentStore}
+ closeCheckoutDialog={() => {this.props.endCheckoutSequence()}}
  />
  )
  }

Extension/src/contentScript/contentScript.tsx

@@ -66,6 +66,8 @@ class App extends Component<AppProps, AppState> {
  }
  }
  return;
+ case SupportedSites.APPLESTORE:
+ return;
  default:
  return;
  }
@@ -141,6 +143,7 @@ class App extends Component<AppProps, AppState> {
  { (this.state.checkoutState &&
  this.state.startScenarioSquence) ?
  <CheckoutContainer
+ currentStore={this.state.currentStore}
  isPermissionGranted={this.state.isPermissionGranted}
  setPermission={() => {this.setState({isPermissionGranted: true})}}
  endCheckoutSequence={() => {this.setState({startScenarioSquence: false})}}

Extension/src/static/manifest.json

@@ -20,7 +20,7 @@
  },
  "content_scripts": [
  {
- "matches": ["https://www.instacart.com/store/*"],
+ "matches": ["https://www.instacart.com/store/*", "https://www.apple.com/ca/shop/*"],
  "js": ["contentScript.js"]
  }
  ]

Extension/src/stores/getAppleCart.js

@@ -0,0 +1,20 @@
+import { analyzeCart } from "./storesCommon"
+
+export function getAppleCart() {
+ const _cartItems = document.getElementsByClassName("row rs-iteminfo");
+ let cart = [];
+ for(let i = 0; i < _cartItems.length; i++){
+ const priceItem = _cartItems[i].getElementsByClassName("rs-iteminfo-price");
+ const textItem = _cartItems[i].getElementsByClassName("rs-iteminfo-title");
+ const imageItem = _cartItems[i].getElementsByClassName("as-util-relatedlink")[0].childNodes[0];
+ let _item = {
+ id: textItem[0].childNodes[0].innerText,
+ name: textItem[0].childNodes[0].innerText,
+ price: priceItem[0].childNodes[0].childNodes[1].innerText,
+ image: imageItem
+ }
+ cart.push(_item);
+ }
+ cart = analyzeCart(cart);
+ return cart;
+}

Extension/src/stores/getInstaCart.js

@@ -1,8 +1,10 @@
+import { analyzeCart } from "./storesCommon"
+
 export function getInstaCart() {
  // const _html = document.all[0].outerHTML;
  const _cart = document.getElementsByClassName("cart-container-inner")[0];
- const _cartItems = _cart.getElementsByClassName("clamped-name");
- const _cartImages = _cart.getElementsByTagName("img");
+ const _cartItems = _cart ? _cart.getElementsByClassName("clamped-name") : [];
+ const _cartImages = _cart ? _cart.getElementsByTagName("img") : [];
 
  let cart = [];
  let imageIndex = 0;
@@ -21,34 +23,4 @@ export function getInstaCart() {
  }
  cart = analyzeCart(cart);
  return cart;
-}
-
-export function analyzeCart(cart) {
- for(let item of cart){
- item.score = Math.floor(Math.random() * 10) + 1;
- if(item.score > 10){
- item.score = 10;
- }
- }
- cart.sort(compare);
- return cart;
-}
-
-function compare( a, b ) {
- if ( a.score < b.score ){
- return 1;
- }
- if ( a.score > b.score ){
- return -1;
- }
- return 0;
-}
-
-export function getCartAverageRating(cart) {
- let score = 0;
- for(let item of cart){
- score += item.score;
- }
- console.log(score);
- return score/cart.length;
 }

Extension/src/stores/storesCommon.js

@@ -0,0 +1,46 @@
+import { getInstaCart } from "./getInstaCart"
+import { getAppleCart } from "./getAppleCart"
+import { SupportedSites } from "../utils/constants"
+
+export function getCart(currentStore) {
+ switch(currentStore){
+ case SupportedSites.INSTACART:
+ return getInstaCart();
+ case SupportedSites.APPLESTORE:
+ return getAppleCart();
+ default:
+ return [];
+ }
+}
+
+export function analyzeCart(cart) {
+ for(let item of cart){
+ item.score = Math.floor(Math.random() * 10) + 1;
+ if(item.score > 10){
+ item.score = 10;
+ }
+ }
+ cart.sort(compare);
+ return cart;
+}
+
+export function compare( a, b ) {
+ if ( a.score < b.score ){
+ return 1;
+ }
+ if ( a.score > b.score ){
+ return -1;
+ }
+ return 0;
+}
+
+export function getCartAverageRating(cart) {
+ let score = 0;
+ for(let item of cart){
+ score += item.score;
+ }
+ if(cart.length <= 0 ){
+ return 0;
+ }
+ return score/cart.length;
+}

Extension/src/utils/constants.ts

@@ -8,9 +8,11 @@ export enum Messages {
 }
 
 export const SupportedSites = {
- INSTACART: "INSTACART"
+ INSTACART: "INSTACART",
+ APPLESTORE: "APPLESTORE"
 }
 
 export const SupportedSitesUrls = {
- [SupportedSites.INSTACART]: "instacart.com"
+ [SupportedSites.INSTACART]: "instacart.com",
+ [SupportedSites.APPLESTORE]: "apple.com"
 }