@@ -26,7 +26,7 @@ export async function searchProblem(): Promise<void> {
2626return ;
2727}
2828const problems :IProblem [ ] = await list . listProblems ( ) ;
29- const choice :IQuickItemEx < string > | undefined = await vscode . window . showQuickPick (
29+ const choice :IQuickItemEx < IProblem > | undefined = await vscode . window . showQuickPick (
3030parseProblemsToPicks ( problems ) ,
3131{
3232matchOnDetail :true ,
@@ -36,7 +36,7 @@ export async function searchProblem(): Promise<void> {
3636if ( ! choice ) {
3737return ;
3838}
39- await showProblemInternal ( problems . find ( ( problem : IProblem ) => problem . id === choice . value ) as IProblem ) ;
39+ await showProblemInternal ( choice . value ) ;
4040}
4141
4242async function showProblemInternal ( node :IProblem ) :Promise < void > {
@@ -57,12 +57,22 @@ async function showProblemInternal(node: IProblem): Promise<void> {
5757if ( outputPath ) {
5858switch ( outputPath [ 1 ] . toLowerCase ( ) ) {
5959case "tag" :
60- const closestTag :string = node . tags . reduce ( ( prev :string , curr :string ) => {
61- return curr . length > prev . length ?
62- curr :
63- prev ;
64- } , "" ) ;
65- outDir = path . join ( outDir , closestTag ) ;
60+ let tag :string | undefined ;
61+ if ( node . tags . length === 1 ) {
62+ tag = node . tags [ 0 ] ;
63+ } else {
64+ tag = await vscode . window . showQuickPick (
65+ node . tags ,
66+ {
67+ matchOnDetail :true ,
68+ placeHolder :"Select one tag" ,
69+ } ,
70+ ) ;
71+ }
72+ if ( ! tag ) {
73+ return ;
74+ }
75+ outDir = path . join ( outDir , tag ) ;
6676break ;
6777case "language" :
6878outDir = path . join ( outDir , language ) ;
@@ -108,13 +118,13 @@ async function showProblemInternal(node: IProblem): Promise<void> {
108118}
109119}
110120
111- async function parseProblemsToPicks ( p :IProblem [ ] ) :Promise < Array < IQuickItemEx < string > > > {
112- return new Promise ( async ( resolve :( res :Array < IQuickItemEx < string > > ) => void ) :Promise < void > => {
113- const picks :Array < IQuickItemEx < string > > = p . map ( ( problem :IProblem ) => Object . assign ( { } , {
121+ async function parseProblemsToPicks ( p :IProblem [ ] ) :Promise < Array < IQuickItemEx < IProblem > > > {
122+ return new Promise ( async ( resolve :( res :Array < IQuickItemEx < IProblem > > ) => void ) :Promise < void > => {
123+ const picks :Array < IQuickItemEx < IProblem > > = p . map ( ( problem :IProblem ) => Object . assign ( { } , {
114124label :`${ parseProblemDecorator ( problem . state , problem . locked ) } ${ problem . id } ${ problem . name } ,
115125description :"" ,
116126detail :`AC rate:${ problem . passRate } ${ problem . difficulty } ,
117- value :problem . id ,
127+ value :problem ,
118128} ) ) ;
119129resolve ( picks ) ;
120130} ) ;