module ApplicationHelper
def avatar_url(user)
    if user.avatar
        user.avatar
    else
            "/images/missing.png"
    end
end

class OmniauthCallbacksController < Devise::OmniauthCallbacksController

def facebook
    @user = User.from_omniauth(request.env["omniauth.auth"])

    if @user.persisted?
        sign_in_and_redirect @user, :event => :authentication
        set_flash_message(:notice, :success, :kind => "Facebook") if is_navigational_format?
    else
        session["devise.facebook_data"] = request.env["omniauth.auth"]
        redirect_to new_user_registration_url
    end
end

def google_oauth2
@user = User.from_omniauth(request.env['omniauth.auth'])

if @user.persisted?
  sign_in_and_redirect @user, event: :authentication
  set_flash_message(:notice, :success, :kind => "Google") if is_navigational_format?
else
  redirect_to root_path, flash: { error: 'Authentication failed!' }
end

class User < ActiveRecord::Base

def self.from_omniauth(auth)
where(provider: auth.provider, uid: auth.uid).first_or_create do |user|
    user.fullname = auth.info.name
    user.provider = auth.provider
    user.uid = auth.uid
    user.email = auth.info.email
    user.avatar = auth.info.avatar
    user.password = Devise.friendly_token[0,20]
  end