├── .gitignore ├── LICENCE ├── Makefile ├── README.md ├── ascii_generator ├── coffee.go ├── gen.go ├── row.go └── tree.go ├── assets ├── coffee.gif ├── coffee.png ├── demo-small-screen.png ├── demo.png ├── flow.gif ├── flow.png ├── logo.png ├── logo_with_background.png ├── pomossh.png ├── pomossh_text.png ├── sshaimftpsh.png ├── sshpomosairashgautamcomnp.png ├── step2.png ├── step3.png ├── transparent.png ├── tree.gif └── tree.png ├── go.mod ├── go.sum ├── helper └── helper.go └── main.go /.gitignore: -------------------------------------------------------------------------------- 1 | binary 2 | .ssh -------------------------------------------------------------------------------- /LICENCE: -------------------------------------------------------------------------------- 1 | GNU AFFERO GENERAL PUBLIC LICENSE 2 | Version 3, 19 November 2007 3 | 4 | Copyright (C) 2007 Free Software Foundation, Inc. 5 | Everyone is permitted to copy and distribute verbatim copies 6 | of this license document, but changing it is not allowed. 7 | 8 | Preamble 9 | 10 | The GNU Affero General Public License is a free, copyleft license for 11 | software and other kinds of works, specifically designed to ensure 12 | cooperation with the community in the case of network server software. 13 | 14 | The licenses for most software and other practical works are designed 15 | to take away your freedom to share and change the works. By contrast, 16 | our General Public Licenses are intended to guarantee your freedom to 17 | share and change all versions of a program--to make sure it remains free 18 | software for all its users. 19 | 20 | When we speak of free software, we are referring to freedom, not 21 | price. Our General Public Licenses are designed to make sure that you 22 | have the freedom to distribute copies of free software (and charge for 23 | them if you wish), that you receive source code or can get it if you 24 | want it, that you can change the software or use pieces of it in new 25 | free programs, and that you know you can do these things. 26 | 27 | Developers that use our General Public Licenses protect your rights 28 | with two steps: (1) assert copyright on the software, and (2) offer 29 | you this License which gives you legal permission to copy, distribute 30 | and/or modify the software. 31 | 32 | A secondary benefit of defending all users' freedom is that 33 | improvements made in alternate versions of the program, if they 34 | receive widespread use, become available for other developers to 35 | incorporate. Many developers of free software are heartened and 36 | encouraged by the resulting cooperation. However, in the case of 37 | software used on network servers, this result may fail to come about. 38 | The GNU General Public License permits making a modified version and 39 | letting the public access it on a server without ever releasing its 40 | source code to the public. 41 | 42 | The GNU Affero General Public License is designed specifically to 43 | ensure that, in such cases, the modified source code becomes available 44 | to the community. It requires the operator of a network server to 45 | provide the source code of the modified version running there to the 46 | users of that server. Therefore, public use of a modified version, on 47 | a publicly accessible server, gives the public access to the source 48 | code of the modified version. 49 | 50 | An older license, called the Affero General Public License and 51 | published by Affero, was designed to accomplish similar goals. This is 52 | a different license, not a version of the Affero GPL, but Affero has 53 | released a new version of the Affero GPL which permits relicensing under 54 | this license. 55 | 56 | The precise terms and conditions for copying, distribution and 57 | modification follow. 58 | 59 | TERMS AND CONDITIONS 60 | 61 | 0. Definitions. 62 | 63 | "This License" refers to version 3 of the GNU Affero General Public License. 64 | 65 | "Copyright" also means copyright-like laws that apply to other kinds of 66 | works, such as semiconductor masks. 67 | 68 | "The Program" refers to any copyrightable work licensed under this 69 | License. Each licensee is addressed as "you". "Licensees" and 70 | "recipients" may be individuals or organizations. 71 | 72 | To "modify" a work means to copy from or adapt all or part of the work 73 | in a fashion requiring copyright permission, other than the making of an 74 | exact copy. The resulting work is called a "modified version" of the 75 | earlier work or a work "based on" the earlier work. 76 | 77 | A "covered work" means either the unmodified Program or a work based 78 | on the Program. 79 | 80 | To "propagate" a work means to do anything with it that, without 81 | permission, would make you directly or secondarily liable for 82 | infringement under applicable copyright law, except executing it on a 83 | computer or modifying a private copy. Propagation includes copying, 84 | distribution (with or without modification), making available to the 85 | public, and in some countries other activities as well. 86 | 87 | To "convey" a work means any kind of propagation that enables other 88 | parties to make or receive copies. Mere interaction with a user through 89 | a computer network, with no transfer of a copy, is not conveying. 90 | 91 | An interactive user interface displays "Appropriate Legal Notices" 92 | to the extent that it includes a convenient and prominently visible 93 | feature that (1) displays an appropriate copyright notice, and (2) 94 | tells the user that there is no warranty for the work (except to the 95 | extent that warranties are provided), that licensees may convey the 96 | work under this License, and how to view a copy of this License. If 97 | the interface presents a list of user commands or options, such as a 98 | menu, a prominent item in the list meets this criterion. 99 | 100 | 1. Source Code. 101 | 102 | The "source code" for a work means the preferred form of the work 103 | for making modifications to it. "Object code" means any non-source 104 | form of a work. 105 | 106 | A "Standard Interface" means an interface that either is an official 107 | standard defined by a recognized standards body, or, in the case of 108 | interfaces specified for a particular programming language, one that 109 | is widely used among developers working in that language. 110 | 111 | The "System Libraries" of an executable work include anything, other 112 | than the work as a whole, that (a) is included in the normal form of 113 | packaging a Major Component, but which is not part of that Major 114 | Component, and (b) serves only to enable use of the work with that 115 | Major Component, or to implement a Standard Interface for which an 116 | implementation is available to the public in source code form. A 117 | "Major Component", in this context, means a major essential component 118 | (kernel, window system, and so on) of the specific operating system 119 | (if any) on which the executable work runs, or a compiler used to 120 | produce the work, or an object code interpreter used to run it. 121 | 122 | The "Corresponding Source" for a work in object code form means all 123 | the source code needed to generate, install, and (for an executable 124 | work) run the object code and to modify the work, including scripts to 125 | control those activities. However, it does not include the work's 126 | System Libraries, or general-purpose tools or generally available free 127 | programs which are used unmodified in performing those activities but 128 | which are not part of the work. For example, Corresponding Source 129 | includes interface definition files associated with source files for 130 | the work, and the source code for shared libraries and dynamically 131 | linked subprograms that the work is specifically designed to require, 132 | such as by intimate data communication or control flow between those 133 | subprograms and other parts of the work. 134 | 135 | The Corresponding Source need not include anything that users 136 | can regenerate automatically from other parts of the Corresponding 137 | Source. 138 | 139 | The Corresponding Source for a work in source code form is that 140 | same work. 141 | 142 | 2. Basic Permissions. 143 | 144 | All rights granted under this License are granted for the term of 145 | copyright on the Program, and are irrevocable provided the stated 146 | conditions are met. This License explicitly affirms your unlimited 147 | permission to run the unmodified Program. The output from running a 148 | covered work is covered by this License only if the output, given its 149 | content, constitutes a covered work. This License acknowledges your 150 | rights of fair use or other equivalent, as provided by copyright law. 151 | 152 | You may make, run and propagate covered works that you do not 153 | convey, without conditions so long as your license otherwise remains 154 | in force. You may convey covered works to others for the sole purpose 155 | of having them make modifications exclusively for you, or provide you 156 | with facilities for running those works, provided that you comply with 157 | the terms of this License in conveying all material for which you do 158 | not control copyright. Those thus making or running the covered works 159 | for you must do so exclusively on your behalf, under your direction 160 | and control, on terms that prohibit them from making any copies of 161 | your copyrighted material outside their relationship with you. 162 | 163 | Conveying under any other circumstances is permitted solely under 164 | the conditions stated below. Sublicensing is not allowed; section 10 165 | makes it unnecessary. 166 | 167 | 3. Protecting Users' Legal Rights From Anti-Circumvention Law. 168 | 169 | No covered work shall be deemed part of an effective technological 170 | measure under any applicable law fulfilling obligations under article 171 | 11 of the WIPO copyright treaty adopted on 20 December 1996, or 172 | similar laws prohibiting or restricting circumvention of such 173 | measures. 174 | 175 | When you convey a covered work, you waive any legal power to forbid 176 | circumvention of technological measures to the extent such circumvention 177 | is effected by exercising rights under this License with respect to 178 | the covered work, and you disclaim any intention to limit operation or 179 | modification of the work as a means of enforcing, against the work's 180 | users, your or third parties' legal rights to forbid circumvention of 181 | technological measures. 182 | 183 | 4. Conveying Verbatim Copies. 184 | 185 | You may convey verbatim copies of the Program's source code as you 186 | receive it, in any medium, provided that you conspicuously and 187 | appropriately publish on each copy an appropriate copyright notice; 188 | keep intact all notices stating that this License and any 189 | non-permissive terms added in accord with section 7 apply to the code; 190 | keep intact all notices of the absence of any warranty; and give all 191 | recipients a copy of this License along with the Program. 192 | 193 | You may charge any price or no price for each copy that you convey, 194 | and you may offer support or warranty protection for a fee. 195 | 196 | 5. Conveying Modified Source Versions. 197 | 198 | You may convey a work based on the Program, or the modifications to 199 | produce it from the Program, in the form of source code under the 200 | terms of section 4, provided that you also meet all of these conditions: 201 | 202 | a) The work must carry prominent notices stating that you modified 203 | it, and giving a relevant date. 204 | 205 | b) The work must carry prominent notices stating that it is 206 | released under this License and any conditions added under section 207 | 7. This requirement modifies the requirement in section 4 to 208 | "keep intact all notices". 209 | 210 | c) You must license the entire work, as a whole, under this 211 | License to anyone who comes into possession of a copy. This 212 | License will therefore apply, along with any applicable section 7 213 | additional terms, to the whole of the work, and all its parts, 214 | regardless of how they are packaged. This License gives no 215 | permission to license the work in any other way, but it does not 216 | invalidate such permission if you have separately received it. 217 | 218 | d) If the work has interactive user interfaces, each must display 219 | Appropriate Legal Notices; however, if the Program has interactive 220 | interfaces that do not display Appropriate Legal Notices, your 221 | work need not make them do so. 222 | 223 | A compilation of a covered work with other separate and independent 224 | works, which are not by their nature extensions of the covered work, 225 | and which are not combined with it such as to form a larger program, 226 | in or on a volume of a storage or distribution medium, is called an 227 | "aggregate" if the compilation and its resulting copyright are not 228 | used to limit the access or legal rights of the compilation's users 229 | beyond what the individual works permit. Inclusion of a covered work 230 | in an aggregate does not cause this License to apply to the other 231 | parts of the aggregate. 232 | 233 | 6. Conveying Non-Source Forms. 234 | 235 | You may convey a covered work in object code form under the terms 236 | of sections 4 and 5, provided that you also convey the 237 | machine-readable Corresponding Source under the terms of this License, 238 | in one of these ways: 239 | 240 | a) Convey the object code in, or embodied in, a physical product 241 | (including a physical distribution medium), accompanied by the 242 | Corresponding Source fixed on a durable physical medium 243 | customarily used for software interchange. 244 | 245 | b) Convey the object code in, or embodied in, a physical product 246 | (including a physical distribution medium), accompanied by a 247 | written offer, valid for at least three years and valid for as 248 | long as you offer spare parts or customer support for that product 249 | model, to give anyone who possesses the object code either (1) a 250 | copy of the Corresponding Source for all the software in the 251 | product that is covered by this License, on a durable physical 252 | medium customarily used for software interchange, for a price no 253 | more than your reasonable cost of physically performing this 254 | conveying of source, or (2) access to copy the 255 | Corresponding Source from a network server at no charge. 256 | 257 | c) Convey individual copies of the object code with a copy of the 258 | written offer to provide the Corresponding Source. This 259 | alternative is allowed only occasionally and noncommercially, and 260 | only if you received the object code with such an offer, in accord 261 | with subsection 6b. 262 | 263 | d) Convey the object code by offering access from a designated 264 | place (gratis or for a charge), and offer equivalent access to the 265 | Corresponding Source in the same way through the same place at no 266 | further charge. You need not require recipients to copy the 267 | Corresponding Source along with the object code. If the place to 268 | copy the object code is a network server, the Corresponding Source 269 | may be on a different server (operated by you or a third party) 270 | that supports equivalent copying facilities, provided you maintain 271 | clear directions next to the object code saying where to find the 272 | Corresponding Source. Regardless of what server hosts the 273 | Corresponding Source, you remain obligated to ensure that it is 274 | available for as long as needed to satisfy these requirements. 275 | 276 | e) Convey the object code using peer-to-peer transmission, provided 277 | you inform other peers where the object code and Corresponding 278 | Source of the work are being offered to the general public at no 279 | charge under subsection 6d. 280 | 281 | A separable portion of the object code, whose source code is excluded 282 | from the Corresponding Source as a System Library, need not be 283 | included in conveying the object code work. 284 | 285 | A "User Product" is either (1) a "consumer product", which means any 286 | tangible personal property which is normally used for personal, family, 287 | or household purposes, or (2) anything designed or sold for incorporation 288 | into a dwelling. In determining whether a product is a consumer product, 289 | doubtful cases shall be resolved in favor of coverage. For a particular 290 | product received by a particular user, "normally used" refers to a 291 | typical or common use of that class of product, regardless of the status 292 | of the particular user or of the way in which the particular user 293 | actually uses, or expects or is expected to use, the product. A product 294 | is a consumer product regardless of whether the product has substantial 295 | commercial, industrial or non-consumer uses, unless such uses represent 296 | the only significant mode of use of the product. 297 | 298 | "Installation Information" for a User Product means any methods, 299 | procedures, authorization keys, or other information required to install 300 | and execute modified versions of a covered work in that User Product from 301 | a modified version of its Corresponding Source. The information must 302 | suffice to ensure that the continued functioning of the modified object 303 | code is in no case prevented or interfered with solely because 304 | modification has been made. 305 | 306 | If you convey an object code work under this section in, or with, or 307 | specifically for use in, a User Product, and the conveying occurs as 308 | part of a transaction in which the right of possession and use of the 309 | User Product is transferred to the recipient in perpetuity or for a 310 | fixed term (regardless of how the transaction is characterized), the 311 | Corresponding Source conveyed under this section must be accompanied 312 | by the Installation Information. But this requirement does not apply 313 | if neither you nor any third party retains the ability to install 314 | modified object code on the User Product (for example, the work has 315 | been installed in ROM). 316 | 317 | The requirement to provide Installation Information does not include a 318 | requirement to continue to provide support service, warranty, or updates 319 | for a work that has been modified or installed by the recipient, or for 320 | the User Product in which it has been modified or installed. Access to a 321 | network may be denied when the modification itself materially and 322 | adversely affects the operation of the network or violates the rules and 323 | protocols for communication across the network. 324 | 325 | Corresponding Source conveyed, and Installation Information provided, 326 | in accord with this section must be in a format that is publicly 327 | documented (and with an implementation available to the public in 328 | source code form), and must require no special password or key for 329 | unpacking, reading or copying. 330 | 331 | 7. Additional Terms. 332 | 333 | "Additional permissions" are terms that supplement the terms of this 334 | License by making exceptions from one or more of its conditions. 335 | Additional permissions that are applicable to the entire Program shall 336 | be treated as though they were included in this License, to the extent 337 | that they are valid under applicable law. If additional permissions 338 | apply only to part of the Program, that part may be used separately 339 | under those permissions, but the entire Program remains governed by 340 | this License without regard to the additional permissions. 341 | 342 | When you convey a copy of a covered work, you may at your option 343 | remove any additional permissions from that copy, or from any part of 344 | it. (Additional permissions may be written to require their own 345 | removal in certain cases when you modify the work.) You may place 346 | additional permissions on material, added by you to a covered work, 347 | for which you have or can give appropriate copyright permission. 348 | 349 | Notwithstanding any other provision of this License, for material you 350 | add to a covered work, you may (if authorized by the copyright holders of 351 | that material) supplement the terms of this License with terms: 352 | 353 | a) Disclaiming warranty or limiting liability differently from the 354 | terms of sections 15 and 16 of this License; or 355 | 356 | b) Requiring preservation of specified reasonable legal notices or 357 | author attributions in that material or in the Appropriate Legal 358 | Notices displayed by works containing it; or 359 | 360 | c) Prohibiting misrepresentation of the origin of that material, or 361 | requiring that modified versions of such material be marked in 362 | reasonable ways as different from the original version; or 363 | 364 | d) Limiting the use for publicity purposes of names of licensors or 365 | authors of the material; or 366 | 367 | e) Declining to grant rights under trademark law for use of some 368 | trade names, trademarks, or service marks; or 369 | 370 | f) Requiring indemnification of licensors and authors of that 371 | material by anyone who conveys the material (or modified versions of 372 | it) with contractual assumptions of liability to the recipient, for 373 | any liability that these contractual assumptions directly impose on 374 | those licensors and authors. 375 | 376 | All other non-permissive additional terms are considered "further 377 | restrictions" within the meaning of section 10. If the Program as you 378 | received it, or any part of it, contains a notice stating that it is 379 | governed by this License along with a term that is a further 380 | restriction, you may remove that term. If a license document contains 381 | a further restriction but permits relicensing or conveying under this 382 | License, you may add to a covered work material governed by the terms 383 | of that license document, provided that the further restriction does 384 | not survive such relicensing or conveying. 385 | 386 | If you add terms to a covered work in accord with this section, you 387 | must place, in the relevant source files, a statement of the 388 | additional terms that apply to those files, or a notice indicating 389 | where to find the applicable terms. 390 | 391 | Additional terms, permissive or non-permissive, may be stated in the 392 | form of a separately written license, or stated as exceptions; 393 | the above requirements apply either way. 394 | 395 | 8. Termination. 396 | 397 | You may not propagate or modify a covered work except as expressly 398 | provided under this License. Any attempt otherwise to propagate or 399 | modify it is void, and will automatically terminate your rights under 400 | this License (including any patent licenses granted under the third 401 | paragraph of section 11). 402 | 403 | However, if you cease all violation of this License, then your 404 | license from a particular copyright holder is reinstated (a) 405 | provisionally, unless and until the copyright holder explicitly and 406 | finally terminates your license, and (b) permanently, if the copyright 407 | holder fails to notify you of the violation by some reasonable means 408 | prior to 60 days after the cessation. 409 | 410 | Moreover, your license from a particular copyright holder is 411 | reinstated permanently if the copyright holder notifies you of the 412 | violation by some reasonable means, this is the first time you have 413 | received notice of violation of this License (for any work) from that 414 | copyright holder, and you cure the violation prior to 30 days after 415 | your receipt of the notice. 416 | 417 | Termination of your rights under this section does not terminate the 418 | licenses of parties who have received copies or rights from you under 419 | this License. If your rights have been terminated and not permanently 420 | reinstated, you do not qualify to receive new licenses for the same 421 | material under section 10. 422 | 423 | 9. Acceptance Not Required for Having Copies. 424 | 425 | You are not required to accept this License in order to receive or 426 | run a copy of the Program. Ancillary propagation of a covered work 427 | occurring solely as a consequence of using peer-to-peer transmission 428 | to receive a copy likewise does not require acceptance. However, 429 | nothing other than this License grants you permission to propagate or 430 | modify any covered work. These actions infringe copyright if you do 431 | not accept this License. Therefore, by modifying or propagating a 432 | covered work, you indicate your acceptance of this License to do so. 433 | 434 | 10. Automatic Licensing of Downstream Recipients. 435 | 436 | Each time you convey a covered work, the recipient automatically 437 | receives a license from the original licensors, to run, modify and 438 | propagate that work, subject to this License. You are not responsible 439 | for enforcing compliance by third parties with this License. 440 | 441 | An "entity transaction" is a transaction transferring control of an 442 | organization, or substantially all assets of one, or subdividing an 443 | organization, or merging organizations. If propagation of a covered 444 | work results from an entity transaction, each party to that 445 | transaction who receives a copy of the work also receives whatever 446 | licenses to the work the party's predecessor in interest had or could 447 | give under the previous paragraph, plus a right to possession of the 448 | Corresponding Source of the work from the predecessor in interest, if 449 | the predecessor has it or can get it with reasonable efforts. 450 | 451 | You may not impose any further restrictions on the exercise of the 452 | rights granted or affirmed under this License. For example, you may 453 | not impose a license fee, royalty, or other charge for exercise of 454 | rights granted under this License, and you may not initiate litigation 455 | (including a cross-claim or counterclaim in a lawsuit) alleging that 456 | any patent claim is infringed by making, using, selling, offering for 457 | sale, or importing the Program or any portion of it. 458 | 459 | 11. Patents. 460 | 461 | A "contributor" is a copyright holder who authorizes use under this 462 | License of the Program or a work on which the Program is based. The 463 | work thus licensed is called the contributor's "contributor version". 464 | 465 | A contributor's "essential patent claims" are all patent claims 466 | owned or controlled by the contributor, whether already acquired or 467 | hereafter acquired, that would be infringed by some manner, permitted 468 | by this License, of making, using, or selling its contributor version, 469 | but do not include claims that would be infringed only as a 470 | consequence of further modification of the contributor version. For 471 | purposes of this definition, "control" includes the right to grant 472 | patent sublicenses in a manner consistent with the requirements of 473 | this License. 474 | 475 | Each contributor grants you a non-exclusive, worldwide, royalty-free 476 | patent license under the contributor's essential patent claims, to 477 | make, use, sell, offer for sale, import and otherwise run, modify and 478 | propagate the contents of its contributor version. 479 | 480 | In the following three paragraphs, a "patent license" is any express 481 | agreement or commitment, however denominated, not to enforce a patent 482 | (such as an express permission to practice a patent or covenant not to 483 | sue for patent infringement). To "grant" such a patent license to a 484 | party means to make such an agreement or commitment not to enforce a 485 | patent against the party. 486 | 487 | If you convey a covered work, knowingly relying on a patent license, 488 | and the Corresponding Source of the work is not available for anyone 489 | to copy, free of charge and under the terms of this License, through a 490 | publicly available network server or other readily accessible means, 491 | then you must either (1) cause the Corresponding Source to be so 492 | available, or (2) arrange to deprive yourself of the benefit of the 493 | patent license for this particular work, or (3) arrange, in a manner 494 | consistent with the requirements of this License, to extend the patent 495 | license to downstream recipients. "Knowingly relying" means you have 496 | actual knowledge that, but for the patent license, your conveying the 497 | covered work in a country, or your recipient's use of the covered work 498 | in a country, would infringe one or more identifiable patents in that 499 | country that you have reason to believe are valid. 500 | 501 | If, pursuant to or in connection with a single transaction or 502 | arrangement, you convey, or propagate by procuring conveyance of, a 503 | covered work, and grant a patent license to some of the parties 504 | receiving the covered work authorizing them to use, propagate, modify 505 | or convey a specific copy of the covered work, then the patent license 506 | you grant is automatically extended to all recipients of the covered 507 | work and works based on it. 508 | 509 | A patent license is "discriminatory" if it does not include within 510 | the scope of its coverage, prohibits the exercise of, or is 511 | conditioned on the non-exercise of one or more of the rights that are 512 | specifically granted under this License. You may not convey a covered 513 | work if you are a party to an arrangement with a third party that is 514 | in the business of distributing software, under which you make payment 515 | to the third party based on the extent of your activity of conveying 516 | the work, and under which the third party grants, to any of the 517 | parties who would receive the covered work from you, a discriminatory 518 | patent license (a) in connection with copies of the covered work 519 | conveyed by you (or copies made from those copies), or (b) primarily 520 | for and in connection with specific products or compilations that 521 | contain the covered work, unless you entered into that arrangement, 522 | or that patent license was granted, prior to 28 March 2007. 523 | 524 | Nothing in this License shall be construed as excluding or limiting 525 | any implied license or other defenses to infringement that may 526 | otherwise be available to you under applicable patent law. 527 | 528 | 12. No Surrender of Others' Freedom. 529 | 530 | If conditions are imposed on you (whether by court order, agreement or 531 | otherwise) that contradict the conditions of this License, they do not 532 | excuse you from the conditions of this License. If you cannot convey a 533 | covered work so as to satisfy simultaneously your obligations under this 534 | License and any other pertinent obligations, then as a consequence you may 535 | not convey it at all. For example, if you agree to terms that obligate you 536 | to collect a royalty for further conveying from those to whom you convey 537 | the Program, the only way you could satisfy both those terms and this 538 | License would be to refrain entirely from conveying the Program. 539 | 540 | 13. Remote Network Interaction; Use with the GNU General Public License. 541 | 542 | Notwithstanding any other provision of this License, if you modify the 543 | Program, your modified version must prominently offer all users 544 | interacting with it remotely through a computer network (if your version 545 | supports such interaction) an opportunity to receive the Corresponding 546 | Source of your version by providing access to the Corresponding Source 547 | from a network server at no charge, through some standard or customary 548 | means of facilitating copying of software. This Corresponding Source 549 | shall include the Corresponding Source for any work covered by version 3 550 | of the GNU General Public License that is incorporated pursuant to the 551 | following paragraph. 552 | 553 | Notwithstanding any other provision of this License, you have 554 | permission to link or combine any covered work with a work licensed 555 | under version 3 of the GNU General Public License into a single 556 | combined work, and to convey the resulting work. The terms of this 557 | License will continue to apply to the part which is the covered work, 558 | but the work with which it is combined will remain governed by version 559 | 3 of the GNU General Public License. 560 | 561 | 14. Revised Versions of this License. 562 | 563 | The Free Software Foundation may publish revised and/or new versions of 564 | the GNU Affero General Public License from time to time. Such new versions 565 | will be similar in spirit to the present version, but may differ in detail to 566 | address new problems or concerns. 567 | 568 | Each version is given a distinguishing version number. If the 569 | Program specifies that a certain numbered version of the GNU Affero General 570 | Public License "or any later version" applies to it, you have the 571 | option of following the terms and conditions either of that numbered 572 | version or of any later version published by the Free Software 573 | Foundation. If the Program does not specify a version number of the 574 | GNU Affero General Public License, you may choose any version ever published 575 | by the Free Software Foundation. 576 | 577 | If the Program specifies that a proxy can decide which future 578 | versions of the GNU Affero General Public License can be used, that proxy's 579 | public statement of acceptance of a version permanently authorizes you 580 | to choose that version for the Program. 581 | 582 | Later license versions may give you additional or different 583 | permissions. However, no additional obligations are imposed on any 584 | author or copyright holder as a result of your choosing to follow a 585 | later version. 586 | 587 | 15. Disclaimer of Warranty. 588 | 589 | THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY 590 | APPLICABLE LAW. EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT 591 | HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY 592 | OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, 593 | THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 594 | PURPOSE. THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM 595 | IS WITH YOU. SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF 596 | ALL NECESSARY SERVICING, REPAIR OR CORRECTION. 597 | 598 | 16. Limitation of Liability. 599 | 600 | IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING 601 | WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS 602 | THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY 603 | GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE 604 | USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF 605 | DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD 606 | PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS), 607 | EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF 608 | SUCH DAMAGES. 609 | 610 | 17. Interpretation of Sections 15 and 16. 611 | 612 | If the disclaimer of warranty and limitation of liability provided 613 | above cannot be given local legal effect according to their terms, 614 | reviewing courts shall apply local law that most closely approximates 615 | an absolute waiver of all civil liability in connection with the 616 | Program, unless a warranty or assumption of liability accompanies a 617 | copy of the Program in return for a fee. 618 | 619 | END OF TERMS AND CONDITIONS 620 | 621 | How to Apply These Terms to Your New Programs 622 | 623 | If you develop a new program, and you want it to be of the greatest 624 | possible use to the public, the best way to achieve this is to make it 625 | free software which everyone can redistribute and change under these terms. 626 | 627 | To do so, attach the following notices to the program. It is safest 628 | to attach them to the start of each source file to most effectively 629 | state the exclusion of warranty; and each file should have at least 630 | the "copyright" line and a pointer to where the full notice is found. 631 | 632 | 633 | Copyright (C) 634 | 635 | This program is free software: you can redistribute it and/or modify 636 | it under the terms of the GNU Affero General Public License as published 637 | by the Free Software Foundation, either version 3 of the License, or 638 | (at your option) any later version. 639 | 640 | This program is distributed in the hope that it will be useful, 641 | but WITHOUT ANY WARRANTY; without even the implied warranty of 642 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 643 | GNU Affero General Public License for more details. 644 | 645 | You should have received a copy of the GNU Affero General Public License 646 | along with this program. If not, see . 647 | 648 | Also add information on how to contact you by electronic and paper mail. 649 | 650 | If your software can interact with users remotely through a computer 651 | network, you should also make sure that it provides a way for users to 652 | get its source. For example, if your program is a web application, its 653 | interface could display a "Source" link that leads users to an archive 654 | of the code. There are many ways you could offer source, and different 655 | solutions will be better for different programs; see section 13 for the 656 | specific requirements. 657 | 658 | You should also get your employer (if you work as a programmer) or school, 659 | if any, to sign a "copyright disclaimer" for the program, if necessary. 660 | For more information on this, and how to apply and follow the GNU AGPL, see 661 | . -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- 1 | OUTPUT_DIR := binary 2 | BINARY_NAME := aimssh 3 | 4 | PLATFORMS := linux/amd64 linux/arm64 darwin/amd64 darwin/arm64 windows/amd64 5 | 6 | build: 7 | go build -o $(OUTPUT_DIR)/$(BINARY_NAME) 8 | 9 | ssh: build 10 | ./$(OUTPUT_DIR)/$(BINARY_NAME) -ssh true 11 | 12 | pomo: build 13 | ./$(OUTPUT_DIR)/$(BINARY_NAME) 14 | 15 | test: 16 | go test ./... -v 17 | 18 | cross-build: $(PLATFORMS) 19 | 20 | $(PLATFORMS): 21 | @mkdir -p $(OUTPUT_DIR) 22 | GOOS=$(word 1,$(subst /, ,$@)) GOARCH=$(word 2,$(subst /, ,$@)) go build -o $(OUTPUT_DIR)/$(BINARY_NAME)-$(word 1,$(subst /, ,$@))-$(word 2,$(subst /, ,$@))$(if $(findstring windows,$@),.exe,) . 23 | 24 | clean: 25 | rm -rf $(OUTPUT_DIR) 26 | 27 | .PHONY: build ssh pomo test cross-build clean $(PLATFORMS) 28 | 29 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | 2 | 3 | # Aim SSH 4 | Mulitple Productive __`Terminal SSH Application`__. 5 | 6 | [Official Website](https://aim.ftp.sh) 7 | 8 | 9 | To use the application: 10 | ``` sh 11 | ssh aim.ftp.sh 12 | ``` 13 | 14 |
15 |
16 | 17 | 18 | ![GitHub Release](https://img.shields.io/github/v/release/sairash/aimssh) ![](https://img.shields.io/github/go-mod/go-version/sairash/aimssh) ![](https://img.shields.io/github/languages/code-size/sairash/aimssh) ![](https://img.shields.io/github/license/sairash/aimssh) [![Go Report Card](https://goreportcard.com/badge/github.com/sairash/aimssh)](https://goreportcard.com/report/github.com/sairash/aimssh) 19 | 20 | 21 | 22 | ![Demo](./assets/demo-small-screen.png) 23 | 24 | 25 | Introducing a fresh take on productivity, a unique terminal productivity app designed specifically for the terminal enthusiasts and tech-savvy professionals. 26 | 27 | This minimalist design keeps your aim on tasks without the clutter of traditional apps. 28 | 29 | ### Planned and Added 30 | - ✅ Timer 31 | - ❌ Todo List similar to todist 32 | - ❌ Notes Taking 33 | - ❌ Stats 34 | 35 | 36 | --- 37 | 38 | __Visual Options:__ 39 | 40 | __Note:__ The screenshots are before the rebranding (Updating soon) 41 | 42 | 43 | Visual Effects are the visual aspects of the timer to make every sessions intresting. Here are some of the available visual options. 44 | 45 | 46 |
47 | 🌲 Tree 48 |
49 | 50 | 51 | 52 | A random procedurally generated tree everytime you start a session. 53 |
54 | 55 |
56 | 🛶 Flow 57 |
58 | 59 | 60 | 61 | A guy who is rowing through the _"Time River"_. 62 |
63 | 64 | 65 |
66 | ☕ Coffee 67 |
68 | 69 | 70 | 71 | A coffee mug that filles up over time. 72 |
73 | 74 | 75 | --- 76 | 77 | ### How to use aimssh: 78 | 79 | __STEP [1]:__ 80 | 81 | Run the following comamnd in your terminal 82 | 83 | 84 | 85 | __STEP [2]:__ 86 | 87 | Enter the time in minutes and also session "Title" 88 | 89 | 90 | 91 | 92 | __STEP [3]:__ 93 | 94 | Select a visual option for the session 95 | 96 | 97 | 98 | __STEP [4]:__ 99 | 100 | Work! 101 | 102 | 103 | 104 | ---- 105 | 106 | 107 | ### How to install aimssh locally: 108 | 109 | To install aimssh locally run the following command. 110 | 111 | __Linux/MAC:__ 112 | 113 | ``` sh 114 | curl -sSL https://gist.githubusercontent.com/sairash/f07c0d194c755fdd6c4fe39d0010ec30/raw | bash 115 | ``` 116 | 117 | __Windows:__ 118 | 119 | ``` sh 120 | curl -sSL https://gist.githubusercontent.com/sairash/d6ce0c6a627f932dd105f17209d1b0e2/raw/20c42bfbafb09bf495cda7a77fe33fcab0055e6a/install_pomo.ps1 | powershell -c - 121 | ``` 122 | 123 | 124 | Or use it directly 125 | 126 | 127 | 128 | ``` sh 129 | git clone https://github.com/sairash/aimssh 130 | ``` 131 | 132 | ``` sh 133 | cd aimssh 134 | ``` 135 | 136 | ``` sh 137 | go build 138 | ``` 139 | 140 | And, 141 | 142 | __Run aimssh as client:__ 143 | ```sh 144 | aimssh 145 | ``` 146 | 147 | __Run aimssh as server:__ 148 | ```sh 149 | aimssh -ssh true 150 | ``` 151 |
152 | 153 | Made With ❤️ by Sairash. 154 | -------------------------------------------------------------------------------- /ascii_generator/coffee.go: -------------------------------------------------------------------------------- 1 | // inspiration from: https://www.youtube.com/watch?v=I5Q03I-ybXQ 2 | 3 | package ascii_generator 4 | 5 | import ( 6 | "aimssh/helper" 7 | "math" 8 | "time" 9 | 10 | "github.com/charmbracelet/lipgloss" 11 | ) 12 | 13 | type Coffee struct { 14 | width, height, curFrame, prevPercent int 15 | cachedTop string 16 | cachedBottom string 17 | } 18 | 19 | func (c *Coffee) fill_amount(percent, width, height int) [][]rune { 20 | totalCells := width * height 21 | fillCells := int(math.Ceil(float64(percent*totalCells) / 100.0)) 22 | 23 | grid := make([][]rune, height) 24 | for i := range grid { 25 | grid[i] = make([]rune, width) 26 | for j := range grid[i] { 27 | grid[i][j] = ' ' 28 | } 29 | } 30 | 31 | remaining := fillCells 32 | for i := height - 1; i >= 0 && remaining > 0; i-- { 33 | for j := 0; j < width && remaining > 0; j++ { 34 | grid[i][j] = '#' 35 | remaining-- 36 | } 37 | } 38 | 39 | return grid 40 | } 41 | 42 | var ( 43 | cup_color = lipgloss.NewStyle().Foreground(lipgloss.Color("#f78dbb")) 44 | coffee_color = lipgloss.NewStyle().Foreground(lipgloss.Color("#967259")) 45 | 46 | steam_animation = [][]string{ 47 | { 48 | " {", " { }", 49 | " }" + cup_color.Render("_") + "{ " + cup_color.Render("__") + "{", 50 | cup_color.Render(" .-") + "{ } }" + cup_color.Render("-."), 51 | cup_color.Render(" (") + " } { " + cup_color.Render(")"), 52 | }, 53 | { 54 | " }", " { } ", 55 | " { " + cup_color.Render("_") + "} " + cup_color.Render("__") + "{", 56 | cup_color.Render(" .-") + "} { {" + cup_color.Render("-."), 57 | cup_color.Render(" (") + " { } " + cup_color.Render(")"), 58 | }, 59 | { 60 | " {", " { }", 61 | " }" + cup_color.Render("_") + "{ " + cup_color.Render("__") + "{", 62 | cup_color.Render(" .-") + "{ } }" + cup_color.Render("-."), 63 | cup_color.Render(" (") + " } { " + cup_color.Render(")"), 64 | }, 65 | { 66 | " }", " { } ", 67 | " { " + cup_color.Render("_") + "} " + cup_color.Render("__") + "{", 68 | cup_color.Render(" .-") + "} { {" + cup_color.Render("-."), 69 | cup_color.Render(" (") + " { } " + cup_color.Render(")"), 70 | }, 71 | } 72 | 73 | cup_head = cup_color.Render(helper.Center("|'-.._____..-'|", 40)) 74 | 75 | cup_body = [][]string{ 76 | {cup_color.Render(" | "), cup_color.Render(" ;--.")}, 77 | {cup_color.Render(" | "), cup_color.Render(" (__ \\")}, 78 | {cup_color.Render(" | "), cup_color.Render(" | ) )")}, 79 | {cup_color.Render(" | "), cup_color.Render(" | / /")}, 80 | {cup_color.Render(" | "), cup_color.Render(" |/ / ")}, 81 | {cup_color.Render(" | "), cup_color.Render(" ( / ")}, 82 | {cup_color.Render(" \\ "), cup_color.Render(" y` ")}, 83 | } 84 | cup_bottom = cup_color.Render(" `-.._____..-'") 85 | ) 86 | 87 | func (t *Coffee) Width() int { 88 | return t.width 89 | } 90 | 91 | func (t *Coffee) Height() int { 92 | return t.height 93 | } 94 | 95 | func (c *Coffee) NextAndString(percent int) string { 96 | c.Next(percent) 97 | return c.cachedTop + c.cachedBottom 98 | } 99 | 100 | func (c *Coffee) Next(percent int) bool { 101 | if c.prevPercent == percent { 102 | return false 103 | } 104 | 105 | grid := c.fill_amount(percent, c.width, c.height) 106 | 107 | total_body := cup_head + "\n" 108 | for k, v := range grid { 109 | total_body += cup_body[k][0] + coffee_color.Render(string(v)) + cup_body[k][1] + "\n" 110 | } 111 | total_body += cup_bottom 112 | c.cachedBottom = total_body + "\n\n\n" 113 | c.prevPercent = percent 114 | 115 | return true 116 | } 117 | 118 | func (c *Coffee) steam() int { 119 | c.curFrame++ 120 | 121 | if c.curFrame >= len(steam_animation) { 122 | c.curFrame = 0 123 | } 124 | 125 | top := "" 126 | for _, v := range steam_animation[c.curFrame] { 127 | top += helper.Center(v, 40) + "\n" 128 | } 129 | c.cachedTop = "\n\n\n" + top 130 | 131 | return c.curFrame 132 | } 133 | 134 | func GenerateCoffee() *Coffee { 135 | c := &Coffee{ 136 | width: 11, 137 | height: 7, 138 | curFrame: 0, 139 | prevPercent: -1, 140 | cachedTop: "", 141 | cachedBottom: "", 142 | } 143 | 144 | c.Next(0) 145 | go func() { 146 | for { 147 | t := time.NewTicker(2 * time.Second) 148 | c.steam() 149 | <-t.C 150 | } 151 | }() 152 | 153 | return c 154 | } 155 | -------------------------------------------------------------------------------- /ascii_generator/gen.go: -------------------------------------------------------------------------------- 1 | package ascii_generator 2 | 3 | type AsciiArt interface { 4 | Width() int 5 | Height() int 6 | NextAndString(int) string 7 | } 8 | -------------------------------------------------------------------------------- /ascii_generator/row.go: -------------------------------------------------------------------------------- 1 | // Row Animation, inspiration: https://stonestoryrpg.com/ 2 | 3 | package ascii_generator 4 | 5 | import ( 6 | "aimssh/helper" 7 | "math/rand" 8 | "sync" 9 | "time" 10 | ) 11 | 12 | type Row struct { 13 | cached string 14 | background [][]helper.Cell 15 | cur_animation, x, y, width, height int 16 | mu sync.RWMutex 17 | } 18 | 19 | var ( 20 | row_animation = []string{ 21 | (" o |\n _ \\ | __\n_ __ _ _ ___( ___\\| __ `)"), 22 | " o _\n _ \\ / _ \n_ __ ___ __ ( ___\\/ __ `)\n /_", 23 | " o \n _ V _ \n_ __ _ ___ _( _ / ____ `)\n _/ ~", 24 | " o \n _ (/ , _ \n_ __ _ ___( _ / ____ `)\n __ ~/ ", 25 | " o / \n _ = len(row_animation) { 72 | r.cur_animation = 0 73 | } 74 | 75 | return r.cur_animation 76 | } 77 | 78 | func randomBlue() string { 79 | x := rand.Float32() 80 | if x >= 0.9 { 81 | return navyBlueColor 82 | } else if x > 0.6 { 83 | return deepBlueColor 84 | } 85 | return skyblueColor 86 | } 87 | 88 | func (r *Row) backgroundCreator() { 89 | r.mu.Lock() 90 | defer r.mu.Unlock() 91 | 92 | r.background = [][]helper.Cell{} 93 | for i := 0; i < r.height; i++ { 94 | cell := []helper.Cell{} 95 | for j := 0; j < r.width; j++ { 96 | if rand.Float32() > 0.92 { 97 | cell = append(cell, helper.Cell{Ch: '_', Color: randomBlue()}) 98 | } else { 99 | cell = append(cell, helper.Cell{Ch: ' '}) 100 | } 101 | } 102 | r.background = append(r.background, cell) 103 | } 104 | } 105 | 106 | func GenerateRow(width, height int) *Row { 107 | r := &Row{ 108 | x: -30, 109 | y: rand.Intn(height - 4), 110 | width: width, 111 | height: height, 112 | } 113 | 114 | r.backgroundCreator() 115 | 116 | go func() { 117 | t := time.NewTicker(5 * time.Second) 118 | for { 119 | r.backgroundCreator() 120 | <-t.C 121 | } 122 | }() 123 | 124 | go func() { 125 | t := time.NewTicker(200 * time.Millisecond) 126 | for { 127 | r.mu.RLock() 128 | r.cached = helper.LayerCanvas(r.background, row_animation_cell[r.nextFrame()], r.x, r.y, true) 129 | r.mu.RUnlock() 130 | <-t.C 131 | if r.x > r.width { 132 | r.x = -30 133 | r.y = rand.Intn(height - 4) 134 | } 135 | r.x++ 136 | } 137 | }() 138 | 139 | return r 140 | } 141 | -------------------------------------------------------------------------------- /ascii_generator/tree.go: -------------------------------------------------------------------------------- 1 | // inspiration: https://gitlab.com/jallbrit/cbonsai 2 | 3 | package ascii_generator 4 | 5 | import ( 6 | "aimssh/helper" 7 | "math" 8 | "math/rand" 9 | "strings" 10 | "sync" 11 | ) 12 | 13 | const ( 14 | branchColor = "\033[38;5;94m" 15 | orangeColor = "\033[38;5;208m" 16 | lightSkinColor = "\033[38;5;223m" 17 | whiteColor = "\033[38;5;15m" 18 | darkRedColor = "\033[38;5;124m" 19 | lightBrownColor = "\033[38;5;137m" 20 | skyblueColor = "\033[38;5;69m" 21 | navyBlueColor = "\033[38;5;4m" 22 | deepBlueColor = "\033[38;5;39m" 23 | leafColor = "\033[32m" 24 | resetColor = "\033[0m" 25 | skipHeight = 2 26 | ) 27 | 28 | var ( 29 | cacheMutex sync.Mutex 30 | ) 31 | 32 | type branchParams struct { 33 | x, y, angle, length float64 34 | depth int 35 | } 36 | 37 | type Tree struct { 38 | Canvas [][]helper.Cell 39 | width int 40 | height int 41 | branchesQueue []branchParams 42 | currentLevel int 43 | totalLevels int 44 | cachedOutput string // Cached string output 45 | dirty bool // Whether cache needs invalidation 46 | lastPercent int // Last requested percentage 47 | } 48 | 49 | func initCanvas(width, height int) Tree { 50 | t := Tree{ 51 | height: height, 52 | width: width, 53 | dirty: true, 54 | } 55 | t.Canvas = make([][]helper.Cell, height) 56 | for i := 0; i < height; i++ { 57 | t.Canvas[i] = make([]helper.Cell, width) 58 | append_char := ' ' 59 | color := "" 60 | if i == t.height-skipHeight { 61 | append_char = '#' 62 | color = branchColor 63 | } 64 | for j := 0; j < width; j++ { 65 | t.Canvas[i][j] = helper.Cell{Ch: append_char, Color: color} 66 | } 67 | } 68 | 69 | return t 70 | } 71 | 72 | func (t *Tree) drawLine(x0, y0, x1, y1 float64, ch rune, col string) { 73 | dx := x1 - x0 74 | dy := y1 - y0 75 | steps := int(math.Max(math.Abs(dx), math.Abs(dy))) 76 | 77 | for i := 0; i <= steps; i++ { 78 | x := x0 + dx*float64(i)/float64(steps) 79 | y := y0 + dy*float64(i)/float64(steps) 80 | ix := int(math.Round(x)) 81 | iy := int(math.Round(y)) 82 | 83 | if ix >= 0 && ix < t.width && iy >= 0 && iy < t.height-skipHeight { 84 | // Only mark dirty if cell actually changes 85 | if t.Canvas[iy][ix].Ch != ch || t.Canvas[iy][ix].Color != col { 86 | t.Canvas[iy][ix] = helper.Cell{Ch: ch, Color: col} 87 | t.dirty = true 88 | } 89 | } 90 | } 91 | } 92 | 93 | func (t *Tree) Next(percentage int) bool { 94 | cacheMutex.Lock() 95 | defer cacheMutex.Unlock() 96 | 97 | // Return cached result if percentage hasn't changed 98 | if percentage == t.lastPercent && !t.dirty { 99 | return len(t.branchesQueue) > 0 100 | } 101 | 102 | t.lastPercent = percentage 103 | if percentage < 0 { 104 | percentage = 0 105 | } else if percentage > 100 { 106 | percentage = 100 107 | } 108 | 109 | desiredLevels := (percentage * t.totalLevels) / 100 110 | levelsToProcess := desiredLevels - t.currentLevel 111 | if levelsToProcess <= 0 { 112 | return len(t.branchesQueue) > 0 113 | } 114 | 115 | processedAny := false 116 | for i := 0; i < levelsToProcess; i++ { 117 | if len(t.branchesQueue) == 0 { 118 | break 119 | } 120 | 121 | nextQueue := make([]branchParams, 0, len(t.branchesQueue)*2) 122 | for _, bp := range t.branchesQueue { 123 | if bp.depth == 0 || bp.length < 1 { 124 | ix, iy := int(math.Round(bp.x)), int(math.Round(bp.y)) 125 | if ix >= 0 && ix < t.width && iy >= 0 && iy < t.height { 126 | // Only mark dirty if cell changes 127 | if t.Canvas[iy][ix].Ch != '#' || t.Canvas[iy][ix].Color != leafColor { 128 | t.Canvas[iy][ix] = helper.Cell{Ch: '#', Color: leafColor} 129 | t.dirty = true 130 | } 131 | } 132 | continue 133 | } 134 | 135 | xEnd := bp.x + bp.length*math.Cos(bp.angle) 136 | yEnd := bp.y - bp.length*math.Sin(bp.angle) 137 | 138 | var ch rune 139 | diff := math.Abs(bp.angle - math.Pi/2) 140 | if diff < 0.2 { 141 | ch = '|' 142 | } else if bp.angle < math.Pi/2 { 143 | ch = '/' 144 | } else { 145 | ch = '\\' 146 | } 147 | 148 | t.drawLine(bp.x, bp.y, xEnd, yEnd, ch, branchColor) 149 | 150 | newLength := bp.length * (0.7 + 0.1*rand.Float64()) 151 | leftAngle := bp.angle + (0.3 + 0.2*rand.Float64()) 152 | rightAngle := bp.angle - (0.3 + 0.2*rand.Float64()) 153 | 154 | nextDepth := bp.depth - 1 155 | nextQueue = append(nextQueue, branchParams{ 156 | x: xEnd, 157 | y: yEnd, 158 | angle: leftAngle, 159 | length: newLength, 160 | depth: nextDepth, 161 | }, branchParams{ 162 | x: xEnd, 163 | y: yEnd, 164 | angle: rightAngle, 165 | length: newLength, 166 | depth: nextDepth, 167 | }) 168 | processedAny = true 169 | } 170 | 171 | t.branchesQueue = nextQueue 172 | t.currentLevel++ 173 | } 174 | 175 | if processedAny { 176 | t.dirty = true 177 | } 178 | return len(t.branchesQueue) > 0 179 | } 180 | 181 | func (t *Tree) StringPrint() string { 182 | cacheMutex.Lock() 183 | defer cacheMutex.Unlock() 184 | 185 | if !t.dirty && t.cachedOutput != "" { 186 | return t.cachedOutput 187 | } 188 | 189 | var builder strings.Builder 190 | builder.Grow(t.height * (t.width + 1)) // Pre-allocate memory 191 | 192 | for i := 0; i < t.height; i++ { 193 | line := make([]byte, 0, t.width) 194 | for j := 0; j < t.width; j++ { 195 | cell := t.Canvas[i][j] 196 | if cell.Color != "" { 197 | line = append(line, cell.Color...) 198 | line = append(line, byte(cell.Ch)) 199 | line = append(line, resetColor...) 200 | } else { 201 | line = append(line, byte(cell.Ch)) 202 | } 203 | } 204 | builder.Write(line) 205 | builder.WriteByte('\n') 206 | } 207 | 208 | t.cachedOutput = builder.String() 209 | t.dirty = false 210 | return t.cachedOutput 211 | } 212 | 213 | func (t *Tree) NextAndString(percent int) string { 214 | t.Next(percent) 215 | return t.StringPrint() + "\n" 216 | } 217 | 218 | func (t *Tree) Width() int { 219 | return t.width 220 | } 221 | 222 | func (t *Tree) Height() int { 223 | return t.height 224 | } 225 | 226 | func GenerateTree(width, height int) *Tree { 227 | t := initCanvas(width, height) 228 | 229 | startX := float64(width / 2) 230 | startY := float64(height - 1) 231 | initialLength := float64(height) / 4 232 | 233 | t.totalLevels = 10 // Corresponds to the initial depth of 10 234 | t.branchesQueue = []branchParams{ 235 | { 236 | x: startX, 237 | y: startY, 238 | angle: math.Pi / 2, 239 | length: initialLength, 240 | depth: 10, 241 | }, 242 | } 243 | 244 | return &t 245 | } 246 | -------------------------------------------------------------------------------- /assets/coffee.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/coffee.gif -------------------------------------------------------------------------------- /assets/coffee.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/coffee.png -------------------------------------------------------------------------------- /assets/demo-small-screen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/demo-small-screen.png -------------------------------------------------------------------------------- /assets/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/demo.png -------------------------------------------------------------------------------- /assets/flow.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/flow.gif -------------------------------------------------------------------------------- /assets/flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/flow.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/logo.png -------------------------------------------------------------------------------- /assets/logo_with_background.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/logo_with_background.png -------------------------------------------------------------------------------- /assets/pomossh.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/pomossh.png -------------------------------------------------------------------------------- /assets/pomossh_text.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/pomossh_text.png -------------------------------------------------------------------------------- /assets/sshaimftpsh.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/sshaimftpsh.png -------------------------------------------------------------------------------- /assets/sshpomosairashgautamcomnp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/sshpomosairashgautamcomnp.png -------------------------------------------------------------------------------- /assets/step2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/step2.png -------------------------------------------------------------------------------- /assets/step3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/step3.png -------------------------------------------------------------------------------- /assets/transparent.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/transparent.png -------------------------------------------------------------------------------- /assets/tree.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/tree.gif -------------------------------------------------------------------------------- /assets/tree.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sairash/aimssh/8f12d05d28128f2543990387ec390bf36f6370af/assets/tree.png -------------------------------------------------------------------------------- /go.mod: -------------------------------------------------------------------------------- 1 | module aimssh 2 | 3 | go 1.24.0 4 | 5 | require ( 6 | github.com/charmbracelet/bubbles v0.20.0 7 | github.com/charmbracelet/bubbletea v1.3.4 8 | github.com/charmbracelet/lipgloss v1.0.0 9 | github.com/charmbracelet/log v0.4.0 10 | github.com/charmbracelet/ssh v0.0.0-20250128164007-98fd5ae11894 11 | github.com/charmbracelet/wish v1.4.6 12 | github.com/gen2brain/beeep v0.0.0-20240516210008-9c006672e7f4 13 | ) 14 | 15 | require ( 16 | github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be // indirect 17 | github.com/atotto/clipboard v0.1.4 // indirect 18 | github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect 19 | github.com/charmbracelet/keygen v0.5.1 // indirect 20 | github.com/charmbracelet/x/ansi v0.8.0 // indirect 21 | github.com/charmbracelet/x/conpty v0.1.0 // indirect 22 | github.com/charmbracelet/x/errors v0.0.0-20240508181413-e8d8b6e2de86 // indirect 23 | github.com/charmbracelet/x/input v0.3.1 // indirect 24 | github.com/charmbracelet/x/term v0.2.1 // indirect 25 | github.com/charmbracelet/x/termios v0.1.0 // indirect 26 | github.com/charmbracelet/x/windows v0.2.0 // indirect 27 | github.com/creack/pty v1.1.21 // indirect 28 | github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect 29 | github.com/go-logfmt/logfmt v0.6.0 // indirect 30 | github.com/go-toast/toast v0.0.0-20190211030409-01e6764cf0a4 // indirect 31 | github.com/godbus/dbus/v5 v5.1.0 // indirect 32 | github.com/lucasb-eyer/go-colorful v1.2.0 // indirect 33 | github.com/mattn/go-isatty v0.0.20 // indirect 34 | github.com/mattn/go-localereader v0.0.1 // indirect 35 | github.com/mattn/go-runewidth v0.0.16 // indirect 36 | github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect 37 | github.com/muesli/cancelreader v0.2.2 // indirect 38 | github.com/muesli/termenv v0.15.3-0.20240509142007-81b8f94111d5 // indirect 39 | github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d // indirect 40 | github.com/rivo/uniseg v0.4.7 // indirect 41 | github.com/sahilm/fuzzy v0.1.1 // indirect 42 | github.com/tadvi/systray v0.0.0-20190226123456-11a2b8fa57af // indirect 43 | github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect 44 | golang.org/x/crypto v0.32.0 // indirect 45 | golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56 // indirect 46 | golang.org/x/sync v0.11.0 // indirect 47 | golang.org/x/sys v0.30.0 // indirect 48 | golang.org/x/text v0.21.0 // indirect 49 | ) 50 | -------------------------------------------------------------------------------- /go.sum: -------------------------------------------------------------------------------- 1 | github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be h1:9AeTilPcZAjCFIImctFaOjnTIavg87rW78vTPkQqLI8= 2 | github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be/go.mod h1:ySMOLuWl6zY27l47sB3qLNK6tF2fkHG55UZxx8oIVo4= 3 | github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4= 4 | github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI= 5 | github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k= 6 | github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8= 7 | github.com/charmbracelet/bubbles v0.20.0 h1:jSZu6qD8cRQ6k9OMfR1WlM+ruM8fkPWkHvQWD9LIutE= 8 | github.com/charmbracelet/bubbles v0.20.0/go.mod h1:39slydyswPy+uVOHZ5x/GjwVAFkCsV8IIVy+4MhzwwU= 9 | github.com/charmbracelet/bubbletea v1.3.4 h1:kCg7B+jSCFPLYRA52SDZjr51kG/fMUEoPoZrkaDHyoI= 10 | github.com/charmbracelet/bubbletea v1.3.4/go.mod h1:dtcUCyCGEX3g9tosuYiut3MXgY/Jsv9nKVdibKKRRXo= 11 | github.com/charmbracelet/keygen v0.5.1 h1:zBkkYPtmKDVTw+cwUyY6ZwGDhRxXkEp0Oxs9sqMLqxI= 12 | github.com/charmbracelet/keygen v0.5.1/go.mod h1:zznJVmK/GWB6dAtjluqn2qsttiCBhA5MZSiwb80fcHw= 13 | github.com/charmbracelet/lipgloss v1.0.0 h1:O7VkGDvqEdGi93X+DeqsQ7PKHDgtQfF8j8/O2qFMQNg= 14 | github.com/charmbracelet/lipgloss v1.0.0/go.mod h1:U5fy9Z+C38obMs+T+tJqst9VGzlOYGj4ri9reL3qUlo= 15 | github.com/charmbracelet/log v0.4.0 h1:G9bQAcx8rWA2T3pWvx7YtPTPwgqpk7D68BX21IRW8ZM= 16 | github.com/charmbracelet/log v0.4.0/go.mod h1:63bXt/djrizTec0l11H20t8FDSvA4CRZJ1KH22MdptM= 17 | github.com/charmbracelet/ssh v0.0.0-20250128164007-98fd5ae11894 h1:Ffon9TbltLGBsT6XE//YvNuu4OAaThXioqalhH11xEw= 18 | github.com/charmbracelet/ssh v0.0.0-20250128164007-98fd5ae11894/go.mod h1:hg+I6gvlMl16nS9ZzQNgBIrrCasGwEw0QiLsDcP01Ko= 19 | github.com/charmbracelet/wish v1.4.6 h1:27WRqMTUmyFoZASoaAaEe78Je7LTU4VqyoBxnl4d9XA= 20 | github.com/charmbracelet/wish v1.4.6/go.mod h1:RRy2LFW3WQ3tlPmMMGgEeSMDVlFd5yqklGBVZWQSHmk= 21 | github.com/charmbracelet/x/ansi v0.8.0 h1:9GTq3xq9caJW8ZrBTe0LIe2fvfLR/bYXKTx2llXn7xE= 22 | github.com/charmbracelet/x/ansi v0.8.0/go.mod h1:wdYl/ONOLHLIVmQaxbIYEC/cRKOQyjTkowiI4blgS9Q= 23 | github.com/charmbracelet/x/conpty v0.1.0 h1:4zc8KaIcbiL4mghEON8D72agYtSeIgq8FSThSPQIb+U= 24 | github.com/charmbracelet/x/conpty v0.1.0/go.mod h1:rMFsDJoDwVmiYM10aD4bH2XiRgwI7NYJtQgl5yskjEQ= 25 | github.com/charmbracelet/x/errors v0.0.0-20240508181413-e8d8b6e2de86 h1:JSt3B+U9iqk37QUU2Rvb6DSBYRLtWqFqfxf8l5hOZUA= 26 | github.com/charmbracelet/x/errors v0.0.0-20240508181413-e8d8b6e2de86/go.mod h1:2P0UgXMEa6TsToMSuFqKFQR+fZTO9CNGUNokkPatT/0= 27 | github.com/charmbracelet/x/input v0.3.1 h1:TE4s3fTRj+OUpJ86dKphrN99+NgBnto//EkWncMJQIg= 28 | github.com/charmbracelet/x/input v0.3.1/go.mod h1:4w9jS/NW62WrHSdmjbpzydvnbqkd+mtyK8WOWbHCdvs= 29 | github.com/charmbracelet/x/term v0.2.1 h1:AQeHeLZ1OqSXhrAWpYUtZyX1T3zVxfpZuEQMIQaGIAQ= 30 | github.com/charmbracelet/x/term v0.2.1/go.mod h1:oQ4enTYFV7QN4m0i9mzHrViD7TQKvNEEkHUMCmsxdUg= 31 | github.com/charmbracelet/x/termios v0.1.0 h1:y4rjAHeFksBAfGbkRDmVinMg7x7DELIGAFbdNvxg97k= 32 | github.com/charmbracelet/x/termios v0.1.0/go.mod h1:H/EVv/KRnrYjz+fCYa9bsKdqF3S8ouDK0AZEbG7r+/U= 33 | github.com/charmbracelet/x/windows v0.2.0 h1:ilXA1GJjTNkgOm94CLPeSz7rar54jtFatdmoiONPuEw= 34 | github.com/charmbracelet/x/windows v0.2.0/go.mod h1:ZibNFR49ZFqCXgP76sYanisxRyC+EYrBE7TTknD8s1s= 35 | github.com/creack/pty v1.1.21 h1:1/QdRyBaHHJP61QkWMXlOIBfsgdDeeKfK8SYVUWJKf0= 36 | github.com/creack/pty v1.1.21/go.mod h1:MOBLtS5ELjhRRrroQr9kyvTxUAFNvYEK993ew/Vr4O4= 37 | github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= 38 | github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= 39 | github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f h1:Y/CXytFA4m6baUTXGLOoWe4PQhGxaX0KpnayAqC48p4= 40 | github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f/go.mod h1:vw97MGsxSvLiUE2X8qFplwetxpGLQrlU1Q9AUEIzCaM= 41 | github.com/gen2brain/beeep v0.0.0-20240516210008-9c006672e7f4 h1:ygs9POGDQpQGLJPlq4+0LBUmMBNox1N4JSpw+OETcvI= 42 | github.com/gen2brain/beeep v0.0.0-20240516210008-9c006672e7f4/go.mod h1:0W7dI87PvXJ1Sjs0QPvWXKcQmNERY77e8l7GFhZB/s4= 43 | github.com/go-logfmt/logfmt v0.6.0 h1:wGYYu3uicYdqXVgoYbvnkrPVXkuLM1p1ifugDMEdRi4= 44 | github.com/go-logfmt/logfmt v0.6.0/go.mod h1:WYhtIu8zTZfxdn5+rREduYbwxfcBr/Vr6KEVveWlfTs= 45 | github.com/go-toast/toast v0.0.0-20190211030409-01e6764cf0a4 h1:qZNfIGkIANxGv/OqtnntR4DfOY2+BgwR60cAcu/i3SE= 46 | github.com/go-toast/toast v0.0.0-20190211030409-01e6764cf0a4/go.mod h1:kW3HQ4UdaAyrUCSSDR4xUzBKW6O2iA4uHhk7AtyYp10= 47 | github.com/godbus/dbus/v5 v5.1.0 h1:4KLkAxT3aOY8Li4FRJe/KvhoNFFxo0m6fNuFUO8QJUk= 48 | github.com/godbus/dbus/v5 v5.1.0/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= 49 | github.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc= 50 | github.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw= 51 | github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY= 52 | github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0= 53 | github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= 54 | github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= 55 | github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4= 56 | github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88= 57 | github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc= 58 | github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= 59 | github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI= 60 | github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo= 61 | github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA= 62 | github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo= 63 | github.com/muesli/termenv v0.15.3-0.20240509142007-81b8f94111d5 h1:NiONcKK0EV5gUZcnCiPMORaZA0eBDc+Fgepl9xl4lZ8= 64 | github.com/muesli/termenv v0.15.3-0.20240509142007-81b8f94111d5/go.mod h1:hxSnBBYLK21Vtq/PHd0S2FYCxBXzBua8ov5s1RobyRQ= 65 | github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d h1:VhgPp6v9qf9Agr/56bj7Y/xa04UccTW04VP0Qed4vnQ= 66 | github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d/go.mod h1:YUTz3bUH2ZwIWBy3CJBeOBEugqcmXREj14T+iG/4k4U= 67 | github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= 68 | github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= 69 | github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= 70 | github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ= 71 | github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88= 72 | github.com/sahilm/fuzzy v0.1.1 h1:ceu5RHF8DGgoi+/dR5PsECjCDH1BE3Fnmpo7aVXOdRA= 73 | github.com/sahilm/fuzzy v0.1.1/go.mod h1:VFvziUEIMCrT6A6tw2RFIXPXXmzXbOsSHF0DOI8ZK9Y= 74 | github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg= 75 | github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= 76 | github.com/tadvi/systray v0.0.0-20190226123456-11a2b8fa57af h1:6yITBqGTE2lEeTPG04SN9W+iWHCRyHqlVYILiSXziwk= 77 | github.com/tadvi/systray v0.0.0-20190226123456-11a2b8fa57af/go.mod h1:4F09kP5F+am0jAwlQLddpoMDM+iewkxxt6nxUQ5nq5o= 78 | github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no= 79 | github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM= 80 | golang.org/x/crypto v0.32.0 h1:euUpcYgM8WcP71gNpTqQCn6rC2t6ULUPiOzfWaXVVfc= 81 | golang.org/x/crypto v0.32.0/go.mod h1:ZnnJkOaASj8g0AjIduWNlq2NRxL0PlBrbKVyZ6V/Ugc= 82 | golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56 h1:2dVuKD2vS7b0QIHQbpyTISPd0LeHDbnYEryqj5Q1ug8= 83 | golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56/go.mod h1:M4RDyNAINzryxdtnbRXRL/OHtkFuWGRjvuhBJpk2IlY= 84 | golang.org/x/sync v0.11.0 h1:GGz8+XQP4FvTTrjZPzNKTMFtSXH80RAzG+5ghFPgK9w= 85 | golang.org/x/sync v0.11.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= 86 | golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= 87 | golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= 88 | golang.org/x/sys v0.30.0 h1:QjkSwP/36a20jFYWkSue1YwXzLmsV5Gfq7Eiy72C1uc= 89 | golang.org/x/sys v0.30.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= 90 | golang.org/x/term v0.28.0 h1:/Ts8HFuMR2E6IP/jlo7QVLZHggjKQbhu/7H0LJFr3Gg= 91 | golang.org/x/term v0.28.0/go.mod h1:Sw/lC2IAUZ92udQNf3WodGtn4k/XoLyZoh8v/8uiwek= 92 | golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo= 93 | golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ= 94 | gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= 95 | gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= 96 | -------------------------------------------------------------------------------- /helper/helper.go: -------------------------------------------------------------------------------- 1 | package helper 2 | 3 | import "strings" 4 | 5 | var ( 6 | resetColor = "\033[0m" 7 | ) 8 | 9 | const ( 10 | TimerEndedTitle = "Timer Ended" 11 | TimerStartedTitle = "Timer Started" 12 | TimerRestartedTitle = "Timer Restarted" 13 | ) 14 | 15 | type Cell struct { 16 | Ch rune 17 | Color string 18 | } 19 | 20 | func Center(s string, total_width int) string { 21 | if len(s) >= total_width { 22 | return s 23 | } 24 | n := total_width - len(s) 25 | div := n / 2 26 | return strings.Repeat(" ", div) + s 27 | } 28 | 29 | func LayerString(background, top string, x, y int, hideOverflow bool) string { 30 | bgLines := strings.Split(background, "\n") 31 | topLines := strings.Split(top, "\n") 32 | 33 | if y < 0 { 34 | trim := -y 35 | if trim >= len(topLines) { 36 | return background 37 | } 38 | topLines = topLines[trim:] 39 | y = 0 40 | } 41 | 42 | trimLeft := 0 43 | if x < 0 { 44 | trimLeft = -x 45 | x = 0 46 | } 47 | for i := range topLines { 48 | if trimLeft > len(topLines[i]) { 49 | topLines[i] = "" 50 | } else { 51 | topLines[i] = topLines[i][trimLeft:] 52 | } 53 | } 54 | 55 | for i, topLine := range topLines { 56 | targetLine := y + i 57 | 58 | for len(bgLines) <= targetLine { 59 | bgLines = append(bgLines, "") 60 | } 61 | 62 | original := bgLines[targetLine] 63 | var start string 64 | 65 | if len(original) >= x { 66 | start = original[:x] 67 | } else { 68 | start = original + strings.Repeat(" ", x-len(original)) 69 | } 70 | 71 | availableSpace := len(original) - x 72 | if availableSpace < 0 { 73 | availableSpace = 0 74 | } 75 | 76 | if hideOverflow { 77 | if len(topLine) > availableSpace { 78 | if availableSpace <= 0 { 79 | topLine = "" 80 | } else { 81 | topLine = topLine[:availableSpace] 82 | } 83 | } 84 | } 85 | 86 | endStart := x + len(topLine) 87 | var end string 88 | if len(original) > endStart { 89 | end = original[endStart:] 90 | } 91 | 92 | bgLines[targetLine] = start + topLine + end 93 | } 94 | 95 | return strings.Join(bgLines, "\n") 96 | } 97 | 98 | func cellsToString(grid [][]Cell) string { 99 | lines := make([]string, len(grid)) 100 | for i, row := range grid { 101 | var sb strings.Builder 102 | for _, cell := range row { 103 | if cell.Ch == 0 { 104 | sb.WriteRune(' ') 105 | } else { 106 | sb.WriteString(cell.Color) 107 | sb.WriteRune(cell.Ch) 108 | sb.WriteString(resetColor) 109 | } 110 | } 111 | lines[i] = sb.String() 112 | } 113 | return strings.Join(lines, "\n") + "\n" 114 | } 115 | 116 | func LayerCanvas(background, topCan [][]Cell, x, y int, hideOverflow bool) string { 117 | bg := make([][]Cell, len(background)) 118 | for i := range background { 119 | bg[i] = make([]Cell, len(background[i])) 120 | copy(bg[i], background[i]) 121 | } 122 | 123 | top := make([][]Cell, len(topCan)) 124 | for i := range topCan { 125 | top[i] = make([]Cell, len(topCan[i])) 126 | copy(top[i], topCan[i]) 127 | } 128 | 129 | if y < 0 { 130 | if trim := -y; trim < len(top) { 131 | top = top[trim:] 132 | } else { 133 | return cellsToString(bg) 134 | } 135 | y = 0 136 | } 137 | 138 | trimLeft := 0 139 | if x < 0 { 140 | trimLeft = -x 141 | x = 0 142 | } 143 | for i := range top { 144 | if trimLeft < len(top[i]) { 145 | top[i] = top[i][trimLeft:] 146 | } else { 147 | top[i] = []Cell{} 148 | } 149 | } 150 | 151 | for i, tline := range top { 152 | targetY := y + i 153 | 154 | for len(bg) <= targetY { 155 | bg = append(bg, []Cell{}) 156 | } 157 | 158 | bgline := bg[targetY] 159 | available := len(bgline) - x 160 | if available < 0 { 161 | available = 0 162 | } 163 | 164 | visible := tline 165 | if hideOverflow && len(tline) > available { 166 | if available > 0 { 167 | visible = tline[:available] 168 | } else { 169 | visible = []Cell{} 170 | } 171 | } 172 | 173 | if !hideOverflow && x+len(visible) > len(bgline) { 174 | newLen := x + len(visible) 175 | if cap(bgline) >= newLen { 176 | bgline = bgline[:newLen] 177 | } else { 178 | newBgline := make([]Cell, newLen) 179 | copy(newBgline, bgline) 180 | bgline = newBgline 181 | } 182 | } 183 | 184 | if x > len(bgline) { 185 | padding := make([]Cell, x-len(bgline)) 186 | for j := range padding { 187 | padding[j].Ch = ' ' 188 | } 189 | bgline = append(bgline, padding...) 190 | } 191 | 192 | for j, cell := range visible { 193 | pos := x + j 194 | if pos < len(bgline) { 195 | bgline[pos] = cell 196 | } else if !hideOverflow { 197 | bgline = append(bgline, cell) 198 | } 199 | } 200 | 201 | bg[targetY] = bgline 202 | } 203 | 204 | return cellsToString(bg) 205 | } 206 | -------------------------------------------------------------------------------- /main.go: -------------------------------------------------------------------------------- 1 | package main 2 | 3 | import ( 4 | "aimssh/ascii_generator" 5 | "aimssh/helper" 6 | "context" 7 | "errors" 8 | "flag" 9 | "fmt" 10 | "io" 11 | "net" 12 | "os" 13 | "os/signal" 14 | "strconv" 15 | "strings" 16 | "syscall" 17 | "time" 18 | 19 | "github.com/charmbracelet/bubbles/help" 20 | "github.com/charmbracelet/bubbles/key" 21 | "github.com/charmbracelet/bubbles/list" 22 | "github.com/charmbracelet/bubbles/textinput" 23 | "github.com/charmbracelet/bubbles/timer" 24 | tea "github.com/charmbracelet/bubbletea" 25 | "github.com/charmbracelet/lipgloss" 26 | "github.com/charmbracelet/log" 27 | "github.com/charmbracelet/ssh" 28 | "github.com/charmbracelet/wish" 29 | "github.com/charmbracelet/wish/activeterm" 30 | "github.com/charmbracelet/wish/bubbletea" 31 | "github.com/charmbracelet/wish/logging" 32 | "github.com/gen2brain/beeep" 33 | ) 34 | 35 | type sessionState int 36 | 37 | const ( 38 | dotChar = " • " 39 | app_width = 50 40 | host = "localhost" 41 | port = "13234" 42 | logo = ` _ ___ __ __ ____ ____ _ _ 43 | / \ |_ _| \/ | / ___/ ___|| | | | 44 | / _ \ | || |\/| | \___ \___ \| |_| | 45 | / ___ \ | || | | | ___) |__) | _ | 46 | /_/ \_\___|_| |_| |____/____/|_| |_| 47 | ` 48 | 49 | inputView sessionState = iota 50 | logoView 51 | listView 52 | timerView 53 | ) 54 | 55 | var ( 56 | appStyle = lipgloss.NewStyle().Padding(1, 2).Border(lipgloss.RoundedBorder(), true, true, true, true).Width(app_width) 57 | heightThing = lipgloss.NewStyle().Height(21) 58 | paddingleft = lipgloss.NewStyle().PaddingLeft(2) 59 | titleStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("#49beaa")).Bold(true).SetString(helper.Center(``, app_width-4)).AlignHorizontal(lipgloss.Center) 60 | listTitleStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("#bfedc1")).PaddingLeft(-10) 61 | itemStyle = lipgloss.NewStyle().PaddingLeft(4) 62 | selectedItemStyle = lipgloss.NewStyle().PaddingLeft(2).Foreground(lipgloss.Color("#CFF27E")) 63 | 64 | greenColor = lipgloss.NewStyle().Foreground(lipgloss.Color("#bfedc1")).PaddingLeft(2).Faint(true) 65 | dotStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("236")).Render(dotChar) 66 | // subtleStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("241")) 67 | 68 | run_as_ssh bool 69 | gitlink = greenColor.Render("https://github.com/sairash/aimssh") 70 | 71 | brownColor = lipgloss.NewStyle().Foreground(lipgloss.Color("#967969")) 72 | end_info = fmt.Sprintf("\n Thanks for using %s! \n Give a star %s \n Made By %s\n", lipgloss.NewStyle().Foreground(lipgloss.Color("#49beaa")).Bold(true).Render(""), gitlink, greenColor.Render("https://sairashgautam.com.np/")) 73 | ) 74 | 75 | type item string 76 | 77 | func (i item) FilterValue() string { return "" } 78 | 79 | type itemDelegate struct{} 80 | 81 | func (d itemDelegate) Height() int { return 1 } 82 | func (d itemDelegate) Spacing() int { return 0 } 83 | func (d itemDelegate) Update(_ tea.Msg, _ *list.Model) tea.Cmd { return nil } 84 | func (d itemDelegate) Render(w io.Writer, m list.Model, index int, listItem list.Item) { 85 | i, ok := listItem.(item) 86 | if !ok { 87 | return 88 | } 89 | 90 | str := fmt.Sprintf("%d. %s", index+1, i) 91 | 92 | fn := itemStyle.Render 93 | if index == m.Index() { 94 | fn = func(s ...string) string { 95 | return selectedItemStyle.Render("> " + strings.Join(s, " ")) 96 | } 97 | } 98 | 99 | fmt.Fprint(w, fn(str)) 100 | } 101 | 102 | type model struct { 103 | state sessionState 104 | input textinput.Model 105 | workingon textinput.Model 106 | list list.Model 107 | minute time.Duration 108 | selectedItem string 109 | timer timer.Model 110 | loadingTimer timer.Model 111 | keymap keymap 112 | help help.Model 113 | err error 114 | width int 115 | height int 116 | session string 117 | asciiArt ascii_generator.AsciiArt 118 | quitting bool 119 | timedOut bool 120 | sessionSsh ssh.Session 121 | } 122 | 123 | type keymap struct { 124 | start key.Binding 125 | stop key.Binding 126 | reset key.Binding 127 | quit key.Binding 128 | new key.Binding 129 | 130 | back key.Binding 131 | enter key.Binding 132 | ctrlc key.Binding 133 | } 134 | 135 | func initialModel(session interface{}) model { 136 | ti := textinput.New() 137 | ti.PlaceholderStyle = lipgloss.NewStyle().Faint(true) 138 | ti.Placeholder = "10" 139 | ti.Focus() 140 | ti.CharLimit = 50 141 | ti.Width = 30 142 | ti.Prompt = "- " 143 | 144 | woI := textinput.New() 145 | woI.PlaceholderStyle = lipgloss.NewStyle().Faint(true) 146 | woI.Placeholder = "Work" 147 | woI.CharLimit = 50 148 | woI.Width = 30 149 | woI.Prompt = "- " 150 | 151 | items := []list.Item{ 152 | item("Tree"), 153 | item("Flow"), 154 | item("Coffee"), 155 | } 156 | 157 | l := list.New(items, itemDelegate{}, 30, 11) 158 | l.Title = "Select a visual option: " 159 | l.SetShowStatusBar(false) 160 | l.Styles.Title = listTitleStyle 161 | // l.SetShowHelp(false) 162 | l.SetHeight(23) 163 | // l.SetShowTitle(false) 164 | 165 | m := model{ 166 | state: logoView, 167 | input: ti, 168 | loadingTimer: timer.NewWithInterval(800*time.Millisecond, time.Millisecond), 169 | workingon: woI, 170 | list: l, 171 | err: nil, 172 | session: "Work", 173 | timedOut: false, 174 | keymap: keymap{ 175 | start: key.NewBinding( 176 | key.WithKeys(" ", "s"), 177 | key.WithHelp("space", "start"), 178 | ), 179 | stop: key.NewBinding( 180 | key.WithKeys(" ", "s"), 181 | key.WithHelp("space", "stop"), 182 | ), 183 | reset: key.NewBinding( 184 | key.WithKeys("r"), 185 | key.WithHelp("r", "restart"), 186 | ), 187 | quit: key.NewBinding( 188 | key.WithKeys("q", "ctrl+c"), 189 | key.WithHelp("q", "quit"), 190 | ), 191 | new: key.NewBinding( 192 | key.WithKeys("n"), 193 | key.WithHelp("n", "new"), 194 | ), 195 | back: key.NewBinding( 196 | key.WithKeys("ctrl+b"), 197 | key.WithHelp("ctrl+b", "back"), 198 | ), 199 | enter: key.NewBinding( 200 | key.WithKeys("enter"), 201 | key.WithHelp("↩", "enter"), 202 | ), 203 | ctrlc: key.NewBinding( 204 | key.WithKeys("ctrl+c"), 205 | key.WithHelp("ctrl+c", "quit"), 206 | ), 207 | }, 208 | help: help.New(), 209 | } 210 | 211 | switch session := session.(type) { 212 | case ssh.Session: 213 | m.sessionSsh = session 214 | } 215 | 216 | return m 217 | 218 | } 219 | 220 | func (m model) Init() tea.Cmd { 221 | return tea.Batch(m.loadingTimer.Init(), m.loadingTimer.Start()) 222 | } 223 | 224 | func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) { 225 | switch msg := msg.(type) { 226 | case tea.WindowSizeMsg: 227 | m.width = msg.Width 228 | m.height = msg.Height 229 | case tea.KeyMsg: 230 | 231 | switch msg.Type { 232 | case tea.KeyCtrlC: 233 | m.quitting = true 234 | return m, tea.Quit 235 | 236 | case tea.KeyCtrlB: 237 | 238 | switch m.state { 239 | case inputView: 240 | if m.workingon.Focused() { 241 | m.workingon.Blur() 242 | m.input.Focus() 243 | } 244 | 245 | case listView: 246 | m.state = inputView 247 | m.input.Blur() 248 | m.workingon.Focus() 249 | case timerView: 250 | m.state = listView 251 | } 252 | 253 | } 254 | 255 | } 256 | 257 | switch m.state { 258 | case logoView: 259 | return updateLogo(msg, m) 260 | case inputView: 261 | return updateInput(msg, m) 262 | case listView: 263 | return updateList(msg, m) 264 | case timerView: 265 | return updateTimer(msg, m) 266 | default: 267 | return m, nil 268 | } 269 | } 270 | 271 | func updateLogo(msg tea.Msg, m model) (tea.Model, tea.Cmd) { 272 | switch msg.(type) { 273 | case timer.TickMsg: 274 | var cmd tea.Cmd 275 | m.loadingTimer, cmd = m.loadingTimer.Update(msg) 276 | return m, cmd 277 | case timer.TimeoutMsg: 278 | m.loadingTimer.Stop() 279 | m.state = inputView 280 | return m, textinput.Blink 281 | } 282 | return m, nil 283 | } 284 | 285 | func updateInput(msg tea.Msg, m model) (tea.Model, tea.Cmd) { 286 | var cmd tea.Cmd 287 | 288 | switch msg := msg.(type) { 289 | case tea.KeyMsg: 290 | switch msg.Type { 291 | case tea.KeyEnter: 292 | if m.input.Focused() { 293 | if m.input.Value() != "" { 294 | 295 | min, err := strconv.Atoi(m.input.Value()) 296 | 297 | if err != nil { 298 | m.err = err 299 | m.quitting = true 300 | return m, tea.Quit 301 | } 302 | 303 | m.minute = time.Duration(min) * time.Minute 304 | m.input.Blur() 305 | return m, m.workingon.Focus() 306 | } 307 | } else { 308 | if m.workingon.Value() != "" { 309 | m.session = m.workingon.Value() 310 | } 311 | m.state = listView 312 | } 313 | } 314 | } 315 | 316 | if m.input.Focused() { 317 | m.input, cmd = m.input.Update(msg) 318 | } else { 319 | m.workingon, cmd = m.workingon.Update(msg) 320 | } 321 | 322 | return m, cmd 323 | } 324 | 325 | func (m model) generate_ascii() ascii_generator.AsciiArt { 326 | switch m.selectedItem { 327 | case "Coffee": 328 | return ascii_generator.GenerateCoffee() 329 | case "Tree": 330 | return ascii_generator.GenerateTree(40, 18) 331 | default: 332 | return ascii_generator.GenerateRow(40, 17) 333 | } 334 | } 335 | 336 | func updateList(msg tea.Msg, m model) (tea.Model, tea.Cmd) { 337 | var cmd tea.Cmd 338 | 339 | switch msg := msg.(type) { 340 | case tea.KeyMsg: 341 | switch msg.Type { 342 | case tea.KeyEnter: 343 | if selected, ok := m.list.SelectedItem().(item); ok { 344 | m.selectedItem = string(selected) 345 | m.timer = timer.NewWithInterval(m.minute, time.Millisecond) 346 | m.state = timerView 347 | m.asciiArt = m.generate_ascii() 348 | // +brownColor.Render(strings.Repeat("░", app_width-8)) 349 | m.keymap.start.SetEnabled(false) 350 | 351 | notificaiton_body := fmt.Sprintf("Timer set for %d minutes.", int(m.timer.Timeout.Minutes())) 352 | if !run_as_ssh { 353 | beeep.Alert(helper.TimerStartedTitle, notificaiton_body, "assets/logo.png") 354 | } else { 355 | switch m.sessionSsh.Context().Value("operating_system") { 356 | case "linux": 357 | wish.Command(m.sessionSsh, "notify-send", "-a", "Aimssh", helper.TimerStartedTitle, notificaiton_body).Run() 358 | } 359 | } 360 | 361 | return m, m.timer.Init() 362 | } 363 | } 364 | } 365 | 366 | m.list, cmd = m.list.Update(msg) 367 | return m, cmd 368 | } 369 | 370 | func format(n int) string { 371 | var b [2]byte 372 | if n < 10 { 373 | b[0], b[1] = '0', byte(n)+'0' 374 | } else { 375 | b[0], b[1] = byte(n/10)+'0', byte(n%10)+'0' 376 | } 377 | return string(b[:]) 378 | } 379 | 380 | func updateTimer(msg tea.Msg, m model) (tea.Model, tea.Cmd) { 381 | switch msg := msg.(type) { 382 | case timer.TickMsg: 383 | var cmd tea.Cmd 384 | m.timer, cmd = m.timer.Update(msg) 385 | return m, cmd 386 | case timer.TimeoutMsg: 387 | m.timedOut = true 388 | 389 | notificaiton_body := "The timer has ended." 390 | if !run_as_ssh { 391 | beeep.Alert(helper.TimerEndedTitle, notificaiton_body, "assets/logo.png") 392 | } else { 393 | switch m.sessionSsh.Context().Value("operating_system") { 394 | case "linux": 395 | wish.Command(m.sessionSsh, "notify-send", "-a", "Aimssh", helper.TimerEndedTitle, notificaiton_body).Run() 396 | 397 | } 398 | } 399 | 400 | // return m, tea.SetWindowTitle("Done") 401 | 402 | case timer.StartStopMsg: 403 | var cmd tea.Cmd 404 | m.timer, cmd = m.timer.Update(msg) 405 | m.keymap.stop.SetEnabled(m.timer.Running()) 406 | m.keymap.start.SetEnabled(!m.timer.Running()) 407 | return m, cmd 408 | 409 | // case timer.TimeoutMsg: 410 | // m.quitting = true 411 | // return m, tea.Quit 412 | 413 | case tea.KeyMsg: 414 | switch { 415 | case key.Matches(msg, m.keymap.quit): 416 | m.timedOut = false 417 | m.quitting = true 418 | return m, tea.Quit 419 | case key.Matches(msg, m.keymap.reset): 420 | m.timedOut = false 421 | m.asciiArt = m.generate_ascii() 422 | m.timer.Timeout = m.minute 423 | 424 | notificaiton_body := fmt.Sprintf("Timer set for %d minutes.", int(m.timer.Timeout.Minutes())) 425 | if !run_as_ssh { 426 | beeep.Alert(helper.TimerRestartedTitle, notificaiton_body, "assets/logo.png") 427 | } else { 428 | switch m.sessionSsh.Context().Value("operating_system") { 429 | case "linux": 430 | wish.Command(m.sessionSsh, "notify-send", "-a", "Aimssh", helper.TimerRestartedTitle, notificaiton_body).Run() 431 | } 432 | } 433 | 434 | return m, m.timer.Start() 435 | case key.Matches(msg, m.keymap.new): 436 | m.timedOut = false 437 | m.state = inputView 438 | m.workingon.Blur() 439 | m.input.Focus() 440 | m.timer.Stop() 441 | return m, nil 442 | case key.Matches(msg, m.keymap.start, m.keymap.stop): 443 | return m, m.timer.Toggle() 444 | } 445 | } 446 | 447 | return m, nil 448 | } 449 | 450 | func (m model) helpView() string { 451 | switch m.state { 452 | case inputView: 453 | return "\n" + m.help.ShortHelpView([]key.Binding{ 454 | m.keymap.enter, 455 | m.keymap.back, 456 | m.keymap.ctrlc, 457 | }) 458 | } 459 | 460 | return "\n" + m.help.ShortHelpView([]key.Binding{ 461 | m.keymap.start, 462 | m.keymap.stop, 463 | m.keymap.reset, 464 | m.keymap.quit, 465 | m.keymap.new, 466 | }) 467 | } 468 | 469 | func (m model) DrawAscii(a, b time.Duration) string { 470 | n := m.asciiArt.Height() 471 | if n == 0 { 472 | return "" 473 | } 474 | if m.selectedItem != "Flow" { 475 | return m.asciiArt.NextAndString(int(percentageDifference(a, b))) 476 | } 477 | return "\n" + greenColor.Render(m.asciiArt.NextAndString(0)) + "\n" 478 | // camel := ` 479 | 480 | // _v ,, 481 | // /'| &&. 482 | // '-\'-&&&&&. 483 | // \&&&&&&&\ 484 | // &&#""&& \ 485 | // / / |\ x 486 | // / / / / 487 | 488 | // ` 489 | // return ascii_generator.BrownColor.Render(camel) + "\n" 490 | } 491 | 492 | func (m model) View() string { 493 | if m.quitting { 494 | return "" 495 | } 496 | 497 | var view string 498 | switch m.state { 499 | case logoView: 500 | view = fmt.Sprintf("\n%s \n\n\n %s%s\n\n Loading...\n", titleStyle.SetString(logo).Render(), greenColor.Bold(true).Render("ssh"), selectedItemStyle.Underline(true).PaddingLeft(2).Render("aim.ftp.sh")) 501 | case inputView: 502 | view = fmt.Sprintf( 503 | "%s \n\n%s", 504 | titleStyle.Render(), 505 | paddingleft.Render( 506 | fmt.Sprintf("%s\n%s", 507 | heightThing.Render( 508 | fmt.Sprintf("%s \n\n%s\n\n%s \n\n%s", 509 | listTitleStyle.Render("Time in minute: *"), 510 | m.input.View(), 511 | brownColor.Render("Session:"), 512 | m.workingon.View(), 513 | ), 514 | ), 515 | m.helpView()))) 516 | case listView: 517 | view = fmt.Sprintf( 518 | "%s \n\n%s%s%s", 519 | titleStyle.Render(), 520 | // listTitleStyle.Render("Select a visual option: "), 521 | m.list.View(), 522 | dotStyle, 523 | m.list.Help.ShortHelpView([]key.Binding{m.keymap.back}), 524 | ) 525 | case timerView: 526 | text := "Session Ended, Press (r) or (n)" 527 | if !m.timedOut { 528 | text = format(int(m.timer.Timeout.Minutes())) + " : " + format(int(m.timer.Timeout.Seconds())%60) 529 | } 530 | view = fmt.Sprintf( 531 | "%s \n\n %s", 532 | titleStyle.Render(), 533 | paddingleft.Render(fmt.Sprintf("%s\n%s%s\n%s", helper.Center(text, app_width-10), 534 | m.DrawAscii(m.minute, m.timer.Timeout), 535 | helper.Center("Session: "+m.session, app_width-8), 536 | // m.generated_thing, 537 | m.helpView()))) 538 | 539 | } 540 | 541 | return lipgloss.Place(m.width, m.height, lipgloss.Center, lipgloss.Center, appStyle.Render(view)) 542 | } 543 | 544 | func percentageDifference(a, b time.Duration) float64 { 545 | if a == 0 && b == 0 { 546 | return 0.0 547 | } 548 | 549 | return ((a.Seconds() - b.Seconds()) / a.Seconds()) * 100 550 | } 551 | 552 | func wish_server() { 553 | s, err := wish.NewServer( 554 | wish.WithAddress(net.JoinHostPort(host, port)), 555 | wish.WithHostKeyPath(".ssh/id_ed25519"), 556 | wish.WithMiddleware( 557 | bubbletea.Middleware(teaHandler), 558 | activeterm.Middleware(), 559 | logging.Middleware(), 560 | // After Bubbletea quit. 561 | func(next ssh.Handler) ssh.Handler { 562 | return func(sess ssh.Session) { 563 | // pty, _, _ := 564 | wish.Println(sess, end_info) 565 | next(sess) 566 | } 567 | }, 568 | ), 569 | ) 570 | if err != nil { 571 | log.Error("Could not start server", "error", err) 572 | } 573 | 574 | done := make(chan os.Signal, 1) 575 | signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM) 576 | log.Info("Starting SSH server", "host", host, "port", port) 577 | go func() { 578 | if err = s.ListenAndServe(); err != nil && !errors.Is(err, ssh.ErrServerClosed) { 579 | log.Error("Could not start server", "error", err) 580 | done <- nil 581 | } 582 | }() 583 | 584 | <-done 585 | log.Info("Stopping SSH server") 586 | ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second) 587 | defer func() { cancel() }() 588 | if err := s.Shutdown(ctx); err != nil && !errors.Is(err, ssh.ErrServerClosed) { 589 | log.Error("Could not stop server", "error", err) 590 | } 591 | } 592 | 593 | func teaHandler(s ssh.Session) (tea.Model, []tea.ProgramOption) { 594 | s.Context().SetValue("operating_system", detectOS(s)) 595 | return initialModel(s), []tea.ProgramOption{tea.WithAltScreen()} 596 | } 597 | 598 | func detectOS(s ssh.Session) string { 599 | // if wish.Command(s, "sw_vers").Run() == nil { 600 | // return "darwin" 601 | // } 602 | 603 | if wish.Command(s, "paplay", "--help").Run() == nil { 604 | return "linux" 605 | } 606 | 607 | return "windows" 608 | } 609 | 610 | func main() { 611 | 612 | flag.BoolVar(&run_as_ssh, "ssh", false, "run as ssh server?") 613 | flag.Parse() 614 | 615 | if run_as_ssh { 616 | wish_server() 617 | return 618 | } 619 | 620 | p := tea.NewProgram(initialModel(nil), tea.WithAltScreen()) 621 | m, err := p.Run() 622 | if err != nil { 623 | fmt.Println("Error running program:", err) 624 | os.Exit(1) 625 | } 626 | 627 | if model, ok := m.(model); ok && model.quitting { 628 | if model.err != nil { 629 | fmt.Printf("Error Occoured: %s \n", model.err.Error()) 630 | return 631 | } 632 | fmt.Println(end_info) 633 | } 634 | } 635 | --------------------------------------------------------------------------------