| 
				
			 | 
			
			 | 
			@@ -1,17 +1,46 @@ | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			import logging | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			import time | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from collections.abc import Mapping | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from typing import Any | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			import click | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from celery import shared_task | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from flask import render_template_string | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from jinja2.runtime import Context | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from jinja2.sandbox import ImmutableSandboxedEnvironment | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from configs import dify_config | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from configs.feature import TemplateMode | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from extensions.ext_mail import mail | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			from libs.email_i18n import get_email_i18n_service | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			logger = logging.getLogger(__name__) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			class SandboxedEnvironment(ImmutableSandboxedEnvironment): | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    def __init__(self, timeout: int, *args: Any, **kwargs: Any): | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        self._timeout_time = time.time() + timeout | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        super().__init__(*args, **kwargs) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    def call(self, context: Context, obj: Any, *args: Any, **kwargs: Any) -> Any: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        if time.time() > self._timeout_time: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			            raise TimeoutError("Template rendering timeout") | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        return super().call(context, obj, *args, **kwargs) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			def _render_template_with_strategy(body: str, substitutions: Mapping[str, str]) -> str: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    mode = dify_config.MAIL_TEMPLATING_MODE | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    timeout = dify_config.MAIL_TEMPLATING_TIMEOUT | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    if mode == TemplateMode.UNSAFE: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        return render_template_string(body, **substitutions) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    if mode == TemplateMode.SANDBOX: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        tmpl = SandboxedEnvironment(timeout=timeout).from_string(body) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        return tmpl.render(substitutions) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    if mode == TemplateMode.DISABLED: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        return body | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    raise ValueError(f"Unsupported mail templating mode: {mode}") | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			@shared_task(queue="mail") | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			def send_inner_email_task(to: list[str], subject: str, body: str, substitutions: Mapping[str, str]): | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    if not mail.is_inited(): | 
		
		
	
	
		
			
			| 
				
			 | 
			
			 | 
			@@ -21,7 +50,7 @@ def send_inner_email_task(to: list[str], subject: str, body: str, substitutions: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    start_at = time.perf_counter() | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			    try: | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        html_content = render_template_string(body, **substitutions) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        html_content = _render_template_with_strategy(body, substitutions) | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			
  | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        email_service = get_email_i18n_service() | 
		
		
	
		
			
			 | 
			 | 
			
			 | 
			        email_service.send_raw_email(to=to, subject=subject, html_content=html_content) |