shopify-cli/lib/shopify_cli/commands/login.rb at 0c1dc46115f85d8e2c0a15eb204a1c32e80af497 · Shopify/shopify-cli · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
require "shopify_cli"

module ShopifyCLI
  module Commands
    class Login < ShopifyCLI::Command
      PROTOCOL_REGEX = /^https?\:\/\//
      PERMANENT_DOMAIN_SUFFIX = /\.myshopify\.(com|io)$/

      options do |parser, flags|
        parser.on("-s", "--store=STORE") { |url| flags[:shop] = url }
        # backwards compatibility allow 'shop' for now
        parser.on("--shop=SHOP") { |url| flags[:shop] = url }
        parser.on("--password=PASSWORD") { |password| flags[:password] = password }
      end

      def call(*)
        shop = (options.flags[:shop] || @ctx.getenv("SHOPIFY_SHOP" || nil))
        ShopifyCLI::DB.set(shop: self.class.validate_shop(shop, context: @ctx)) unless shop.nil?

        if shop.nil? && Shopifolk.check
          Shopifolk.reset
          @ctx.puts(@ctx.message("core.tasks.select_org_and_shop.identified_as_shopify"))
          message = @ctx.message("core.tasks.select_org_and_shop.first_party")
          if CLI::UI::Prompt.confirm(message, default: false)
            Shopifolk.act_as_shopify_organization
          end
        end

        password = options.flags[:password] || @ctx.getenv("SHOPIFY_PASSWORD")
        if !password.nil? && !password.empty?
          ShopifyCLI::DB.set(shopify_exchange_token: password)
        else
          IdentityAuth.new(ctx: @ctx).authenticate(spinner: true)
          org = select_organization
          ShopifyCLI::DB.set(organization_id: org["id"].to_i) unless org.nil?
          Whoami.call([], "whoami")
        end
      end

      def self.help
        ShopifyCLI::Context.message("core.login.help", ShopifyCLI::TOOL_NAME)
      end

      def self.validate_shop(shop, context:)
        permanent_domain = shop_to_permanent_domain(shop)
        context.abort(context.message("core.login.invalid_shop", shop)) unless permanent_domain
        permanent_domain
      end

      def self.shop_to_permanent_domain(shop)
        url = if PROTOCOL_REGEX =~ shop
          shop
        elsif shop.include?(".")
          "https://#{shop}"
        else
          "https://#{shop}.myshopify.com"
        end

        # Make a request to see if it exists or if we get redirected to the permanent domain one
        uri = URI.parse(url)
        Net::HTTP.start(uri.host, use_ssl: true) do |http|
          response = http.request_head("/admin")
          case response
          when Net::HTTPSuccess, Net::HTTPSeeOther
            uri.host
          when Net::HTTPFound
            domain = URI.parse(response["location"]).host
            if PERMANENT_DOMAIN_SUFFIX =~ domain
              domain
            end
          end
        end
      end

      private

      def select_organization
        organizations = []
        CLI::UI::Spinner.spin(@ctx.message("core.login.spinner.loading_organizations")) do
          organizations = ShopifyCLI::PartnersAPI::Organizations.fetch_all(@ctx)
        end

        if organizations.count == 0
          nil
        elsif organizations.count == 1
          organizations.first
        else
          org_id = CLI::UI::Prompt.ask(@ctx.message("core.tasks.select_org_and_shop.organization_select")) do |handler|
            organizations.each do |o|
              handler.option(@ctx.message("core.partners_api.org_name_and_id", o["businessName"], o["id"])) { o["id"] }
            end
          end
          organizations.find { |o| o["id"] == org_id }
        end
      end
    end
  end
end