square_sandbox_simulator 0.1.0

data/lib/square_sandbox_simulator/generators/entity_generator.rb

@@ -0,0 +1,248 @@
+# frozen_string_literal: true
+
+module SquareSandboxSimulator
+  module Generators
+    # Creates entities in Square (categories, items, discounts, taxes, team members, customers)
+    class EntityGenerator
+      # Configuration constants
+      DEFAULT_TEAM_MEMBER_COUNT = 5
+      DEFAULT_CUSTOMER_COUNT = 20
+      LOG_SEPARATOR = "=" * 60
+
+      attr_reader :services, :data, :logger
+
+      def initialize(services: nil, business_type: :restaurant)
+        @services = services || Services::Square::ServicesManager.new
+        @data = DataLoader.new(business_type: business_type)
+        @logger = SquareSandboxSimulator.logger
+      end
+
+      # Set up all entities (categories, items, discounts, etc.)
+      # @return [Hash] Hash of created entities with their Square IDs
+      def setup_all
+        logger.info LOG_SEPARATOR
+        logger.info "Setting up entities in Square..."
+        logger.info LOG_SEPARATOR
+
+        results = {
+          categories: setup_categories,
+          items: setup_items,
+          discounts: setup_discounts,
+          tax_rates: setup_tax_rates,
+          team_members: setup_team_members,
+          customers: setup_customers,
+        }
+
+        logger.info LOG_SEPARATOR
+        logger.info "Entity setup complete!"
+        logger.info "  Categories:    #{results[:categories].size}"
+        logger.info "  Items:         #{results[:items].size}"
+        logger.info "  Discounts:     #{results[:discounts].size}"
+        logger.info "  Tax Rates:     #{results[:tax_rates].size}"
+        logger.info "  Team Members:  #{results[:team_members].size}"
+        logger.info "  Customers:     #{results[:customers].size}"
+        logger.info LOG_SEPARATOR
+
+        results
+      end
+
+      # Create categories from data file (idempotent via name matching).
+      def setup_categories
+        logger.info "Setting up categories..."
+
+        existing = services.catalog.list_catalog(types: "CATEGORY")
+        existing_names = existing.map { |c| c.dig("category_data", "name")&.downcase }.compact.to_set
+
+        created = []
+        data.categories.each do |cat_data|
+          name = cat_data["name"]
+
+          if existing_names.include?(name&.downcase)
+            logger.debug "Category '#{name}' already exists, skipping"
+            match = existing.find { |c| c.dig("category_data", "name")&.downcase == name&.downcase }
+            created << match if match
+          else
+            result = services.catalog.upsert_category(name: name)
+            obj = result&.dig("catalog_object")
+            created << obj if obj
+          end
+        end
+
+        logger.info "Categories ready: #{created.size}"
+        created
+      end
+
+      # Create items from data file.
+      # Each item is created with a single "Regular" variation containing the price.
+      # Tracks variation catalog_object_ids (needed for orders).
+      def setup_items
+        logger.info "Setting up items..."
+
+        # Fetch current categories from Square for association
+        categories = services.catalog.list_catalog(types: "CATEGORY")
+        category_by_name = categories.each_with_object({}) do |cat, hash|
+          name = cat.dig("category_data", "name")
+          hash[name.downcase] = cat["id"] if name
+        end
+
+        # Fetch existing items for idempotent matching
+        existing = services.catalog.list_catalog(types: "ITEM")
+        existing_names = existing.map { |i| i.dig("item_data", "name")&.downcase }.compact.to_set
+
+        created = []
+        data.items.each do |item_data|
+          name = item_data["name"]
+
+          if existing_names.include?(name&.downcase)
+            logger.debug "Item '#{name}' already exists, skipping"
+            match = existing.find { |i| i.dig("item_data", "name")&.downcase == name&.downcase }
+            created << match if match
+          else
+            category_id = category_by_name[item_data["category"]&.downcase]
+            price_cents = item_data["price"] || 0
+
+            result = services.catalog.upsert_item(
+              name: name,
+              price_cents: price_cents,
+              category_id: category_id,
+            )
+            obj = result&.dig("catalog_object")
+            created << obj if obj
+          end
+        end
+
+        logger.info "Items ready: #{created.size}"
+        created
+      end
+
+      # Create discounts from data file (idempotent via name matching).
+      def setup_discounts
+        logger.info "Setting up discounts..."
+
+        existing = services.catalog.list_catalog(types: "DISCOUNT")
+        existing_names = existing.map { |d| d.dig("discount_data", "name")&.downcase }.compact.to_set
+
+        created = []
+        data.discounts.each do |disc_data|
+          name = disc_data["name"]
+
+          if existing_names.include?(name&.downcase)
+            logger.debug "Discount '#{name}' already exists, skipping"
+            match = existing.find { |d| d.dig("discount_data", "name")&.downcase == name&.downcase }
+            created << match if match
+          else
+            result = if disc_data["percentage"]
+                       services.catalog.upsert_discount(
+                         name: name,
+                         percentage: disc_data["percentage"],
+                       )
+                     else
+                       services.catalog.upsert_discount(
+                         name: name,
+                         amount_cents: disc_data["amount"] || 0,
+                       )
+                     end
+            obj = result&.dig("catalog_object")
+            created << obj if obj
+          end
+        end
+
+        logger.info "Discounts ready: #{created.size}"
+        created
+      end
+
+      # Create tax rates from data file (idempotent via name matching).
+      def setup_tax_rates
+        logger.info "Setting up tax rates..."
+
+        existing = services.catalog.list_catalog(types: "TAX")
+        existing_names = existing.map { |t| t.dig("tax_data", "name")&.downcase }.compact.to_set
+
+        created = []
+        data.tax_rates.each do |rate_data|
+          name = rate_data["name"]
+
+          if existing_names.include?(name&.downcase)
+            logger.debug "Tax rate '#{name}' already exists, skipping"
+            match = existing.find { |t| t.dig("tax_data", "name")&.downcase == name&.downcase }
+            created << match if match
+          else
+            # Square expects percentage as string (e.g. "8.25")
+            # Clover stores rate as integer (e.g. 825000 = 8.25%)
+            percentage = if rate_data["rate"]
+                           (rate_data["rate"] / 100_000.0).to_s
+                         else
+                           rate_data["percentage"]&.to_s || "0"
+                         end
+
+            result = services.catalog.upsert_tax(
+              name: name,
+              percentage: percentage,
+            )
+            obj = result&.dig("catalog_object")
+            created << obj if obj
+          end
+        end
+
+        logger.info "Tax rates ready: #{created.size}"
+        created
+      end
+
+      # Ensure team members exist (idempotent).
+      def setup_team_members
+        logger.info "Setting up team members..."
+        members = services.team.ensure_team_members(count: DEFAULT_TEAM_MEMBER_COUNT)
+        logger.info "Team members ready: #{members.size}"
+        members
+      end
+
+      # Ensure customers exist (idempotent).
+      def setup_customers
+        logger.info "Setting up customers..."
+        customers = services.customer.ensure_customers(count: DEFAULT_CUSTOMER_COUNT)
+        logger.info "Customers ready: #{customers.size}"
+        customers
+      end
+
+      # Delete all entities (for clean slate)
+      def delete_all
+        logger.warn "=" * 60
+        logger.warn "DELETING ALL ENTITIES..."
+        logger.warn "=" * 60
+
+        # Delete catalog objects (items, categories, discounts, taxes)
+        services.catalog.delete_all
+
+        # Delete customers
+        customers = services.customer.list_customers
+        customers.each do |c|
+          services.customer.delete_customer(c["id"])
+        end
+
+        logger.info "All entities deleted"
+      end
+
+      # Extract variation IDs from catalog items (needed for creating orders).
+      # Square orders reference item variation IDs, not item IDs.
+      #
+      # @param items [Array<Hash>] Array of catalog item objects
+      # @return [Array<Hash>] Array of { item_id:, variation_id:, name:, price_cents: }
+      def self.extract_variation_ids(items)
+        items.filter_map do |item|
+          variations = item.dig("item_data", "variations") || []
+          next if variations.empty?
+
+          variation = variations.first
+          price = variation.dig("item_variation_data", "price_money", "amount") || 0
+
+          {
+            item_id: item["id"],
+            variation_id: variation["id"],
+            name: item.dig("item_data", "name"),
+            price_cents: price,
+          }
+        end
+      end
+    end
+  end
+end