commit
0cf9a4ea20
@ -1,3 +1,5 @@
|
||||
/target
|
||||
chain.dat
|
||||
*~
|
||||
chain.dat
|
||||
flamegraph.svg
|
||||
|
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,674 @@
|
||||
GNU GENERAL PUBLIC LICENSE
|
||||
Version 3, 29 June 2007
|
||||
|
||||
Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
|
||||
Everyone is permitted to copy and distribute verbatim copies
|
||||
of this license document, but changing it is not allowed.
|
||||
|
||||
Preamble
|
||||
|
||||
The GNU General Public License is a free, copyleft license for
|
||||
software and other kinds of works.
|
||||
|
||||
The licenses for most software and other practical works are designed
|
||||
to take away your freedom to share and change the works. By contrast,
|
||||
the GNU General Public License is intended to guarantee your freedom to
|
||||
share and change all versions of a program--to make sure it remains free
|
||||
software for all its users. We, the Free Software Foundation, use the
|
||||
GNU General Public License for most of our software; it applies also to
|
||||
any other work released this way by its authors. You can apply it to
|
||||
your programs, too.
|
||||
|
||||
When we speak of free software, we are referring to freedom, not
|
||||
price. Our General Public Licenses are designed to make sure that you
|
||||
have the freedom to distribute copies of free software (and charge for
|
||||
them if you wish), that you receive source code or can get it if you
|
||||
want it, that you can change the software or use pieces of it in new
|
||||
free programs, and that you know you can do these things.
|
||||
|
||||
To protect your rights, we need to prevent others from denying you
|
||||
these rights or asking you to surrender the rights. Therefore, you have
|
||||
certain responsibilities if you distribute copies of the software, or if
|
||||
you modify it: responsibilities to respect the freedom of others.
|
||||
|
||||
For example, if you distribute copies of such a program, whether
|
||||
gratis or for a fee, you must pass on to the recipients the same
|
||||
freedoms that you received. You must make sure that they, too, receive
|
||||
or can get the source code. And you must show them these terms so they
|
||||
know their rights.
|
||||
|
||||
Developers that use the GNU GPL protect your rights with two steps:
|
||||
(1) assert copyright on the software, and (2) offer you this License
|
||||
giving you legal permission to copy, distribute and/or modify it.
|
||||
|
||||
For the developers' and authors' protection, the GPL clearly explains
|
||||
that there is no warranty for this free software. For both users' and
|
||||
authors' sake, the GPL requires that modified versions be marked as
|
||||
changed, so that their problems will not be attributed erroneously to
|
||||
authors of previous versions.
|
||||
|
||||
Some devices are designed to deny users access to install or run
|
||||
modified versions of the software inside them, although the manufacturer
|
||||
can do so. This is fundamentally incompatible with the aim of
|
||||
protecting users' freedom to change the software. The systematic
|
||||
pattern of such abuse occurs in the area of products for individuals to
|
||||
use, which is precisely where it is most unacceptable. Therefore, we
|
||||
have designed this version of the GPL to prohibit the practice for those
|
||||
products. If such problems arise substantially in other domains, we
|
||||
stand ready to extend this provision to those domains in future versions
|
||||
of the GPL, as needed to protect the freedom of users.
|
||||
|
||||
Finally, every program is threatened constantly by software patents.
|
||||
States should not allow patents to restrict development and use of
|
||||
software on general-purpose computers, but in those that do, we wish to
|
||||
avoid the special danger that patents applied to a free program could
|
||||
make it effectively proprietary. To prevent this, the GPL assures that
|
||||
patents cannot be used to render the program non-free.
|
||||
|
||||
The precise terms and conditions for copying, distribution and
|
||||
modification follow.
|
||||
|
||||
TERMS AND CONDITIONS
|
||||
|
||||
0. Definitions.
|
||||
|
||||
"This License" refers to version 3 of the GNU General Public License.
|
||||
|
||||
"Copyright" also means copyright-like laws that apply to other kinds of
|
||||
works, such as semiconductor masks.
|
||||
|
||||
"The Program" refers to any copyrightable work licensed under this
|
||||
License. Each licensee is addressed as "you". "Licensees" and
|
||||
"recipients" may be individuals or organizations.
|
||||
|
||||
To "modify" a work means to copy from or adapt all or part of the work
|
||||
in a fashion requiring copyright permission, other than the making of an
|
||||
exact copy. The resulting work is called a "modified version" of the
|
||||
earlier work or a work "based on" the earlier work.
|
||||
|
||||
A "covered work" means either the unmodified Program or a work based
|
||||
on the Program.
|
||||
|
||||
To "propagate" a work means to do anything with it that, without
|
||||
permission, would make you directly or secondarily liable for
|
||||
infringement under applicable copyright law, except executing it on a
|
||||
computer or modifying a private copy. Propagation includes copying,
|
||||
distribution (with or without modification), making available to the
|
||||
public, and in some countries other activities as well.
|
||||
|
||||
To "convey" a work means any kind of propagation that enables other
|
||||
parties to make or receive copies. Mere interaction with a user through
|
||||
a computer network, with no transfer of a copy, is not conveying.
|
||||
|
||||
An interactive user interface displays "Appropriate Legal Notices"
|
||||
to the extent that it includes a convenient and prominently visible
|
||||
feature that (1) displays an appropriate copyright notice, and (2)
|
||||
tells the user that there is no warranty for the work (except to the
|
||||
extent that warranties are provided), that licensees may convey the
|
||||
work under this License, and how to view a copy of this License. If
|
||||
the interface presents a list of user commands or options, such as a
|
||||
menu, a prominent item in the list meets this criterion.
|
||||
|
||||
1. Source Code.
|
||||
|
||||
The "source code" for a work means the preferred form of the work
|
||||
for making modifications to it. "Object code" means any non-source
|
||||
form of a work.
|
||||
|
||||
A "Standard Interface" means an interface that either is an official
|
||||
standard defined by a recognized standards body, or, in the case of
|
||||
interfaces specified for a particular programming language, one that
|
||||
is widely used among developers working in that language.
|
||||
|
||||
The "System Libraries" of an executable work include anything, other
|
||||
than the work as a whole, that (a) is included in the normal form of
|
||||
packaging a Major Component, but which is not part of that Major
|
||||
Component, and (b) serves only to enable use of the work with that
|
||||
Major Component, or to implement a Standard Interface for which an
|
||||
implementation is available to the public in source code form. A
|
||||
"Major Component", in this context, means a major essential component
|
||||
(kernel, window system, and so on) of the specific operating system
|
||||
(if any) on which the executable work runs, or a compiler used to
|
||||
produce the work, or an object code interpreter used to run it.
|
||||
|
||||
The "Corresponding Source" for a work in object code form means all
|
||||
the source code needed to generate, install, and (for an executable
|
||||
work) run the object code and to modify the work, including scripts to
|
||||
control those activities. However, it does not include the work's
|
||||
System Libraries, or general-purpose tools or generally available free
|
||||
programs which are used unmodified in performing those activities but
|
||||
which are not part of the work. For example, Corresponding Source
|
||||
includes interface definition files associated with source files for
|
||||
the work, and the source code for shared libraries and dynamically
|
||||
linked subprograms that the work is specifically designed to require,
|
||||
such as by intimate data communication or control flow between those
|
||||
subprograms and other parts of the work.
|
||||
|
||||
The Corresponding Source need not include anything that users
|
||||
can regenerate automatically from other parts of the Corresponding
|
||||
Source.
|
||||
|
||||
The Corresponding Source for a work in source code form is that
|
||||
same work.
|
||||
|
||||
2. Basic Permissions.
|
||||
|
||||
All rights granted under this License are granted for the term of
|
||||
copyright on the Program, and are irrevocable provided the stated
|
||||
conditions are met. This License explicitly affirms your unlimited
|
||||
permission to run the unmodified Program. The output from running a
|
||||
covered work is covered by this License only if the output, given its
|
||||
content, constitutes a covered work. This License acknowledges your
|
||||
rights of fair use or other equivalent, as provided by copyright law.
|
||||
|
||||
You may make, run and propagate covered works that you do not
|
||||
convey, without conditions so long as your license otherwise remains
|
||||
in force. You may convey covered works to others for the sole purpose
|
||||
of having them make modifications exclusively for you, or provide you
|
||||
with facilities for running those works, provided that you comply with
|
||||
the terms of this License in conveying all material for which you do
|
||||
not control copyright. Those thus making or running the covered works
|
||||
for you must do so exclusively on your behalf, under your direction
|
||||
and control, on terms that prohibit them from making any copies of
|
||||
your copyrighted material outside their relationship with you.
|
||||
|
||||
Conveying under any other circumstances is permitted solely under
|
||||
the conditions stated below. Sublicensing is not allowed; section 10
|
||||
makes it unnecessary.
|
||||
|
||||
3. Protecting Users' Legal Rights From Anti-Circumvention Law.
|
||||
|
||||
No covered work shall be deemed part of an effective technological
|
||||
measure under any applicable law fulfilling obligations under article
|
||||
11 of the WIPO copyright treaty adopted on 20 December 1996, or
|
||||
similar laws prohibiting or restricting circumvention of such
|
||||
measures.
|
||||
|
||||
When you convey a covered work, you waive any legal power to forbid
|
||||
circumvention of technological measures to the extent such circumvention
|
||||
is effected by exercising rights under this License with respect to
|
||||
the covered work, and you disclaim any intention to limit operation or
|
||||
modification of the work as a means of enforcing, against the work's
|
||||
users, your or third parties' legal rights to forbid circumvention of
|
||||
technological measures.
|
||||
|
||||
4. Conveying Verbatim Copies.
|
||||
|
||||
You may convey verbatim copies of the Program's source code as you
|
||||
receive it, in any medium, provided that you conspicuously and
|
||||
appropriately publish on each copy an appropriate copyright notice;
|
||||
keep intact all notices stating that this License and any
|
||||
non-permissive terms added in accord with section 7 apply to the code;
|
||||
keep intact all notices of the absence of any warranty; and give all
|
||||
recipients a copy of this License along with the Program.
|
||||
|
||||
You may charge any price or no price for each copy that you convey,
|
||||
and you may offer support or warranty protection for a fee.
|
||||
|
||||
5. Conveying Modified Source Versions.
|
||||
|
||||
You may convey a work based on the Program, or the modifications to
|
||||
produce it from the Program, in the form of source code under the
|
||||
terms of section 4, provided that you also meet all of these conditions:
|
||||
|
||||
a) The work must carry prominent notices stating that you modified
|
||||
it, and giving a relevant date.
|
||||
|
||||
b) The work must carry prominent notices stating that it is
|
||||
released under this License and any conditions added under section
|
||||
7. This requirement modifies the requirement in section 4 to
|
||||
"keep intact all notices".
|
||||
|
||||
c) You must license the entire work, as a whole, under this
|
||||
License to anyone who comes into possession of a copy. This
|
||||
License will therefore apply, along with any applicable section 7
|
||||
additional terms, to the whole of the work, and all its parts,
|
||||
regardless of how they are packaged. This License gives no
|
||||
permission to license the work in any other way, but it does not
|
||||
invalidate such permission if you have separately received it.
|
||||
|
||||
d) If the work has interactive user interfaces, each must display
|
||||
Appropriate Legal Notices; however, if the Program has interactive
|
||||
interfaces that do not display Appropriate Legal Notices, your
|
||||
work need not make them do so.
|
||||
|
||||
A compilation of a covered work with other separate and independent
|
||||
works, which are not by their nature extensions of the covered work,
|
||||
and which are not combined with it such as to form a larger program,
|
||||
in or on a volume of a storage or distribution medium, is called an
|
||||
"aggregate" if the compilation and its resulting copyright are not
|
||||
used to limit the access or legal rights of the compilation's users
|
||||
beyond what the individual works permit. Inclusion of a covered work
|
||||
in an aggregate does not cause this License to apply to the other
|
||||
parts of the aggregate.
|
||||
|
||||
6. Conveying Non-Source Forms.
|
||||
|
||||
You may convey a covered work in object code form under the terms
|
||||
of sections 4 and 5, provided that you also convey the
|
||||
machine-readable Corresponding Source under the terms of this License,
|
||||
in one of these ways:
|
||||
|
||||
a) Convey the object code in, or embodied in, a physical product
|
||||
(including a physical distribution medium), accompanied by the
|
||||
Corresponding Source fixed on a durable physical medium
|
||||
customarily used for software interchange.
|
||||
|
||||
b) Convey the object code in, or embodied in, a physical product
|
||||
(including a physical distribution medium), accompanied by a
|
||||
written offer, valid for at least three years and valid for as
|
||||
long as you offer spare parts or customer support for that product
|
||||
model, to give anyone who possesses the object code either (1) a
|
||||
copy of the Corresponding Source for all the software in the
|
||||
product that is covered by this License, on a durable physical
|
||||
medium customarily used for software interchange, for a price no
|
||||
more than your reasonable cost of physically performing this
|
||||
conveying of source, or (2) access to copy the
|
||||
Corresponding Source from a network server at no charge.
|
||||
|
||||
c) Convey individual copies of the object code with a copy of the
|
||||
written offer to provide the Corresponding Source. This
|
||||
alternative is allowed only occasionally and noncommercially, and
|
||||
only if you received the object code with such an offer, in accord
|
||||
with subsection 6b.
|
||||
|
||||
d) Convey the object code by offering access from a designated
|
||||
place (gratis or for a charge), and offer equivalent access to the
|
||||
Corresponding Source in the same way through the same place at no
|
||||
further charge. You need not require recipients to copy the
|
||||
Corresponding Source along with the object code. If the place to
|
||||
copy the object code is a network server, the Corresponding Source
|
||||
may be on a different server (operated by you or a third party)
|
||||
that supports equivalent copying facilities, provided you maintain
|
||||
clear directions next to the object code saying where to find the
|
||||
Corresponding Source. Regardless of what server hosts the
|
||||
Corresponding Source, you remain obligated to ensure that it is
|
||||
available for as long as needed to satisfy these requirements.
|
||||
|
||||
e) Convey the object code using peer-to-peer transmission, provided
|
||||
you inform other peers where the object code and Corresponding
|
||||
Source of the work are being offered to the general public at no
|
||||
charge under subsection 6d.
|
||||
|
||||
A separable portion of the object code, whose source code is excluded
|
||||
from the Corresponding Source as a System Library, need not be
|
||||
included in conveying the object code work.
|
||||
|
||||
A "User Product" is either (1) a "consumer product", which means any
|
||||
tangible personal property which is normally used for personal, family,
|
||||
or household purposes, or (2) anything designed or sold for incorporation
|
||||
into a dwelling. In determining whether a product is a consumer product,
|
||||
doubtful cases shall be resolved in favor of coverage. For a particular
|
||||
product received by a particular user, "normally used" refers to a
|
||||
typical or common use of that class of product, regardless of the status
|
||||
of the particular user or of the way in which the particular user
|
||||
actually uses, or expects or is expected to use, the product. A product
|
||||
is a consumer product regardless of whether the product has substantial
|
||||
commercial, industrial or non-consumer uses, unless such uses represent
|
||||
the only significant mode of use of the product.
|
||||
|
||||
"Installation Information" for a User Product means any methods,
|
||||
procedures, authorization keys, or other information required to install
|
||||
and execute modified versions of a covered work in that User Product from
|
||||
a modified version of its Corresponding Source. The information must
|
||||
suffice to ensure that the continued functioning of the modified object
|
||||
code is in no case prevented or interfered with solely because
|
||||
modification has been made.
|
||||
|
||||
If you convey an object code work under this section in, or with, or
|
||||
specifically for use in, a User Product, and the conveying occurs as
|
||||
part of a transaction in which the right of possession and use of the
|
||||
User Product is transferred to the recipient in perpetuity or for a
|
||||
fixed term (regardless of how the transaction is characterized), the
|
||||
Corresponding Source conveyed under this section must be accompanied
|
||||
by the Installation Information. But this requirement does not apply
|
||||
if neither you nor any third party retains the ability to install
|
||||
modified object code on the User Product (for example, the work has
|
||||
been installed in ROM).
|
||||
|
||||
The requirement to provide Installation Information does not include a
|
||||
requirement to continue to provide support service, warranty, or updates
|
||||
for a work that has been modified or installed by the recipient, or for
|
||||
the User Product in which it has been modified or installed. Access to a
|
||||
network may be denied when the modification itself materially and
|
||||
adversely affects the operation of the network or violates the rules and
|
||||
protocols for communication across the network.
|
||||
|
||||
Corresponding Source conveyed, and Installation Information provided,
|
||||
in accord with this section must be in a format that is publicly
|
||||
documented (and with an implementation available to the public in
|
||||
source code form), and must require no special password or key for
|
||||
unpacking, reading or copying.
|
||||
|
||||
7. Additional Terms.
|
||||
|
||||
"Additional permissions" are terms that supplement the terms of this
|
||||
License by making exceptions from one or more of its conditions.
|
||||
Additional permissions that are applicable to the entire Program shall
|
||||
be treated as though they were included in this License, to the extent
|
||||
that they are valid under applicable law. If additional permissions
|
||||
apply only to part of the Program, that part may be used separately
|
||||
under those permissions, but the entire Program remains governed by
|
||||
this License without regard to the additional permissions.
|
||||
|
||||
When you convey a copy of a covered work, you may at your option
|
||||
remove any additional permissions from that copy, or from any part of
|
||||
it. (Additional permissions may be written to require their own
|
||||
removal in certain cases when you modify the work.) You may place
|
||||
additional permissions on material, added by you to a covered work,
|
||||
for which you have or can give appropriate copyright permission.
|
||||
|
||||
Notwithstanding any other provision of this License, for material you
|
||||
add to a covered work, you may (if authorized by the copyright holders of
|
||||
that material) supplement the terms of this License with terms:
|
||||
|
||||
a) Disclaiming warranty or limiting liability differently from the
|
||||
terms of sections 15 and 16 of this License; or
|
||||
|
||||
b) Requiring preservation of specified reasonable legal notices or
|
||||
author attributions in that material or in the Appropriate Legal
|
||||
Notices displayed by works containing it; or
|
||||
|
||||
c) Prohibiting misrepresentation of the origin of that material, or
|
||||
requiring that modified versions of such material be marked in
|
||||
reasonable ways as different from the original version; or
|
||||
|
||||
d) Limiting the use for publicity purposes of names of licensors or
|
||||
authors of the material; or
|
||||
|
||||
e) Declining to grant rights under trademark law for use of some
|
||||
trade names, trademarks, or service marks; or
|
||||
|
||||
f) Requiring indemnification of licensors and authors of that
|
||||
material by anyone who conveys the material (or modified versions of
|
||||
it) with contractual assumptions of liability to the recipient, for
|
||||
any liability that these contractual assumptions directly impose on
|
||||
those licensors and authors.
|
||||
|
||||
All other non-permissive additional terms are considered "further
|
||||
restrictions" within the meaning of section 10. If the Program as you
|
||||
received it, or any part of it, contains a notice stating that it is
|
||||
governed by this License along with a term that is a further
|
||||
restriction, you may remove that term. If a license document contains
|
||||
a further restriction but permits relicensing or conveying under this
|
||||
License, you may add to a covered work material governed by the terms
|
||||
of that license document, provided that the further restriction does
|
||||
not survive such relicensing or conveying.
|
||||
|
||||
If you add terms to a covered work in accord with this section, you
|
||||
must place, in the relevant source files, a statement of the
|
||||
additional terms that apply to those files, or a notice indicating
|
||||
where to find the applicable terms.
|
||||
|
||||
Additional terms, permissive or non-permissive, may be stated in the
|
||||
form of a separately written license, or stated as exceptions;
|
||||
the above requirements apply either way.
|
||||
|
||||
8. Termination.
|
||||
|
||||
You may not propagate or modify a covered work except as expressly
|
||||
provided under this License. Any attempt otherwise to propagate or
|
||||
modify it is void, and will automatically terminate your rights under
|
||||
this License (including any patent licenses granted under the third
|
||||
paragraph of section 11).
|
||||
|
||||
However, if you cease all violation of this License, then your
|
||||
license from a particular copyright holder is reinstated (a)
|
||||
provisionally, unless and until the copyright holder explicitly and
|
||||
finally terminates your license, and (b) permanently, if the copyright
|
||||
holder fails to notify you of the violation by some reasonable means
|
||||
prior to 60 days after the cessation.
|
||||
|
||||
Moreover, your license from a particular copyright holder is
|
||||
reinstated permanently if the copyright holder notifies you of the
|
||||
violation by some reasonable means, this is the first time you have
|
||||
received notice of violation of this License (for any work) from that
|
||||
copyright holder, and you cure the violation prior to 30 days after
|
||||
your receipt of the notice.
|
||||
|
||||
Termination of your rights under this section does not terminate the
|
||||
licenses of parties who have received copies or rights from you under
|
||||
this License. If your rights have been terminated and not permanently
|
||||
reinstated, you do not qualify to receive new licenses for the same
|
||||
material under section 10.
|
||||
|
||||
9. Acceptance Not Required for Having Copies.
|
||||
|
||||
You are not required to accept this License in order to receive or
|
||||
run a copy of the Program. Ancillary propagation of a covered work
|
||||
occurring solely as a consequence of using peer-to-peer transmission
|
||||
to receive a copy likewise does not require acceptance. However,
|
||||
nothing other than this License grants you permission to propagate or
|
||||
modify any covered work. These actions infringe copyright if you do
|
||||
not accept this License. Therefore, by modifying or propagating a
|
||||
covered work, you indicate your acceptance of this License to do so.
|
||||
|
||||
10. Automatic Licensing of Downstream Recipients.
|
||||
|
||||
Each time you convey a covered work, the recipient automatically
|
||||
receives a license from the original licensors, to run, modify and
|
||||
propagate that work, subject to this License. You are not responsible
|
||||
for enforcing compliance by third parties with this License.
|
||||
|
||||
An "entity transaction" is a transaction transferring control of an
|
||||
organization, or substantially all assets of one, or subdividing an
|
||||
organization, or merging organizations. If propagation of a covered
|
||||
work results from an entity transaction, each party to that
|
||||
transaction who receives a copy of the work also receives whatever
|
||||
licenses to the work the party's predecessor in interest had or could
|
||||
give under the previous paragraph, plus a right to possession of the
|
||||
Corresponding Source of the work from the predecessor in interest, if
|
||||
the predecessor has it or can get it with reasonable efforts.
|
||||
|
||||
You may not impose any further restrictions on the exercise of the
|
||||
rights granted or affirmed under this License. For example, you may
|
||||
not impose a license fee, royalty, or other charge for exercise of
|
||||
rights granted under this License, and you may not initiate litigation
|
||||
(including a cross-claim or counterclaim in a lawsuit) alleging that
|
||||
any patent claim is infringed by making, using, selling, offering for
|
||||
sale, or importing the Program or any portion of it.
|
||||
|
||||
11. Patents.
|
||||
|
||||
A "contributor" is a copyright holder who authorizes use under this
|
||||
License of the Program or a work on which the Program is based. The
|
||||
work thus licensed is called the contributor's "contributor version".
|
||||
|
||||
A contributor's "essential patent claims" are all patent claims
|
||||
owned or controlled by the contributor, whether already acquired or
|
||||
hereafter acquired, that would be infringed by some manner, permitted
|
||||
by this License, of making, using, or selling its contributor version,
|
||||
but do not include claims that would be infringed only as a
|
||||
consequence of further modification of the contributor version. For
|
||||
purposes of this definition, "control" includes the right to grant
|
||||
patent sublicenses in a manner consistent with the requirements of
|
||||
this License.
|
||||
|
||||
Each contributor grants you a non-exclusive, worldwide, royalty-free
|
||||
patent license under the contributor's essential patent claims, to
|
||||
make, use, sell, offer for sale, import and otherwise run, modify and
|
||||
propagate the contents of its contributor version.
|
||||
|
||||
In the following three paragraphs, a "patent license" is any express
|
||||
agreement or commitment, however denominated, not to enforce a patent
|
||||
(such as an express permission to practice a patent or covenant not to
|
||||
sue for patent infringement). To "grant" such a patent license to a
|
||||
party means to make such an agreement or commitment not to enforce a
|
||||
patent against the party.
|
||||
|
||||
If you convey a covered work, knowingly relying on a patent license,
|
||||
and the Corresponding Source of the work is not available for anyone
|
||||
to copy, free of charge and under the terms of this License, through a
|
||||
publicly available network server or other readily accessible means,
|
||||
then you must either (1) cause the Corresponding Source to be so
|
||||
available, or (2) arrange to deprive yourself of the benefit of the
|
||||
patent license for this particular work, or (3) arrange, in a manner
|
||||
consistent with the requirements of this License, to extend the patent
|
||||
license to downstream recipients. "Knowingly relying" means you have
|
||||
actual knowledge that, but for the patent license, your conveying the
|
||||
covered work in a country, or your recipient's use of the covered work
|
||||
in a country, would infringe one or more identifiable patents in that
|
||||
country that you have reason to believe are valid.
|
||||
|
||||
If, pursuant to or in connection with a single transaction or
|
||||
arrangement, you convey, or propagate by procuring conveyance of, a
|
||||
covered work, and grant a patent license to some of the parties
|
||||
receiving the covered work authorizing them to use, propagate, modify
|
||||
or convey a specific copy of the covered work, then the patent license
|
||||
you grant is automatically extended to all recipients of the covered
|
||||
work and works based on it.
|
||||
|
||||
A patent license is "discriminatory" if it does not include within
|
||||
the scope of its coverage, prohibits the exercise of, or is
|
||||
conditioned on the non-exercise of one or more of the rights that are
|
||||
specifically granted under this License. You may not convey a covered
|
||||
work if you are a party to an arrangement with a third party that is
|
||||
in the business of distributing software, under which you make payment
|
||||
to the third party based on the extent of your activity of conveying
|
||||
the work, and under which the third party grants, to any of the
|
||||
parties who would receive the covered work from you, a discriminatory
|
||||
patent license (a) in connection with copies of the covered work
|
||||
conveyed by you (or copies made from those copies), or (b) primarily
|
||||
for and in connection with specific products or compilations that
|
||||
contain the covered work, unless you entered into that arrangement,
|
||||
or that patent license was granted, prior to 28 March 2007.
|
||||
|
||||
Nothing in this License shall be construed as excluding or limiting
|
||||
any implied license or other defenses to infringement that may
|
||||
otherwise be available to you under applicable patent law.
|
||||
|
||||
12. No Surrender of Others' Freedom.
|
||||
|
||||
If conditions are imposed on you (whether by court order, agreement or
|
||||
otherwise) that contradict the conditions of this License, they do not
|
||||
excuse you from the conditions of this License. If you cannot convey a
|
||||
covered work so as to satisfy simultaneously your obligations under this
|
||||
License and any other pertinent obligations, then as a consequence you may
|
||||
not convey it at all. For example, if you agree to terms that obligate you
|
||||
to collect a royalty for further conveying from those to whom you convey
|
||||
the Program, the only way you could satisfy both those terms and this
|
||||
License would be to refrain entirely from conveying the Program.
|
||||
|
||||
13. Use with the GNU Affero General Public License.
|
||||
|
||||
Notwithstanding any other provision of this License, you have
|
||||
permission to link or combine any covered work with a work licensed
|
||||
under version 3 of the GNU Affero General Public License into a single
|
||||
combined work, and to convey the resulting work. The terms of this
|
||||
License will continue to apply to the part which is the covered work,
|
||||
but the special requirements of the GNU Affero General Public License,
|
||||
section 13, concerning interaction through a network will apply to the
|
||||
combination as such.
|
||||
|
||||
14. Revised Versions of this License.
|
||||
|
||||
The Free Software Foundation may publish revised and/or new versions of
|
||||
the GNU General Public License from time to time. Such new versions will
|
||||
be similar in spirit to the present version, but may differ in detail to
|
||||
address new problems or concerns.
|
||||
|
||||
Each version is given a distinguishing version number. If the
|
||||
Program specifies that a certain numbered version of the GNU General
|
||||
Public License "or any later version" applies to it, you have the
|
||||
option of following the terms and conditions either of that numbered
|
||||
version or of any later version published by the Free Software
|
||||
Foundation. If the Program does not specify a version number of the
|
||||
GNU General Public License, you may choose any version ever published
|
||||
by the Free Software Foundation.
|
||||
|
||||
If the Program specifies that a proxy can decide which future
|
||||
versions of the GNU General Public License can be used, that proxy's
|
||||
public statement of acceptance of a version permanently authorizes you
|
||||
to choose that version for the Program.
|
||||
|
||||
Later license versions may give you additional or different
|
||||
permissions. However, no additional obligations are imposed on any
|
||||
author or copyright holder as a result of your choosing to follow a
|
||||
later version.
|
||||
|
||||
15. Disclaimer of Warranty.
|
||||
|
||||
THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY
|
||||
APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT
|
||||
HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY
|
||||
OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO,
|
||||
THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||
PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM
|
||||
IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF
|
||||
ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
|
||||
|
||||
16. Limitation of Liability.
|
||||
|
||||
IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
|
||||
WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS
|
||||
THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY
|
||||
GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE
|
||||
USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF
|
||||
DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD
|
||||
PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS),
|
||||
EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF
|
||||
SUCH DAMAGES.
|
||||
|
||||
17. Interpretation of Sections 15 and 16.
|
||||
|
||||
If the disclaimer of warranty and limitation of liability provided
|
||||
above cannot be given local legal effect according to their terms,
|
||||
reviewing courts shall apply local law that most closely approximates
|
||||
an absolute waiver of all civil liability in connection with the
|
||||
Program, unless a warranty or assumption of liability accompanies a
|
||||
copy of the Program in return for a fee.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
How to Apply These Terms to Your New Programs
|
||||
|
||||
If you develop a new program, and you want it to be of the greatest
|
||||
possible use to the public, the best way to achieve this is to make it
|
||||
free software which everyone can redistribute and change under these terms.
|
||||
|
||||
To do so, attach the following notices to the program. It is safest
|
||||
to attach them to the start of each source file to most effectively
|
||||
state the exclusion of warranty; and each file should have at least
|
||||
the "copyright" line and a pointer to where the full notice is found.
|
||||
|
||||
<one line to give the program's name and a brief idea of what it does.>
|
||||
Copyright (C) <year> <name of author>
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
|
||||
Also add information on how to contact you by electronic and paper mail.
|
||||
|
||||
If the program does terminal interaction, make it output a short
|
||||
notice like this when it starts in an interactive mode:
|
||||
|
||||
<program> Copyright (C) <year> <name of author>
|
||||
This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
|
||||
This is free software, and you are welcome to redistribute it
|
||||
under certain conditions; type `show c' for details.
|
||||
|
||||
The hypothetical commands `show w' and `show c' should show the appropriate
|
||||
parts of the General Public License. Of course, your program's commands
|
||||
might be different; for a GUI interface, you would use an "about box".
|
||||
|
||||
You should also get your employer (if you work as a programmer) or school,
|
||||
if any, to sign a "copyright disclaimer" for the program, if necessary.
|
||||
For more information on this, and how to apply and follow the GNU GPL, see
|
||||
<https://www.gnu.org/licenses/>.
|
||||
|
||||
The GNU General Public License does not permit incorporating your program
|
||||
into proprietary programs. If your program is a subroutine library, you
|
||||
may consider it more useful to permit linking proprietary applications with
|
||||
the library. If this is what you want to do, use the GNU Lesser General
|
||||
Public License instead of this License. But first, please read
|
||||
<https://www.gnu.org/licenses/why-not-lgpl.html>.
|
@ -0,0 +1,22 @@
|
||||
FEATURES:="api,always-aggregate"
|
||||
|
||||
markov:
|
||||
cargo build --release --features $(FEATURES)
|
||||
strip target/release/markov
|
||||
|
||||
install:
|
||||
-rc-service markov shutdown && sleep 0.6
|
||||
-rc-service markov stop
|
||||
cp -f target/release/markov /usr/local/bin/markov
|
||||
rc-service markov start
|
||||
|
||||
reinstall: uninstall
|
||||
cp -f target/release/markov /usr/local/bin/markov
|
||||
rm -f /var/nginx/markov.dat
|
||||
rc-service markov start
|
||||
sleep 0.2
|
||||
curl -X PUT --data-binary @default http://127.0.0.1:8001/put
|
||||
|
||||
uninstall:
|
||||
-rc-service markov stop
|
||||
rm -f /usr/local/bin/markov
|
@ -1,6 +0,0 @@
|
||||
Generate strings from markov chain of stdin
|
||||
|
||||
Usage:
|
||||
|
||||
$ cat corpus | markov
|
||||
$ cat corpus | markov <n of outputs to generate>
|
@ -0,0 +1,75 @@
|
||||
# genmarkov
|
||||
HTTP server connecting to a Markov chain
|
||||
|
||||
# Build requirements
|
||||
Unix & Rust nightly are currently requirements to build, for now.
|
||||
|
||||
# Configuration
|
||||
When ran with no arguments, `markov` will attempt to load the config file at `markov.toml`. If it does not exist, it will use the default configuration. (On debug builds, it will also create the default `markov.toml`.)
|
||||
|
||||
An example default configuration file is provided at [./markov.toml](markov.toml).
|
||||
|
||||
When ran with an argument specifying the config file however, it will attempt to load that. If it fails to load the file, the default will be used.
|
||||
|
||||
## Config file entries
|
||||
| Name | Description | Default | Optional |
|
||||
|-------------------------|---------------------------------------------------------|------------------|----------|
|
||||
| `bindpoint` | Address or Unix domain socket for the server to bind to | `127.0.0.1:8001` | No |
|
||||
| `file` | File to save and load the chain from | `chain.dat` | No |
|
||||
| `max_content_length` | Max request body length to allow | `4194304` (4MB) | No |
|
||||
| `max_gen_size` | Max number of strings for a request to generate at once | `256` | No |
|
||||
| `save_interval_secs` | Number of seconds to ensure waiting before saving chain | `2` | Yes |
|
||||
| `trust_x_forwarded_for` | Trust the `X-Forwarded-For` HTTP header | `false` | No |
|
||||
| `filter` | Remove characters from incoming and/or outgoing text | None | Yes |
|
||||
|
||||
### AF_UNIX note
|
||||
When binding to a Unix domain socket, prefix the path with `unix:/` (e.g. `unix://var/markov.socket`)
|
||||
The server will not attempt to remove already existing sockets at the path, so ensure there isn't one before launching.
|
||||
|
||||
## Logging
|
||||
|
||||
Set the `RUST_LOG` environment variable to one of the following to switch runtime logging levels.
|
||||
* `trace` - Most verbose
|
||||
* `debug` - Verbose
|
||||
* `info` - Show input and output to/from the chain and requests
|
||||
* `warn` - Only show warnings (default)
|
||||
* `error` - Only show errors
|
||||
|
||||
## Signals
|
||||
On Unix systems at runtime, some signals are trapped:
|
||||
|
||||
| Signal | Description |
|
||||
|-----------|------------------------------------------------------------------------|
|
||||
| `SIGUSR1` | Immediately save the chain |
|
||||
| `SIGUSR2` | Immediately load the chain |
|
||||
| `SIGQUIT` | Ensure the chain is properly saved and then immediately call `abort()` |
|
||||
| `SIGINT` | Perform a full graceful shutdown |
|
||||
|
||||
# Usage
|
||||
The server exposes several paths for access of the chain
|
||||
|
||||
## Feeding
|
||||
### `PUT /put`
|
||||
Request body is fed to the chain
|
||||
|
||||
#### NOTE
|
||||
Strings fed to the chain must be valid UTF-8 and a size below the value specified in the config file.
|
||||
|
||||
## Generating
|
||||
### `GET /get`
|
||||
Generate a string from the chain
|
||||
|
||||
### `GET /get/<number>`
|
||||
Generate `<number>` strings from the chain
|
||||
|
||||
### `GET /get/sentance`
|
||||
Generate a single sentance from the chain
|
||||
|
||||
### `GET /get/sentance/<number>`
|
||||
Generate `<number>` sentances from the chain
|
||||
|
||||
#### NOTE
|
||||
Number of strings/sentances must be lower than the value specified in the config file.
|
||||
|
||||
# License
|
||||
GPL'd with <3
|
@ -0,0 +1,10 @@
|
||||
bindpoint = '127.0.0.1:8001'
|
||||
file = 'chain.dat'
|
||||
max_content_length = 4194304
|
||||
max_gen_size = 256
|
||||
save_interval_secs = 2
|
||||
trust_x_forwarded_for = false
|
||||
|
||||
[filter]
|
||||
inbound = ''
|
||||
outbound = ''
|
@ -0,0 +1,55 @@
|
||||
//! API errors
|
||||
//use super::*;
|
||||
use std::{
|
||||
error,
|
||||
fmt,
|
||||
};
|
||||
use warp::{
|
||||
Rejection,
|
||||
Reply,
|
||||
};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum ApiError {
|
||||
Body,
|
||||
}
|
||||
|
||||
impl ApiError
|
||||
{
|
||||
#[inline] fn error_code(&self) -> warp::http::StatusCode
|
||||
{
|
||||
status!(match self {
|
||||
Self::Body => 422,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
impl warp::reject::Reject for ApiError{}
|
||||
impl error::Error for ApiError{}
|
||||
impl std::fmt::Display for ApiError
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
match self {
|
||||
Self::Body => write!(f, "invalid data in request body"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<std::str::Utf8Error> for ApiError
|
||||
{
|
||||
fn from(_: std::str::Utf8Error) -> Self
|
||||
{
|
||||
Self::Body
|
||||
}
|
||||
}
|
||||
|
||||
// Handles API rejections
|
||||
pub async fn rejection(err: Rejection) -> Result<impl Reply, Rejection>
|
||||
{
|
||||
if let Some(api) = err.find::<ApiError>() {
|
||||
Ok(warp::reply::with_status(format!("ApiError: {}\n", api), api.error_code()))
|
||||
} else {
|
||||
Err(err)
|
||||
}
|
||||
}
|
@ -0,0 +1,32 @@
|
||||
//! For API calls if enabled
|
||||
use super::*;
|
||||
use std::{
|
||||
iter,
|
||||
convert::Infallible,
|
||||
};
|
||||
use futures::{
|
||||
stream::{
|
||||
self,
|
||||
BoxStream,
|
||||
StreamExt,
|
||||
},
|
||||
};
|
||||
pub mod error;
|
||||
use error::ApiError;
|
||||
|
||||
mod single;
|
||||
|
||||
#[inline] fn aggregate(mut body: impl Buf) -> Result<String, std::str::Utf8Error>
|
||||
{
|
||||
std::str::from_utf8(&body.to_bytes()).map(ToOwned::to_owned)
|
||||
}
|
||||
|
||||
pub async fn single(host: IpAddr, num: Option<usize>, body: impl Buf) -> Result<impl warp::Reply, warp::reject::Rejection>
|
||||
{
|
||||
single::single_stream(host, num, body).await
|
||||
.map(|rx| Response::new(Body::wrap_stream(rx.map(move |x| {
|
||||
info!("{} <- {:?}", host, x);
|
||||
x
|
||||
}))))
|
||||
.map_err(warp::reject::custom)
|
||||
}
|
@ -0,0 +1,47 @@
|
||||
//! Handler for /single/
|
||||
use super::*;
|
||||
|
||||
//TODO: Change to stream impl like normal `feed` has, instead of taking aggregate?
|
||||
pub async fn single_stream(host: IpAddr, num: Option<usize>, body: impl Buf) -> Result<BoxStream<'static, Result<String, Infallible>>, ApiError>
|
||||
{
|
||||
let body = aggregate(body)?;
|
||||
info!("{} <- {:?}", host, &body[..]);
|
||||
|
||||
let mut chain = Chain::new();
|
||||
|
||||
if_debug! {
|
||||
let timer = std::time::Instant::now();
|
||||
}
|
||||
cfg_if! {
|
||||
if #[cfg(feature="split-newlines")] {
|
||||
for body in body.split('\n').filter(|line| !line.trim().is_empty()) {
|
||||
feed::feed(&mut chain, body, 1..);
|
||||
}
|
||||
}else {
|
||||
feed::feed(&mut chain, body, 1..);
|
||||
}
|
||||
}
|
||||
if_debug!{
|
||||
trace!("Write took {}ms", timer.elapsed().as_millis());
|
||||
}
|
||||
if chain.is_empty() {
|
||||
Ok(stream::empty().boxed())
|
||||
} else {
|
||||
match num {
|
||||
|
||||
None => Ok(stream::iter(iter::once(Ok(chain.generate_str()))).boxed()),
|
||||
Some(num) => {
|
||||
let (mut tx, rx) = mpsc::channel(num);
|
||||
tokio::spawn(async move {
|
||||
for string in chain.str_iter_for(num) {
|
||||
if let Err(e) = tx.send(string).await {
|
||||
error!("Failed to send string to body, aborting: {:?}", e.0);
|
||||
break;
|
||||
}
|
||||
}
|
||||
});
|
||||
Ok(StreamExt::map(rx, |x| Ok::<_, Infallible>(x)).boxed())
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,171 @@
|
||||
//! For binding to sockets
|
||||
use super::*;
|
||||
use futures::{
|
||||
prelude::*,
|
||||
};
|
||||
use std::{
|
||||
marker::{
|
||||
Send,
|
||||
Unpin,
|
||||
},
|
||||
fmt,
|
||||
error,
|
||||
path::{
|
||||
Path,
|
||||
PathBuf,
|
||||
},
|
||||
};
|
||||
use tokio::{
|
||||
io::{
|
||||
self,
|
||||
AsyncRead,
|
||||
AsyncWrite,
|
||||
},
|
||||
};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum BindError<E>
|
||||
{
|
||||
IO(io::Error),
|
||||
Warp(warp::Error),
|
||||
Other(E),
|
||||
}
|
||||
|
||||
impl<E: error::Error + 'static> error::Error for BindError<E>
|
||||
{
|
||||
fn source(&self) -> Option<&(dyn error::Error + 'static)> {
|
||||
Some(match &self {
|
||||
Self::IO(io) => io,
|
||||
Self::Other(o) => o,
|
||||
Self::Warp(w) => w,
|
||||
})
|
||||
}
|
||||
}
|
||||
impl<E: fmt::Display> fmt::Display for BindError<E>
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
match self {
|
||||
Self::IO(io) => write!(f, "io error: {}", io),
|
||||
Self::Other(other) => write!(f, "{}", other),
|
||||
Self::Warp(warp) => write!(f, "server error: {}", warp),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct BindpointParseError;
|
||||
|
||||
impl error::Error for BindpointParseError{}
|
||||
impl fmt::Display for BindpointParseError
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
write!(f, "Failed to parse bindpoint as IP or unix domain socket")
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize, PartialOrd)]
|
||||
pub enum Bindpoint
|
||||
{
|
||||
Unix(PathBuf),
|
||||
TCP(SocketAddr),
|
||||
}
|
||||
|
||||
impl fmt::Display for Bindpoint
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
match self {
|
||||
Self::Unix(unix) => write!(f, "unix:/{}", unix.to_string_lossy()),
|
||||
Self::TCP(tcp) => write!(f, "{}", tcp),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl std::str::FromStr for Bindpoint
|
||||
{
|
||||
type Err = BindpointParseError;
|
||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||
Ok(if let Ok(ip) = s.parse::<SocketAddr>() {
|
||||
Self::TCP(ip)
|
||||
} else if s.starts_with("unix:/") {
|
||||
Self::Unix(PathBuf::from(&s[6..].to_owned()))
|
||||
} else {
|
||||
return Err(BindpointParseError);
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
fn bind_unix(to: impl AsRef<Path>) -> io::Result<impl TryStream<Ok= impl AsyncRead + AsyncWrite + Send + Unpin + 'static + Send, Error = impl Into<Box<dyn std::error::Error + Send + Sync>>>>
|
||||
{
|
||||
debug!("Binding to AF_UNIX: {:?}", to.as_ref());
|
||||
let listener = tokio::net::UnixListener::bind(to)?;
|
||||
Ok(listener)
|
||||
}
|
||||
|
||||
pub fn serve<F>(server: warp::Server<F>, bind: Bindpoint, signal: impl Future<Output=()> + Send + 'static) -> Result<(Bindpoint, BoxFuture<'static, ()>), BindError<std::convert::Infallible>>
|
||||
where F: Filter + Clone + Send + Sync + 'static,
|
||||
<F::Future as TryFuture>::Ok: warp::Reply,
|
||||
{
|
||||
Ok(match bind {
|
||||
Bindpoint::TCP(sock) => server.try_bind_with_graceful_shutdown(sock, signal).map(|(sock, fut)| (Bindpoint::TCP(sock), fut.boxed())).map_err(BindError::Warp)?,
|
||||
Bindpoint::Unix(unix) => {
|
||||
(Bindpoint::Unix(unix.clone()),
|
||||
server.serve_incoming_with_graceful_shutdown(bind_unix(unix).map_err(BindError::IO)?, signal).boxed())
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
impl From<SocketAddr> for Bindpoint
|
||||
{
|
||||
fn from(from: SocketAddr) -> Self
|
||||
{
|
||||
Self::TCP(from)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn try_serve<F>(server: warp::Server<F>, bind: impl TryBindpoint, signal: impl Future<Output=()> + Send + 'static) -> Result<(Bindpoint, BoxFuture<'static, ()>), BindError<impl error::Error + 'static>>
|
||||
where F: Filter + Clone + Send + Sync + 'static,
|
||||
<F::Future as TryFuture>::Ok: warp::Reply,
|
||||
{
|
||||
serve(server, bind.try_parse().map_err(BindError::Other)?, signal).map_err(BindError::coerce)
|
||||
}
|
||||
|
||||
pub trait TryBindpoint: Sized
|
||||
{
|
||||
type Err: error::Error + 'static;
|
||||
fn try_parse(self) -> Result<Bindpoint, Self::Err>;
|
||||
}
|
||||
|
||||
impl TryBindpoint for Bindpoint
|
||||
{
|
||||
type Err = std::convert::Infallible;
|
||||
fn try_parse(self) -> Result<Bindpoint, Self::Err>
|
||||
{
|
||||
Ok(self)
|
||||
}
|
||||
}
|
||||
|
||||
impl<T: AsRef<str>> TryBindpoint for T
|
||||
{
|
||||
type Err = BindpointParseError;
|
||||
fn try_parse(self) -> Result<Bindpoint, Self::Err>
|
||||
{
|
||||
self.as_ref().parse()
|
||||
}
|
||||
}
|
||||
|
||||
impl BindError<std::convert::Infallible>
|
||||
{
|
||||
pub fn coerce<T>(self) -> BindError<T>
|
||||
{
|
||||
match self {
|
||||
Self::Warp(w) => BindError::Warp(w),
|
||||
Self::IO(w) => BindError::IO(w),
|
||||
#[cold] _ => unreachable!(),
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,28 @@
|
||||
use libc::{
|
||||
c_void,
|
||||
};
|
||||
/// Copy slice of bytes only
|
||||
///
|
||||
/// # Notes
|
||||
/// `dst` and `src` must not overlap. See [move_slice].
|
||||
pub fn copy_slice(dst: &mut [u8], src: &[u8]) -> usize
|
||||
{
|
||||
let sz = std::cmp::min(dst.len(),src.len());
|
||||
unsafe {
|
||||
libc::memcpy(&mut dst[0] as *mut u8 as *mut c_void, &src[0] as *const u8 as *const c_void, sz);
|
||||
}
|
||||
sz
|
||||
}
|
||||
|
||||
/// Move slice of bytes only
|
||||
///
|
||||
/// # Notes
|
||||
/// `dst` and `src` can overlap.
|
||||
pub fn move_slice(dst: &mut [u8], src: &[u8]) -> usize
|
||||
{
|
||||
let sz = std::cmp::min(dst.len(),src.len());
|
||||
unsafe {
|
||||
libc::memmove(&mut dst[0] as *mut u8 as *mut c_void, &src[0] as *const u8 as *const c_void, sz);
|
||||
}
|
||||
sz
|
||||
}
|
@ -0,0 +1,175 @@
|
||||
//! Stream related things
|
||||
use super::*;
|
||||
use std::{
|
||||
task::{
|
||||
Poll,
|
||||
Context,
|
||||
},
|
||||
pin::Pin,
|
||||
};
|
||||
use tokio::{
|
||||
io::{
|
||||
AsyncBufRead,
|
||||
AsyncRead,
|
||||
},
|
||||
prelude::*,
|
||||
};
|
||||
use futures::{
|
||||
stream::{
|
||||
Stream,
|
||||
StreamExt,
|
||||
Fuse,
|
||||
},
|
||||
};
|
||||
use pin_project::pin_project;
|
||||
|
||||
/// Converts a stream of byte-containing objects into an `AsyncRead` and `AsyncBufRead`er.
|
||||
#[pin_project]
|
||||
pub struct StreamReader<I, T>
|
||||
where I: Stream<Item=T>
|
||||
{
|
||||
#[pin]
|
||||
source: Fuse<I>,
|
||||
buffer: Vec<u8>,
|
||||
}
|
||||
|
||||
impl<T, I> StreamReader<I, T>
|
||||
where I: Stream<Item=T>,
|
||||
T: AsRef<[u8]>
|
||||
{
|
||||
/// The current buffer
|
||||
pub fn buffer(&self) -> &[u8]
|
||||
{
|
||||
&self.buffer[..]
|
||||
}
|
||||
/// Consume into the original stream
|
||||
pub fn into_inner(self) -> I
|
||||
{
|
||||
self.source.into_inner()
|
||||
}
|
||||
/// Create a new instance with a buffer capacity
|
||||
pub fn with_capacity(source: I, cap: usize) -> Self
|
||||
{
|
||||
Self {
|
||||
source: source.fuse(),
|
||||
buffer: Vec::with_capacity(cap)
|
||||
}
|
||||
}
|
||||
/// Create a new instance from this stream
|
||||
pub fn new(source: I) -> Self
|
||||
{
|
||||
Self {
|
||||
source: source.fuse(),
|
||||
buffer: Vec::new(),
|
||||
}
|
||||
}
|
||||
/// Attempt to add to this buffer
|
||||
#[cold] fn poll_next(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<usize>
|
||||
{
|
||||
let this = self.project();
|
||||
match this.source.poll_next(cx) {
|
||||
Poll::Ready(None) => Poll::Ready(0),
|
||||
Poll::Ready(Some(buf)) if buf.as_ref().len() > 0 => {
|
||||
let buf = buf.as_ref();
|
||||
this.buffer.extend_from_slice(buf);
|
||||
Poll::Ready(buf.len())
|
||||
},
|
||||
_ => Poll::Pending,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl<T: AsRef<[u8]>, I: Stream<Item=T>> AsyncRead for StreamReader<I,T>
|
||||
{
|
||||
fn poll_read(self: Pin<&mut Self>, cx: &mut Context<'_>, buf: &mut [u8]) -> Poll<io::Result<usize>> {
|
||||
let this = self.project();
|
||||
if this.buffer.len() != 0 {
|
||||
// We can fill the whole buffer, do it.
|
||||
Poll::Ready(Ok(bytes::copy_slice(buf, this.buffer.drain(..buf.len()).as_slice())))
|
||||
} else {
|
||||
// Buffer is empty, try to fill it
|
||||
match match this.source.poll_next(cx) {
|
||||
Poll::Ready(None) => Poll::Ready(0),
|
||||
Poll::Ready(Some(buf)) if buf.as_ref().len() > 0 => {
|
||||
let buf = buf.as_ref();
|
||||
this.buffer.extend_from_slice(buf);
|
||||
Poll::Ready(buf.len())
|
||||
},
|
||||
_ => Poll::Pending,
|
||||
} {
|
||||
Poll::Ready(0) => Poll::Ready(Ok(0)),
|
||||
Poll::Ready(x) => {
|
||||
// x has been written
|
||||
Poll::Ready(Ok(bytes::copy_slice(buf, this.buffer.drain(..x).as_slice())))
|
||||
},
|
||||
_ => Poll::Pending,
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
impl<T: AsRef<[u8]>, I: Stream<Item=T>> AsyncBufRead for StreamReader<I,T>
|
||||
{
|
||||
fn poll_fill_buf(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<io::Result<&[u8]>> {
|
||||
let this = self.project();
|
||||
if this.buffer.len() < 1 {
|
||||
// Fetch more into buffer
|
||||
match match this.source.poll_next(cx) {
|
||||
Poll::Ready(None) => Poll::Ready(0),
|
||||
Poll::Ready(Some(buf)) if buf.as_ref().len() > 0 => {
|
||||
let buf = buf.as_ref();
|
||||
this.buffer.extend_from_slice(buf);
|
||||
Poll::Ready(buf.len())
|
||||
},
|
||||
_ => Poll::Pending,
|
||||
} {
|
||||
Poll::Ready(0) => Poll::Ready(Ok(&[])), // should we return EOF error here?
|
||||
Poll::Ready(x) => Poll::Ready(Ok(&this.buffer[..x])),
|
||||
_ => Poll::Pending
|
||||
}
|
||||
} else {
|
||||
Poll::Ready(Ok(&this.buffer[..]))
|
||||
}
|
||||
}
|
||||
fn consume(self: Pin<&mut Self>, amt: usize) {
|
||||
self.project().buffer.drain(..amt);
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests
|
||||
{
|
||||
use super::*;
|
||||
use tokio::{
|
||||
sync::{
|
||||
mpsc,
|
||||
},
|
||||
};
|
||||
#[tokio::test]
|
||||
async fn stream_of_vec()
|
||||
{
|
||||
let (mut tx, rx) = mpsc::channel(16);
|
||||
let sender = tokio::spawn(async move {
|
||||
tx.send("Hello ").await.unwrap();
|
||||
tx.send("world").await.unwrap();
|
||||
tx.send("\n").await.unwrap();
|
||||
tx.send("How ").await.unwrap();
|
||||
tx.send("are ").await.unwrap();
|
||||
tx.send("you").await.unwrap();
|
||||
});
|
||||
|
||||
let mut reader = StreamReader::new(rx);
|
||||
|
||||
let mut output = String::new();
|
||||
let mut read;
|
||||
while {read = reader.read_line(&mut output).await.expect("Failed to read"); read!=0} {
|
||||
println!("Read: {}", read);
|
||||
}
|
||||
|
||||
println!("Done: {:?}", output);
|
||||
sender.await.expect("Child panic");
|
||||
assert_eq!(&output[..], "Hello world\nHow are you");
|
||||
}
|
||||
}
|
@ -0,0 +1,141 @@
|
||||
//! Server config
|
||||
use super::*;
|
||||
use std::{
|
||||
net::SocketAddr,
|
||||
path::Path,
|
||||
io,
|
||||
borrow::Cow,
|
||||
num::NonZeroU64,
|
||||
};
|
||||
use tokio::{
|
||||
fs::OpenOptions,
|
||||
prelude::*,
|
||||
time::Duration,
|
||||
io::BufReader,
|
||||
};
|
||||
|
||||
pub const DEFAULT_FILE_LOCATION: &'static str = "markov.toml";
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Hash, Serialize, Deserialize)]
|
||||
pub struct Config
|
||||
{
|
||||
pub bindpoint: String,
|
||||
pub file: String,
|
||||
pub max_content_length: u64,
|
||||
pub max_gen_size: usize,
|
||||
pub save_interval_secs: Option<NonZeroU64>,
|
||||
pub trust_x_forwarded_for: bool,
|
||||
#[serde(default)]
|
||||
pub filter: FilterConfig,
|
||||
}
|
||||
|
||||
#[derive(Debug, Default, Clone, PartialEq, Eq, PartialOrd, Hash, Serialize, Deserialize)]
|
||||
pub struct FilterConfig
|
||||
{
|
||||
inbound: String,
|
||||
#[serde(default)]
|
||||
outbound: String,
|
||||
}
|
||||
|
||||
impl FilterConfig
|
||||
{
|
||||
pub fn get_inbound_filter(&self) -> sanitise::filter::Filter
|
||||
{
|
||||
let filt: sanitise::filter::Filter = self.inbound.parse().unwrap();
|
||||
if !filt.is_empty()
|
||||
{
|
||||
info!("Loaded inbound filter: {:?}", filt.iter().collect::<String>());
|
||||
}
|
||||
filt
|
||||
}
|
||||
pub fn get_outbound_filter(&self) -> sanitise::filter::Filter
|
||||
{
|
||||
let filt: sanitise::filter::Filter = self.outbound.parse().unwrap();
|
||||
if !filt.is_empty()
|
||||
{
|
||||
info!("Loaded outbound filter: {:?}", filt.iter().collect::<String>());
|
||||
}
|
||||
filt
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for Config
|
||||
{
|
||||
#[inline]
|
||||
fn default() -> Self
|
||||
{
|
||||
Self {
|
||||
bindpoint: SocketAddr::from(([127,0,0,1], 8001)).to_string(),
|
||||
file: "chain.dat".to_owned(),
|
||||
max_content_length: 1024 * 1024 * 4,
|
||||
max_gen_size: 256,
|
||||
save_interval_secs: Some(unsafe{NonZeroU64::new_unchecked(2)}),
|
||||
trust_x_forwarded_for: false,
|
||||
filter: Default::default(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Config
|
||||
{
|
||||
pub fn save_interval(&self) -> Option<Duration>
|
||||
{
|
||||
self.save_interval_secs.map(|x| Duration::from_secs(x.into()))
|
||||
}
|
||||
pub async fn load(from: impl AsRef<Path>) -> io::Result<Self>
|
||||
{
|
||||
let file = OpenOptions::new()
|
||||
.read(true)
|
||||
.open(from).await?;
|
||||
|
||||
let mut buffer= String::new();
|
||||
let reader = BufReader::new(file);
|
||||
let mut lines = reader.lines();
|
||||
while let Some(line) = lines.next_line().await? {
|
||||
buffer.push_str(&line[..]);
|
||||
buffer.push('\n');
|
||||
}
|
||||
toml::de::from_str(&buffer[..]).map_err(|e| io::Error::new(io::ErrorKind::InvalidInput, e))
|
||||
}
|
||||
|
||||
pub async fn save(&self, to: impl AsRef<Path>) -> io::Result<()>
|
||||
{
|
||||
let config = toml::ser::to_string_pretty(self).map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?;
|
||||
let mut file = OpenOptions::new()
|
||||
.write(true)
|
||||
.create(true)
|
||||
.truncate(true)
|
||||
.open(to).await?;
|
||||
file.write_all(config.as_bytes()).await?;
|
||||
file.shutdown().await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// Try to load config file specified by args, or default config file
|
||||
pub fn load() -> impl futures::future::Future<Output =Option<Config>>
|
||||
{
|
||||
load_args(std::env::args().skip(1))
|
||||
}
|
||||
|
||||
async fn load_args<I: Iterator<Item=String>>(mut from: I) -> Option<Config>
|
||||
{
|
||||
let place = if let Some(arg) = from.next() {
|
||||
trace!("File {:?} provided", arg);
|
||||
Cow::Owned(arg)
|
||||
} else {
|
||||
warn!("No config file provided. Using default location {:?}", DEFAULT_FILE_LOCATION);
|
||||
Cow::Borrowed(DEFAULT_FILE_LOCATION)
|
||||
};
|
||||
|
||||
match Config::load(place.as_ref()).await {
|
||||
Ok(cfg) => {
|
||||
info!("Loaded config file {:?}", place);
|
||||
Some(cfg)
|
||||
},
|
||||
Err(err) => {
|
||||
error!("Failed to load config file from {:?}: {}", place, err);
|
||||
None
|
||||
},
|
||||
}
|
||||
}
|
@ -0,0 +1,152 @@
|
||||
//! Extensions
|
||||
use std::{
|
||||
iter,
|
||||
ops::{
|
||||
Range,
|
||||
Deref,DerefMut,
|
||||
},
|
||||
marker::{
|
||||
PhantomData,
|
||||
Send,
|
||||
},
|
||||
};
|
||||
|
||||
pub trait StringJoinExt: Sized
|
||||
{
|
||||
fn join<P: AsRef<str>>(self, sep: P) -> String;
|
||||
}
|
||||
|
||||
impl<I,T> StringJoinExt for I
|
||||
where I: IntoIterator<Item=T>,
|
||||
T: AsRef<str>
|
||||
{
|
||||
fn join<P: AsRef<str>>(self, sep: P) -> String
|
||||
{
|
||||
let mut string = String::new();
|
||||
for (first, s) in iter::successors(Some(true), |_| Some(false)).zip(self.into_iter())
|
||||
{
|
||||
if !first {
|
||||
string.push_str(sep.as_ref());
|
||||
}
|
||||
string.push_str(s.as_ref());
|
||||
}
|
||||
string
|
||||
}
|
||||
}
|
||||
|
||||
pub trait FindSliceBounds
|
||||
{
|
||||
type SliceType: ?Sized;
|
||||
fn slice_bounds(&self, from: &Self::SliceType) -> Range<usize>;
|
||||
}
|
||||
|
||||
impl<T: ?Sized + AsRef<str>> FindSliceBounds for T
|
||||
{
|
||||
type SliceType = str;
|
||||
fn slice_bounds(&self, from: &Self::SliceType) -> Range<usize>{
|
||||
let this = self.as_ref();
|
||||
unsafe {
|
||||
let sptr = from.as_ptr();
|
||||
let eptr = sptr.add(from.len());
|
||||
|
||||
let ssptr = this.as_ptr();
|
||||
let septr = ssptr.add(this.len());
|
||||
|
||||
let sptr = sptr as usize;
|
||||
let ssptr = ssptr as usize;
|
||||
let eptr = eptr as usize;
|
||||
let septr = septr as usize;
|
||||
|
||||
assert!(sptr >= ssptr && sptr <= septr, "Start index of slice is outside the bounds of self");
|
||||
assert!(eptr >= ssptr && eptr <= septr, "End index of slice is outside the bounds of self");
|
||||
|
||||
(sptr - ssptr)..(eptr - ssptr)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub trait SliceInPlace
|
||||
{
|
||||
fn drain_inverse<R: std::ops::RangeBounds<usize>>(&mut self, slice: R);
|
||||
}
|
||||
|
||||
impl SliceInPlace for String
|
||||
{
|
||||
fn drain_inverse<R: std::ops::RangeBounds<usize>>(&mut self, slice: R)
|
||||
{
|
||||
use std::ops::Bound;
|
||||
match slice.end_bound() {
|
||||
Bound::Excluded(&ex) => drop(self.drain(ex..)),
|
||||
Bound::Included(&inc) => drop(self.drain(inc+1..)),
|
||||
_ => (),
|
||||
};
|
||||
match slice.start_bound() {
|
||||
Bound::Included(&ex) => drop(self.drain(..ex)),
|
||||
Bound::Excluded(&ex) => drop(..ex+1),
|
||||
_ => ()
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
pub trait TrimInPlace
|
||||
{
|
||||
fn trim_in_place(&mut self) -> &mut Self;
|
||||
}
|
||||
|
||||
impl TrimInPlace for String
|
||||
{
|
||||
fn trim_in_place(&mut self) -> &mut Self {
|
||||
let bounds = self.slice_bounds(self.trim());
|
||||
self.drain_inverse(bounds);
|
||||
self
|
||||
}
|
||||
}
|
||||
|
||||
pub trait MapTuple2<T,U>
|
||||
{
|
||||
fn map<V,W, F: FnOnce((T,U)) -> (V,W)>(self, fun: F) -> (V,W);
|
||||
}
|
||||
|
||||
impl<T,U> MapTuple2<T,U> for (T,U)
|
||||
{
|
||||
#[inline] fn map<V,W, F: FnOnce((T,U)) -> (V,W)>(self, fun: F) -> (V,W)
|
||||
{
|
||||
fun(self)
|
||||
}
|
||||
}
|
||||
|
||||
/// To make sure we don't keep this data across an `await` boundary.
|
||||
#[repr(transparent)]
|
||||
pub struct AssertNotSend<T>(pub T, PhantomData<*const T>);
|
||||
|
||||
impl<T> AssertNotSend<T>
|
||||
{
|
||||
pub const fn new(from :T) -> Self
|
||||
{
|
||||
Self(from, PhantomData)
|
||||
}
|
||||
pub fn into_inner(self) -> T
|
||||
{
|
||||
self.0
|
||||
}
|
||||
}
|
||||
|
||||
/// Require a future is Send
|
||||
#[inline(always)] pub fn require_send<T: Send>(t: T) -> T
|
||||
{
|
||||
t
|
||||
}
|
||||
|
||||
impl<T> Deref for AssertNotSend<T>
|
||||
{
|
||||
type Target = T;
|
||||
fn deref(&self) -> &Self::Target {
|
||||
&self.0
|
||||
}
|
||||
}
|
||||
impl<T> DerefMut for AssertNotSend<T>
|
||||
{
|
||||
fn deref_mut(&mut self) -> &mut Self::Target {
|
||||
&mut self.0
|
||||
}
|
||||
}
|
@ -0,0 +1,130 @@
|
||||
//! Feeding the chain
|
||||
use super::*;
|
||||
use sanitise::Sentance;
|
||||
|
||||
const FEED_BOUNDS: std::ops::RangeFrom<usize> = 2..; //TODO: Add to config somehow
|
||||
|
||||
|
||||
pub fn feed(chain: &mut Chain<String>, what: impl AsRef<str>, bounds: impl std::ops::RangeBounds<usize>)
|
||||
{
|
||||
cfg_if! {
|
||||
if #[cfg(feature="split-sentance")] {
|
||||
let map = Sentance::new_iter(&what) //get each sentance in string
|
||||
.map(|what| what.words()
|
||||
.map(|s| s.to_owned()).collect::<Vec<_>>());
|
||||
debug_assert!(!bounds.contains(&0), "Cannot allow 0 size feeds");
|
||||
for map in map {// feed each sentance seperately
|
||||
if bounds.contains(&map.len()) {
|
||||
chain.feed(map);
|
||||
}
|
||||
else {
|
||||
debug!("Ignoring feed of invalid length {}", map.len());
|
||||
}
|
||||
}
|
||||
} else {
|
||||
let map = Sentance::new_iter(&what) //get each sentance in string
|
||||
.map(|what| what.words())
|
||||
.flatten() // add all into one buffer
|
||||
.map(|s| s.to_owned()).collect::<Vec<_>>();
|
||||
debug_assert!(!bounds.contains(&0), "Cannot allow 0 size feeds");
|
||||
if bounds.contains(&map.len()) {
|
||||
chain.feed(map);
|
||||
}
|
||||
else {
|
||||
debug!("Ignoring feed of invalid length {}", map.len());
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn full(who: &IpAddr, state: State, body: impl Unpin + Stream<Item = Result<impl Buf, impl std::error::Error + 'static>>) -> Result<usize, FillBodyError> {
|
||||
|
||||
let mut written = 0usize;
|
||||
if_debug! {
|
||||
let timer = std::time::Instant::now();
|
||||
}
|
||||
macro_rules! feed {
|
||||
($chain:expr, $buffer:ident, $bounds:expr) => {
|
||||
{
|
||||
let buffer = $buffer;
|
||||
feed($chain, &buffer, $bounds)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
cfg_if!{
|
||||
if #[cfg(any(not(feature="split-newlines"), feature="always-aggregate"))] {
|
||||
let mut body = body;
|
||||
let mut buffer = Vec::new();
|
||||
while let Some(buf) = body.next().await {
|
||||
let mut body = buf.map_err(|_| FillBodyError)?;
|
||||
while body.has_remaining() {
|
||||
if body.bytes().len() > 0 {
|
||||
buffer.extend_from_slice(body.bytes());
|
||||
let cnt = body.bytes().len();
|
||||
body.advance(cnt);
|
||||
written += cnt;
|
||||
}
|
||||
}
|
||||
}
|
||||
let buffer = std::str::from_utf8(&buffer[..]).map_err(|_| FillBodyError)?;
|
||||
let buffer = state.inbound_filter().filter_cow(buffer);
|
||||
info!("{} -> {:?}", who, buffer);
|
||||
let mut chain = state.chain().write().await;
|
||||
cfg_if! {
|
||||
if #[cfg(feature="split-newlines")] {
|
||||
for buffer in buffer.split('\n').filter(|line| !line.trim().is_empty()) {
|
||||
feed!(&mut chain, buffer, FEED_BOUNDS);
|
||||
|
||||
}
|
||||
} else {
|
||||
feed!(&mut chain, buffer, FEED_BOUNDS);
|
||||
|
||||
}
|
||||
}
|
||||
} else {
|
||||
use tokio::prelude::*;
|
||||
|
||||
let reader = chunking::StreamReader::new(body.filter_map(|x| x.map(|mut x| x.to_bytes()).ok()));
|
||||
let mut lines = reader.lines();
|
||||
|
||||
#[cfg(feature="hog-buffer")]
|
||||
let mut chain = state.chain().write().await;
|
||||
while let Some(line) = lines.next_line().await.map_err(|_| FillBodyError)? {
|
||||
let line = state.inbound_filter().filter_cow(&line);
|
||||
let line = line.trim();
|
||||
if !line.is_empty() {
|
||||
#[cfg(not(feature="hog-buffer"))]
|
||||
let mut chain = state.chain().write().await; // Acquire mutex once per line? Is this right?
|
||||
|
||||
feed!(&mut chain, line, FEED_BOUNDS);
|
||||
info!("{} -> {:?}", who, line);
|
||||
}
|
||||
written+=line.len();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if_debug!{
|
||||
trace!("Write took {}ms", timer.elapsed().as_millis());
|
||||
}
|
||||
state.notify_save();
|
||||
Ok(written)
|
||||
|
||||
}
|
||||
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct FillBodyError;
|
||||
|
||||
impl error::Error for FillBodyError{}
|
||||
impl warp::reject::Reject for FillBodyError{}
|
||||
impl fmt::Display for FillBodyError
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
write!(f, "failed to feed chain with this data")
|
||||
}
|
||||
}
|
||||
|
@ -0,0 +1,74 @@
|
||||
use std::{
|
||||
net::{
|
||||
IpAddr,
|
||||
AddrParseError,
|
||||
},
|
||||
str,
|
||||
error,
|
||||
fmt,
|
||||
};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct XFormatError;
|
||||
|
||||
impl error::Error for XFormatError{}
|
||||
|
||||
impl fmt::Display for XFormatError
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
write!(f, "X-Forwarded-For was not in the correct format")
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialOrd, PartialEq, Eq, Default)]
|
||||
pub struct XForwardedFor(Vec<IpAddr>);
|
||||
|
||||
impl XForwardedFor
|
||||
{
|
||||
pub fn new() -> Self
|
||||
{
|
||||
Self(Vec::new())
|
||||
}
|
||||
pub fn single(ip: impl Into<IpAddr>) -> Self
|
||||
{
|
||||
Self(vec![ip.into()])
|
||||
}
|
||||
pub fn addrs(&self) -> &[IpAddr]
|
||||
{
|
||||
&self.0[..]
|
||||
}
|
||||
|
||||
pub fn into_first(self) -> Option<IpAddr>
|
||||
{
|
||||
self.0.into_iter().next()
|
||||
}
|
||||
|
||||
pub fn into_addrs(self) -> Vec<IpAddr>
|
||||
{
|
||||
self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl str::FromStr for XForwardedFor
|
||||
{
|
||||
type Err = XFormatError;
|
||||
|
||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||
let mut output = Vec::new();
|
||||
for next in s.split(',')
|
||||
{
|
||||
output.push(next.trim().parse()?)
|
||||
}
|
||||
Ok(Self(output))
|
||||
}
|
||||
}
|
||||
|
||||
impl From<AddrParseError> for XFormatError
|
||||
{
|
||||
#[inline(always)] fn from(_: AddrParseError) -> Self
|
||||
{
|
||||
Self
|
||||
}
|
||||
}
|
||||
|
@ -0,0 +1,34 @@
|
||||
//! Generating the strings
|
||||
use super::*;
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct GenBodyError(pub String);
|
||||
|
||||
impl error::Error for GenBodyError{}
|
||||
impl fmt::Display for GenBodyError
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
write!(f, "failed to write {:?} to body", self.0)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
pub async fn body(state: State, num: Option<usize>, mut output: mpsc::Sender<String>) -> Result<(), GenBodyError>
|
||||
{
|
||||
let chain = state.chain().read().await;
|
||||
if !chain.is_empty() {
|
||||
let filter = state.outbound_filter();
|
||||
match num {
|
||||
Some(num) if num < state.config().max_gen_size => {
|
||||
//This could DoS `full_body` and writes, potentially.
|
||||
for string in chain.str_iter_for(num) {
|
||||
output.send(filter.filter_owned(string)).await.map_err(|e| GenBodyError(e.0))?;
|
||||
}
|
||||
},
|
||||
_ => output.send(filter.filter_owned(chain.generate_str())).await.map_err(|e| GenBodyError(e.0))?,
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
@ -1,49 +1,293 @@
|
||||
#![feature(split_inclusive)]
|
||||
|
||||
#![allow(dead_code)]
|
||||
|
||||
#[macro_use] extern crate log;
|
||||
|
||||
use chain::{
|
||||
Chain,
|
||||
};
|
||||
use warp::{
|
||||
Filter,
|
||||
Buf,
|
||||
reply::Response,
|
||||
};
|
||||
use hyper::Body;
|
||||
use std::{
|
||||
io::{
|
||||
BufRead,
|
||||
self,
|
||||
sync::Arc,
|
||||
fmt,
|
||||
error,
|
||||
net::{
|
||||
SocketAddr,
|
||||
IpAddr,
|
||||
},
|
||||
};
|
||||
use tokio::{
|
||||
sync::{
|
||||
RwLock,
|
||||
mpsc,
|
||||
Notify,
|
||||
},
|
||||
stream::{Stream,StreamExt,},
|
||||
};
|
||||
use serde::{
|
||||
Serialize,
|
||||
Deserialize
|
||||
};
|
||||
use futures::{
|
||||
future::{
|
||||
FutureExt,
|
||||
BoxFuture,
|
||||
join_all,
|
||||
},
|
||||
};
|
||||
use lazy_static::lazy_static;
|
||||
use cfg_if::cfg_if;
|
||||
|
||||
fn buffered_read_all_lines<T: BufRead+?Sized, F: FnMut(&str) -> io::Result<()>>(input: &mut T, mut then: F) -> io::Result<usize>
|
||||
{
|
||||
let mut buffer = String::new();
|
||||
let mut read;
|
||||
let mut total=0;
|
||||
while {read = input.read_line(&mut buffer)?; read!=0} {
|
||||
if buffer.trim().len() > 0 {
|
||||
then(&buffer[..])?;
|
||||
macro_rules! if_debug {
|
||||
($($tt:tt)*) => {
|
||||
cfg_if::cfg_if!{
|
||||
if #[cfg(debug_assertions)] {
|
||||
$($tt)*
|
||||
}
|
||||
}
|
||||
buffer.clear();
|
||||
total += read;
|
||||
}
|
||||
Ok(total)
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let stdin = io::stdin();
|
||||
let mut stdin = stdin.lock();
|
||||
let mut chain = Chain::new();
|
||||
macro_rules! status {
|
||||
($code:expr) => {
|
||||
::warp::http::status::StatusCode::from_u16($code).unwrap()
|
||||
};
|
||||
}
|
||||
|
||||
buffered_read_all_lines(&mut stdin, |string| {
|
||||
chain.feed(&string.split_whitespace()
|
||||
.filter(|word| !word.is_empty())
|
||||
.map(|s| s.to_owned()).collect::<Vec<_>>());
|
||||
|
||||
Ok(())
|
||||
}).expect("Failed to read from stdin");
|
||||
|
||||
if !chain.is_empty() {
|
||||
if let Some(num) = std::env::args().skip(1).next() {
|
||||
let sz: usize = num.parse().expect("Cannot parse number of tokens to generate");
|
||||
for string in chain.str_iter_for(sz) {
|
||||
println!("{}", string);
|
||||
mod ext;
|
||||
use ext::*;
|
||||
mod util;
|
||||
mod sanitise;
|
||||
mod bytes;
|
||||
mod chunking;
|
||||
#[cfg(feature="api")]
|
||||
mod api;
|
||||
#[cfg(target_family="unix")]
|
||||
mod signals;
|
||||
mod config;
|
||||
mod state;
|
||||
use state::State;
|
||||
mod save;
|
||||
mod forwarded_list;
|
||||
use forwarded_list::XForwardedFor;
|
||||
|
||||
mod feed;
|
||||
mod gen;
|
||||
mod sentance;
|
||||
|
||||
const DEFAULT_LOG_LEVEL: &str = "warn";
|
||||
|
||||
fn init_log()
|
||||
{
|
||||
let level = match std::env::var_os("RUST_LOG") {
|
||||
None => {
|
||||
std::env::set_var("RUST_LOG", DEFAULT_LOG_LEVEL);
|
||||
std::borrow::Cow::Borrowed(std::ffi::OsStr::new(DEFAULT_LOG_LEVEL))
|
||||
},
|
||||
Some(w) => std::borrow::Cow::Owned(w),
|
||||
};
|
||||
pretty_env_logger::init();
|
||||
trace!("Initialising `genmarkov` ({}) v{} with log level {:?}.\n\tMade by {} with <3.\n\tLicensed with GPL v3 or later",
|
||||
std::env::args().next().unwrap(),
|
||||
env!("CARGO_PKG_VERSION"),
|
||||
level,
|
||||
env!("CARGO_PKG_AUTHORS"));
|
||||
}
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
init_log();
|
||||
|
||||
let config = match config::load().await {
|
||||
Some(v) => v,
|
||||
_ => {
|
||||
let cfg = config::Config::default();
|
||||
#[cfg(debug_assertions)]
|
||||
{
|
||||
if let Err(err) = cfg.save(config::DEFAULT_FILE_LOCATION).await {
|
||||
error!("Failed to create default config file: {}", err);
|
||||
}
|
||||
}
|
||||
cfg
|
||||
},
|
||||
};
|
||||
trace!("Using config {:?}", config);
|
||||
|
||||
let chain = Arc::new(RwLock::new(match save::load(&config.file).await {
|
||||
Ok(chain) => {
|
||||
info!("Loaded chain from {:?}", config.file);
|
||||
chain
|
||||
},
|
||||
Err(e) => {
|
||||
warn!("Failed to load chain, creating new");
|
||||
trace!("Error: {}", e);
|
||||
Chain::new()
|
||||
},
|
||||
}));
|
||||
{
|
||||
let mut tasks = Vec::<BoxFuture<'static, ()>>::new();
|
||||
let (state, chain) = {
|
||||
let save_when = Arc::new(Notify::new());
|
||||
|
||||
let state = State::new(config,
|
||||
Arc::clone(&chain),
|
||||
Arc::clone(&save_when));
|
||||
let state2 = state.clone();
|
||||
let saver = tokio::spawn(save::host(state.clone()));
|
||||
let chain = warp::any().map(move || state.clone());
|
||||
|
||||
tasks.push(saver.map(|res| res.expect("Saver panicked")).boxed());
|
||||
(state2, chain)
|
||||
};
|
||||
|
||||
let client_ip = if state.config().trust_x_forwarded_for {
|
||||
warp::header("x-forwarded-for")
|
||||
.map(|ip: XForwardedFor| ip)
|
||||
.and_then(|x: XForwardedFor| async move { x.into_first().ok_or_else(|| warp::reject::not_found()) })
|
||||
.or(warp::filters::addr::remote()
|
||||
.and_then(|x: Option<SocketAddr>| async move { x.map(|x| x.ip()).ok_or_else(|| warp::reject::not_found()) }))
|
||||
.unify().boxed()
|
||||
} else {
|
||||
println!("{}", chain.generate_str());
|
||||
warp::filters::addr::remote().and_then(|x: Option<SocketAddr>| async move {x.map(|x| x.ip()).ok_or_else(|| warp::reject::not_found())}).boxed()
|
||||
};
|
||||
|
||||
let push = warp::put()
|
||||
.and(warp::path("put"))
|
||||
.and(chain.clone())
|
||||
.and(client_ip.clone())
|
||||
.and(warp::body::content_length_limit(state.config().max_content_length))
|
||||
.and(warp::body::stream())
|
||||
.and_then(|state: State, host: IpAddr, buf| {
|
||||
async move {
|
||||
feed::full(&host, state, buf).await
|
||||
.map(|_| warp::reply::with_status(warp::reply(), status!(201)))
|
||||
.map_err(|_| warp::reject::not_found()) //(warp::reject::custom) //TODO: Recover rejection filter down below for custom error return
|
||||
}
|
||||
})
|
||||
.with(warp::log("markov::put"));
|
||||
|
||||
|
||||
cfg_if!{
|
||||
if #[cfg(feature="api")] {
|
||||
let api = {
|
||||
let single = {
|
||||
let msz = state.config().max_gen_size;
|
||||
warp::post()
|
||||
.and(warp::path("single"))
|
||||
.and(client_ip.clone())
|
||||
.and(warp::path::param()
|
||||
.map(move |sz: usize| {
|
||||
if sz == 0 || (2..=msz).contains(&sz) {
|
||||
Some(sz)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
})
|
||||
.or(warp::any().map(|| None))
|
||||
.unify())
|
||||
.and(warp::body::content_length_limit(state.config().max_content_length))
|
||||
.and(warp::body::aggregate())
|
||||
.and_then(api::single)
|
||||
.with(warp::log("markov::api::single"))
|
||||
};
|
||||
warp::path("api")
|
||||
.and(single)
|
||||
.recover(api::error::rejection)
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
let read = warp::get()
|
||||
.and(chain.clone())
|
||||
.and(client_ip.clone())
|
||||
.and(warp::path::param().map(|opt: usize| Some(opt))
|
||||
.or(warp::path::end().map(|| Option::<usize>::None)).unify())
|
||||
.and_then(|state: State, host: IpAddr, num: Option<usize>| {
|
||||
async move {
|
||||
let (tx, rx) = mpsc::channel(state.config().max_gen_size);
|
||||
tokio::spawn(gen::body(state, num, tx));
|
||||
Ok::<_, std::convert::Infallible>(Response::new(Body::wrap_stream(rx.filter_map(move |mut x| {
|
||||
if x.trim_in_place().len() != 0 {
|
||||
info!("{} <- {:?}", host, x);
|
||||
x.push('\n');
|
||||
Some(Ok::<_, std::convert::Infallible>(x))
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}))))
|
||||
}
|
||||
})
|
||||
.with(warp::log("markov::read"));
|
||||
|
||||
let sentance = warp::get()
|
||||
.and(warp::path("sentance")) //TODO: sanitise::Sentance::new_iter the body line
|
||||
.and(chain.clone())
|
||||
.and(client_ip.clone())
|
||||
.and(warp::path::param().map(|opt: usize| Some(opt))
|
||||
.or(warp::path::end().map(|| Option::<usize>::None)).unify())
|
||||
.and_then(|state: State, host: IpAddr, num: Option<usize>| {
|
||||
async move {
|
||||
let (tx, rx) = mpsc::channel(state.config().max_gen_size);
|
||||
tokio::spawn(sentance::body(state, num, tx));
|
||||
Ok::<_, std::convert::Infallible>(Response::new(Body::wrap_stream(rx.filter_map(move |mut x| {
|
||||
if x.trim_in_place().len() != 0 {
|
||||
info!("{} (sentance) <- {:?}", host, x);
|
||||
x.push(' ');
|
||||
Some(Ok::<_, std::convert::Infallible>(x))
|
||||
} else {
|
||||
None
|
||||
}
|
||||
}))))
|
||||
}
|
||||
})
|
||||
.with(warp::log("markov::read::sentance"));
|
||||
|
||||
let read = warp::path("get").and(read.or(sentance));
|
||||
#[cfg(feature="api")]
|
||||
let read = read.or(api);
|
||||
|
||||
#[cfg(target_family="unix")]
|
||||
tasks.push(tokio::spawn(signals::handle(state.clone())).map(|res| res.expect("Signal handler panicked")).boxed());
|
||||
|
||||
require_send(async {
|
||||
let server = {
|
||||
let s2 = AssertNotSend::new(state.clone()); //temp clone the Arcs here for shutdown if server fails to bind, assert they cannot remain cloned across an await boundary.
|
||||
match bind::try_serve(warp::serve(push
|
||||
.or(read)),
|
||||
state.config().bindpoint.clone(),
|
||||
async move {
|
||||
tokio::signal::ctrl_c().await.unwrap();
|
||||
state.shutdown();
|
||||
}) {
|
||||
Ok((addr, server)) => {
|
||||
info!("Server bound on {:?}", addr);
|
||||
server
|
||||
},
|
||||
Err(err) => {
|
||||
error!("Failed to bind server: {}", err);
|
||||
s2.into_inner().shutdown();
|
||||
return;
|
||||
},
|
||||
}
|
||||
};
|
||||
server.await;
|
||||
}).await;
|
||||
|
||||
// Cleanup
|
||||
async move {
|
||||
trace!("Cleanup");
|
||||
|
||||
join_all(tasks).await;
|
||||
}
|
||||
}.await;
|
||||
info!("Shut down gracefully")
|
||||
}
|
||||
|
||||
mod bind;
|
||||
|
@ -0,0 +1,277 @@
|
||||
//! Filter out characters and such
|
||||
use smallmap::Map as SmallMap;
|
||||
use std::{
|
||||
borrow::Cow,
|
||||
fmt,
|
||||
iter::{
|
||||
self,
|
||||
FromIterator,
|
||||
},
|
||||
str,
|
||||
};
|
||||
use once_cell::sync::OnceCell;
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||
pub struct Filter(SmallMap<char, ()>);
|
||||
|
||||
/*
|
||||
impl<const N: usize> From<[char; N]> for Filter
|
||||
{
|
||||
fn from(from: [char; N]) -> Self
|
||||
{
|
||||
let mut map = SmallMap::with_capacity(1 + (N / 256));
|
||||
for &chr in from.iter()
|
||||
{
|
||||
map.insert(chr, ());
|
||||
}
|
||||
Self(map)
|
||||
}
|
||||
}*/
|
||||
|
||||
impl<'a> From<&'a [char]> for Filter
|
||||
{
|
||||
fn from(from: &'a [char]) -> Self
|
||||
{
|
||||
let mut map = SmallMap::new();
|
||||
for &chr in from.iter()
|
||||
{
|
||||
map.insert(chr, ());
|
||||
}
|
||||
Self(map)
|
||||
}
|
||||
}
|
||||
impl<'a> From<&'a str> for Filter
|
||||
{
|
||||
fn from(from: &'a str) -> Self
|
||||
{
|
||||
let mut output = Self::new();
|
||||
output.insert(from.chars());
|
||||
output
|
||||
}
|
||||
}
|
||||
|
||||
impl str::FromStr for Filter
|
||||
{
|
||||
type Err = std::convert::Infallible;
|
||||
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||||
Ok(Self::from(s))
|
||||
}
|
||||
}
|
||||
|
||||
impl fmt::Display for Filter
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
use std::fmt::Write;
|
||||
for chr in self.iter()
|
||||
{
|
||||
f.write_char(chr)?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
pub struct FilterKeyIter<'a>(smallmap::iter::Iter<'a, char, ()>, usize);
|
||||
|
||||
impl<'a> Iterator for FilterKeyIter<'a>
|
||||
{
|
||||
type Item = char;
|
||||
fn next(&mut self) -> Option<Self::Item>
|
||||
{
|
||||
self.0.next().map(|&(x, _)| x)
|
||||
}
|
||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||
(self.1, Some(self.1))
|
||||
}
|
||||
}
|
||||
impl<'a> iter::FusedIterator for FilterKeyIter<'a>{}
|
||||
impl<'a> iter::ExactSizeIterator for FilterKeyIter<'a>{}
|
||||
|
||||
impl Filter
|
||||
{
|
||||
pub fn new() -> Self
|
||||
{
|
||||
Self(SmallMap::new())
|
||||
}
|
||||
pub fn insert<I: IntoIterator<Item=char>>(&mut self, from: I)
|
||||
{
|
||||
for from in from.into_iter()
|
||||
{
|
||||
self.0.insert(from, ());
|
||||
}
|
||||
}
|
||||
|
||||
pub fn remove<I: IntoIterator<Item=char>>(&mut self, from: I)
|
||||
{
|
||||
for from in from.into_iter()
|
||||
{
|
||||
self.0.remove(&from);
|
||||
}
|
||||
}
|
||||
|
||||
pub fn len(&self) -> usize
|
||||
{
|
||||
self.0.len()
|
||||
}
|
||||
|
||||
pub fn is_empty(&self) -> bool
|
||||
{
|
||||
//TODO: impl this in smallmap itself
|
||||
self.len() == 0
|
||||
}
|
||||
|
||||
pub fn iter(&self) -> FilterKeyIter<'_> //impl Iterator<Item=char> + '_
|
||||
{
|
||||
//self.0.iter()
|
||||
// .copied()
|
||||
// .map(|(x, _)| x)
|
||||
FilterKeyIter(self.0.iter(), self.0.len())
|
||||
}
|
||||
|
||||
/// Should this character be filtered?
|
||||
#[inline] pub fn check(&self, chr: char) -> bool
|
||||
{
|
||||
self.0.get(&chr).is_some()
|
||||
}
|
||||
|
||||
pub fn filter_owned(&self, input: impl Into<String>) -> String
|
||||
{
|
||||
let mut input = input.into();
|
||||
self.filter(&mut input);
|
||||
input
|
||||
}
|
||||
|
||||
pub fn filter<'a>(&self, output: &'a mut String) -> &'a mut String
|
||||
{
|
||||
if self.is_empty() {
|
||||
return output;
|
||||
}
|
||||
output.retain(|chr| !self.check(chr));
|
||||
output
|
||||
}
|
||||
|
||||
pub fn filter_iter<'a, I: IntoIterator<Item=char>>(&'a self, from_iter: I) -> FilterIter<'a, I::IntoIter>
|
||||
where I::IntoIter: 'a
|
||||
{
|
||||
FilterIter(&self, from_iter.into_iter().fuse())
|
||||
}
|
||||
|
||||
pub fn filter_cow<'a>(&self, string: &'a (impl AsRef<str> + 'a + ?Sized)) -> Cow<'a, str>
|
||||
{
|
||||
let string = string.as_ref();
|
||||
|
||||
if self.is_empty() {
|
||||
return Cow::Borrowed(string);
|
||||
}
|
||||
|
||||
let mut output = Cow::Borrowed(string);
|
||||
let mut i=0;
|
||||
for chr in string.chars()
|
||||
{
|
||||
if self.check(chr) {
|
||||
output.to_mut().remove(i);
|
||||
} else {
|
||||
i+=1;
|
||||
}
|
||||
}
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
pub fn filter_str<'a, T: AsRef<str>+'a +?Sized>(&'a self, string: &'a T) -> FilterStr<'a>
|
||||
{
|
||||
FilterStr(string.as_ref(), self, OnceCell::new())
|
||||
}
|
||||
}
|
||||
|
||||
impl FromIterator<char> for Filter
|
||||
{
|
||||
fn from_iter<I: IntoIterator<Item=char>>(iter: I) -> Self
|
||||
{
|
||||
let mut output= Self::new();
|
||||
output.insert(iter);
|
||||
output
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> FilterStr<'a>
|
||||
{
|
||||
pub fn as_str(&self) -> &str
|
||||
{
|
||||
fn fmt(this: &FilterStr<'_>) -> String
|
||||
{
|
||||
let chars = this.0.chars();
|
||||
let mut f: String = crate::util::hint_cap(&chars);
|
||||
for chr in chars {
|
||||
if !this.1.check(chr) {
|
||||
f.push(chr);
|
||||
}
|
||||
}
|
||||
f
|
||||
}
|
||||
&self.2.get_or_init(|| fmt(&self))[..]
|
||||
}
|
||||
}
|
||||
|
||||
pub struct FilterStr<'a>(&'a str, &'a Filter, OnceCell<String>);
|
||||
impl<'a> fmt::Display for FilterStr<'a>
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
write!(f, "{}", self.as_str())
|
||||
}
|
||||
}
|
||||
impl<'a> FilterStr<'a>
|
||||
{
|
||||
pub fn filter(&self) -> &Filter
|
||||
{
|
||||
&self.1
|
||||
}
|
||||
}
|
||||
|
||||
pub struct FilterIter<'a, I>(&'a Filter, iter::Fuse<I>);
|
||||
|
||||
impl<'a, I: Iterator<Item=char>> Iterator for FilterIter<'a, I>
|
||||
{
|
||||
type Item = char;
|
||||
fn next(&mut self) -> Option<Self::Item>
|
||||
{
|
||||
loop {
|
||||
break match self.1.next() {
|
||||
Some(chr) if !self.0.check(chr) => Some(chr),
|
||||
None => None,
|
||||
_ => continue,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn size_hint(&self) -> (usize, Option<usize>) {
|
||||
let (_, high) = self.1.size_hint();
|
||||
(0, high)
|
||||
}
|
||||
}
|
||||
impl<'a, I> FilterIter<'a, I>
|
||||
{
|
||||
pub fn filter(&self) -> &Filter
|
||||
{
|
||||
self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a, I: Iterator<Item=char>> iter::FusedIterator for FilterIter<'a, I>{}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests
|
||||
{
|
||||
use super::*;
|
||||
#[test]
|
||||
fn filter_cow()
|
||||
{
|
||||
let filter: Filter = " hi".chars().collect();
|
||||
|
||||
let string = "abcdef ghi jk1\nhian";
|
||||
|
||||
assert_eq!(filter.filter_str(&string).to_string(), filter.filter_cow(&string).to_string());
|
||||
assert_eq!(filter.filter_cow(&string).to_string(), filter.filter(string.chars()).collect::<String>());
|
||||
}
|
||||
}
|
@ -0,0 +1,76 @@
|
||||
//! Sanitisers
|
||||
use super::*;
|
||||
use std::{
|
||||
error,
|
||||
fmt,
|
||||
};
|
||||
mod sentance;
|
||||
pub use sentance::*;
|
||||
mod word;
|
||||
pub use word::*;
|
||||
|
||||
pub mod filter;
|
||||
|
||||
/*
|
||||
pub fn take_sentance<T: AsyncBufRead+ ?Sized + Unpin, U: AsyncWrite + ?Sized + Unpin>(from: &mut T, to: &mut U) -> Result<usize, Error>
|
||||
{
|
||||
todo!()
|
||||
}*/
|
||||
|
||||
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum Error {
|
||||
Word(WordError),
|
||||
Sentance(SentanceError),
|
||||
}
|
||||
|
||||
impl error::Error for Error{}
|
||||
|
||||
impl fmt::Display for Error
|
||||
{
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
match self {
|
||||
Self::Word(_) => write!(f, "couldn't extract word"),
|
||||
Self::Sentance(_) => write!(f, "couldn't extract sentance"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<WordError> for Error
|
||||
{
|
||||
#[inline] fn from(from: WordError) -> Self
|
||||
{
|
||||
Self::Word(from)
|
||||
}
|
||||
}
|
||||
|
||||
impl From<SentanceError> for Error
|
||||
{
|
||||
#[inline] fn from(from: SentanceError) -> Self
|
||||
{
|
||||
Self::Sentance(from)
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests
|
||||
{
|
||||
use super::*;
|
||||
#[test]
|
||||
fn sentance()
|
||||
{
|
||||
let string = r#"Hello world.
|
||||
I am a string, that is a string. Strings, I love them!!!
|
||||
|
||||
Owo uwu"#;
|
||||
let sentances = Sentance::new_iter(string);
|
||||
for sentance in sentances {
|
||||
let words = Word::new(sentance);
|
||||
println!("Word in {:?} -> {:?}", sentance, words);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
}
|
@ -0,0 +1,146 @@
|
||||
//! Sentance splitting
|
||||
use super::*;
|
||||
use std::{
|
||||
borrow::{
|
||||
Borrow,
|
||||
ToOwned,
|
||||
},
|
||||
ops::{
|
||||
Deref,DerefMut,
|
||||
},
|
||||
};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct SentanceError;
|
||||
|
||||
/// A sentance
|
||||
#[derive(Debug, PartialEq, Eq)]
|
||||
#[repr(transparent)]
|
||||
pub struct Sentance(str);
|
||||
|
||||
|
||||
macro_rules! new {
|
||||
($str:expr) => {
|
||||
unsafe {Sentance::new_unchecked($str)}
|
||||
};
|
||||
}
|
||||
|
||||
const DEFAULT_BOUNDARIES: &[char] = &['\n', '.', ':', '!', '?'];
|
||||
|
||||
lazy_static! {
|
||||
static ref BOUNDARIES: smallmap::Map<char, ()> = {
|
||||
let mut map = smallmap::Map::new();
|
||||
for &chr in DEFAULT_BOUNDARIES.iter() {
|
||||
map.insert(chr, ());
|
||||
}
|
||||
map
|
||||
};
|
||||
}
|
||||
|
||||
#[inline] pub fn is_sentance_boundary(chr: char) -> bool
|
||||
{
|
||||
BOUNDARIES.contains_key(&chr)
|
||||
}
|
||||
|
||||
impl Sentance
|
||||
{
|
||||
/// Create a new word reference without checking for sentance boundaries
|
||||
pub unsafe fn new_unchecked<'a>(from: &'a str) -> &'a Self
|
||||
{
|
||||
std::mem::transmute(from)
|
||||
}
|
||||
|
||||
/// Create a single sentance
|
||||
pub fn single<'a>(from: &'a (impl AsRef<str> + 'a + ?Sized)) -> Result<&'a Self, SentanceError>
|
||||
{
|
||||
let from = from.as_ref();
|
||||
match from.find(is_sentance_boundary) {
|
||||
Some(_) => Err(SentanceError),
|
||||
_ => Ok(new!(from)),
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a new section of sentances from this string
|
||||
#[inline] pub fn new<'a>(from: &'a (impl AsRef<str> + 'a + ?Sized)) -> Vec<&'a Self>
|
||||
{
|
||||
Self::new_iter(from)
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// Create a new iterator over sentances from this string.
|
||||
pub fn new_iter<'a>(from: &'a (impl AsRef<str> +'a + ?Sized)) -> impl Iterator<Item = &'a Self> + Clone
|
||||
{
|
||||
let from = from.as_ref();
|
||||
from.split_inclusive(is_sentance_boundary)
|
||||
.map(|x| new!(x.trim()))
|
||||
.filter(|x| !x.is_empty())
|
||||
}
|
||||
|
||||
/// Get the words in this sentance
|
||||
#[inline] pub fn words(&self) -> impl Iterator<Item = &'_ Word>
|
||||
{
|
||||
Word::new_iter(self)
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> From<&'a str> for &'a Sentance
|
||||
{
|
||||
fn from(from: &'a str) -> Self
|
||||
{
|
||||
new!(from)
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<str> for Sentance
|
||||
{
|
||||
fn as_ref(&self) -> &str
|
||||
{
|
||||
&self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<Sentance> for str
|
||||
{
|
||||
fn as_ref(&self) -> &Sentance
|
||||
{
|
||||
new!(self)
|
||||
}
|
||||
}
|
||||
|
||||
impl Borrow<Sentance> for String
|
||||
{
|
||||
fn borrow(&self) -> &Sentance {
|
||||
new!(&self[..])
|
||||
}
|
||||
}
|
||||
|
||||
impl ToOwned for Sentance
|
||||
{
|
||||
type Owned = String;
|
||||
fn to_owned(&self) -> Self::Owned {
|
||||
self.0.to_owned()
|
||||
}
|
||||
}
|
||||
|
||||
impl Deref for Sentance
|
||||
{
|
||||
type Target = str;
|
||||
fn deref(&self) -> &Self::Target {
|
||||
&self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl DerefMut for Sentance
|
||||
{
|
||||
fn deref_mut(&mut self) -> &mut Self::Target {
|
||||
&mut self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<Sentance> for Sentance
|
||||
{
|
||||
#[inline] fn as_ref(&self) -> &Sentance
|
||||
{
|
||||
self
|
||||
}
|
||||
}
|
@ -0,0 +1,141 @@
|
||||
//! Word splitting
|
||||
use super::*;
|
||||
use std::{
|
||||
borrow::{
|
||||
Borrow,
|
||||
ToOwned,
|
||||
},
|
||||
ops::{
|
||||
Deref,DerefMut,
|
||||
},
|
||||
};
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct WordError;
|
||||
|
||||
/// A word is a non-whitespace containing string representing part of a sentance
|
||||
#[derive(Debug, PartialEq, Eq)]
|
||||
#[repr(transparent)]
|
||||
pub struct Word(str);
|
||||
|
||||
|
||||
macro_rules! new {
|
||||
($str:expr) => {
|
||||
unsafe {Word::new_unchecked($str)}
|
||||
};
|
||||
}
|
||||
|
||||
const DEFAULT_BOUNDARIES: &[char] = &['!', '.', ','];
|
||||
|
||||
lazy_static! {
|
||||
static ref BOUNDARIES: smallmap::Map<char, ()> = {
|
||||
let mut map = smallmap::Map::new();
|
||||
for &chr in DEFAULT_BOUNDARIES.iter() {
|
||||
map.insert(chr, ());
|
||||
}
|
||||
map
|
||||
};
|
||||
}
|
||||
|
||||
#[inline] pub fn is_word_boundary(chr: char) -> bool
|
||||
{
|
||||
chr.is_whitespace() || BOUNDARIES.contains_key(&chr)
|
||||
}
|
||||
|
||||
impl Word
|
||||
{
|
||||
/// Create a new word reference without checking for whitespace
|
||||
pub unsafe fn new_unchecked<'a>(from: &'a str) -> &'a Self
|
||||
{
|
||||
std::mem::transmute(from)
|
||||
}
|
||||
|
||||
/// Create a single word
|
||||
pub fn single<'a>(from: &'a (impl AsRef<Sentance> +?Sized +'a)) -> Result<&'a Self, WordError>
|
||||
{
|
||||
let from = from.as_ref();
|
||||
match from.find(is_word_boundary) {
|
||||
Some(_) => Err(WordError),
|
||||
_ => Ok(new!(from)),
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a new section of words from this sentance
|
||||
pub fn new<'a>(from: &'a (impl AsRef<Sentance> +?Sized+'a)) -> Vec<&'a Self>
|
||||
{
|
||||
Self::new_iter(from)
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// Create a new iterator over words from this sentance.
|
||||
pub fn new_iter<'a>(from: &'a (impl AsRef<Sentance> +?Sized+'a)) -> impl Iterator<Item = &'a Self>
|
||||
{
|
||||
let from = from.as_ref();
|
||||
from.split_inclusive(is_word_boundary)
|
||||
.map(|x| x.trim())
|
||||
.filter(|x| !x.is_empty())
|
||||
.map(|x| new!(x))
|
||||
}
|
||||
}
|
||||
|
||||
impl<'a> From<&'a str> for &'a Word
|
||||
{
|
||||
fn from(from: &'a str) -> Self
|
||||
{
|
||||
new!(from)
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<str> for Word
|
||||
{
|
||||
fn as_ref(&self) -> &str
|
||||
{
|
||||
&self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<Word> for str
|
||||
{
|
||||
fn as_ref(&self) -> &Word
|
||||
{
|
||||
new!(self)
|
||||
}
|
||||
}
|
||||
|
||||
impl Borrow<Word> for String
|
||||
{
|
||||
fn borrow(&self) -> &Word {
|
||||
new!(&self[..])
|
||||
}
|
||||
}
|
||||
|
||||
impl ToOwned for Word
|
||||
{
|
||||
type Owned = String;
|
||||
fn to_owned(&self) -> Self::Owned {
|
||||
self.0.to_owned()
|
||||
}
|
||||
}
|
||||
|
||||
impl Deref for Word
|
||||
{
|
||||
type Target = str;
|
||||
fn deref(&self) -> &Self::Target {
|
||||
&self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl DerefMut for Word
|
||||
{
|
||||
fn deref_mut(&mut self) -> &mut Self::Target {
|
||||
&mut self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl AsRef<Word> for Word
|
||||
{
|
||||
#[inline] fn as_ref(&self) -> &Word
|
||||
{
|
||||
self
|
||||
}
|
||||
}
|
@ -0,0 +1,125 @@
|
||||
//! Saving and loading chain
|
||||
use super::*;
|
||||
use std::{
|
||||
sync::Arc,
|
||||
path::{
|
||||
Path,
|
||||
},
|
||||
io,
|
||||
};
|
||||
use tokio::{
|
||||
time::{
|
||||
self,
|
||||
Duration,
|
||||
},
|
||||
fs::{
|
||||
OpenOptions,
|
||||
},
|
||||
prelude::*,
|
||||
};
|
||||
use futures::{
|
||||
future::{
|
||||
OptionFuture,
|
||||
},
|
||||
};
|
||||
#[cfg(feature="compress-chain")]
|
||||
use async_compression::{
|
||||
tokio_02::{
|
||||
write::{
|
||||
BzEncoder,
|
||||
BzDecoder,
|
||||
},
|
||||
},
|
||||
};
|
||||
|
||||
const SAVE_INTERVAL: Option<Duration> = Some(Duration::from_secs(2));
|
||||
|
||||
|
||||
#[cfg(feature="compress-chain")]
|
||||
type Compressor<T> = BzEncoder<T>;
|
||||
#[cfg(feature="compress-chain")]
|
||||
type Decompressor<T> = BzDecoder<T>;
|
||||
|
||||
|
||||
pub async fn save_now(state: &State) -> io::Result<()>
|
||||
{
|
||||
let chain = state.chain().read().await;
|
||||
use std::ops::Deref;
|
||||
let to = &state.config().file;
|
||||
save_now_to(chain.deref(),to).await
|
||||
}
|
||||
|
||||
async fn save_now_to(chain: &Chain<String>, to: impl AsRef<Path>) -> io::Result<()>
|
||||
{
|
||||
debug!("Saving chain to {:?}", to.as_ref());
|
||||
let mut file = OpenOptions::new()
|
||||
.write(true)
|
||||
.create(true)
|
||||
.truncate(true)
|
||||
.open(to).await?;
|
||||
let chain = serde_cbor::to_vec(chain).map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))?;
|
||||
{
|
||||
#[cfg(feature="compress-chain")]
|
||||
let mut file = Compressor::new(&mut file);
|
||||
file.write_all(&chain[..]).await?;
|
||||
#[cfg(feature="compress-chain")]
|
||||
file.flush().await?;
|
||||
#[cfg(feature="compress-chain")]
|
||||
file.shutdown().await?;
|
||||
}
|
||||
file.flush().await?;
|
||||
file.shutdown().await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Start the save loop for this chain
|
||||
pub async fn host(mut state: State)
|
||||
{
|
||||
let to = state.config().file.to_owned();
|
||||
let interval = state.config().save_interval();
|
||||
while Arc::strong_count(state.when()) > 1 {
|
||||
{
|
||||
let chain = state.chain().read().await;
|
||||
use std::ops::Deref;
|
||||
if let Err(e) = save_now_to(chain.deref(), &to).await {
|
||||
error!("Failed to save chain: {}", e);
|
||||
} else {
|
||||
info!("Saved chain to {:?}", to);
|
||||
}
|
||||
}
|
||||
|
||||
tokio::select!{
|
||||
_ = OptionFuture::from(interval.map(|interval| time::delay_for(interval))) => {},
|
||||
_ = state.on_shutdown() => {
|
||||
break;
|
||||
}
|
||||
}
|
||||
state.when().notified().await;
|
||||
if state.has_shutdown() {
|
||||
break;
|
||||
}
|
||||
}
|
||||
trace!("Saver exiting");
|
||||
}
|
||||
|
||||
/// Try to load a chain from this path
|
||||
pub async fn load(from: impl AsRef<Path>) -> io::Result<Chain<String>>
|
||||
{
|
||||
debug!("Loading chain from {:?}", from.as_ref());
|
||||
#[allow(unused_mut)]
|
||||
let mut file = OpenOptions::new()
|
||||
.read(true)
|
||||
.open(from).await?;
|
||||
#[allow(unused_mut)]
|
||||
let mut whole = Vec::new();
|
||||
#[cfg(feature="compress-chain")]
|
||||
let mut whole = Decompressor::new(whole);
|
||||
tokio::io::copy(&mut file, &mut whole).await?;
|
||||
whole.flush().await?;
|
||||
#[cfg(feature="compress-chain")]
|
||||
whole.shutdown().await?;
|
||||
#[cfg(feature="compress-chain")]
|
||||
let whole = whole.into_inner();
|
||||
serde_cbor::from_slice(&whole[..])
|
||||
.map_err(|e| io::Error::new(io::ErrorKind::InvalidInput, e))
|
||||
}
|
@ -0,0 +1,34 @@
|
||||
//! /sentance/
|
||||
use super::*;
|
||||
|
||||
pub async fn body(state: State, num: Option<usize>, mut output: mpsc::Sender<String>) -> Result<(), gen::GenBodyError>
|
||||
{
|
||||
let string = {
|
||||
let chain = state.chain().read().await;
|
||||
if chain.is_empty() {
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
match num {
|
||||
None => chain.generate_str(),
|
||||
Some(num) => (0..num).map(|_| chain.generate_str()).join("\n"),
|
||||
}
|
||||
};
|
||||
|
||||
debug!("Taking {:?} from {:?}" ,num, string);
|
||||
let filter = state.outbound_filter();
|
||||
if let Some(num) = num {
|
||||
for sen in sanitise::Sentance::new_iter(&string).take(num)
|
||||
{
|
||||
output.send(filter.filter_owned(sen.to_owned())).await.map_err(|e| gen::GenBodyError(e.0))?;
|
||||
}
|
||||
} else {
|
||||
output.send(filter.filter_owned(match sanitise::Sentance::new_iter(&string)
|
||||
.max_by_key(|x| x.len()) {
|
||||
Some(x) => x,
|
||||
#[cold] None => return Ok(()),
|
||||
}.to_owned())).await.map_err(|e| gen::GenBodyError(e.0))?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
@ -0,0 +1,59 @@
|
||||
//! Unix signals
|
||||
use super::*;
|
||||
use tokio::{
|
||||
signal::unix::{
|
||||
self,
|
||||
SignalKind,
|
||||
},
|
||||
};
|
||||
|
||||
|
||||
|
||||
pub async fn handle(mut state: State)
|
||||
{
|
||||
let mut usr1 = unix::signal(SignalKind::user_defined1()).expect("Failed to hook SIGUSR1");
|
||||
let mut usr2 = unix::signal(SignalKind::user_defined2()).expect("Failed to hook SIGUSR2");
|
||||
let mut quit = unix::signal(SignalKind::quit()).expect("Failed to hook SIGQUIT");
|
||||
|
||||
loop {
|
||||
tokio::select! {
|
||||
_ = state.on_shutdown() => {
|
||||
break;
|
||||
}
|
||||
_ = usr1.recv() => {
|
||||
info!("Got SIGUSR1. Saving chain immediately.");
|
||||
if let Err(e) = save::save_now(&state).await {
|
||||
error!("Failed to save chain: {}", e);
|
||||
} else{
|
||||
trace!("Saved chain okay");
|
||||
}
|
||||
},
|
||||
_ = usr2.recv() => {
|
||||
info!("Got SIGUSR1. Loading chain immediately.");
|
||||
match save::load(&state.config().file).await {
|
||||
Ok(new) => {
|
||||
{
|
||||
let mut chain = state.chain().write().await;
|
||||
*chain = new;
|
||||
}
|
||||
trace!("Replaced with read chain");
|
||||
},
|
||||
Err(e) => {
|
||||
error!("Failed to load chain from file, keeping current: {}", e);
|
||||
},
|
||||
}
|
||||
},
|
||||
_ = quit.recv() => {
|
||||
warn!("Got SIGQUIT. Saving chain then aborting.");
|
||||
if let Err(e) = save::save_now(&state).await {
|
||||
error!("Failed to save chain: {}", e);
|
||||
} else{
|
||||
trace!("Saved chain okay.");
|
||||
}
|
||||
error!("Aborting");
|
||||
std::process::abort()
|
||||
},
|
||||
}
|
||||
}
|
||||
trace!("Graceful shutdown");
|
||||
}
|
@ -0,0 +1,86 @@
|
||||
//! State
|
||||
use super::*;
|
||||
use tokio::{
|
||||
sync::{
|
||||
watch,
|
||||
},
|
||||
};
|
||||
use config::Config;
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct State
|
||||
{
|
||||
config: Arc<Config>, //to avoid cloning config
|
||||
exclude: Arc<(sanitise::filter::Filter, sanitise::filter::Filter)>,
|
||||
chain: Arc<RwLock<Chain<String>>>,
|
||||
save: Arc<Notify>,
|
||||
|
||||
shutdown: Arc<watch::Sender<bool>>,
|
||||
shutdown_recv: watch::Receiver<bool>,
|
||||
}
|
||||
|
||||
impl State
|
||||
{
|
||||
pub fn inbound_filter(&self) -> &sanitise::filter::Filter
|
||||
{
|
||||
&self.exclude.0
|
||||
}
|
||||
pub fn outbound_filter(&self) -> &sanitise::filter::Filter
|
||||
{
|
||||
&self.exclude.1
|
||||
}
|
||||
|
||||
pub fn new(config: Config, chain: Arc<RwLock<Chain<String>>>, save: Arc<Notify>) -> Self
|
||||
{
|
||||
let (shutdown, shutdown_recv) = watch::channel(false);
|
||||
Self {
|
||||
exclude: Arc::new((config.filter.get_inbound_filter(),
|
||||
config.filter.get_outbound_filter())),
|
||||
config: Arc::new(config),
|
||||
chain,
|
||||
save,
|
||||
shutdown: Arc::new(shutdown),
|
||||
shutdown_recv,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn config(&self) -> &Config
|
||||
{
|
||||
self.config.as_ref()
|
||||
}
|
||||
|
||||
pub fn notify_save(&self)
|
||||
{
|
||||
self.save.notify();
|
||||
}
|
||||
|
||||
pub fn chain(&self) -> &RwLock<Chain<String>>
|
||||
{
|
||||
&self.chain.as_ref()
|
||||
}
|
||||
|
||||
pub fn when(&self) -> &Arc<Notify>
|
||||
{
|
||||
&self.save
|
||||
}
|
||||
|
||||
pub fn shutdown(self)
|
||||
{
|
||||
self.shutdown.broadcast(true).expect("Failed to communicate shutdown");
|
||||
self.save.notify();
|
||||
}
|
||||
|
||||
pub fn has_shutdown(&self) -> bool
|
||||
{
|
||||
*self.shutdown_recv.borrow()
|
||||
}
|
||||
|
||||
pub async fn on_shutdown(&mut self)
|
||||
{
|
||||
if !self.has_shutdown() {
|
||||
while let Some(false) = self.shutdown_recv.recv().await {
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,41 @@
|
||||
//! Utils
|
||||
|
||||
pub trait NewCapacity: Sized
|
||||
{
|
||||
fn new() -> Self;
|
||||
fn with_capacity(cap: usize) -> Self;
|
||||
}
|
||||
|
||||
impl NewCapacity for String
|
||||
{
|
||||
fn new() -> Self
|
||||
{
|
||||
Self::new()
|
||||
}
|
||||
|
||||
fn with_capacity(cap: usize) -> Self
|
||||
{
|
||||
Self::with_capacity(cap)
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> NewCapacity for Vec<T>
|
||||
{
|
||||
fn new() -> Self
|
||||
{
|
||||
Self::new()
|
||||
}
|
||||
|
||||
fn with_capacity(cap: usize) -> Self
|
||||
{
|
||||
Self::with_capacity(cap)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn hint_cap<T: NewCapacity, I: Iterator>(iter: &I) -> T
|
||||
{
|
||||
match iter.size_hint() {
|
||||
(0, Some(0)) | (0, None) => T::new(),
|
||||
(_, Some(x)) | (x, _) => T::with_capacity(x)
|
||||
}
|
||||
}
|
Loading…
Reference in new issue