Revert "Remove old search API which is no longer in use"

endpoints/api/search.py

@@ -1,6 +1,6 @@
 """ Conduct searches against all registry context. """
 
-from endpoints.api import (ApiResource, parse_args, query_param, nickname, resource,
+from endpoints.api import (ApiResource, parse_args, query_param, truthy_bool, nickname, resource,
                            require_scope, path_param)
 from data import model
 from auth.permissions import (OrganizationMemberPermission, ReadRepositoryPermission,
@@ -15,6 +15,86 @@ from util.names import parse_robot_username
 
 import math
 
+@resource('/v1/entities/<prefix>')
+class EntitySearch(ApiResource):
+  """ Resource for searching entities. """
+  @path_param('prefix', 'The prefix of the entities being looked up')
+  @parse_args()
+  @query_param('namespace', 'Namespace to use when querying for org entities.', type=str,
+               default='')
+  @query_param('includeTeams', 'Whether to include team names.', type=truthy_bool, default=False)
+  @query_param('includeOrgs', 'Whether to include orgs names.', type=truthy_bool, default=False)
+  @nickname('getMatchingEntities')
+  def get(self, prefix, parsed_args):
+    """ Get a list of entities that match the specified prefix. """
+    teams = []
+    org_data = []
+
+    namespace_name = parsed_args['namespace']
+    robot_namespace = None
+    organization = None
+
+    try:
+      organization = model.organization.get_organization(namespace_name)
+
+      # namespace name was an org
+      permission = OrganizationMemberPermission(namespace_name)
+      if permission.can():
+        robot_namespace = namespace_name
+
+        if parsed_args['includeTeams']:
+          teams = model.team.get_matching_teams(prefix, organization)
+
+        if (parsed_args['includeOrgs'] and AdministerOrganizationPermission(namespace_name) and
+            namespace_name.startswith(prefix)):
+          org_data = [{
+            'name': namespace_name,
+            'kind': 'org',
+            'is_org_member': True,
+            'avatar': avatar.get_data_for_org(organization),
+          }]
+
+    except model.organization.InvalidOrganizationException:
+      # namespace name was a user
+      user = get_authenticated_user()
+      if user and user.username == namespace_name:
+        # Check if there is admin user permissions (login only)
+        admin_permission = UserAdminPermission(user.username)
+        if admin_permission.can():
+          robot_namespace = namespace_name
+
+    users = model.user.get_matching_users(prefix, robot_namespace, organization)
+
+    def entity_team_view(team):
+      result = {
+        'name': team.name,
+        'kind': 'team',
+        'is_org_member': True,
+        'avatar': avatar.get_data_for_team(team)
+      }
+      return result
+
+    def user_view(user):
+      user_json = {
+        'name': user.username,
+        'kind': 'user',
+        'is_robot': user.robot,
+        'avatar': avatar.get_data_for_user(user)
+      }
+
+      if organization is not None:
+        user_json['is_org_member'] = user.robot or user.is_org_member
+
+      return user_json
+
+    team_data = [entity_team_view(team) for team in teams]
+    user_data = [user_view(user) for user in users]
+
+    return {
+      'results': team_data + user_data + org_data
+    }
+
+
 def search_entity_view(username, entity, get_short_name=None):
   kind = 'user'
   avatar_data = avatar.get_data_for_user(entity)