mains_controller.rb in phcmemberspro-6.0.0

- old
+ new

@@ -8,11 +8,11 @@
 		layout :phcmemberspro_membersmain_layouts_resolver
 		before_action :set_members_main, only: [:show, :edit, :update, :destroy]
 
 		# Member Profile List
 		def index
-			@members_mains = Members::Main.where(user_id: current_user.id)
+			@members_mains = Members::Main.where(membership_id: membership_info.id)
 		end
 
 		# Detailed Member Profile
 		def show
 		end
@@ -26,21 +26,26 @@
 		def edit
 		end
 
 		# Create Member Profile Action
 		def create
-			@members_main = Members::Main.new(members_main_params)
 			@members_main.user_id = current_user.id
+			@members_main.membership_id = membership_info.id
+			@members_main.oganization_id = membership_info.org_id
+			@members_main = Members::Main.new(members_main_params)
 			if @members_main.save
 				redirect_to members_mains_path, notice: 'Member profile has been successfully created.'
 				else
 					render :new
 			end
 		end
 
 		# Update Member Profile Action
 		def update
+			@members_main.user_id = current_user.id
+			@members_main.membership_id = membership_info.id
+			@members_main.oganization_id = membership_info.org_id
 			if @members_main.update(members_main_params)
 				redirect_to members_mains_path, notice: 'Member profile has been successfully updated.'
 				else
 					render :edit
 			end
@@ -57,17 +62,22 @@
 		# Grab User Session Key (For ID)
 		def current_user
 			@_current_user ||= AuthRocket::Session.from_token(session[:ar_token]).try(:user)
 		end
 
+		# Get Current User from Above and Get Membership Info
+		def membership_info
+		    AuthRocket::Membership.all(user_id: current_user.id).first
+		end
+
 		# Common Callbacks
 		def set_members_main
 			@members_main = Members::Main.find(params[:id])
 		end
 
 		# White List
 		def members_main_params
-			params.require(:members_main).permit(:mfirstname, :mlastname, :mtitle, :memail, :mphone, :mnotes, :smfacebook, :smtwitter, :smlinkedin, :user_id)
+			params.require(:members_main).permit(:mfirstname, :mlastname, :mtitle, :memail, :mphone, :mnotes, :smfacebook, :smtwitter, :smlinkedin, :user_id, :membership_id, :oganization_id)
 		end
 
 	end
 end