Blame view

PFE06/src/main/java/com/PFE/ServerManager/AppConfig.java 2.27 KB
a13d6d0f   Unknown   Fichier de config...
1
2
3
4
5
  package com.PFE.ServerManager;
  
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
  import org.springframework.stereotype.Component;
a13d6d0f   Unknown   Fichier de config...
6
7
8
9
10
  import org.yaml.snakeyaml.Yaml;
  import org.yaml.snakeyaml.constructor.Constructor;
  
  import javax.annotation.PostConstruct;
  import java.io.*;
a13d6d0f   Unknown   Fichier de config...
11
12
13
14
15
16
17
18
19
20
21
  
  @Component
  public class AppConfig {
  
      @Autowired
      CustomerRepository customerRepository;
  
      @Autowired
      RoleRepository roleRepository;
  
      @Autowired
65f1240c   sfeutrie   Ajout des groupes...
22
23
24
      TeamRepository teamRepository;
  
      @Autowired
a13d6d0f   Unknown   Fichier de config...
25
26
      BCryptPasswordEncoder bCryptPasswordEncoder;
  
a13d6d0f   Unknown   Fichier de config...
27
28
29
30
      @PostConstruct
      public void init() {
          System.out.println("============= Configuration de l'application =============");
  
cb678ccc   Unknown   Changement pour l...
31
32
33
34
35
          /* Ajout des rôles si non présent */
  
          if(roleRepository.count() != 2) {
              roleRepository.deleteAll();
              Role roleAdmin = new Role();
65f1240c   sfeutrie   Ajout des groupes...
36
              roleAdmin.setRoleId(1);
cb678ccc   Unknown   Changement pour l...
37
38
39
              roleAdmin.setRole("ADMIN");
              roleRepository.save(roleAdmin);
              Role roleUser = new Role();
65f1240c   sfeutrie   Ajout des groupes...
40
              roleUser.setRoleId(2);
cb678ccc   Unknown   Changement pour l...
41
42
43
44
45
46
              roleUser.setRole("USER");
              roleRepository.save(roleUser);
          }
  
          /* Ajout d'un utilisateur ADMIN avec les informations du fichier de conf*/
  
a13d6d0f   Unknown   Fichier de config...
47
48
49
50
51
52
53
54
55
56
57
58
          Yaml yaml = new Yaml(new Constructor(ConfYAML.class));
          InputStream inputStream = null;
          try {
              inputStream = new FileInputStream(new File("app.conf"));
          } catch (FileNotFoundException e) {
              e.printStackTrace();
          }
          ConfYAML conf = yaml.load(inputStream);
  
          Customer n = new Customer();
          n.setEmail(conf.getAdminEmail());
          n.setPassword(bCryptPasswordEncoder.encode(conf.getAdminPassword()));
0e18634e   sfeutrie   modifications min...
59
          n.setCustomerId((int)(customerRepository.count() + 1));
a13d6d0f   Unknown   Fichier de config...
60
61
62
63
          n.setActive(1);
  
          Customer temp = customerRepository.findByEmail(conf.getAdminEmail());
          Role userRole = roleRepository.findByRole("ADMIN");
0e18634e   sfeutrie   modifications min...
64
          n.addRole(userRole);
a13d6d0f   Unknown   Fichier de config...
65
66
67
  
          if(temp == null) {
              customerRepository.save(n);
a13d6d0f   Unknown   Fichier de config...
68
          }
65f1240c   sfeutrie   Ajout des groupes...
69
70
71
72
          /* Partie sur les groupes */
          Team g = new Team();
          g.setTeamId((int)(teamRepository.count() +1));
          g.setTeam("Labo1");
0e18634e   sfeutrie   modifications min...
73
          g.addCustomer(n);
65f1240c   sfeutrie   Ajout des groupes...
74
          teamRepository.save(g);
a13d6d0f   Unknown   Fichier de config...
75
76
      }
  }