From dcbe1f2f2ffcedbae0f6b69ea75db7ffbd75c4ec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=82=B1=E5=8D=9A=E4=BA=9E?= <naruto0426@rulingcom.com>
Date: Mon, 8 May 2023 23:45:15 +0800
Subject: [PATCH] fix error

---
 app/controllers/admin/register_api_controller.rb | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/app/controllers/admin/register_api_controller.rb b/app/controllers/admin/register_api_controller.rb
index ddc3454..c7844b6 100644
--- a/app/controllers/admin/register_api_controller.rb
+++ b/app/controllers/admin/register_api_controller.rb
@@ -59,7 +59,12 @@ class Admin::RegisterApiController < ApplicationController
 
 	def re_register_url
 		site_token = params[:store_token]
+		send_token = false
 		site = RegisteredSite.find_by(:uid => site_token) rescue nil
+		if site.nil? && site_token.blank?
+			send_token = true
+			site = RegisteredSite.find_by(:site_domain => params[:site_domain]) rescue nil
+		end
 		if !site.nil?
 			site.site_domain = params[:site_domain]
 			site.save
@@ -77,7 +82,11 @@ class Admin::RegisterApiController < ApplicationController
 												end
 												{title: v.title,content: v.content,see_more_link: v.see_more_link}
 										end
-			render  :json => {"success" => true,'site_anncs'=>site_anncs}.to_json
+			if send_token
+				render  :json => {"success" => true, 'site_anncs'=>site_anncs, 'store_token' => site_token}.to_json
+			else
+				render  :json => {"success" => true, 'site_anncs'=>site_anncs}.to_json
+			end
 		else
 			render  :json => {"success" => false}.to_json
 		end