application_controller.rb 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369
  1. # Copyright (C) 2012-2014 Zammad Foundation, http://zammad-foundation.org/
  2. class ApplicationController < ActionController::Base
  3. # http_basic_authenticate_with :name => "test", :password => "ttt"
  4. helper_method :current_user,
  5. :authentication_check,
  6. :authentication_check_action_token,
  7. :config_frontend,
  8. :role?,
  9. :model_create_render,
  10. :model_update_render,
  11. :model_restory_render,
  12. :mode_show_rendeder,
  13. :model_index_render
  14. skip_before_action :verify_authenticity_token
  15. before_action :set_user, :session_update
  16. before_action :cors_preflight_check
  17. after_action :set_access_control_headers
  18. after_action :trigger_events
  19. # For all responses in this controller, return the CORS access control headers.
  20. def set_access_control_headers
  21. headers['Access-Control-Allow-Origin'] = '*'
  22. headers['Access-Control-Allow-Methods'] = 'POST, GET, PUT, DELETE, OPTIONS'
  23. headers['Access-Control-Max-Age'] = '1728000'
  24. headers['Access-Control-Allow-Headers'] = 'Content-Type, Depth, User-Agent, X-File-Size, X-Requested-With, If-Modified-Since, X-File-Name, Cache-Control, Accept-Language'
  25. headers['Access-Control-Allow-Credentials'] = 'true'
  26. end
  27. # If this is a preflight OPTIONS request, then short-circuit the
  28. # request, return only the necessary headers and return an empty
  29. # text/plain.
  30. def cors_preflight_check
  31. return if request.method != 'OPTIONS'
  32. headers['Access-Control-Allow-Origin'] = '*'
  33. headers['Access-Control-Allow-Methods'] = 'POST, GET, PUT, DELETE, OPTIONS'
  34. headers['Access-Control-Allow-Headers'] = 'Content-Type, Depth, User-Agent, X-File-Size, X-Requested-With, If-Modified-Since, X-File-Name, Cache-Control, Accept-Language'
  35. headers['Access-Control-Max-Age'] = '1728000'
  36. headers['Access-Control-Allow-Credentials'] = 'true'
  37. render text: '', content_type: 'text/plain'
  38. false
  39. end
  40. private
  41. # execute events
  42. def trigger_events
  43. Observer::Ticket::Notification.transaction
  44. end
  45. # Finds the User with the ID stored in the session with the key
  46. # :current_user_id This is a common way to handle user login in
  47. # a Rails application; logging in sets the session value and
  48. # logging out removes it.
  49. def current_user
  50. return @_current_user if @_current_user
  51. return if !session[:user_id]
  52. @_current_user = User.find( session[:user_id] )
  53. end
  54. def current_user_set(user)
  55. session[:user_id] = user.id
  56. @_current_user = user
  57. set_user
  58. end
  59. # Sets the current user into a named Thread location so that it can be accessed
  60. # by models and observers
  61. def set_user
  62. return if !current_user
  63. UserInfo.current_user_id = current_user.id
  64. end
  65. # update session updated_at
  66. def session_update
  67. #sleep 0.6
  68. session[:ping] = Time.zone.now.iso8601
  69. # check if remote ip need to be updated
  70. if !session[:remote_id] || session[:remote_id] != request.remote_ip
  71. session[:remote_id] = request.remote_ip
  72. session[:geo] = GeoIp.location( request.remote_ip )
  73. end
  74. # fill user agent
  75. return if session[:user_agent]
  76. session[:user_agent] = request.env['HTTP_USER_AGENT']
  77. end
  78. def authentication_check_only
  79. logger.debug 'authentication_check'
  80. session[:request_type] = 1
  81. #logger.debug params.inspect
  82. #logger.debug session.inspect
  83. #logger.debug cookies.inspect
  84. # check http basic auth
  85. authenticate_with_http_basic do |username, password|
  86. logger.debug 'http basic auth check'
  87. session[:request_type] = 2
  88. userdata = User.authenticate( username, password )
  89. message = ''
  90. if !userdata
  91. message = 'authentication failed'
  92. end
  93. # return auth ok
  94. if message == ''
  95. # remember user
  96. session[:user_id] = userdata.id
  97. # set basic auth user to current user
  98. current_user_set(userdata)
  99. return {
  100. auth: true
  101. }
  102. end
  103. # return auth not ok
  104. return {
  105. auth: false,
  106. message: message,
  107. }
  108. end
  109. # check logon session
  110. if params['logon_session']
  111. logon_session = ActiveRecord::SessionStore::Session.where( session_id: params['logon_session'] ).first
  112. if logon_session
  113. userdata = User.find( logon_session.data[:user_id] )
  114. end
  115. session[:request_type] = 3
  116. # set logon session user to current user
  117. current_user_set(userdata)
  118. return {
  119. auth: true
  120. }
  121. end
  122. # check sso
  123. if !session[:user_id]
  124. user = User.sso(params)
  125. # Log the authorizing user in.
  126. if user
  127. session[:user_id] = user.id
  128. end
  129. end
  130. # return auth not ok (no session exists)
  131. if !session[:user_id]
  132. logger.debug 'no valid session, user_id'
  133. message = 'no valid session, user_id'
  134. return {
  135. auth: false,
  136. message: message,
  137. }
  138. end
  139. {
  140. auth: true
  141. }
  142. end
  143. def authentication_check
  144. result = authentication_check_only
  145. # return auth not ok
  146. if result[:auth] == false
  147. render(
  148. json: {
  149. error: result[:message],
  150. },
  151. status: :unauthorized
  152. )
  153. return false
  154. end
  155. # return auth ok
  156. true
  157. end
  158. def authentication_check_action_token(action)
  159. user = Token.check(
  160. action: action,
  161. name: params[:action_token],
  162. )
  163. if !user
  164. logger.debug params.inspect
  165. response_access_deny
  166. return
  167. end
  168. current_user_set( user )
  169. true
  170. end
  171. def role?( role_name )
  172. return false if !current_user
  173. current_user.role?( role_name )
  174. end
  175. def ticket_permission(ticket)
  176. return true if ticket.permission( current_user: current_user )
  177. response_access_deny
  178. false
  179. end
  180. def deny_if_not_role( role_name )
  181. return false if role?( role_name )
  182. response_access_deny
  183. true
  184. end
  185. def valid_session_with_user
  186. return true if current_user
  187. render json: { message: 'No session user!' }, status: :unprocessable_entity
  188. false
  189. end
  190. def response_access_deny
  191. render(
  192. json: {},
  193. status: :unauthorized
  194. )
  195. false
  196. end
  197. def config_frontend
  198. # config
  199. config = {}
  200. Setting.select('name').where( frontend: true ).each { |setting|
  201. config[setting.name] = Setting.get(setting.name)
  202. }
  203. # get all time zones
  204. config['timezones'] = {}
  205. TZInfo::Timezone.all.each { |t|
  206. # ignore the following time zones
  207. next if t.name =~ /^GMT/
  208. next if t.name =~ /^Etc/
  209. next if t.name =~ /^MET/
  210. next if t.name =~ /^MST/
  211. next if t.name =~ /^ROC/
  212. next if t.name =~ /^ROK/
  213. diff = t.current_period.utc_total_offset / 60 / 60
  214. config['timezones'][ t.name ] = diff
  215. }
  216. if session[:switched_from_user_id]
  217. config['switch_back_to_possible'] = true
  218. end
  219. config
  220. end
  221. # model helper
  222. def model_create_render (object, params)
  223. # create object
  224. generic_object = object.new( object.param_cleanup( params[object.to_app_model_url], true ) )
  225. # save object
  226. generic_object.save!
  227. # set relations
  228. generic_object.param_set_associations( params )
  229. model_create_render_item(generic_object)
  230. rescue Exception => e
  231. logger.error e.message
  232. logger.error e.backtrace.inspect
  233. render json: { error: e.message }, status: :unprocessable_entity
  234. end
  235. def model_create_render_item (generic_object)
  236. render json: generic_object.attributes_with_associations, status: :created
  237. end
  238. def model_update_render (object, params)
  239. # find object
  240. generic_object = object.find( params[:id] )
  241. # save object
  242. generic_object.update_attributes!( object.param_cleanup( params[object.to_app_model_url] ) )
  243. # set relations
  244. generic_object.param_set_associations( params )
  245. model_update_render_item( generic_object )
  246. rescue Exception => e
  247. logger.error e.message
  248. logger.error e.backtrace.inspect
  249. render json: { error: e.message }, status: :unprocessable_entity
  250. end
  251. def model_update_render_item (generic_object)
  252. render json: generic_object.attributes_with_associations, status: :ok
  253. end
  254. def model_destory_render (object, params)
  255. generic_object = object.find( params[:id] )
  256. generic_object.destroy
  257. model_destory_render_item()
  258. rescue Exception => e
  259. logger.error e.message
  260. logger.error e.backtrace.inspect
  261. render json: { error: e.message }, status: :unprocessable_entity
  262. end
  263. def model_destory_render_item ()
  264. render json: {}, status: :ok
  265. end
  266. def model_show_render (object, params)
  267. if params[:full]
  268. generic_object_full = object.full( params[:id] )
  269. render json: generic_object_full, status: :ok
  270. return
  271. end
  272. generic_object = object.find( params[:id] )
  273. model_show_render_item(generic_object)
  274. rescue Exception => e
  275. logger.error e.message
  276. logger.error e.backtrace.inspect
  277. render json: { error: e.message }, status: :unprocessable_entity
  278. end
  279. def model_show_render_item (generic_object)
  280. render json: generic_object.attributes_with_associations, status: :ok
  281. end
  282. def model_index_render (object, _params)
  283. generic_objects = object.all
  284. model_index_render_result( generic_objects )
  285. rescue Exception => e
  286. logger.error e.message
  287. logger.error e.backtrace.inspect
  288. render json: { error: e.message }, status: :unprocessable_entity
  289. end
  290. def model_index_render_result (generic_objects)
  291. render json: generic_objects, status: :ok
  292. end
  293. end