require 'shopify_cli'

module ShopifyCli
  class AdminAPI
    class Schema < Hash
      def self.get(ctx)
        unless ShopifyCli::DB.exists?(:shopify_admin_schema)
          shop = Project.current.env.shop || get_shop(@ctx)
          schema = AdminAPI.query(ctx, 'admin_introspection', shop: shop)
          ShopifyCli::DB.set(shopify_admin_schema: JSON.dump(schema))
        end
        # This is ruby magic for making a new hash with another hash.
        # It wraps the JSON in our Schema Class to have the helper methods
        # available
        self[JSON.parse(ShopifyCli::DB.get(:shopify_admin_schema))]
      end

      def type(name)
        data = self["data"]
        schema = data["__schema"]
        schema["types"].find do |object|
          object['name'] == name.to_s
        end
      end

      def get_names_from_type(name)
        type(name)["enumValues"].map do |object|
          object["name"]
        end
      end

      private

      def get_shop(ctx)
        res = ShopifyCli::Tasks::SelectOrgAndShop.call(ctx)
        domain = res[:shop_domain]
        Project.current.env.update(ctx, :shop, domain)
        domain
      end
    end
  end
end