Commit 3ecf41ec authored by Yorick Peterse's avatar Yorick Peterse

Added docs for the new EE injection methods

parent cc6619a8
...@@ -125,20 +125,24 @@ This also applies to views. ...@@ -125,20 +125,24 @@ This also applies to views.
### EE features based on CE features ### EE features based on CE features
For features that build on existing CE features, write a module in the `EE` For features that build on existing CE features, write a module in the `EE`
namespace and `prepend` it in the CE class, on the last line of the file that namespace and inject it in the CE class, on the last line of the file that the
the class resides in. This makes conflicts less likely to happen during CE to EE class resides in. This makes conflicts less likely to happen during CE to EE
merges because only one line is added to the CE class - the `prepend` line. For merges because only one line is added to the CE class - the line that injects
example, to prepend a module into the `User` class you would use the following the module. For example, to prepend a module into the `User` class you would use
approach: the following approach:
```ruby ```ruby
class User < ActiveRecord::Base class User < ActiveRecord::Base
# ... lots of code here ... # ... lots of code here ...
end end
User.prepend(EE::User) User.prepend_if_ee('EE::User')
``` ```
Do not use methods such as `prepend`, `extend`, and `include`. Instead, use
`prepend_if_ee`, `extend_if_ee`, or `include_if_ee`. These methods take a
_String_ containing the full module name as the argument, not the module itself.
Since the module would require an `EE` namespace, the file should also be Since the module would require an `EE` namespace, the file should also be
put in an `ee/` sub-directory. For example, we want to extend the user model put in an `ee/` sub-directory. For example, we want to extend the user model
in EE, so we have a module called `::EE::User` put inside in EE, so we have a module called `::EE::User` put inside
...@@ -255,7 +259,7 @@ class ApplicationController < ActionController::Base ...@@ -255,7 +259,7 @@ class ApplicationController < ActionController::Base
# ... # ...
end end
ApplicationController.prepend(EE::ApplicationController) ApplicationController.prepend_if_ee('EE::ApplicationController')
``` ```
And create a new file in the `ee/` sub-directory with the altered And create a new file in the `ee/` sub-directory with the altered
...@@ -504,9 +508,9 @@ EE-specific LDAP classes in `ee/lib/ee/gitlab/ldap`. ...@@ -504,9 +508,9 @@ EE-specific LDAP classes in `ee/lib/ee/gitlab/ldap`.
### Code in `lib/api/` ### Code in `lib/api/`
It can be very tricky to extend EE features by a single line of `prepend`, It can be very tricky to extend EE features by a single line of `prepend_if_ee`,
and for each different [Grape](https://github.com/ruby-grape/grape) feature, and for each different [Grape](https://github.com/ruby-grape/grape) feature, we
we might need different strategies to extend it. To apply different strategies might need different strategies to extend it. To apply different strategies
easily, we would use `extend ActiveSupport::Concern` in the EE module. easily, we would use `extend ActiveSupport::Concern` in the EE module.
Put the EE module files following Put the EE module files following
...@@ -543,12 +547,12 @@ constants. ...@@ -543,12 +547,12 @@ constants.
We can define `params` and utilize `use` in another `params` definition to We can define `params` and utilize `use` in another `params` definition to
include params defined in EE. However, we need to define the "interface" first include params defined in EE. However, we need to define the "interface" first
in CE in order for EE to override it. We don't have to do this in other places in CE in order for EE to override it. We don't have to do this in other places
due to `prepend`, but Grape is complex internally and we couldn't easily do due to `prepend_if_ee`, but Grape is complex internally and we couldn't easily
that, so we'll follow regular object-oriented practices that we define the do that, so we'll follow regular object-oriented practices that we define the
interface first here. interface first here.
For example, suppose we have a few more optional params for EE. We can move the For example, suppose we have a few more optional params for EE. We can move the
params out of the `Grape::API` class to a helper module, so we can `prepend` it params out of the `Grape::API` class to a helper module, so we can inject it
before it would be used in the class. before it would be used in the class.
```ruby ```ruby
...@@ -583,7 +587,7 @@ module API ...@@ -583,7 +587,7 @@ module API
end end
end end
API::Helpers::ProjectsHelpers.prepend(EE::API::Helpers::ProjectsHelpers) API::Helpers::ProjectsHelpers.prepend_if_ee('EE::API::Helpers::ProjectsHelpers')
``` ```
We could override it in EE module: We could override it in EE module:
...@@ -624,7 +628,7 @@ module API ...@@ -624,7 +628,7 @@ module API
end end
end end
API::JobArtifacts.prepend(EE::API::JobArtifacts) API::JobArtifacts.prepend_if_ee('EE::API::JobArtifacts')
``` ```
And then we can follow regular object-oriented practices to override it: And then we can follow regular object-oriented practices to override it:
...@@ -677,7 +681,7 @@ module API ...@@ -677,7 +681,7 @@ module API
end end
end end
API::MergeRequests.prepend(EE::API::MergeRequests) API::MergeRequests.prepend_if_ee('EE::API::MergeRequests')
``` ```
Note that `update_merge_request_ee` doesn't do anything in CE, but Note that `update_merge_request_ee` doesn't do anything in CE, but
...@@ -717,8 +721,8 @@ Sometimes we need to use different arguments for a particular API route, and we ...@@ -717,8 +721,8 @@ Sometimes we need to use different arguments for a particular API route, and we
can't easily extend it with an EE module because Grape has different context in can't easily extend it with an EE module because Grape has different context in
different blocks. In order to overcome this, we need to move the data to a class different blocks. In order to overcome this, we need to move the data to a class
method that resides in a separate module or class. This allows us to extend that method that resides in a separate module or class. This allows us to extend that
module or class before its data is used, without having to place a `prepend` in module or class before its data is used, without having to place a
the middle of CE code. `prepend_if_ee` in the middle of CE code.
For example, in one place we need to pass an extra argument to For example, in one place we need to pass an extra argument to
`at_least_one_of` so that the API could consider an EE-only argument as the `at_least_one_of` so that the API could consider an EE-only argument as the
...@@ -739,7 +743,7 @@ module API ...@@ -739,7 +743,7 @@ module API
end end
end end
API::MergeRequests::Parameters.prepend(EE::API::MergeRequests::Parameters) API::MergeRequests::Parameters.prepend_if_ee('EE::API::MergeRequests::Parameters')
# api/merge_requests.rb # api/merge_requests.rb
module API module API
...@@ -789,7 +793,7 @@ class Identity < ActiveRecord::Base ...@@ -789,7 +793,7 @@ class Identity < ActiveRecord::Base
[:provider] [:provider]
end end
prepend EE::Identity prepend_if_ee('EE::Identity')
validates :extern_uid, validates :extern_uid,
allow_blank: true, allow_blank: true,
...@@ -841,7 +845,7 @@ class Identity < ActiveRecord::Base ...@@ -841,7 +845,7 @@ class Identity < ActiveRecord::Base
end end
end end
Identity::UniquenessScopes.prepend(EE::Identity::UniquenessScopes) Identity::UniquenessScopes.prepend_if_ee('EE::Identity::UniquenessScopes')
# app/models/identity.rb # app/models/identity.rb
class Identity < ActiveRecord::Base class Identity < ActiveRecord::Base
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment