1+ import { WebApi } from 'azure-devops-node-api' ;
2+ import axios from 'axios' ;
3+ import {
4+ AzureDevOpsError ,
5+ AzureDevOpsResourceNotFoundError ,
6+ AzureDevOpsValidationError ,
7+ AzureDevOpsPermissionError ,
8+ } from '../../../shared/errors' ;
9+ import {
10+ SearchWikiOptions ,
11+ WikiSearchRequest ,
12+ WikiSearchResponse ,
13+ } from '../types' ;
14+
15+ /**
16+ * Search for wiki pages in Azure DevOps projects
17+ *
18+ * @param connection The Azure DevOps WebApi connection
19+ * @param options Parameters for searching wiki pages
20+ * @returns Search results for wiki pages
21+ */
22+ export async function searchWiki (
23+ connection : WebApi ,
24+ options : SearchWikiOptions ,
25+ ) : Promise < WikiSearchResponse > {
26+ try {
27+ // Prepare the search request
28+ const searchRequest : WikiSearchRequest = {
29+ searchText : options . searchText ,
30+ $skip : options . skip ,
31+ $top : options . top ,
32+ filters : {
33+ Project : [ options . projectId ] ,
34+ } ,
35+ includeFacets : options . includeFacets ,
36+ } ;
37+
38+ // Add custom filters if provided
39+ if ( options . filters && options . filters . Project && options . filters . Project . length > 0 ) {
40+ if ( searchRequest . filters && searchRequest . filters . Project ) {
41+ searchRequest . filters . Project = [
42+ ...searchRequest . filters . Project ,
43+ ...options . filters . Project ,
44+ ] ;
45+ }
46+ }
47+
48+ // Get the authorization header from the connection
49+ const authHeader = await getAuthorizationHeader ( connection ) ;
50+
51+ // Extract organization and project from the connection URL
52+ const { organization, project } = extractOrgAndProject (
53+ connection ,
54+ options . projectId ,
55+ ) ;
56+
57+ // Make the search API request
58+ const searchUrl = `https://almsearch.dev.azure.com/${ organization } ${ project } ;
59+ const searchResponse = await axios . post < WikiSearchResponse > (
60+ searchUrl ,
61+ searchRequest ,
62+ {
63+ headers : {
64+ Authorization : authHeader ,
65+ 'Content-Type' : 'application/json' ,
66+ } ,
67+ } ,
68+ ) ;
69+
70+ return searchResponse . data ;
71+ } catch ( error ) {
72+ // If it's already an AzureDevOpsError, rethrow it
73+ if ( error instanceof AzureDevOpsError ) {
74+ throw error ;
75+ }
76+
77+ // Handle axios errors
78+ if ( axios . isAxiosError ( error ) ) {
79+ const status = error . response ?. status ;
80+ const message = error . response ?. data ?. message || error . message ;
81+
82+ if ( status === 404 ) {
83+ throw new AzureDevOpsResourceNotFoundError (
84+ `Resource not found: ${ message } ,
85+ ) ;
86+ } else if ( status === 400 ) {
87+ throw new AzureDevOpsValidationError (
88+ `Invalid request: ${ message } ,
89+ error . response ?. data ,
90+ ) ;
91+ } else if ( status === 401 || status === 403 ) {
92+ throw new AzureDevOpsPermissionError ( `Permission denied: ${ message } ) ;
93+ } else {
94+ // For other axios errors, wrap in a generic AzureDevOpsError
95+ throw new AzureDevOpsError ( `Azure DevOps API error: ${ message } ) ;
96+ }
97+
98+ // This return is never reached but helps TypeScript understand the control flow
99+ return null as never ;
100+ }
101+
102+ // Otherwise, wrap it in a generic error
103+ throw new AzureDevOpsError (
104+ `Failed to search wiki: ${ error instanceof Error ? error . message : String ( error ) } ,
105+ ) ;
106+ }
107+ }
108+
109+ /**
110+ * Extract organization and project from the connection URL
111+ *
112+ * @param connection The Azure DevOps WebApi connection
113+ * @param projectId The project ID or name
114+ * @returns The organization and project
115+ */
116+ function extractOrgAndProject (
117+ connection : WebApi ,
118+ projectId : string ,
119+ ) : { organization : string ; project : string } {
120+ // Extract organization from the connection URL
121+ const url = connection . serverUrl ;
122+ const match = url . match ( / h t t p s ? : \/ \/ d e v \. a z u r e \. c o m \/ ( [ ^ / ] + ) / ) ;
123+ const organization = match ? match [ 1 ] : '' ;
124+
125+ if ( ! organization ) {
126+ throw new AzureDevOpsValidationError (
127+ 'Could not extract organization from connection URL' ,
128+ ) ;
129+ }
130+
131+ return {
132+ organization,
133+ project : projectId ,
134+ } ;
135+ }
136+
137+ /**
138+ * Get the authorization header from the connection
139+ *
140+ * @param connection The Azure DevOps WebApi connection
141+ * @returns The authorization header
142+ */
143+ async function getAuthorizationHeader ( connection : WebApi ) : Promise < string > {
144+ try {
145+ // For PAT authentication, we can construct the header directly
146+ if (
147+ process . env . AZURE_DEVOPS_AUTH_METHOD ?. toLowerCase ( ) === 'pat' &&
148+ process . env . AZURE_DEVOPS_PAT
149+ ) {
150+ // For PAT auth, we can construct the Basic auth header directly
151+ const token = process . env . AZURE_DEVOPS_PAT ;
152+ const base64Token = Buffer . from ( `:${ token } ) . toString ( 'base64' ) ;
153+ return `Basic ${ base64Token } ;
154+ }
155+
156+ // For other auth methods, we'll make a simple API call to get a valid token
157+ // This is a workaround since we can't directly access the auth handler's token
158+ const coreApi = await connection . getCoreApi ( ) ;
159+ await coreApi . getProjects ( ) ;
160+
161+ // At this point, the connection should have made a request and we can
162+ // extract the auth header from the most recent request
163+ // If this fails, we'll fall back to a default approach
164+ return `Basic ${ Buffer . from ( ':' + process . env . AZURE_DEVOPS_PAT ) . toString ( 'base64' ) } ;
165+ } catch ( error ) {
166+ throw new AzureDevOpsValidationError (
167+ `Failed to get authorization header: ${ error instanceof Error ? error . message : String ( error ) } ,
168+ ) ;
169+ }
170+ }
0 commit comments