Retour à la ligne dans les blocs de code pour éviter la barre de défilement horizontale

Est-il possible d’activer le retour à la ligne pour les blocs de code afin d’éviter la barre de défilement horizontale ?

Encore mieux si cela peut être configuré pour n’affecter que les blocs de code d’un langage spécifique.

.lang-jai.hljs {
  /* Désactiver la hauteur maximale pour éviter la barre de défilement verticale ; afficher toujours les blocs de code en entier */
  max-height: none;

  /* Activer le retour à la ligne dans le code jai pour éviter la barre de défilement horizontale. */
  overflow-x: auto; 
  white-space: pre-wrap;
}