wechat oauth: avatar download

app/libs/util.rb

@@ -1,3 +1,5 @@
+require 'open-uri'
+
 module Util
   module_function
 
@@ -29,6 +31,16 @@ module Util
     end
   end
 
+  def download_file(url, save_path)
+    data = open(url, &:read)
+    file = File.new(save_path, 'w+')
+    file.binmode
+    file << data
+    file.flush
+    file.close
+    file
+  end
+
   def logger_error(exception)
     Rails.logger.error(exception.message)
     exception.backtrace.each { |message| Rails.logger.error(message) }

app/services/oauth/create_or_find_wechat_account_service.rb

@@ -33,6 +33,10 @@ class Oauth::CreateOrFindWechatAccountService < ApplicationService
 
         gender = result['sex'].to_i == 1 ? 0 : 1
         user.create_user_extension!(gender: gender)
+
+        # 下载头像
+        avatar_path = Util::FileManage.source_disk_filename(user)
+        Util.download_file(result['headimgurl'], avatar_path)
       end
 
       new_open_user= OpenUsers::Wechat.create!(user: user, uid: result['unionid'])