cjhi UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZmZmZmZmZmZddlmZddlmZddlmZdd lmZerdd lm Z ed d Z!edd Z"ejFe$Z%dZ&de'd<e jPdk7rdZ)ndZ)Gdd Z*Gdde*Z+Gdde*Z,Gdde,Z-Gdde,Z.Gdde*ee!Z/Gdd e j`Z1y)!ae The MIT License (MIT) Copyright (c) 2015-2021 Rapptz Copyright (c) 2021-present Pycord Development Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. ) annotationsN)IO TYPE_CHECKINGAnyCallableGenericTypeVar)ClientException) OggStream)Encoder)MISSING) VoiceClientAT AudioSource)boundFTFFmpegOpusAudio)rPCMAudio FFmpegAudioFFmpegPCMAudiorPCMVolumeTransformerintCREATE_NO_WINDOWwin32ic0eZdZdZddZddZd dZd dZy) raRepresents an audio stream. The audio stream can be Opus encoded or not, however if the audio stream is not Opus encoded then the audio format must be 16-bit 48KHz stereo PCM. .. warning:: The audio source reads are done in a separate thread. ct)aWReads 20ms worth of audio. Subclasses must implement this. If the audio is complete, then returning an empty :term:`py:bytes-like object` to signal this is the way to do so. If :meth:`~AudioSource.is_opus` method returns ``True``, then it must return 20ms worth of Opus encoded audio. Otherwise, it must be 20ms worth of 16-bit 48KHz stereo PCM, which is about 3,840 bytes per frame (20ms worth of audio). Returns ------- :class:`bytes` A bytes like object that represents the PCM or Opus data. )NotImplementedErrorselfs A/var/www/html/venv/lib/python3.12/site-packages/discord/player.pyreadzAudioSource.readTs $"!cy)z6Checks if the audio source is already encoded in Opus.Frs r!is_opuszAudioSource.is_opushsr#cy)zCalled when clean-up is needed to be done. Useful for clearing buffer data or processes after it is done playing audio. Nr%rs r!cleanupzAudioSource.cleanuplsr#c$|jyN)r(rs r!__del__zAudioSource.__del__ss  r#Nreturnbytesr-boolr-None)__name__ __module__ __qualname____doc__r"r&r(r+r%r#r!rrIs"( r#c eZdZdZddZddZy)rzRepresents raw 16-bit 48KHz stereo PCM audio source. Attributes ---------- stream: :term:`py:file object` A file-like object that reads byte data representing raw PCM. c||_yr*)stream)r r9s r!__init__zPCMAudio.__init__s )/ r#c|jjtj}t |tjk7ry|SNr#)r9r" OpusEncoder FRAME_SIZElenr rets r!r"z PCMAudio.reads6kk{556 s8{-- - r#N)r9io.BufferedIOBaser-r2r,)r3r4r5r6r:r"r%r#r!rrws0r#rcLeZdZdZdd d dZd dZd dZd dZd dZy )rzRepresents an FFmpeg (or AVConv) based AudioSource. User created AudioSources using FFmpeg differently from how :class:`FFmpegPCMAudio` and :class:`FFmpegOpusAudio` work should subclass this. .. versionadded:: 1.3 ffmpeg) executablec 4|jdtjk(}|rt|tr t d|g|}dtji}|j ||j|fi||_|jj|_ d|_ d|_ |rndt|d}|jj|_ tj |j"|fd||_ |jj%yy)NstdinzNparameter conflict: 'source' parameter cannot be a string when piping to stdinstdoutzpopen-stdin-writer:z#xT)targetargsdaemonname)get subprocessPIPE isinstancestr TypeErrorupdate_spawn_process_processrH_stdout_stdin _pipe_threadidrG threadingThread _pipe_writerstart)r sourcerErJsubprocess_kwargspipingkwargsns r!r:zFFmpegAudio.__init__s#&&w/:??B j-  "T"JOO, '(*=$*=*=d*Mf*M "&--"6"6 (, 59 %bhr]3A----DK ) 0 0((yA!D     # # % r#c B tj|fdti|}|S#t$r9t |t r|j ddn|d}t|ddtj$r+}td|jjd||d}~wwxYw)N creationflags rz was not found.zPopen failed: z: ) rNPopenrFileNotFoundErrorrPrQ partitionr SubprocessError __class__r3)r rJr_processrEexcs r!rTzFFmpegAudio._spawn_processs  &&$48IGN! L3=dC3H,Q/dSTgJ!ZL"@At K)) ! !7!7 83%@  s AB3&BBc|j}|turytjd|j |j |j\tjd|j|jtjd|j|jytjd|j|jy#t $r#tjd|jYwxYw)Nz)Preparing to terminate ffmpeg process %s.z3Ignoring error attempting to kill ffmpeg process %sz=ffmpeg process %s has not terminated. Waiting to terminate...zBffmpeg process %s should have terminated with a return code of %s.zAffmpeg process %s successfully terminated with return code of %s.) rUr_loginfopidkill Exception exceptionpoll communicate returncode)r procs r! _kill_processzFFmpegAudio._kill_processs}} 7?  =txxH  IIK 99;  IIO      IIT  IIS #  NNEtxx  sC)D  D cN|jrX|jd}|s|jjy |jj ||jrWyy#t $r5t jd|d|jjYywxYw)Ni z2Write error for %s, this is probably not a problemTexc_info) rUr"rWclosewriterrrndebug terminate)r r^datas r!r\zFFmpegAudio._pipe_writersmm;;t$D !!#  !!$'mm  H!  '') sA&&;B$#B$cV|jtx|_x|_|_yr*)rxrrUrVrWrs r!r(zFFmpegAudio.cleanups# 5<< < t{r#N)r^str | io.BufferedIOBaserErQrJrr_r)rJrr_rr-zsubprocess.Popenr1)r^rBr-r2) r3r4r5r6r:rTrxr\r(r%r#r!rrsM# &'& &  & ! &@ @&=r#rc\eZdZdZdddddd d fdZd dZd dZxZS) raAn audio source from FFmpeg (or AVConv). This launches a sub-process to a specific input file given. .. warning:: You must have the ffmpeg or avconv executable in your path environment variable in order for this to work. Parameters ---------- source: Union[:class:`str`, :class:`io.BufferedIOBase`] The input that ffmpeg will take and convert to PCM bytes. If ``pipe`` is ``True`` then this is a file-like object that is passed to the stdin of ffmpeg. executable: :class:`str` The executable name (and path) to use. Defaults to ``ffmpeg``. pipe: :class:`bool` If ``True``, denotes that ``source`` parameter will be passed to the stdin of ffmpeg. Defaults to ``False``. stderr: Optional[:term:`py:file object`] A file-like object to pass to the Popen constructor. Could also be an instance of ``subprocess.PIPE``. before_options: Optional[:class:`str`] Extra command line arguments to pass to ffmpeg before the ``-i`` flag. options: Optional[:class:`str`] Extra command line arguments to pass to ffmpeg after the ``-i`` flag. Raises ------ ClientException The subprocess failed to be created. rDFN)rEpipestderrbefore_optionsoptionscg}|rtjntj|d}t|tr$|j t j||jd|j|rdn||j dt|tr$|j t j||jdt |(|f||d|y)NrGr-i-)-fs16le-ar48000-ac2 -loglevelwarningpipe:1rErJ) rNrODEVNULLrPrQextendshlexsplitappendsuperr:) r r^rErrrrrJr_rjs r!r:zFFmpegPCMAudio.__init__s(,Z__*2D2D  nc * KK N3 4 D 4CV, WX gs # KK G, - H WJTWEVWr#c|jjtj}t |tjk7ry|Sr<)rVr"r=r>r?r@s r!r"zFFmpegPCMAudio.read;s6ll 6 67 s8{-- - r#cyNFr%rs r!r&zFFmpegPCMAudio.is_opusAsr#)r^rrErQrr0rzIO[str] | Noner str | Nonerrr-r2r,r/)r3r4r5r6r:r"r& __classcell__rjs@r!rrsu L#!%%)"X'X X  X  X#XX X< r#rceZdZdZdddddddd dfdZedd dd Zeddd  dd Ze d dd Z e d dd Z ddZ ddZ xZ S)ra An audio source from FFmpeg (or AVConv). This launches a sub-process to a specific input file given. However, rather than producing PCM packets like :class:`FFmpegPCMAudio` does that need to be encoded to Opus, this class produces Opus packets, skipping the encoding step done by the library. Alternatively, instead of instantiating this class directly, you can use :meth:`FFmpegOpusAudio.from_probe` to probe for bitrate and codec information. This can be used to opportunistically skip pointless re-encoding of existing Opus audio data for a boost in performance at the cost of a short initial delay to gather the information. The same can be achieved by passing ``copy`` to the ``codec`` parameter, but only if you know that the input source is Opus encoded beforehand. .. versionadded:: 1.3 .. warning:: You must have the ffmpeg or avconv executable in your path environment variable in order for this to work. Parameters ---------- source: Union[:class:`str`, :class:`io.BufferedIOBase`] The input that ffmpeg will take and convert to Opus bytes. If ``pipe`` is ``True`` then this is a file-like object that is passed to the stdin of ffmpeg. bitrate: :class:`int` The bitrate in kbps to encode the output to. Defaults to ``128``. codec: Optional[:class:`str`] The codec to use to encode the audio data. Normally this would be just ``libopus``, but is used by :meth:`FFmpegOpusAudio.from_probe` to opportunistically skip pointlessly re-encoding Opus audio data by passing ``copy`` as the codec value. Any values other than ``copy``, ``opus``, or ``libopus`` will be considered ``libopus``. Defaults to ``libopus``. .. warning:: Do not provide this parameter unless you are certain that the audio input is already Opus encoded. For typical use :meth:`FFmpegOpusAudio.from_probe` should be used to determine the proper value for this parameter. executable: :class:`str` The executable name (and path) to use. Defaults to ``ffmpeg``. pipe: :class:`bool` If ``True``, denotes that ``source`` parameter will be passed to the stdin of ffmpeg. Defaults to ``False``. stderr: Optional[:term:`py:file object`] A file-like object to pass to the Popen constructor. Could also be an instance of ``subprocess.PIPE``. before_options: Optional[:class:`str`] Extra command line arguments to pass to ffmpeg before the ``-i`` flag. options: Optional[:class:`str`] Extra command line arguments to pass to ffmpeg after the ``-i`` flag. Raises ------ ClientException The subprocess failed to be created. NrDF)bitratecodecrErrrrc\g} |rtjntj|d} t|tr$| j t j|| jd| j|rdn||dvrdnd}| j ddd d d |d d ddd|dddft|tr$| j t j|| jdt |(|f|| d| t|jj|_y)Nrrr)opuslibopuscopyrz -map_metadataz-1rrz-c:arrrrz-b:akrrrr)rNrOrrPrQrrrrrr:r rV iter_packets _packet_iter) r r^rrrErrrrrJr_rjs r!r:zFFmpegOpusAudio.__init__s(,Z__*2D2D  nc * KK N3 4 D 4CV,#66I )1   & gs # KK G, - H WJTWEVW%dll3@@Br#)methodcK|jd}|j|||d{\}}||f||d|S7w)a|coro| A factory method that creates a :class:`FFmpegOpusAudio` after probing the input source for audio codec and bitrate information. Parameters ---------- source Identical to the ``source`` parameter for the constructor. method: Optional[Union[:class:`str`, Callable[:class:`str`, :class:`str`]]] The probing method used to determine bitrate and codec information. As a string, valid values are ``native`` to use ffprobe (or avprobe) and ``fallback`` to use ffmpeg (or avconv). As a callable, it must take two string arguments, ``source`` and ``executable``. Both parameters are the same values passed to this factory function. ``executable`` will default to ``ffmpeg`` if not provided as a keyword argument. kwargs The remaining parameters to be passed to the :class:`FFmpegOpusAudio` constructor, excluding ``bitrate`` and ``codec``. Returns ------- :class:`FFmpegOpusAudio` An instance of this class. Raises ------ AttributeError Invalid probe method, must be ``'native'`` or ``'fallback'``. TypeError Invalid value for ``probe`` parameter, must be :class:`str` or a callable. Examples -------- Use this function to create an :class:`FFmpegOpusAudio` instance instead of the constructor: :: source = await discord.FFmpegOpusAudio.from_probe("song.webm") voice_client.play(source) If you are on Windows and don't have ffprobe installed, use the ``fallback`` method to probe using ffmpeg instead: :: source = await discord.FFmpegOpusAudio.from_probe("song.webm", method='fallback') voice_client.play(source) Using a custom method of determining codec and bitrate: :: def custom_probe(source, executable): # some analysis code here return codec, bitrate source = await discord.FFmpegOpusAudio.from_probe("song.webm", method=custom_probe) voice_client.play(source) rErrEN)rr)rMprobe)clsr^rrarErrs r! from_probezFFmpegOpusAudio.from_probesO~ZZ - "yy:yVVw6B7%B6BBWs)A?ArcTK|xsd}xsddxt|tr;t|d|dtd||jurJ|j n=t |r||j n#td|jjddx}}tj} |jdfdd{\}}tjd ||||fS7"#t$rstj!d |Y||fStj!d | |jdfd d{7\}}tjd ||n%#t$rtj!dYnwxYwYwxYw#||fccYSxYww)aO|coro| Probes the input source for bitrate and codec information. Parameters ---------- source Identical to the ``source`` parameter for :class:`FFmpegOpusAudio`. method Identical to the ``method`` parameter for :meth:`FFmpegOpusAudio.from_probe`. executable: :class:`str` Identical to the ``executable`` parameter for :class:`FFmpegOpusAudio`. Returns ------- Optional[Tuple[Optional[:class:`str`], Optional[:class:`int`]]] A 2-tuple with the codec and bitrate of the input source. Raises ------ AttributeError Invalid probe method, must be ``'native'`` or ``'fallback'``. TypeError Invalid value for ``probe`` parameter, must be :class:`str` or a callable. nativerDN _probe_codec_zInvalid probe method z5Expected str or callable for parameter 'probe', not ''cSr*r%)rE probefuncr^sr!z'FFmpegOpusAudio.probe..4siPVXbFcr#z Probe found codec=%s, bitrate=%szProbe '%s' using '%s' failedz-Probe '%s' using '%s' failed, trying fallbackcSr*r%)rEfallbackr^sr!rz'FFmpegOpusAudio.probe..>s(SY[eJfr#z)Fallback probe found codec=%s, bitrate=%sz Fallback probe using '%s' failed)rPrQgetattrAttributeError_probe_codec_native_probe_codec_fallbackcallablerRrjr3asyncioget_event_looprun_in_executorrnrorrrs) rr^rrErrlooprrs ` ` @@r!rzFFmpegOpusAudio.probesD#8+8 ## H fc "}VH%=tDI $'c#ddNE7 II8% I'> !%e W=vzR'> ! NN?  W'+';';DBf'g!g!gw EugV OA:N O W"'> !sB1F(8C7C5C7F1F(5C77"FFF(F6E1EE1F1FFFFFFFF%#F(c 8|dvr|dddn|}|dddddd d |g }tj|d }dx}}|ratj|}|d d}|j d}t |j dd}t t|dz d}||fS)N>avconvrDrz-vquietz -print_formatjsonz -show_streamsz-select_streamsza:0timeoutstreamsr codec_namebit_ratei)rN check_outputrloadsrMrmaxround) r^rEexerJoutputrrr streamdatas r!rz#FFmpegOpusAudio._probe_codec_nativeHs 11"1~e $            ((r: ::f%Di+JNN<0E*..Q78G%$/5Gg~r#c|dd|g}tj|ttjtj}|j d\}}|j d}dx}}tjd|} | r| jd}tjd |} | r$tt| jdd }||fS) Nz -hide_bannerr)rdrHrrrutf8zStream #0.*?Audio: (\w+)r z (\d+) [kK]b/sr) rNrfrrOSTDOUTrudecoderesearchgrouprr) r^rErJrwout_rrr codec_matchbr_matchs r!rz%FFmpegOpusAudio._probe_codec_fallbackjsND&9 *??$$   !!"!-QF#ii ;VD %%a(E99-v6 #hnnQ/0#6Gg~r#c.t|jdSr<)nextrrs r!r"zFFmpegOpusAudio.readsD%%s++r#cy)NTr%rs r!r&zFFmpegOpusAudio.is_opussr#) r^rrrrrrErQr-r2) rztype[FT]r^rQr@str | Callable[[str, str], tuple[str | None, int | None]] | Nonerarr-r)r^rQrrrErr-tuple[str | None, int | None])rD)rErQr-rr,r/)r3r4r5r6r: classmethodrr staticmethodrrr"r&rrs@r!rrEsZ:@ " 3C'3C 3C  3C  3C 3Cj TX @C @C@CQ @C  @C @C@CD TX!% K"K"Q K"  K" ' K"K"Z"* &B"* &0,r#cbeZdZdZdd dZed dZejd dZd dZd dZ y)ra Transforms a previous :class:`AudioSource` to have volume controls. This does not work on audio sources that have :meth:`AudioSource.is_opus` set to ``True``. Parameters ---------- original: :class:`AudioSource` The original AudioSource to transform. volume: :class:`float` The initial volume to set it to. See :attr:`volume` for more info. Raises ------ TypeError Not an audio source. ClientException The audio source is opus encoded. ct|ts#td|jjd|j r t d||_||_y)Nzexpected AudioSource not .z%AudioSource must not be Opus encoded.) rPrrRrjr3r&r originalvolume)r rrs r!r:zPCMVolumeTransformer.__init__sU(K078J8J8S8S7TTUVW W    !"IJ J$  r#c|jS)zTRetrieves or sets the volume as a floating point percentage (e.g. ``1.0`` for 100%).)_volumers r!rzPCMVolumeTransformer.volumes||r#c&t|d|_y)Ng)rr)r values r!rzPCMVolumeTransformer.volumes5# r#c8|jjyr*)rr(rs r!r(zPCMVolumeTransformer.cleanups r#c|jj}tj|dt |j dS)Nrg@)rr"audioopmulminrr@s r!r"zPCMVolumeTransformer.reads2mm  "{{33t||S#9::r#N)g?)rrrfloat)r-r)rrr-r2r1r,) r3r4r5r6r:propertyrsetterr(r"r%r#r!rrsB* ]]'' ;r#rceZdZUejdz Zded<ddddZddZddZ dd Z dd Z d d dd Z d d ddZ ddZddZddZddZy) AudioPlayerg@@rDELAYN)afterctjj|d|_||_||_||_tj|_tj|_ |jjd|_ |j|_ tj|_|t|s t!dyy)NTz.Expected a callable for the "after" parameter.)rZr[r:rKr^clientrEvent_end_resumedset_current_error _connectedLock_lockrrR)r r^rrs r!r:zAudioPlayer.__init__s!!$'  #) #) ?D %.__%6 )2):  04+1+<+<%.^^%5  Xe_LM M&5 r#cd|_tj|_|jj }|j d|jjst|jjs|jjP|jjs:|jjd|_tj|_|xjdz c_|jj}|s|jy|||jj |j|j |jzz}t#d|j |tjz z}tj$||jjssyy)NrTr )encode)loopstime perf_counter_startrsend_audio_packet_speakris_setrwaitrr^r"stopr&rrsleep)r play_audior next_timedelays r!_do_runzAudioPlayer._do_runsI '') [[22  D))""$=='') ""$??))+$$& "//1 JJ!OJ;;##%D  t (;(;(=$= > djj4::&==I4::T5F5F5H)HIJE JJu 3))""$$r#c2 |j|jj |j y#t$r!}||_|jYd}~Pd}~wwxYw#|jj |j wxYwr*)rrrrrr^r( _call_after)r rls r!runzAudioPlayer.runsr  LLN KK   !      "%D  IIKK  KK   !    s'= A'A"A*"A''A**,Bc|j}|j |j|y|rld|j}tj ||t|tjt jt|||jyy#t$rP}tj d||_t jt|||jYd}~yd}~wwxYw)Nz"Calling the after function failed.zException in voice thread rz)file)rrrrrnrs __context__ tracebackprint_exceptiontype __traceback__rLprintsysr)r errorrlmsgs r!rzAudioPlayer._call_afters## :: ! M 5! .tyyk:C NN3N / #CJJ '  % %d5k5%:M:M N   MCD"'))$s)S#:K:KLL MsB C4$AC//C4c|jj|jj|jdyr)rrrrrs r!rzAudioPlayer.stop s+   Er#Tupdate_speakingc`|jj|r|jdyyr)rclearrr r)s r!pausezAudioPlayer.pauses&   KK  r#cd|_tj|_|jj |r|j dyy)NrT)r r r rrrrr,s r!resumezAudioPlayer.resumes= '')    KK  r#cp|jjxr|jj Sr*)rrrrs r! is_playingzAudioPlayer.is_playings*}}##%@dii.>.>.@*@@r#cr|jj xr|jj Sr*)rrrrs r! is_pausedzAudioPlayer.is_paused s-99##%%Ddmm.B.B.D*DDr#c|j5|jd||_|jddddy#1swYyxYw)NFr()rr-r^r/)r r^s r! _set_sourcezAudioPlayer._set_source#sA ZZ / JJuJ - DK KKK . / / /s ,AA c tj|jjj ||jj y#t $r }tjd|Yd}~yd}~wwxYw)Nz"Speaking call in player failed: %s) rrun_coroutine_threadsaferwsspeakrrrrnro)r speakinges r!rzAudioPlayer._speak)s[ ?  , , $$X. 0@0@  ? II:A > > ?sA A A9A44A9)r^rrrr1)r)r0r-r2r/)r^rr-r2)r:r0r-r2)r3r4r5r= FRAME_LENGTHr__annotations__r:rrrrr-r/r1r3r5rr%r#r!rrs`++f4E54JNN"!FO  04 15AE/ ?r#r)2r6 __future__rrriorloggingrrrNr$rZr rtypingrrrrrr errorsr oggparser rr r=utilsr voice_clientrrr getLoggerr3rn__all__r=platformrrrrrrrr[rr%r#r!rIs 2#    EE#() T' T*+w" <<7!++\{&m=+m=`H[HVBkBJ .;; .;bt?)""t?r#