feature/Upload user avatar from github

Author: LewisYoul

app/controllers/omniauth_callbacks_controller.rb

@@ -3,16 +3,12 @@ class OmniauthCallbacksController < Devise::OmniauthCallbacksController
 	skip_before_action :verify_authenticity_token, only: :github
 
 	def github
-		# You need to implement the method below in your model (e.g. app/models/user.rb)
 		@user = User.from_omniauth(request.env["omniauth.auth"])
 
 		if @user.persisted?
-			flash[:notice] = 'Signed in from GitHub'
-			sign_in_and_redirect @user# this will throw if @user is not activated
+			sign_in_and_redirect @user
 		else
-			# byebug
-			# session["devise.github_data"] = request.env["omniauth.auth"].except(:extra) # Removing extra as it can overflow some session stores
-			redirect_to new_user_registration_url(notice: 'Unable to sign in with Github. Perhaps you already signed up with the associated email?')
+			redirect_to new_user_registration_url(notice: 'Unable to sign in with Github.')
 		end
 	end
 

app/models/user.rb

@@ -1,3 +1,5 @@
+require 'open-uri'
+
 class User < ApplicationRecord
   include Rails.application.routes.url_helpers
 
@@ -142,7 +144,7 @@ def self.authenticate(params)
   end
 
   def self.from_omniauth(auth)
-    where(provider: auth.provider, uid: auth.uid).first_or_create do |user|
+    user = where(provider: auth.provider, uid: auth.uid).first_or_create do |user|
       user.email = auth.info.email
       user.password = Devise.friendly_token[0, 20]
       user.name = auth.info.nickname   # assuming the user model has a name
@@ -151,10 +153,30 @@ def self.from_omniauth(auth)
       # uncomment the line below to skip the confirmation emails.
       user.skip_confirmation!
     end
+
+    byebug
+
+    if user.persisted?
+      user.attach_avatar_from_omniauth(auth.info.image)
+    end
+
+    byebug
+
+    user
+  end
+
+  def attach_avatar_from_omniauth(avatar_url)
+    begin
+      image = open(avatar_url)
+
+      avatar.attach(io: image, filename: 'whocares')
+    rescue
+    end
   end
 
   private
 
+
   def create_default_folder
     folders.create!(name: DEFAULT_FOLDER_NAME)
   end