application_controller.rb 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423
  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. :config_frontend,
  7. :role?,
  8. :model_create_render,
  9. :model_update_render,
  10. :model_restory_render,
  11. :mode_show_rendeder,
  12. :model_index_render
  13. skip_before_action :verify_authenticity_token
  14. before_action :set_user, :session_update
  15. before_action :cors_preflight_check
  16. after_action :user_device_update, :set_access_control_headers
  17. after_action :trigger_events
  18. # For all responses in this controller, return the CORS access control headers.
  19. def set_access_control_headers
  20. headers['Access-Control-Allow-Origin'] = '*'
  21. headers['Access-Control-Allow-Methods'] = 'POST, GET, PUT, DELETE, OPTIONS'
  22. headers['Access-Control-Max-Age'] = '1728000'
  23. 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'
  24. headers['Access-Control-Allow-Credentials'] = 'true'
  25. end
  26. # If this is a preflight OPTIONS request, then short-circuit the
  27. # request, return only the necessary headers and return an empty
  28. # text/plain.
  29. def cors_preflight_check
  30. return if request.method != 'OPTIONS'
  31. headers['Access-Control-Allow-Origin'] = '*'
  32. headers['Access-Control-Allow-Methods'] = 'POST, GET, PUT, DELETE, OPTIONS'
  33. 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'
  34. headers['Access-Control-Max-Age'] = '1728000'
  35. headers['Access-Control-Allow-Credentials'] = 'true'
  36. render text: '', content_type: 'text/plain'
  37. false
  38. end
  39. private
  40. # execute events
  41. def trigger_events
  42. Observer::Ticket::Notification.transaction
  43. end
  44. # Finds the User with the ID stored in the session with the key
  45. # :current_user_id This is a common way to handle user login in
  46. # a Rails application; logging in sets the session value and
  47. # logging out removes it.
  48. def current_user
  49. return @_current_user if @_current_user
  50. return if !session[:user_id]
  51. @_current_user = User.lookup(id: session[:user_id])
  52. end
  53. def current_user_set(user)
  54. session[:user_id] = user.id
  55. @_current_user = user
  56. set_user
  57. end
  58. # Sets the current user into a named Thread location so that it can be accessed
  59. # by models and observers
  60. def set_user
  61. return if !current_user
  62. UserInfo.current_user_id = current_user.id
  63. end
  64. # update session updated_at
  65. def session_update
  66. #sleep 0.6
  67. session[:ping] = Time.zone.now.iso8601
  68. # check if remote ip need to be updated
  69. if !session[:remote_ip] || session[:remote_ip] != request.remote_ip
  70. session[:remote_ip] = request.remote_ip
  71. session[:geo] = Service::GeoIp.location(request.remote_ip)
  72. end
  73. # fill user agent
  74. return if session[:user_agent]
  75. session[:user_agent] = request.env['HTTP_USER_AGENT']
  76. end
  77. # user device recent action update
  78. def user_device_update
  79. # return if we are in switch to user mode
  80. return if session[:switched_from_user_id]
  81. # only if user_id exists
  82. return if !session[:user_id]
  83. # only with user device
  84. if !session[:user_device_id]
  85. if params[:fingerprint]
  86. return false if !user_device_log(current_user, 'session')
  87. end
  88. return
  89. end
  90. # check if entry exists / only if write action
  91. method = request.method
  92. return if method == 'GET'
  93. return if method == 'OPTIONS'
  94. return if method == 'HEAD'
  95. # only update if needed
  96. return if session[:user_device_update_at] && session[:user_device_update_at] > Time.zone.now - 5.minutes
  97. session[:user_device_update_at] = Time.zone.now
  98. UserDevice.action(
  99. session[:user_device_id],
  100. session[:user_agent],
  101. session[:remote_ip],
  102. session[:user_id],
  103. )
  104. end
  105. def user_device_log(user, type)
  106. # return if we are in switch to user mode
  107. return true if session[:switched_from_user_id]
  108. # for sessions we need the fingperprint
  109. if !params[:fingerprint] && type == 'session'
  110. render json: { error: 'Need fingerprint param!' }, status: :unprocessable_entity
  111. return false
  112. end
  113. # add defice if needed
  114. user_device = UserDevice.add(
  115. request.env['HTTP_USER_AGENT'],
  116. request.remote_ip,
  117. user.id,
  118. params[:fingerprint],
  119. type,
  120. )
  121. session[:user_device_id] = user_device.id
  122. end
  123. def authentication_check_only(auth_param)
  124. #logger.debug 'authentication_check'
  125. #logger.debug params.inspect
  126. #logger.debug session.inspect
  127. #logger.debug cookies.inspect
  128. # already logged in, early exit
  129. if session.id && session[:user_id]
  130. userdata = User.lookup(id: session[:user_id])
  131. current_user_set(userdata)
  132. return {
  133. auth: true
  134. }
  135. end
  136. error_message = 'authentication failed'
  137. # check sso based authentication
  138. sso_userdata = User.sso(params)
  139. if sso_userdata
  140. session[:persistent] = true
  141. return {
  142. auth: true
  143. }
  144. end
  145. # check http basic based authentication
  146. authenticate_with_http_basic do |username, password|
  147. logger.debug "http basic auth check '#{username}'"
  148. userdata = User.authenticate(username, password)
  149. next if !userdata
  150. current_user_set(userdata)
  151. user_device_log(userdata, 'basic_auth')
  152. return {
  153. auth: true
  154. }
  155. end
  156. # check http token based authentication
  157. if auth_param[:token_action]
  158. authenticate_with_http_token do |token, _options|
  159. logger.debug "token auth check '#{token}'"
  160. userdata = Token.check(
  161. action: auth_param[:token_action],
  162. name: token,
  163. )
  164. next if !userdata
  165. current_user_set(userdata)
  166. user_device_log(userdata, 'token_auth')
  167. return {
  168. auth: true
  169. }
  170. end
  171. end
  172. logger.debug error_message
  173. {
  174. auth: false,
  175. message: error_message,
  176. }
  177. end
  178. def authentication_check(auth_param = {} )
  179. result = authentication_check_only(auth_param)
  180. # check if basic_auth fallback is possible
  181. if auth_param[:basic_auth_promt] && result[:auth] == false
  182. return request_http_basic_authentication
  183. end
  184. # return auth not ok
  185. if result[:auth] == false
  186. render(
  187. json: {
  188. error: result[:message],
  189. },
  190. status: :unauthorized
  191. )
  192. return false
  193. end
  194. # return auth ok
  195. true
  196. end
  197. def role?(role_name)
  198. return false if !current_user
  199. current_user.role?(role_name)
  200. end
  201. def ticket_permission(ticket)
  202. return true if ticket.permission(current_user: current_user)
  203. response_access_deny
  204. false
  205. end
  206. def deny_if_not_role(role_name)
  207. return false if role?(role_name)
  208. response_access_deny
  209. true
  210. end
  211. def valid_session_with_user
  212. return true if current_user
  213. render json: { message: 'No session user!' }, status: :unprocessable_entity
  214. false
  215. end
  216. def response_access_deny
  217. render(
  218. json: {},
  219. status: :unauthorized
  220. )
  221. false
  222. end
  223. def config_frontend
  224. # config
  225. config = {}
  226. Setting.select('name').where(frontend: true ).each { |setting|
  227. config[setting.name] = Setting.get(setting.name)
  228. }
  229. # remember if we can to swich back to user
  230. if session[:switched_from_user_id]
  231. config['switch_back_to_possible'] = true
  232. end
  233. # remember session_id for websocket logon
  234. config['session_id'] = session.id
  235. config
  236. end
  237. # model helper
  238. def model_create_render (object, params)
  239. # create object
  240. generic_object = object.new(object.param_cleanup(params[object.to_app_model_url], true ))
  241. # save object
  242. generic_object.save!
  243. # set relations
  244. generic_object.param_set_associations(params)
  245. model_create_render_item(generic_object)
  246. rescue => e
  247. logger.error e.message
  248. logger.error e.backtrace.inspect
  249. render json: model_match_error(e.message), status: :unprocessable_entity
  250. end
  251. def model_create_render_item (generic_object)
  252. render json: generic_object.attributes_with_associations, status: :created
  253. end
  254. def model_update_render (object, params)
  255. # find object
  256. generic_object = object.find(params[:id])
  257. # save object
  258. generic_object.update_attributes!(object.param_cleanup(params[object.to_app_model_url]))
  259. # set relations
  260. generic_object.param_set_associations(params)
  261. model_update_render_item(generic_object)
  262. rescue => e
  263. logger.error e.message
  264. logger.error e.backtrace.inspect
  265. render json: model_match_error(e.message), status: :unprocessable_entity
  266. end
  267. def model_update_render_item (generic_object)
  268. render json: generic_object.attributes_with_associations, status: :ok
  269. end
  270. def model_destory_render (object, params)
  271. generic_object = object.find(params[:id])
  272. generic_object.destroy
  273. model_destory_render_item()
  274. rescue => e
  275. logger.error e.message
  276. logger.error e.backtrace.inspect
  277. render json: model_match_error(e.message), status: :unprocessable_entity
  278. end
  279. def model_destory_render_item ()
  280. render json: {}, status: :ok
  281. end
  282. def model_show_render (object, params)
  283. if params[:full]
  284. generic_object_full = object.full(params[:id])
  285. render json: generic_object_full, status: :ok
  286. return
  287. end
  288. generic_object = object.find(params[:id])
  289. model_show_render_item(generic_object)
  290. rescue => e
  291. logger.error e.message
  292. logger.error e.backtrace.inspect
  293. render json: model_match_error(e.message), status: :unprocessable_entity
  294. end
  295. def model_show_render_item (generic_object)
  296. render json: generic_object.attributes_with_associations, status: :ok
  297. end
  298. def model_index_render (object, _params)
  299. generic_objects = object.all
  300. if params[:full]
  301. assets = {}
  302. item_ids = []
  303. generic_objects.each {|item|
  304. item_ids.push item.id
  305. assets = item.assets(assets)
  306. }
  307. render json: {
  308. record_ids: item_ids,
  309. assets: assets,
  310. }, status: :ok
  311. return
  312. end
  313. generic_objects_with_associations = []
  314. generic_objects.each {|item|
  315. generic_objects_with_associations.push item.attributes_with_associations
  316. }
  317. model_index_render_result(generic_objects_with_associations)
  318. rescue => e
  319. logger.error e.message
  320. logger.error e.backtrace.inspect
  321. render json: model_match_error(e.message), status: :unprocessable_entity
  322. end
  323. def model_index_render_result (generic_objects)
  324. render json: generic_objects, status: :ok
  325. end
  326. def model_match_error (error)
  327. data = {
  328. error: error
  329. }
  330. if error =~ /(already exists|duplicate key|duplicate entry)/i
  331. data[:error_human] = 'Object already exists!'
  332. end
  333. data
  334. end
  335. end