```ruby
module ShopifyLoginProtection

  def shopify_session
    if session[:shopify]
      begin
        # session[:shopify] set in LoginController#finalize
        ActiveResource::Base.site = session[:shopify].site
        yield
      ensure 
        ActiveResource::Base.site = nil
      end
    else            
      session[:return_to] = request.request_uri
      redirect_to :controller => 'login'      
    end
  end
  
  def current_shop
    session[:shopify]
  end
    
end```

Use ShopifyAPI instead of ActiveResource
user
```ruby
module ShopifyLoginProtection

  def shopify_session
    if session[:shopify]
      begin
        # session[:shopify] set in LoginController#finalize
        Shopify::Base.site = session[:shopify].site
        yield
      ensure 
        ShopifyAPI::Base.site = nil
      end
    else            
      session[:return_to] = request.request_uri
      redirect_to :controller => 'login'      
    end
  end
  
  def current_shop
    session[:shopify]
  end
    
end```
assistant