18
18
19
19
import jetbrains .buildServer .serverSide .ProjectManager ;
20
20
import jetbrains .buildServer .serverSide .SBuildServer ;
21
- import jetbrains .buildServer .ssh .SecureServerSshKeyManager ;
22
21
import jetbrains .buildServer .ssh .ServerSshKeyManager ;
23
22
import jetbrains .buildServer .ssh .TeamCitySshKey ;
24
23
import jetbrains .buildServer .util .MultiMap ;
25
- import net .kautler .teamcity .ssh_tunnel .common .Constants ;
26
24
import net .kautler .teamcity .ssh_tunnel .common .ParametersHelper ;
27
25
import org .jetbrains .annotations .NotNull ;
26
+ import org .springframework .context .ApplicationContext ;
28
27
29
28
import java .util .List ;
30
29
import java .util .Map ;
@@ -44,12 +43,20 @@ public class ServerParametersHelper extends ParametersHelper {
44
43
@ NotNull
45
44
private final SBuildServer buildServer ;
46
45
46
+ @ SuppressWarnings ("unchecked" )
47
47
public ServerParametersHelper (@ NotNull ProjectManager projectManager ,
48
- @ NotNull SecureServerSshKeyManager sshKeyManager ,
48
+ @ NotNull ApplicationContext applicationContext ,
49
49
@ NotNull SBuildServer buildServer ) {
50
50
this .projectManager = projectManager ;
51
- this .sshKeyManager = sshKeyManager ;
52
51
this .buildServer = buildServer ;
52
+ Class <? extends ServerSshKeyManager > serverSshKeyManagerClass ;
53
+ try {
54
+ serverSshKeyManagerClass = (Class <? extends ServerSshKeyManager >) Class .forName ("jetbrains.buildServer.ssh.SecureServerSshKeyManager" );
55
+ } catch (ClassNotFoundException e ) {
56
+ serverSshKeyManagerClass = ServerSshKeyManager .class ;
57
+ }
58
+ this .sshKeyManager = applicationContext .getBean (serverSshKeyManagerClass );
59
+ System .out .println (this .sshKeyManager );
53
60
}
54
61
55
62
@ Override
0 commit comments